ium_487176/zad1.ipynb

import pandas as pd
import sklearn.model_selection
from datasets import load_dataset

dataset = load_dataset("mstz/wine", "wine")

Found cached dataset wine (C:/Users/macty/.cache/huggingface/datasets/mstz___wine/wine/1.0.0/7c3844cac7ac7a22d5fbbaf60fc1d4e9c9deb1b9b9c4dbae6a7b1a962dbc96d8)
100%|██████████| 1/1 [00:00<00:00, 500.10it/s]

dataset["train"]

Dataset({
    features: ['fixed_acidity', 'volatile_acidity', 'citric_acid', 'residual_sugar', 'chlorides', 'free_sulfur_dioxide', 'total_sulfur_dioxide', 'density', 'pH', 'sulphates', 'alcohol', 'quality', 'is_red'],
    num_rows: 6497
})

wine_dataset = pd.DataFrame(dataset["train"])

wine_dataset.head()# podgląd danych

	fixed_acidity	volatile_acidity	citric_acid	residual_sugar	chlorides	free_sulfur_dioxide	total_sulfur_dioxide	density	pH	sulphates	alcohol	quality	is_red
0	7.4	0.70	0.00	1.9	0.076	11.0	34.0	0.9978	3.51	0.56	9.4	5	0
1	7.8	0.88	0.00	2.6	0.098	25.0	67.0	0.9968	3.20	0.68	9.8	5	0
2	7.8	0.76	0.04	2.3	0.092	15.0	54.0	0.9970	3.26	0.65	9.8	5	0
3	11.2	0.28	0.56	1.9	0.075	17.0	60.0	0.9980	3.16	0.58	9.8	6	0
4	7.4	0.70	0.00	1.9	0.076	11.0	34.0	0.9978	3.51	0.56	9.4	5	0

wine_dataset.describe(include='all')

	fixed_acidity	volatile_acidity	citric_acid	residual_sugar	chlorides	free_sulfur_dioxide	total_sulfur_dioxide	density	pH	sulphates	alcohol	quality	is_red
count	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000
mean	7.215307	0.339666	0.318633	5.443235	0.056034	30.525319	115.744574	0.994697	3.218501	0.531268	10.491801	5.818378	0.753886
std	1.296434	0.164636	0.145318	4.757804	0.035034	17.749400	56.521855	0.002999	0.160787	0.148806	1.192712	0.873255	0.430779
min	3.800000	0.080000	0.000000	0.600000	0.009000	1.000000	6.000000	0.987110	2.720000	0.220000	8.000000	3.000000	0.000000
25%	6.400000	0.230000	0.250000	1.800000	0.038000	17.000000	77.000000	0.992340	3.110000	0.430000	9.500000	5.000000	1.000000
50%	7.000000	0.290000	0.310000	3.000000	0.047000	29.000000	118.000000	0.994890	3.210000	0.510000	10.300000	6.000000	1.000000
75%	7.700000	0.400000	0.390000	8.100000	0.065000	41.000000	156.000000	0.996990	3.320000	0.600000	11.300000	6.000000	1.000000
max	15.900000	1.580000	1.660000	65.800000	0.611000	289.000000	440.000000	1.038980	4.010000	2.000000	14.900000	9.000000	1.000000

wine_dataset["is_red"].value_counts().plot(kind="bar")

<Axes: >

wine_dataset["fixed_acidity"].std()

1.2964337577998153

import numpy as np
np.where(pd.isnull(wine_dataset))## sprawdzanie czy istnieją puste wartości

(array([], dtype=int64), array([], dtype=int64))

for column in wine_dataset.columns:
    wine_dataset[column] = wine_dataset[column]  / wine_dataset[column].abs().max() # normalizacja

wine_dataset.describe(include='all') # sprawdzanie wartości po znormalizowaniu

	fixed_acidity	volatile_acidity	citric_acid	residual_sugar	chlorides	free_sulfur_dioxide	total_sulfur_dioxide	density	pH	sulphates	alcohol	quality	is_red
count	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000	6497.000000
mean	0.453793	0.214978	0.191948	0.082724	0.091708	0.105624	0.263056	0.957378	0.802619	0.265634	0.704148	0.646486	0.753886
std	0.081537	0.104200	0.087541	0.072307	0.057338	0.061417	0.128459	0.002886	0.040097	0.074403	0.080048	0.097028	0.430779
min	0.238994	0.050633	0.000000	0.009119	0.014730	0.003460	0.013636	0.950076	0.678304	0.110000	0.536913	0.333333	0.000000
25%	0.402516	0.145570	0.150602	0.027356	0.062193	0.058824	0.175000	0.955110	0.775561	0.215000	0.637584	0.555556	1.000000
50%	0.440252	0.183544	0.186747	0.045593	0.076923	0.100346	0.268182	0.957564	0.800499	0.255000	0.691275	0.666667	1.000000
75%	0.484277	0.253165	0.234940	0.123100	0.106383	0.141869	0.354545	0.959585	0.827930	0.300000	0.758389	0.666667	1.000000
max	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000

wine_dataset["fixed_acidity"].nlargest(10) #sprawdza czy najwyższe wartości mają sens

652     1.000000
442     0.981132
557     0.981132
554     0.974843
555     0.974843
243     0.943396
244     0.943396
544     0.899371
3125    0.893082
374     0.880503
Name: fixed_acidity, dtype: float64

from sklearn.model_selection import train_test_split
wine_train, wine_test = sklearn.model_selection.train_test_split(wine_dataset, test_size=0.1, random_state=1, stratify=wine_dataset["is_red"])
wine_train["is_red"].value_counts() 
# podzielenie na train i test

1.0    4408
0.0    1439
Name: is_red, dtype: int64

wine_test["is_red"].value_counts()

1.0    490
0.0    160
Name: is_red, dtype: int64

wine_test, wine_val = sklearn.model_selection.train_test_split(wine_test, test_size=0.5, random_state=1, stratify=wine_test["is_red"]) # podzielenie na test i validation

wine_test["is_red"].value_counts()

1.0    245
0.0     80
Name: is_red, dtype: int64

wine_val["is_red"].value_counts()

1.0    245
0.0     80
Name: is_red, dtype: int64

import seaborn as sns
sns.set_theme()

len(wine_dataset.columns)

13

#sns.pairplot(data=wine_dataset, hue="is_red")

wine_test.describe()

	fixed_acidity	volatile_acidity	citric_acid	residual_sugar	chlorides	free_sulfur_dioxide	total_sulfur_dioxide	density	pH	sulphates	alcohol	quality	is_red
count	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000
mean	0.448244	0.217069	0.180630	0.078990	0.088742	0.103024	0.257462	0.957255	0.803553	0.263877	0.703930	0.646154	0.753846
std	0.074301	0.107627	0.078046	0.070045	0.051400	0.054750	0.125165	0.002786	0.039808	0.072275	0.078704	0.095014	0.431433
min	0.314465	0.063291	0.000000	0.012158	0.031097	0.010381	0.020455	0.951116	0.713217	0.130000	0.570470	0.333333	0.000000
25%	0.402516	0.145570	0.144578	0.027356	0.060556	0.058824	0.168182	0.955168	0.775561	0.210000	0.637584	0.555556	1.000000
50%	0.433962	0.177215	0.180723	0.042553	0.078560	0.100346	0.261364	0.957478	0.800499	0.250000	0.691275	0.666667	1.000000
75%	0.471698	0.253165	0.222892	0.113982	0.101473	0.141869	0.343182	0.959354	0.827930	0.300000	0.758389	0.666667	1.000000
max	0.817610	0.569620	0.445783	0.334347	0.679214	0.231834	0.575000	0.965264	0.917706	0.585000	0.939597	1.000000	1.000000

wine_train.describe()

	fixed_acidity	volatile_acidity	citric_acid	residual_sugar	chlorides	free_sulfur_dioxide	total_sulfur_dioxide	density	pH	sulphates	alcohol	quality	is_red
count	5847.000000	5847.000000	5847.000000	5847.000000	5847.000000	5847.000000	5847.000000	5847.000000	5847.000000	5847.000000	5847.000000	5847.000000	5847.000000
mean	0.453848	0.215061	0.192235	0.082331	0.092161	0.105659	0.262894	0.957364	0.802569	0.265798	0.704326	0.646732	0.753891
std	0.081742	0.104315	0.088036	0.071982	0.058619	0.061749	0.128256	0.002882	0.039880	0.074864	0.079852	0.096928	0.430780
min	0.238994	0.050633	0.000000	0.009119	0.014730	0.003460	0.013636	0.950076	0.678304	0.110000	0.536913	0.333333	0.000000
25%	0.402516	0.145570	0.150602	0.027356	0.062193	0.058824	0.176136	0.955071	0.775561	0.215000	0.637584	0.555556	1.000000
50%	0.440252	0.183544	0.186747	0.045593	0.076923	0.100346	0.268182	0.957516	0.800499	0.255000	0.691275	0.666667	1.000000
75%	0.484277	0.253165	0.234940	0.123100	0.106383	0.141869	0.353409	0.959581	0.827930	0.300000	0.758389	0.666667	1.000000
max	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000	1.000000

wine_val.describe()

	fixed_acidity	volatile_acidity	citric_acid	residual_sugar	chlorides	free_sulfur_dioxide	total_sulfur_dioxide	density	pH	sulphates	alcohol	quality	is_red
count	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000	325.000000
mean	0.458355	0.211412	0.198091	0.093521	0.086537	0.107596	0.271556	0.957757	0.802570	0.264446	0.701160	0.642393	0.753846
std	0.084621	0.098749	0.086862	0.079346	0.035141	0.061805	0.135185	0.003031	0.044183	0.068086	0.084939	0.100957	0.431433
min	0.295597	0.056962	0.000000	0.012158	0.019640	0.010381	0.018182	0.950413	0.715711	0.140000	0.563758	0.333333	0.000000
25%	0.402516	0.145570	0.156627	0.030395	0.063830	0.055363	0.179545	0.955456	0.773067	0.215000	0.630872	0.555556	1.000000
50%	0.446541	0.183544	0.186747	0.069149	0.078560	0.100346	0.284091	0.957978	0.800499	0.250000	0.684564	0.666667	1.000000
75%	0.490566	0.253165	0.240964	0.133739	0.098200	0.155709	0.370455	0.960028	0.827930	0.305000	0.758389	0.666667	1.000000
max	0.943396	0.746835	0.445783	0.480243	0.278232	0.266436	0.570455	0.972396	1.000000	0.570000	0.939597	0.888889	1.000000

import torch
from torch import nn
from torch.utils.data import DataLoader, Dataset

class TabularDataset(Dataset):
    def __init__(self, data):
        self.data = data.values.astype('float32')

    def __getitem__(self, index):
        x = torch.tensor(self.data[index, :-1])
        y = torch.tensor(self.data[index, -1])
        return x, y

    def __len__(self):
        return len(self.data)

batch_size = 64
train_dataset = TabularDataset(wine_train)
train_dataloader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True)
test_dataset = TabularDataset(wine_test)
test_dataloader = DataLoader(test_dataset, batch_size=batch_size, shuffle=False)

class TabularModel(nn.Module):
    def __init__(self, input_dim, hidden_dim, output_dim):
        super(TabularModel, self).__init__()
        self.fc1 = nn.Linear(input_dim, hidden_dim)
        self.relu = nn.ReLU()
        self.fc2 = nn.Linear(hidden_dim, output_dim)
        self.softmax = nn.Softmax(dim=1)
        
    def forward(self, x):
        out = self.fc1(x)
        out = self.relu(out)
        out = self.fc2(out)
        out = self.softmax(out)
        return out

input_dim = wine_train.shape[1] - 1
hidden_dim = 32
output_dim = 2
model = TabularModel(input_dim, hidden_dim, output_dim)
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters())

model = TabularModel(input_dim=len(wine_train.columns)-1, hidden_dim=32, output_dim=2)
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=0.01)

num_epochs = 10
for epoch in range(num_epochs):
    running_loss = 0.0
    for i, data in enumerate(train_dataloader, 0):
        inputs, labels = data
        labels = labels.type(torch.LongTensor)
        optimizer.zero_grad()
        outputs = model(inputs)
        loss = criterion(outputs, labels)
        loss.backward()
        optimizer.step()
        running_loss += loss.item()

    # Print the loss every 1000 mini-batches
    if (epoch%2)  == 0:
        print(f'Epoch {epoch + 1}, loss: {running_loss / len(train_dataloader):.4f}')

print('Finished Training')

tensor([[0.3481, 0.6519],
        [0.3542, 0.6458],
        [0.3278, 0.6722],
        [0.3486, 0.6514],
        [0.3409, 0.6591],
        [0.3170, 0.6830],
        [0.3306, 0.6694],
        [0.3409, 0.6591],
        [0.3324, 0.6676],
        [0.3382, 0.6618],
        [0.3430, 0.6570],
        [0.3359, 0.6641],
        [0.3383, 0.6617],
        [0.3464, 0.6536],
        [0.3393, 0.6607],
        [0.3455, 0.6545],
        [0.3454, 0.6546],
        [0.3459, 0.6541],
        [0.3558, 0.6442],
        [0.3278, 0.6722],
        [0.3339, 0.6661],
        [0.3426, 0.6574],
        [0.3383, 0.6617],
        [0.3300, 0.6700],
        [0.3436, 0.6564],
        [0.3370, 0.6630],
        [0.3434, 0.6566],
        [0.3442, 0.6558],
        [0.3330, 0.6670],
        [0.3237, 0.6763],
        [0.3449, 0.6551],
        [0.3453, 0.6547],
        [0.3517, 0.6483],
        [0.3252, 0.6748],
        [0.3510, 0.6490],
        [0.3354, 0.6646],
        [0.3499, 0.6501],
        [0.3472, 0.6528],
        [0.3315, 0.6685],
        [0.3328, 0.6672],
        [0.3271, 0.6729],
        [0.3220, 0.6780],
        [0.3476, 0.6524],
        [0.3432, 0.6568],
        [0.3311, 0.6689],
        [0.3331, 0.6669],
        [0.3446, 0.6554],
        [0.3441, 0.6559],
        [0.3403, 0.6597],
        [0.3218, 0.6782],
        [0.3451, 0.6549],
        [0.3437, 0.6563],
        [0.3394, 0.6606],
        [0.3297, 0.6703],
        [0.3449, 0.6551],
        [0.3566, 0.6434],
        [0.3498, 0.6502],
        [0.3539, 0.6461],
        [0.3251, 0.6749],
        [0.3424, 0.6576],
        [0.3368, 0.6632],
        [0.3565, 0.6435],
        [0.3483, 0.6517],
        [0.3267, 0.6733]], grad_fn=<SoftmaxBackward0>)
tensor([[0.2725, 0.7275],
        [0.2981, 0.7019],
        [0.2908, 0.7092],
        [0.2678, 0.7322],
        [0.2793, 0.7207],
        [0.2871, 0.7129],
        [0.2914, 0.7086],
        [0.2990, 0.7010],
        [0.2877, 0.7123],
        [0.2769, 0.7231],
        [0.2929, 0.7071],
        [0.2950, 0.7050],
        [0.2953, 0.7047],
        [0.2707, 0.7293],
        [0.3063, 0.6937],
        [0.2876, 0.7124],
        [0.2773, 0.7227],
        [0.2883, 0.7117],
        [0.2886, 0.7114],
        [0.2945, 0.7055],
        [0.2842, 0.7158],
        [0.2828, 0.7172],
        [0.3065, 0.6935],
        [0.2837, 0.7163],
        [0.2829, 0.7171],
        [0.2915, 0.7085],
        [0.2802, 0.7198],
        [0.2894, 0.7106],
        [0.2858, 0.7142],
        [0.2850, 0.7150],
        [0.2962, 0.7038],
        [0.2926, 0.7074],
        [0.2897, 0.7103],
        [0.2679, 0.7321],
        [0.2940, 0.7060],
        [0.2908, 0.7092],
        [0.2812, 0.7188],
        [0.2904, 0.7096],
        [0.2894, 0.7106],
        [0.2997, 0.7003],
        [0.2820, 0.7180],
        [0.2849, 0.7151],
        [0.2938, 0.7062],
        [0.2922, 0.7078],
        [0.2906, 0.7094],
        [0.2797, 0.7203],
        [0.2946, 0.7054],
        [0.2894, 0.7106],
        [0.2794, 0.7206],
        [0.2778, 0.7222],
        [0.3066, 0.6934],
        [0.2938, 0.7062],
        [0.2763, 0.7237],
        [0.2706, 0.7294],
        [0.2885, 0.7115],
        [0.2788, 0.7212],
        [0.2847, 0.7153],
        [0.2987, 0.7013],
        [0.2825, 0.7175],
        [0.2892, 0.7108],
        [0.2739, 0.7261],
        [0.2925, 0.7075],
        [0.2758, 0.7242],
        [0.2930, 0.7070]], grad_fn=<SoftmaxBackward0>)
tensor([[0.2494, 0.7506],
        [0.2329, 0.7671],
        [0.2502, 0.7498],
        [0.2373, 0.7627],
        [0.2136, 0.7864],
        [0.2340, 0.7660],
        [0.2182, 0.7818],
        [0.2438, 0.7562],
        [0.2292, 0.7708],
        [0.2286, 0.7714],
        [0.2231, 0.7769],
        [0.2241, 0.7759],
        [0.2480, 0.7520],
        [0.2444, 0.7556],
        [0.2390, 0.7610],
        [0.2265, 0.7735],
        [0.2383, 0.7617],
        [0.2275, 0.7725],
        [0.2388, 0.7612],
        [0.2382, 0.7618],
        [0.2363, 0.7637],
        [0.2262, 0.7738],
        [0.2403, 0.7597],
        [0.2426, 0.7574],
        [0.2342, 0.7658],
        [0.2300, 0.7700],
        [0.2445, 0.7555],
        [0.2317, 0.7683],
        [0.2427, 0.7573],
        [0.2310, 0.7690],
        [0.2518, 0.7482],
        [0.2231, 0.7769],
        [0.2264, 0.7736],
        [0.2307, 0.7693],
        [0.2392, 0.7608],
        [0.2330, 0.7670],
        [0.2426, 0.7574],
        [0.2282, 0.7718],
        [0.2302, 0.7698],
        [0.2423, 0.7577],
        [0.2401, 0.7599],
        [0.2282, 0.7718],
        [0.2415, 0.7585],
        [0.2261, 0.7739],
        [0.2256, 0.7744],
        [0.2413, 0.7587],
        [0.2223, 0.7777],
        [0.2318, 0.7682],
        [0.2415, 0.7585],
        [0.2372, 0.7628],
        [0.2150, 0.7850],
        [0.2383, 0.7617],
        [0.2278, 0.7722],
        [0.2259, 0.7741],
        [0.2328, 0.7672],
        [0.2346, 0.7654],
        [0.2329, 0.7671],
        [0.2227, 0.7773],
        [0.2184, 0.7816],
        [0.2294, 0.7706],
        [0.2310, 0.7690],
        [0.2374, 0.7626],
        [0.2408, 0.7592],
        [0.2466, 0.7534]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1890, 0.8110],
        [0.2030, 0.7970],
        [0.1973, 0.8027],
        [0.2048, 0.7952],
        [0.1955, 0.8045],
        [0.2023, 0.7977],
        [0.1813, 0.8187],
        [0.2066, 0.7934],
        [0.2091, 0.7909],
        [0.1840, 0.8160],
        [0.1879, 0.8121],
        [0.2072, 0.7928],
        [0.1992, 0.8008],
        [0.2170, 0.7830],
        [0.1817, 0.8183],
        [0.1938, 0.8062],
        [0.2000, 0.8000],
        [0.1967, 0.8033],
        [0.1985, 0.8015],
        [0.1934, 0.8066],
        [0.2051, 0.7949],
        [0.1943, 0.8057],
        [0.1872, 0.8128],
        [0.2086, 0.7914],
        [0.2005, 0.7995],
        [0.1930, 0.8070],
        [0.1821, 0.8179],
        [0.1893, 0.8107],
        [0.1915, 0.8085],
        [0.2003, 0.7997],
        [0.2061, 0.7939],
        [0.1992, 0.8008],
        [0.2033, 0.7967],
        [0.1946, 0.8054],
        [0.2031, 0.7969],
        [0.2064, 0.7936],
        [0.1891, 0.8109],
        [0.2103, 0.7897],
        [0.1869, 0.8131],
        [0.1971, 0.8029],
        [0.2013, 0.7987],
        [0.1936, 0.8064],
        [0.1973, 0.8027],
        [0.1936, 0.8064],
        [0.1994, 0.8006],
        [0.1929, 0.8071],
        [0.1879, 0.8121],
        [0.2104, 0.7896],
        [0.1941, 0.8059],
        [0.2035, 0.7965],
        [0.1906, 0.8094],
        [0.2056, 0.7944],
        [0.2014, 0.7986],
        [0.1815, 0.8185],
        [0.2007, 0.7993],
        [0.2013, 0.7987],
        [0.2048, 0.7952],
        [0.2072, 0.7928],
        [0.1868, 0.8132],
        [0.2070, 0.7930],
        [0.1928, 0.8072],
        [0.1880, 0.8120],
        [0.2026, 0.7974],
        [0.1938, 0.8062]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1724, 0.8276],
        [0.1418, 0.8582],
        [0.1563, 0.8437],
        [0.1743, 0.8257],
        [0.1710, 0.8290],
        [0.1512, 0.8488],
        [0.1650, 0.8350],
        [0.1529, 0.8471],
        [0.1441, 0.8559],
        [0.1503, 0.8497],
        [0.1634, 0.8366],
        [0.1618, 0.8382],
        [0.1451, 0.8549],
        [0.1711, 0.8289],
        [0.1539, 0.8461],
        [0.1566, 0.8434],
        [0.1643, 0.8357],
        [0.1593, 0.8407],
        [0.1569, 0.8431],
        [0.1592, 0.8408],
        [0.1671, 0.8329],
        [0.1687, 0.8313],
        [0.1531, 0.8469],
        [0.1473, 0.8527],
        [0.1701, 0.8299],
        [0.1731, 0.8269],
        [0.1623, 0.8377],
        [0.1625, 0.8375],
        [0.1415, 0.8585],
        [0.1622, 0.8378],
        [0.1577, 0.8423],
        [0.1438, 0.8562],
        [0.1500, 0.8500],
        [0.1548, 0.8452],
        [0.1516, 0.8484],
        [0.1557, 0.8443],
        [0.1743, 0.8257],
        [0.1547, 0.8453],
        [0.1635, 0.8365],
        [0.1590, 0.8410],
        [0.1577, 0.8423],
        [0.1591, 0.8409],
        [0.1593, 0.8407],
        [0.1456, 0.8544],
        [0.1569, 0.8431],
        [0.1552, 0.8448],
        [0.1717, 0.8283],
        [0.1522, 0.8478],
        [0.1491, 0.8509],
        [0.1568, 0.8432],
        [0.1617, 0.8383],
        [0.1579, 0.8421],
        [0.1624, 0.8376],
        [0.1624, 0.8376],
        [0.1610, 0.8390],
        [0.1451, 0.8549],
        [0.1615, 0.8385],
        [0.1459, 0.8541],
        [0.1583, 0.8417],
        [0.1602, 0.8398],
        [0.1602, 0.8398],
        [0.1748, 0.8252],
        [0.1730, 0.8270],
        [0.1593, 0.8407]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1227, 0.8773],
        [0.1233, 0.8767],
        [0.1140, 0.8860],
        [0.1197, 0.8803],
        [0.1068, 0.8932],
        [0.1304, 0.8696],
        [0.1312, 0.8688],
        [0.1356, 0.8644],
        [0.1347, 0.8653],
        [0.1106, 0.8894],
        [0.1144, 0.8856],
        [0.1267, 0.8733],
        [0.1228, 0.8772],
        [0.1331, 0.8669],
        [0.1202, 0.8798],
        [0.1251, 0.8749],
        [0.1144, 0.8856],
        [0.1313, 0.8687],
        [0.1263, 0.8737],
        [0.1243, 0.8757],
        [0.1293, 0.8707],
        [0.1174, 0.8826],
        [0.1202, 0.8798],
        [0.1475, 0.8525],
        [0.1355, 0.8645],
        [0.1375, 0.8625],
        [0.1134, 0.8866],
        [0.1189, 0.8811],
        [0.1100, 0.8900],
        [0.1193, 0.8807],
        [0.1278, 0.8722],
        [0.1181, 0.8819],
        [0.1310, 0.8690],
        [0.1217, 0.8783],
        [0.1317, 0.8683],
        [0.1247, 0.8753],
        [0.1310, 0.8690],
        [0.1237, 0.8763],
        [0.1292, 0.8708],
        [0.1176, 0.8824],
        [0.1334, 0.8666],
        [0.1217, 0.8783],
        [0.1358, 0.8642],
        [0.1245, 0.8755],
        [0.1262, 0.8738],
        [0.1292, 0.8708],
        [0.1394, 0.8606],
        [0.1132, 0.8868],
        [0.1192, 0.8808],
        [0.1326, 0.8674],
        [0.1382, 0.8618],
        [0.1215, 0.8785],
        [0.1277, 0.8723],
        [0.1230, 0.8770],
        [0.1156, 0.8844],
        [0.1220, 0.8780],
        [0.1225, 0.8775],
        [0.1393, 0.8607],
        [0.1292, 0.8708],
        [0.1267, 0.8733],
        [0.1103, 0.8897],
        [0.1267, 0.8733],
        [0.1303, 0.8697],
        [0.1159, 0.8841]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1003, 0.8997],
        [0.0971, 0.9029],
        [0.1046, 0.8954],
        [0.0827, 0.9173],
        [0.0996, 0.9004],
        [0.0840, 0.9160],
        [0.0926, 0.9074],
        [0.1031, 0.8969],
        [0.1028, 0.8972],
        [0.0981, 0.9019],
        [0.0949, 0.9051],
        [0.1053, 0.8947],
        [0.1057, 0.8943],
        [0.1084, 0.8916],
        [0.0939, 0.9061],
        [0.0917, 0.9083],
        [0.0882, 0.9118],
        [0.1012, 0.8988],
        [0.0904, 0.9096],
        [0.0900, 0.9100],
        [0.1018, 0.8982],
        [0.0938, 0.9062],
        [0.0831, 0.9169],
        [0.1070, 0.8930],
        [0.1028, 0.8972],
        [0.1037, 0.8963],
        [0.0968, 0.9032],
        [0.0942, 0.9058],
        [0.0842, 0.9158],
        [0.0822, 0.9178],
        [0.0982, 0.9018],
        [0.1131, 0.8869],
        [0.0908, 0.9092],
        [0.0988, 0.9012],
        [0.0880, 0.9120],
        [0.0912, 0.9088],
        [0.1138, 0.8862],
        [0.1180, 0.8820],
        [0.1031, 0.8969],
        [0.0939, 0.9061],
        [0.0884, 0.9116],
        [0.0925, 0.9075],
        [0.0893, 0.9107],
        [0.0985, 0.9015],
        [0.1159, 0.8841],
        [0.0926, 0.9074],
        [0.1156, 0.8844],
        [0.1131, 0.8869],
        [0.0884, 0.9116],
        [0.0939, 0.9061],
        [0.1048, 0.8952],
        [0.0927, 0.9073],
        [0.1048, 0.8952],
        [0.1032, 0.8968],
        [0.0959, 0.9041],
        [0.0955, 0.9045],
        [0.1023, 0.8977],
        [0.0930, 0.9070],
        [0.1002, 0.8998],
        [0.0928, 0.9072],
        [0.0999, 0.9001],
        [0.0948, 0.9052],
        [0.0881, 0.9119],
        [0.0946, 0.9054]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0908, 0.9092],
        [0.0686, 0.9314],
        [0.0842, 0.9158],
        [0.0795, 0.9205],
        [0.0740, 0.9260],
        [0.0632, 0.9368],
        [0.0692, 0.9308],
        [0.0737, 0.9263],
        [0.0652, 0.9348],
        [0.0833, 0.9167],
        [0.0718, 0.9282],
        [0.0768, 0.9232],
        [0.0723, 0.9277],
        [0.0770, 0.9230],
        [0.0790, 0.9210],
        [0.0834, 0.9166],
        [0.0693, 0.9307],
        [0.0639, 0.9361],
        [0.0757, 0.9243],
        [0.0813, 0.9187],
        [0.0673, 0.9327],
        [0.0908, 0.9092],
        [0.0674, 0.9326],
        [0.0621, 0.9379],
        [0.0743, 0.9257],
        [0.0729, 0.9271],
        [0.0885, 0.9115],
        [0.0907, 0.9093],
        [0.0594, 0.9406],
        [0.0799, 0.9201],
        [0.0763, 0.9237],
        [0.0818, 0.9182],
        [0.0625, 0.9375],
        [0.0825, 0.9175],
        [0.0709, 0.9291],
        [0.0721, 0.9279],
        [0.0698, 0.9302],
        [0.0670, 0.9330],
        [0.0698, 0.9302],
        [0.0709, 0.9291],
        [0.0664, 0.9336],
        [0.0731, 0.9269],
        [0.0799, 0.9201],
        [0.0644, 0.9356],
        [0.0681, 0.9319],
        [0.0673, 0.9327],
        [0.0832, 0.9168],
        [0.0683, 0.9317],
        [0.0862, 0.9138],
        [0.0781, 0.9219],
        [0.0713, 0.9287],
        [0.0796, 0.9204],
        [0.0722, 0.9278],
        [0.0675, 0.9325],
        [0.0869, 0.9131],
        [0.0781, 0.9219],
        [0.1014, 0.8986],
        [0.0696, 0.9304],
        [0.0821, 0.9179],
        [0.0747, 0.9253],
        [0.0855, 0.9145],
        [0.0799, 0.9201],
        [0.0678, 0.9322],
        [0.0703, 0.9297]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0807, 0.9193],
        [0.0611, 0.9389],
        [0.0656, 0.9344],
        [0.0600, 0.9400],
        [0.0540, 0.9460],
        [0.0659, 0.9341],
        [0.0481, 0.9519],
        [0.0618, 0.9382],
        [0.0550, 0.9450],
        [0.0628, 0.9372],
        [0.0779, 0.9221],
        [0.0690, 0.9310],
        [0.0526, 0.9474],
        [0.0478, 0.9522],
        [0.0669, 0.9331],
        [0.0581, 0.9419],
        [0.0742, 0.9258],
        [0.0623, 0.9377],
        [0.0710, 0.9290],
        [0.0506, 0.9494],
        [0.0511, 0.9489],
        [0.0540, 0.9460],
        [0.0566, 0.9434],
        [0.0586, 0.9414],
        [0.0537, 0.9463],
        [0.0500, 0.9500],
        [0.0608, 0.9392],
        [0.0546, 0.9454],
        [0.0605, 0.9395],
        [0.0496, 0.9504],
        [0.0583, 0.9417],
        [0.0582, 0.9418],
        [0.0702, 0.9298],
        [0.0643, 0.9357],
        [0.0566, 0.9434],
        [0.0550, 0.9450],
        [0.0785, 0.9215],
        [0.0827, 0.9173],
        [0.0694, 0.9306],
        [0.0793, 0.9207],
        [0.0549, 0.9451],
        [0.0589, 0.9411],
        [0.0562, 0.9438],
        [0.0618, 0.9382],
        [0.0584, 0.9416],
        [0.0566, 0.9434],
        [0.0644, 0.9356],
        [0.0658, 0.9342],
        [0.0657, 0.9343],
        [0.0701, 0.9299],
        [0.0534, 0.9466],
        [0.0552, 0.9448],
        [0.0539, 0.9461],
        [0.0635, 0.9365],
        [0.0565, 0.9435],
        [0.0606, 0.9394],
        [0.0635, 0.9365],
        [0.0582, 0.9418],
        [0.0650, 0.9350],
        [0.0806, 0.9194],
        [0.0623, 0.9377],
        [0.0533, 0.9467],
        [0.0528, 0.9472],
        [0.0603, 0.9397]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0594, 0.9406],
        [0.0530, 0.9470],
        [0.0489, 0.9511],
        [0.0572, 0.9428],
        [0.0432, 0.9568],
        [0.0499, 0.9501],
        [0.0432, 0.9568],
        [0.0660, 0.9340],
        [0.0436, 0.9564],
        [0.0406, 0.9594],
        [0.0452, 0.9548],
        [0.0504, 0.9496],
        [0.0493, 0.9507],
        [0.0500, 0.9500],
        [0.0713, 0.9287],
        [0.0644, 0.9356],
        [0.0485, 0.9515],
        [0.0476, 0.9524],
        [0.0522, 0.9478],
        [0.0479, 0.9521],
        [0.0471, 0.9529],
        [0.0460, 0.9540],
        [0.0481, 0.9519],
        [0.0435, 0.9565],
        [0.0579, 0.9421],
        [0.0494, 0.9506],
        [0.0617, 0.9383],
        [0.0415, 0.9585],
        [0.0560, 0.9440],
        [0.0647, 0.9353],
        [0.0486, 0.9514],
        [0.0468, 0.9532],
        [0.0512, 0.9488],
        [0.0502, 0.9498],
        [0.0687, 0.9313],
        [0.0514, 0.9486],
        [0.0446, 0.9554],
        [0.0424, 0.9576],
        [0.0482, 0.9518],
        [0.0676, 0.9324],
        [0.0441, 0.9559],
        [0.0504, 0.9496],
        [0.0553, 0.9447],
        [0.0530, 0.9470],
        [0.0443, 0.9557],
        [0.0494, 0.9506],
        [0.0472, 0.9528],
        [0.0432, 0.9568],
        [0.0428, 0.9572],
        [0.0491, 0.9509],
        [0.0677, 0.9323],
        [0.0623, 0.9377],
        [0.0553, 0.9447],
        [0.0401, 0.9599],
        [0.0483, 0.9517],
        [0.0534, 0.9466],
        [0.0670, 0.9330],
        [0.0475, 0.9525],
        [0.0403, 0.9597],
        [0.0448, 0.9552],
        [0.0515, 0.9485],
        [0.0477, 0.9523],
        [0.0470, 0.9530],
        [0.0434, 0.9566]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0395, 0.9605],
        [0.0412, 0.9588],
        [0.0430, 0.9570],
        [0.0545, 0.9455],
        [0.0452, 0.9548],
        [0.0340, 0.9660],
        [0.0300, 0.9700],
        [0.0320, 0.9680],
        [0.0418, 0.9582],
        [0.0391, 0.9609],
        [0.0447, 0.9553],
        [0.0372, 0.9628],
        [0.0578, 0.9422],
        [0.0363, 0.9637],
        [0.0353, 0.9647],
        [0.0604, 0.9396],
        [0.0350, 0.9650],
        [0.0373, 0.9627],
        [0.0413, 0.9587],
        [0.0351, 0.9649],
        [0.0436, 0.9564],
        [0.0422, 0.9578],
        [0.0553, 0.9447],
        [0.0374, 0.9626],
        [0.0337, 0.9663],
        [0.0497, 0.9503],
        [0.0379, 0.9621],
        [0.0525, 0.9475],
        [0.0578, 0.9422],
        [0.0447, 0.9553],
        [0.0546, 0.9454],
        [0.0452, 0.9548],
        [0.0556, 0.9444],
        [0.0443, 0.9557],
        [0.0362, 0.9638],
        [0.0504, 0.9496],
        [0.0394, 0.9606],
        [0.0343, 0.9657],
        [0.0350, 0.9650],
        [0.0390, 0.9610],
        [0.0408, 0.9592],
        [0.0475, 0.9525],
        [0.0475, 0.9525],
        [0.0386, 0.9614],
        [0.0396, 0.9604],
        [0.0490, 0.9510],
        [0.0372, 0.9628],
        [0.0373, 0.9627],
        [0.0413, 0.9587],
        [0.0461, 0.9539],
        [0.0452, 0.9548],
        [0.0341, 0.9659],
        [0.0410, 0.9590],
        [0.0388, 0.9612],
        [0.0387, 0.9613],
        [0.0390, 0.9610],
        [0.0358, 0.9642],
        [0.0525, 0.9475],
        [0.0401, 0.9599],
        [0.0395, 0.9605],
        [0.0397, 0.9603],
        [0.0451, 0.9549],
        [0.0386, 0.9614],
        [0.0349, 0.9651]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0283, 0.9717],
        [0.0365, 0.9635],
        [0.0370, 0.9630],
        [0.0298, 0.9702],
        [0.0355, 0.9645],
        [0.0354, 0.9646],
        [0.0469, 0.9531],
        [0.0376, 0.9624],
        [0.0347, 0.9653],
        [0.0525, 0.9475],
        [0.0308, 0.9692],
        [0.0307, 0.9693],
        [0.0496, 0.9504],
        [0.0292, 0.9708],
        [0.0377, 0.9623],
        [0.0335, 0.9665],
        [0.0484, 0.9516],
        [0.0331, 0.9669],
        [0.0310, 0.9690],
        [0.0372, 0.9628],
        [0.0352, 0.9648],
        [0.0350, 0.9650],
        [0.0322, 0.9678],
        [0.0301, 0.9699],
        [0.0372, 0.9628],
        [0.0342, 0.9658],
        [0.0288, 0.9712],
        [0.0559, 0.9441],
        [0.0486, 0.9514],
        [0.0485, 0.9515],
        [0.0410, 0.9590],
        [0.0421, 0.9579],
        [0.0370, 0.9630],
        [0.0434, 0.9566],
        [0.0451, 0.9549],
        [0.0384, 0.9616],
        [0.0508, 0.9492],
        [0.0335, 0.9665],
        [0.0329, 0.9671],
        [0.0567, 0.9433],
        [0.0349, 0.9651],
        [0.0568, 0.9432],
        [0.0479, 0.9521],
        [0.0313, 0.9687],
        [0.0306, 0.9694],
        [0.0354, 0.9646],
        [0.0481, 0.9519],
        [0.0348, 0.9652],
        [0.0299, 0.9701],
        [0.0308, 0.9692],
        [0.0401, 0.9599],
        [0.0352, 0.9648],
        [0.0367, 0.9633],
        [0.0383, 0.9617],
        [0.0375, 0.9625],
        [0.0281, 0.9719],
        [0.0362, 0.9638],
        [0.0427, 0.9573],
        [0.0287, 0.9713],
        [0.0271, 0.9729],
        [0.0393, 0.9607],
        [0.0428, 0.9572],
        [0.0395, 0.9605],
        [0.0342, 0.9658]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0324, 0.9676],
        [0.0379, 0.9621],
        [0.0270, 0.9730],
        [0.0310, 0.9690],
        [0.0313, 0.9687],
        [0.0265, 0.9735],
        [0.0280, 0.9720],
        [0.0268, 0.9732],
        [0.0423, 0.9577],
        [0.0299, 0.9701],
        [0.0313, 0.9687],
        [0.0392, 0.9608],
        [0.0306, 0.9694],
        [0.0420, 0.9580],
        [0.0346, 0.9654],
        [0.0245, 0.9755],
        [0.0295, 0.9705],
        [0.0397, 0.9603],
        [0.0310, 0.9690],
        [0.0275, 0.9725],
        [0.0337, 0.9663],
        [0.0315, 0.9685],
        [0.0277, 0.9723],
        [0.0432, 0.9568],
        [0.0338, 0.9662],
        [0.0323, 0.9677],
        [0.0290, 0.9710],
        [0.0602, 0.9398],
        [0.0446, 0.9554],
        [0.0346, 0.9654],
        [0.0295, 0.9705],
        [0.0280, 0.9720],
        [0.0397, 0.9603],
        [0.0306, 0.9694],
        [0.0395, 0.9605],
        [0.0379, 0.9621],
        [0.0329, 0.9671],
        [0.0323, 0.9677],
        [0.0210, 0.9790],
        [0.0228, 0.9772],
        [0.0317, 0.9683],
        [0.0433, 0.9567],
        [0.0288, 0.9712],
        [0.0283, 0.9717],
        [0.0311, 0.9689],
        [0.0305, 0.9695],
        [0.0423, 0.9577],
        [0.0294, 0.9706],
        [0.0350, 0.9650],
        [0.0297, 0.9703],
        [0.0364, 0.9636],
        [0.0313, 0.9687],
        [0.0427, 0.9573],
        [0.0327, 0.9673],
        [0.0519, 0.9481],
        [0.0439, 0.9561],
        [0.0259, 0.9741],
        [0.0318, 0.9682],
        [0.0390, 0.9610],
        [0.0276, 0.9724],
        [0.0252, 0.9748],
        [0.0291, 0.9709],
        [0.0325, 0.9675],
        [0.0335, 0.9665]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0287, 0.9713],
        [0.0361, 0.9639],
        [0.0263, 0.9737],
        [0.0268, 0.9732],
        [0.0447, 0.9553],
        [0.0369, 0.9631],
        [0.0419, 0.9581],
        [0.0299, 0.9701],
        [0.0293, 0.9707],
        [0.0257, 0.9743],
        [0.0326, 0.9674],
        [0.0309, 0.9691],
        [0.0405, 0.9595],
        [0.0235, 0.9765],
        [0.0276, 0.9724],
        [0.0268, 0.9732],
        [0.0241, 0.9759],
        [0.0228, 0.9772],
        [0.0276, 0.9724],
        [0.0248, 0.9752],
        [0.0365, 0.9635],
        [0.0296, 0.9704],
        [0.0477, 0.9523],
        [0.0264, 0.9736],
        [0.0332, 0.9668],
        [0.0238, 0.9762],
        [0.0229, 0.9771],
        [0.0415, 0.9585],
        [0.0223, 0.9777],
        [0.0264, 0.9736],
        [0.0197, 0.9803],
        [0.0231, 0.9769],
        [0.0301, 0.9699],
        [0.0229, 0.9771],
        [0.0223, 0.9777],
        [0.0206, 0.9794],
        [0.0318, 0.9682],
        [0.0206, 0.9794],
        [0.0315, 0.9685],
        [0.0303, 0.9697],
        [0.0244, 0.9756],
        [0.0378, 0.9622],
        [0.0286, 0.9714],
        [0.0340, 0.9660],
        [0.0355, 0.9645],
        [0.0293, 0.9707],
        [0.0246, 0.9754],
        [0.0238, 0.9762],
        [0.0260, 0.9740],
        [0.0271, 0.9729],
        [0.0201, 0.9799],
        [0.0210, 0.9790],
        [0.0513, 0.9487],
        [0.0312, 0.9688],
        [0.0393, 0.9607],
        [0.0317, 0.9683],
        [0.0203, 0.9797],
        [0.0281, 0.9719],
        [0.0439, 0.9561],
        [0.0268, 0.9732],
        [0.0257, 0.9743],
        [0.0246, 0.9754],
        [0.0245, 0.9755],
        [0.0312, 0.9688]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0270, 0.9730],
        [0.0228, 0.9772],
        [0.0256, 0.9744],
        [0.0196, 0.9804],
        [0.0234, 0.9766],
        [0.0255, 0.9745],
        [0.0508, 0.9492],
        [0.0256, 0.9744],
        [0.0273, 0.9727],
        [0.0454, 0.9546],
        [0.0340, 0.9660],
        [0.0320, 0.9680],
        [0.0210, 0.9790],
        [0.0330, 0.9670],
        [0.0406, 0.9594],
        [0.0211, 0.9789],
        [0.0246, 0.9754],
        [0.0489, 0.9511],
        [0.0217, 0.9783],
        [0.0223, 0.9777],
        [0.0296, 0.9704],
        [0.0267, 0.9733],
        [0.0205, 0.9795],
        [0.0267, 0.9733],
        [0.0342, 0.9658],
        [0.0309, 0.9691],
        [0.0222, 0.9778],
        [0.0224, 0.9776],
        [0.0284, 0.9716],
        [0.0214, 0.9786],
        [0.0332, 0.9668],
        [0.0304, 0.9696],
        [0.0274, 0.9726],
        [0.0227, 0.9773],
        [0.0281, 0.9719],
        [0.0238, 0.9762],
        [0.0272, 0.9728],
        [0.0219, 0.9781],
        [0.0213, 0.9787],
        [0.0456, 0.9544],
        [0.0472, 0.9528],
        [0.0319, 0.9681],
        [0.0423, 0.9577],
        [0.0225, 0.9775],
        [0.0219, 0.9781],
        [0.0319, 0.9681],
        [0.0478, 0.9522],
        [0.0193, 0.9807],
        [0.0412, 0.9588],
        [0.0235, 0.9765],
        [0.0213, 0.9787],
        [0.0257, 0.9743],
        [0.0228, 0.9772],
        [0.0322, 0.9678],
        [0.0277, 0.9723],
        [0.0246, 0.9754],
        [0.0392, 0.9608],
        [0.0289, 0.9711],
        [0.0235, 0.9765],
        [0.0444, 0.9556],
        [0.0274, 0.9726],
        [0.0298, 0.9702],
        [0.0235, 0.9765],
        [0.0272, 0.9728]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0341, 0.9659],
        [0.0244, 0.9756],
        [0.0244, 0.9756],
        [0.0349, 0.9651],
        [0.0202, 0.9798],
        [0.0427, 0.9573],
        [0.0278, 0.9722],
        [0.0278, 0.9722],
        [0.0340, 0.9660],
        [0.0260, 0.9740],
        [0.0223, 0.9777],
        [0.0213, 0.9787],
        [0.0313, 0.9687],
        [0.0277, 0.9723],
        [0.0381, 0.9619],
        [0.0255, 0.9745],
        [0.0244, 0.9756],
        [0.0462, 0.9538],
        [0.0391, 0.9609],
        [0.0153, 0.9847],
        [0.0259, 0.9741],
        [0.0224, 0.9776],
        [0.0196, 0.9804],
        [0.0415, 0.9585],
        [0.0234, 0.9766],
        [0.0393, 0.9607],
        [0.0200, 0.9800],
        [0.0206, 0.9794],
        [0.0299, 0.9701],
        [0.0218, 0.9782],
        [0.0230, 0.9770],
        [0.0230, 0.9770],
        [0.0220, 0.9780],
        [0.0296, 0.9704],
        [0.0170, 0.9830],
        [0.0253, 0.9747],
        [0.0262, 0.9738],
        [0.0259, 0.9741],
        [0.0254, 0.9746],
        [0.0215, 0.9785],
        [0.0302, 0.9698],
        [0.0279, 0.9721],
        [0.0155, 0.9845],
        [0.0213, 0.9787],
        [0.0394, 0.9606],
        [0.0321, 0.9679],
        [0.0182, 0.9818],
        [0.0306, 0.9694],
        [0.0250, 0.9750],
        [0.0415, 0.9585],
        [0.0362, 0.9638],
        [0.0229, 0.9771],
        [0.0259, 0.9741],
        [0.0195, 0.9805],
        [0.0314, 0.9686],
        [0.0226, 0.9774],
        [0.0332, 0.9668],
        [0.0219, 0.9781],
        [0.0235, 0.9765],
        [0.0263, 0.9737],
        [0.0265, 0.9735],
        [0.0337, 0.9663],
        [0.0209, 0.9791],
        [0.0352, 0.9648]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0260, 0.9740],
        [0.0193, 0.9807],
        [0.0307, 0.9693],
        [0.0222, 0.9778],
        [0.0242, 0.9758],
        [0.0372, 0.9628],
        [0.0220, 0.9780],
        [0.0368, 0.9632],
        [0.0218, 0.9782],
        [0.0196, 0.9804],
        [0.0220, 0.9780],
        [0.0293, 0.9707],
        [0.0277, 0.9723],
        [0.0331, 0.9669],
        [0.0232, 0.9768],
        [0.0360, 0.9640],
        [0.0226, 0.9774],
        [0.0221, 0.9779],
        [0.0409, 0.9591],
        [0.0208, 0.9792],
        [0.0171, 0.9829],
        [0.0195, 0.9805],
        [0.0194, 0.9806],
        [0.0546, 0.9454],
        [0.0200, 0.9800],
        [0.0177, 0.9823],
        [0.0250, 0.9750],
        [0.0220, 0.9780],
        [0.0203, 0.9797],
        [0.0454, 0.9546],
        [0.0218, 0.9782],
        [0.0194, 0.9806],
        [0.0254, 0.9746],
        [0.0359, 0.9641],
        [0.0254, 0.9746],
        [0.0251, 0.9749],
        [0.0266, 0.9734],
        [0.0238, 0.9762],
        [0.0249, 0.9751],
        [0.0202, 0.9798],
        [0.0227, 0.9773],
        [0.0238, 0.9762],
        [0.0279, 0.9721],
        [0.0203, 0.9797],
        [0.0185, 0.9815],
        [0.0207, 0.9793],
        [0.0238, 0.9762],
        [0.0324, 0.9676],
        [0.0185, 0.9815],
        [0.0237, 0.9763],
        [0.0276, 0.9724],
        [0.0273, 0.9727],
        [0.0213, 0.9787],
        [0.0429, 0.9571],
        [0.0206, 0.9794],
        [0.0136, 0.9864],
        [0.0219, 0.9781],
        [0.0321, 0.9679],
        [0.0224, 0.9776],
        [0.0153, 0.9847],
        [0.0256, 0.9744],
        [0.0227, 0.9773],
        [0.0312, 0.9688],
        [0.0239, 0.9761]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0403, 0.9597],
        [0.0162, 0.9838],
        [0.0274, 0.9726],
        [0.0445, 0.9555],
        [0.0217, 0.9783],
        [0.0202, 0.9798],
        [0.0229, 0.9771],
        [0.0323, 0.9677],
        [0.0217, 0.9783],
        [0.0364, 0.9636],
        [0.0277, 0.9723],
        [0.0240, 0.9760],
        [0.0164, 0.9836],
        [0.0282, 0.9718],
        [0.0187, 0.9813],
        [0.0312, 0.9688],
        [0.0150, 0.9850],
        [0.0328, 0.9672],
        [0.0393, 0.9607],
        [0.0198, 0.9802],
        [0.0148, 0.9852],
        [0.0192, 0.9808],
        [0.0441, 0.9559],
        [0.0196, 0.9804],
        [0.0478, 0.9522],
        [0.0233, 0.9767],
        [0.0315, 0.9685],
        [0.0241, 0.9759],
        [0.0294, 0.9706],
        [0.0276, 0.9724],
        [0.0322, 0.9678],
        [0.0417, 0.9583],
        [0.0259, 0.9741],
        [0.0262, 0.9738],
        [0.0250, 0.9750],
        [0.0203, 0.9797],
        [0.0212, 0.9788],
        [0.0361, 0.9639],
        [0.0174, 0.9826],
        [0.0264, 0.9736],
        [0.0271, 0.9729],
        [0.0378, 0.9622],
        [0.0213, 0.9787],
        [0.0189, 0.9811],
        [0.0350, 0.9650],
        [0.0241, 0.9759],
        [0.0281, 0.9719],
        [0.0262, 0.9738],
        [0.0331, 0.9669],
        [0.0164, 0.9836],
        [0.0358, 0.9642],
        [0.0209, 0.9791],
        [0.0313, 0.9687],
        [0.0269, 0.9731],
        [0.0186, 0.9814],
        [0.0161, 0.9839],
        [0.0240, 0.9760],
        [0.0173, 0.9827],
        [0.0364, 0.9636],
        [0.0218, 0.9782],
        [0.0205, 0.9795],
        [0.0270, 0.9730],
        [0.0238, 0.9762],
        [0.0195, 0.9805]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0201, 0.9799],
        [0.0249, 0.9751],
        [0.0206, 0.9794],
        [0.0522, 0.9478],
        [0.0292, 0.9708],
        [0.0295, 0.9705],
        [0.0342, 0.9658],
        [0.0318, 0.9682],
        [0.0307, 0.9693],
        [0.0191, 0.9809],
        [0.0228, 0.9772],
        [0.0243, 0.9757],
        [0.0280, 0.9720],
        [0.0167, 0.9833],
        [0.0374, 0.9626],
        [0.0293, 0.9707],
        [0.0188, 0.9812],
        [0.0540, 0.9460],
        [0.0206, 0.9794],
        [0.0222, 0.9778],
        [0.0167, 0.9833],
        [0.0247, 0.9753],
        [0.0276, 0.9724],
        [0.0375, 0.9625],
        [0.0144, 0.9856],
        [0.0211, 0.9789],
        [0.0285, 0.9715],
        [0.0398, 0.9602],
        [0.0205, 0.9795],
        [0.0164, 0.9836],
        [0.0177, 0.9823],
        [0.0223, 0.9777],
        [0.0344, 0.9656],
        [0.0244, 0.9756],
        [0.0423, 0.9577],
        [0.0173, 0.9827],
        [0.0188, 0.9812],
        [0.0337, 0.9663],
        [0.0207, 0.9793],
        [0.0162, 0.9838],
        [0.0251, 0.9749],
        [0.0195, 0.9805],
        [0.0301, 0.9699],
        [0.0275, 0.9725],
        [0.0226, 0.9774],
        [0.0217, 0.9783],
        [0.0123, 0.9877],
        [0.0183, 0.9817],
        [0.0564, 0.9436],
        [0.0253, 0.9747],
        [0.0259, 0.9741],
        [0.0489, 0.9511],
        [0.0363, 0.9637],
        [0.0219, 0.9781],
        [0.0313, 0.9687],
        [0.0206, 0.9794],
        [0.0315, 0.9685],
        [0.0172, 0.9828],
        [0.0206, 0.9794],
        [0.0262, 0.9738],
        [0.0441, 0.9559],
        [0.0174, 0.9826],
        [0.0452, 0.9548],
        [0.0752, 0.9248]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0605, 0.9395],
        [0.0191, 0.9809],
        [0.0301, 0.9699],
        [0.0312, 0.9688],
        [0.0352, 0.9648],
        [0.0216, 0.9784],
        [0.0194, 0.9806],
        [0.0212, 0.9788],
        [0.0242, 0.9758],
        [0.0147, 0.9853],
        [0.0300, 0.9700],
        [0.0549, 0.9451],
        [0.0233, 0.9767],
        [0.0197, 0.9803],
        [0.0297, 0.9703],
        [0.0367, 0.9633],
        [0.0227, 0.9773],
        [0.0195, 0.9805],
        [0.0229, 0.9771],
        [0.0551, 0.9449],
        [0.0287, 0.9713],
        [0.0570, 0.9430],
        [0.0277, 0.9723],
        [0.0248, 0.9752],
        [0.0203, 0.9797],
        [0.0436, 0.9564],
        [0.0286, 0.9714],
        [0.0589, 0.9411],
        [0.0399, 0.9601],
        [0.0195, 0.9805],
        [0.0219, 0.9781],
        [0.0320, 0.9680],
        [0.0204, 0.9796],
        [0.0301, 0.9699],
        [0.0227, 0.9773],
        [0.0232, 0.9768],
        [0.0470, 0.9530],
        [0.0184, 0.9816],
        [0.0308, 0.9692],
        [0.0257, 0.9743],
        [0.0464, 0.9536],
        [0.0226, 0.9774],
        [0.0369, 0.9631],
        [0.0193, 0.9807],
        [0.0569, 0.9431],
        [0.0200, 0.9800],
        [0.0286, 0.9714],
        [0.0219, 0.9781],
        [0.0250, 0.9750],
        [0.0335, 0.9665],
        [0.0273, 0.9727],
        [0.0360, 0.9640],
        [0.0351, 0.9649],
        [0.0306, 0.9694],
        [0.0272, 0.9728],
        [0.0343, 0.9657],
        [0.0559, 0.9441],
        [0.0326, 0.9674],
        [0.0571, 0.9429],
        [0.0152, 0.9848],
        [0.0271, 0.9729],
        [0.0337, 0.9663],
        [0.0523, 0.9477],
        [0.0449, 0.9551]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0230, 0.9770],
        [0.0348, 0.9652],
        [0.0256, 0.9744],
        [0.0319, 0.9681],
        [0.0554, 0.9446],
        [0.0633, 0.9367],
        [0.0335, 0.9665],
        [0.0272, 0.9728],
        [0.0310, 0.9690],
        [0.0389, 0.9611],
        [0.0257, 0.9743],
        [0.0591, 0.9409],
        [0.0228, 0.9772],
        [0.0336, 0.9664],
        [0.1011, 0.8989],
        [0.0217, 0.9783],
        [0.0340, 0.9660],
        [0.0405, 0.9595],
        [0.0382, 0.9618],
        [0.0241, 0.9759],
        [0.0660, 0.9340],
        [0.0270, 0.9730],
        [0.0299, 0.9701],
        [0.0265, 0.9735],
        [0.0299, 0.9701],
        [0.0254, 0.9746],
        [0.0240, 0.9760],
        [0.0235, 0.9765],
        [0.0303, 0.9697],
        [0.0212, 0.9788],
        [0.0203, 0.9797],
        [0.0324, 0.9676],
        [0.0387, 0.9613],
        [0.0407, 0.9593],
        [0.0297, 0.9703],
        [0.0529, 0.9471],
        [0.0501, 0.9499],
        [0.0270, 0.9730],
        [0.0299, 0.9701],
        [0.0182, 0.9818],
        [0.0271, 0.9729],
        [0.0248, 0.9752],
        [0.0243, 0.9757],
        [0.0701, 0.9299],
        [0.0248, 0.9752],
        [0.0338, 0.9662],
        [0.0454, 0.9546],
        [0.0233, 0.9767],
        [0.0221, 0.9779],
        [0.0251, 0.9749],
        [0.0252, 0.9748],
        [0.0301, 0.9699],
        [0.0419, 0.9581],
        [0.0422, 0.9578],
        [0.0186, 0.9814],
        [0.0290, 0.9710],
        [0.0230, 0.9770],
        [0.0258, 0.9742],
        [0.0429, 0.9571],
        [0.0240, 0.9760],
        [0.0516, 0.9484],
        [0.0229, 0.9771],
        [0.0892, 0.9108],
        [0.0202, 0.9798]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0408, 0.9592],
        [0.0371, 0.9629],
        [0.0279, 0.9721],
        [0.0398, 0.9602],
        [0.0294, 0.9706],
        [0.0507, 0.9493],
        [0.0504, 0.9496],
        [0.0640, 0.9360],
        [0.0348, 0.9652],
        [0.0496, 0.9504],
        [0.0838, 0.9162],
        [0.0312, 0.9688],
        [0.0399, 0.9601],
        [0.0242, 0.9758],
        [0.0357, 0.9643],
        [0.0620, 0.9380],
        [0.0270, 0.9730],
        [0.0468, 0.9532],
        [0.0247, 0.9753],
        [0.0242, 0.9758],
        [0.0318, 0.9682],
        [0.0293, 0.9707],
        [0.0906, 0.9094],
        [0.0281, 0.9719],
        [0.0329, 0.9671],
        [0.0329, 0.9671],
        [0.0342, 0.9658],
        [0.0319, 0.9681],
        [0.0532, 0.9468],
        [0.0359, 0.9641],
        [0.0306, 0.9694],
        [0.0284, 0.9716],
        [0.0248, 0.9752],
        [0.0309, 0.9691],
        [0.0257, 0.9743],
        [0.0532, 0.9468],
        [0.0380, 0.9620],
        [0.0613, 0.9387],
        [0.0256, 0.9744],
        [0.0357, 0.9643],
        [0.0256, 0.9744],
        [0.0370, 0.9630],
        [0.0383, 0.9617],
        [0.1200, 0.8800],
        [0.0269, 0.9731],
        [0.0585, 0.9415],
        [0.0441, 0.9559],
        [0.0477, 0.9523],
        [0.0299, 0.9701],
        [0.0750, 0.9250],
        [0.0292, 0.9708],
        [0.0349, 0.9651],
        [0.0786, 0.9214],
        [0.0325, 0.9675],
        [0.0259, 0.9741],
        [0.0433, 0.9567],
        [0.0656, 0.9344],
        [0.0211, 0.9789],
        [0.0677, 0.9323],
        [0.0349, 0.9651],
        [0.0726, 0.9274],
        [0.0342, 0.9658],
        [0.0411, 0.9589],
        [0.0335, 0.9665]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0483, 0.9517],
        [0.1001, 0.8999],
        [0.0423, 0.9577],
        [0.0553, 0.9447],
        [0.0606, 0.9394],
        [0.0728, 0.9272],
        [0.0389, 0.9611],
        [0.0402, 0.9598],
        [0.0407, 0.9593],
        [0.0487, 0.9513],
        [0.0867, 0.9133],
        [0.0457, 0.9543],
        [0.0403, 0.9597],
        [0.0377, 0.9623],
        [0.0791, 0.9209],
        [0.0349, 0.9651],
        [0.0390, 0.9610],
        [0.0285, 0.9715],
        [0.0358, 0.9642],
        [0.0361, 0.9639],
        [0.0421, 0.9579],
        [0.0405, 0.9595],
        [0.0608, 0.9392],
        [0.0584, 0.9416],
        [0.0525, 0.9475],
        [0.0482, 0.9518],
        [0.0237, 0.9763],
        [0.0561, 0.9439],
        [0.0522, 0.9478],
        [0.0574, 0.9426],
        [0.0288, 0.9712],
        [0.0298, 0.9702],
        [0.0619, 0.9381],
        [0.0708, 0.9292],
        [0.0221, 0.9779],
        [0.0637, 0.9363],
        [0.1480, 0.8520],
        [0.1043, 0.8957],
        [0.0689, 0.9311],
        [0.0299, 0.9701],
        [0.0750, 0.9250],
        [0.0288, 0.9712],
        [0.0341, 0.9659],
        [0.0398, 0.9602],
        [0.0960, 0.9040],
        [0.0358, 0.9642],
        [0.0659, 0.9341],
        [0.0385, 0.9615],
        [0.0449, 0.9551],
        [0.0393, 0.9607],
        [0.0639, 0.9361],
        [0.0417, 0.9583],
        [0.0618, 0.9382],
        [0.0419, 0.9581],
        [0.0455, 0.9545],
        [0.0841, 0.9159],
        [0.0943, 0.9057],
        [0.0419, 0.9581],
        [0.0350, 0.9650],
        [0.0551, 0.9449],
        [0.0743, 0.9257],
        [0.0340, 0.9660],
        [0.0569, 0.9431],
        [0.0657, 0.9343]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1209, 0.8791],
        [0.0626, 0.9374],
        [0.0844, 0.9156],
        [0.1534, 0.8466],
        [0.1128, 0.8872],
        [0.0753, 0.9247],
        [0.0652, 0.9348],
        [0.0545, 0.9455],
        [0.0376, 0.9624],
        [0.0670, 0.9330],
        [0.0555, 0.9445],
        [0.0491, 0.9509],
        [0.0438, 0.9562],
        [0.0416, 0.9584],
        [0.0466, 0.9534],
        [0.1406, 0.8594],
        [0.0464, 0.9536],
        [0.0609, 0.9391],
        [0.0500, 0.9500],
        [0.0505, 0.9495],
        [0.0781, 0.9219],
        [0.0583, 0.9417],
        [0.1427, 0.8573],
        [0.0647, 0.9353],
        [0.0614, 0.9386],
        [0.0950, 0.9050],
        [0.0617, 0.9383],
        [0.1212, 0.8788],
        [0.0570, 0.9430],
        [0.0544, 0.9456],
        [0.0510, 0.9490],
        [0.0908, 0.9092],
        [0.0410, 0.9590],
        [0.0856, 0.9144],
        [0.0411, 0.9589],
        [0.0455, 0.9545],
        [0.0393, 0.9607],
        [0.0638, 0.9362],
        [0.0629, 0.9371],
        [0.0479, 0.9521],
        [0.0497, 0.9503],
        [0.0447, 0.9553],
        [0.0420, 0.9580],
        [0.0469, 0.9531],
        [0.0460, 0.9540],
        [0.0431, 0.9569],
        [0.0591, 0.9409],
        [0.0562, 0.9438],
        [0.1953, 0.8047],
        [0.0355, 0.9645],
        [0.0926, 0.9074],
        [0.0428, 0.9572],
        [0.0452, 0.9548],
        [0.0577, 0.9423],
        [0.0927, 0.9073],
        [0.0524, 0.9476],
        [0.0521, 0.9479],
        [0.1326, 0.8674],
        [0.0663, 0.9337],
        [0.0545, 0.9455],
        [0.0755, 0.9245],
        [0.1087, 0.8913],
        [0.0906, 0.9094],
        [0.0709, 0.9291]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0681, 0.9319],
        [0.1653, 0.8347],
        [0.1044, 0.8956],
        [0.0731, 0.9269],
        [0.0535, 0.9465],
        [0.0790, 0.9210],
        [0.0832, 0.9168],
        [0.0562, 0.9438],
        [0.1630, 0.8370],
        [0.1567, 0.8433],
        [0.1087, 0.8913],
        [0.0515, 0.9485],
        [0.0647, 0.9353],
        [0.0644, 0.9356],
        [0.1393, 0.8607],
        [0.0558, 0.9442],
        [0.0666, 0.9334],
        [0.1120, 0.8880],
        [0.1567, 0.8433],
        [0.0723, 0.9277],
        [0.0537, 0.9463],
        [0.0585, 0.9415],
        [0.0744, 0.9256],
        [0.0827, 0.9173],
        [0.0764, 0.9236],
        [0.0535, 0.9465],
        [0.0847, 0.9153],
        [0.1073, 0.8927],
        [0.1646, 0.8354],
        [0.0535, 0.9465],
        [0.0554, 0.9446],
        [0.0618, 0.9382],
        [0.1390, 0.8610],
        [0.0711, 0.9289],
        [0.1668, 0.8332],
        [0.0868, 0.9132],
        [0.0380, 0.9620],
        [0.1463, 0.8537],
        [0.1393, 0.8607],
        [0.1370, 0.8630],
        [0.0725, 0.9275],
        [0.1190, 0.8810],
        [0.0565, 0.9435],
        [0.1412, 0.8588],
        [0.1540, 0.8460],
        [0.0825, 0.9175],
        [0.1203, 0.8797],
        [0.1503, 0.8497],
        [0.0408, 0.9592],
        [0.0718, 0.9282],
        [0.0830, 0.9170],
        [0.1438, 0.8562],
        [0.0692, 0.9309],
        [0.0740, 0.9260],
        [0.1266, 0.8734],
        [0.1059, 0.8941],
        [0.0768, 0.9232],
        [0.0409, 0.9591],
        [0.0563, 0.9437],
        [0.0474, 0.9526],
        [0.1645, 0.8355],
        [0.0853, 0.9147],
        [0.0853, 0.9147],
        [0.0766, 0.9234]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1064, 0.8936],
        [0.0943, 0.9057],
        [0.0950, 0.9050],
        [0.1736, 0.8264],
        [0.0745, 0.9255],
        [0.0638, 0.9362],
        [0.0759, 0.9241],
        [0.0825, 0.9175],
        [0.0723, 0.9277],
        [0.0552, 0.9448],
        [0.0866, 0.9134],
        [0.1238, 0.8762],
        [0.0553, 0.9447],
        [0.0616, 0.9384],
        [0.1205, 0.8795],
        [0.0964, 0.9036],
        [0.0630, 0.9370],
        [0.1219, 0.8781],
        [0.1592, 0.8408],
        [0.0856, 0.9144],
        [0.0863, 0.9137],
        [0.1230, 0.8770],
        [0.0758, 0.9242],
        [0.1471, 0.8529],
        [0.1013, 0.8987],
        [0.0776, 0.9224],
        [0.2689, 0.7311],
        [0.1084, 0.8916],
        [0.2146, 0.7854],
        [0.0789, 0.9211],
        [0.1214, 0.8786],
        [0.0769, 0.9231],
        [0.0726, 0.9274],
        [0.0845, 0.9155],
        [0.1311, 0.8689],
        [0.0935, 0.9065],
        [0.1350, 0.8650],
        [0.2415, 0.7585],
        [0.2030, 0.7970],
        [0.0956, 0.9044],
        [0.1879, 0.8121],
        [0.0849, 0.9151],
        [0.0792, 0.9208],
        [0.2150, 0.7850],
        [0.2746, 0.7254],
        [0.0878, 0.9122],
        [0.2240, 0.7760],
        [0.2482, 0.7518],
        [0.0979, 0.9021],
        [0.1377, 0.8623],
        [0.1095, 0.8905],
        [0.2658, 0.7342],
        [0.0928, 0.9072],
        [0.0860, 0.9140],
        [0.0926, 0.9074],
        [0.0750, 0.9250],
        [0.0902, 0.9098],
        [0.2541, 0.7459],
        [0.2620, 0.7380],
        [0.0967, 0.9033],
        [0.4180, 0.5820],
        [0.1063, 0.8937],
        [0.0797, 0.9203],
        [0.1038, 0.8962]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1348, 0.8652],
        [0.1428, 0.8572],
        [0.1069, 0.8931],
        [0.1303, 0.8697],
        [0.3008, 0.6992],
        [0.1342, 0.8658],
        [0.1291, 0.8709],
        [0.1854, 0.8146],
        [0.2203, 0.7797],
        [0.2891, 0.7109],
        [0.0980, 0.9020],
        [0.1603, 0.8397],
        [0.1381, 0.8619],
        [0.0853, 0.9147],
        [0.1727, 0.8273],
        [0.3695, 0.6305],
        [0.0894, 0.9106],
        [0.1106, 0.8894],
        [0.1338, 0.8662],
        [0.2618, 0.7382],
        [0.2600, 0.7400],
        [0.1124, 0.8876],
        [0.1571, 0.8429],
        [0.0863, 0.9137],
        [0.1207, 0.8793],
        [0.1084, 0.8916],
        [0.1030, 0.8970],
        [0.1556, 0.8444],
        [0.0894, 0.9106],
        [0.3626, 0.6374],
        [0.3685, 0.6315],
        [0.3576, 0.6424],
        [0.2609, 0.7391],
        [0.2296, 0.7704],
        [0.2122, 0.7878],
        [0.1120, 0.8880],
        [0.1777, 0.8223],
        [0.1103, 0.8897],
        [0.1390, 0.8610],
        [0.1628, 0.8372],
        [0.0998, 0.9002],
        [0.2771, 0.7229],
        [0.1517, 0.8483],
        [0.1390, 0.8610],
        [0.1525, 0.8475],
        [0.1439, 0.8561],
        [0.0982, 0.9018],
        [0.2262, 0.7738],
        [0.1122, 0.8878],
        [0.1112, 0.8888],
        [0.1941, 0.8059],
        [0.1393, 0.8607],
        [0.1116, 0.8884],
        [0.1369, 0.8631],
        [0.1114, 0.8886],
        [0.1753, 0.8247],
        [0.1114, 0.8886],
        [0.1549, 0.8451],
        [0.1065, 0.8935],
        [0.1011, 0.8989],
        [0.1164, 0.8836],
        [0.1332, 0.8668],
        [0.1061, 0.8939],
        [0.2183, 0.7817]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1345, 0.8655],
        [0.1244, 0.8756],
        [0.0991, 0.9009],
        [0.1924, 0.8076],
        [0.2684, 0.7316],
        [0.6916, 0.3084],
        [0.4160, 0.5840],
        [0.2009, 0.7991],
        [0.1921, 0.8079],
        [0.1374, 0.8626],
        [0.3978, 0.6022],
        [0.1281, 0.8719],
        [0.1558, 0.8442],
        [0.1190, 0.8810],
        [0.1995, 0.8005],
        [0.3372, 0.6628],
        [0.5116, 0.4884],
        [0.3541, 0.6459],
        [0.1158, 0.8842],
        [0.2310, 0.7690],
        [0.1158, 0.8842],
        [0.3027, 0.6973],
        [0.1808, 0.8192],
        [0.1323, 0.8677],
        [0.1759, 0.8241],
        [0.1841, 0.8159],
        [0.1354, 0.8646],
        [0.1329, 0.8671],
        [0.1382, 0.8618],
        [0.1974, 0.8026],
        [0.1948, 0.8052],
        [0.0948, 0.9052],
        [0.3837, 0.6163],
        [0.3981, 0.6019],
        [0.1578, 0.8422],
        [0.1558, 0.8442],
        [0.0811, 0.9189],
        [0.1818, 0.8182],
        [0.1261, 0.8739],
        [0.1185, 0.8815],
        [0.3653, 0.6347],
        [0.1621, 0.8379],
        [0.1790, 0.8210],
        [0.1079, 0.8921],
        [0.1452, 0.8548],
        [0.0849, 0.9151],
        [0.3631, 0.6369],
        [0.2698, 0.7302],
        [0.1465, 0.8535],
        [0.1426, 0.8574],
        [0.1154, 0.8846],
        [0.2190, 0.7810],
        [0.1737, 0.8263],
        [0.1001, 0.8999],
        [0.1694, 0.8306],
        [0.1054, 0.8946],
        [0.1589, 0.8411],
        [0.1607, 0.8393],
        [0.3222, 0.6778],
        [0.1967, 0.8033],
        [0.1247, 0.8753],
        [0.1566, 0.8434],
        [0.1798, 0.8202],
        [0.2246, 0.7754]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1598, 0.8402],
        [0.1566, 0.8434],
        [0.1262, 0.8738],
        [0.1407, 0.8593],
        [0.1641, 0.8359],
        [0.1374, 0.8626],
        [0.2092, 0.7908],
        [0.2708, 0.7292],
        [0.1483, 0.8517],
        [0.3862, 0.6138],
        [0.1243, 0.8757],
        [0.1850, 0.8150],
        [0.0991, 0.9009],
        [0.1972, 0.8028],
        [0.3336, 0.6664],
        [0.1634, 0.8366],
        [0.1796, 0.8204],
        [0.1755, 0.8245],
        [0.3081, 0.6919],
        [0.2071, 0.7929],
        [0.1063, 0.8937],
        [0.1900, 0.8100],
        [0.1191, 0.8809],
        [0.1127, 0.8873],
        [0.2000, 0.8000],
        [0.2384, 0.7616],
        [0.4557, 0.5443],
        [0.1267, 0.8733],
        [0.2079, 0.7921],
        [0.1642, 0.8358],
        [0.1768, 0.8232],
        [0.2196, 0.7804],
        [0.2708, 0.7292],
        [0.6693, 0.3307],
        [0.1853, 0.8147],
        [0.2076, 0.7924],
        [0.1705, 0.8295],
        [0.1465, 0.8535],
        [0.3794, 0.6206],
        [0.1733, 0.8267],
        [0.3961, 0.6039],
        [0.1920, 0.8080],
        [0.1221, 0.8779],
        [0.1398, 0.8602],
        [0.1731, 0.8269],
        [0.1526, 0.8474],
        [0.1894, 0.8106],
        [0.4197, 0.5803],
        [0.1561, 0.8439],
        [0.1334, 0.8666],
        [0.1857, 0.8143],
        [0.1201, 0.8799],
        [0.2145, 0.7855],
        [0.2447, 0.7553],
        [0.1707, 0.8293],
        [0.1622, 0.8378],
        [0.1147, 0.8853],
        [0.2033, 0.7967],
        [0.4033, 0.5967],
        [0.4092, 0.5908],
        [0.1860, 0.8140],
        [0.1935, 0.8065],
        [0.1897, 0.8103],
        [0.1654, 0.8346]], grad_fn=<SoftmaxBackward0>)
tensor([[0.3825, 0.6175],
        [0.1644, 0.8356],
        [0.3236, 0.6764],
        [0.5148, 0.4852],
        [0.1022, 0.8978],
        [0.1022, 0.8978],
        [0.1125, 0.8875],
        [0.1119, 0.8881],
        [0.1364, 0.8636],
        [0.1677, 0.8323],
        [0.1591, 0.8409],
        [0.1625, 0.8375],
        [0.3593, 0.6407],
        [0.1575, 0.8425],
        [0.1731, 0.8269],
        [0.2325, 0.7675],
        [0.1779, 0.8221],
        [0.1962, 0.8038],
        [0.1928, 0.8072],
        [0.3243, 0.6757],
        [0.1854, 0.8146],
        [0.1134, 0.8866],
        [0.2949, 0.7051],
        [0.1484, 0.8516],
        [0.1316, 0.8684],
        [0.2626, 0.7374],
        [0.1032, 0.8968],
        [0.1305, 0.8695],
        [0.1833, 0.8167],
        [0.3190, 0.6810],
        [0.4160, 0.5840],
        [0.3115, 0.6885],
        [0.4383, 0.5617],
        [0.1336, 0.8664],
        [0.1229, 0.8771],
        [0.2483, 0.7517],
        [0.1291, 0.8709],
        [0.1092, 0.8908],
        [0.3480, 0.6520],
        [0.1116, 0.8884],
        [0.1312, 0.8688],
        [0.3959, 0.6041],
        [0.3214, 0.6786],
        [0.1622, 0.8378],
        [0.0247, 0.9753],
        [0.1310, 0.8690],
        [0.1592, 0.8408],
        [0.2980, 0.7020],
        [0.1724, 0.8276],
        [0.1227, 0.8773],
        [0.2838, 0.7162],
        [0.1301, 0.8699],
        [0.6172, 0.3828],
        [0.2318, 0.7682],
        [0.1760, 0.8240],
        [0.1742, 0.8258],
        [0.2017, 0.7983],
        [0.0644, 0.9356],
        [0.1333, 0.8667],
        [0.1328, 0.8672],
        [0.1389, 0.8611],
        [0.1514, 0.8486],
        [0.1720, 0.8280],
        [0.3097, 0.6903]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1663, 0.8337],
        [0.1102, 0.8898],
        [0.1201, 0.8799],
        [0.1034, 0.8966],
        [0.1740, 0.8260],
        [0.1132, 0.8868],
        [0.0898, 0.9102],
        [0.1300, 0.8700],
        [0.4374, 0.5626],
        [0.1543, 0.8457],
        [0.3849, 0.6151],
        [0.2849, 0.7151],
        [0.0925, 0.9075],
        [0.1094, 0.8906],
        [0.1506, 0.8494],
        [0.1005, 0.8995],
        [0.1308, 0.8692],
        [0.1271, 0.8729],
        [0.0860, 0.9140],
        [0.1017, 0.8983],
        [0.1502, 0.8498],
        [0.0763, 0.9237],
        [0.1376, 0.8624],
        [0.1471, 0.8529],
        [0.1318, 0.8682],
        [0.1628, 0.8372],
        [0.3063, 0.6937],
        [0.1333, 0.8667],
        [0.6774, 0.3226],
        [0.1851, 0.8149],
        [0.1735, 0.8265],
        [0.0748, 0.9252],
        [0.1429, 0.8571],
        [0.1888, 0.8112],
        [0.1300, 0.8700],
        [0.1180, 0.8820],
        [0.4890, 0.5110],
        [0.4123, 0.5877],
        [0.1543, 0.8457],
        [0.4867, 0.5133],
        [0.1383, 0.8617],
        [0.0881, 0.9119],
        [0.1630, 0.8370],
        [0.1411, 0.8589],
        [0.1954, 0.8046],
        [0.6071, 0.3929],
        [0.1377, 0.8623],
        [0.0525, 0.9475],
        [0.0648, 0.9352],
        [0.3620, 0.6380],
        [0.1600, 0.8400],
        [0.3739, 0.6261],
        [0.2011, 0.7989],
        [0.0854, 0.9146],
        [0.2423, 0.7577],
        [0.3878, 0.6122],
        [0.1180, 0.8820],
        [0.1713, 0.8287],
        [0.1202, 0.8798],
        [0.0562, 0.9438],
        [0.4291, 0.5709],
        [0.1145, 0.8855],
        [0.1473, 0.8527],
        [0.0936, 0.9064]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1235, 0.8765],
        [0.0958, 0.9042],
        [0.2463, 0.7537],
        [0.1021, 0.8979],
        [0.1596, 0.8404],
        [0.0985, 0.9015],
        [0.1573, 0.8427],
        [0.0620, 0.9380],
        [0.0829, 0.9171],
        [0.0903, 0.9097],
        [0.1025, 0.8975],
        [0.1047, 0.8953],
        [0.1112, 0.8888],
        [0.0722, 0.9278],
        [0.3825, 0.6175],
        [0.0597, 0.9403],
        [0.1311, 0.8689],
        [0.0886, 0.9114],
        [0.1065, 0.8935],
        [0.0872, 0.9128],
        [0.0580, 0.9420],
        [0.0901, 0.9099],
        [0.3702, 0.6298],
        [0.3539, 0.6461],
        [0.0700, 0.9300],
        [0.1547, 0.8453],
        [0.3179, 0.6821],
        [0.0944, 0.9056],
        [0.1424, 0.8576],
        [0.4442, 0.5558],
        [0.1237, 0.8763],
        [0.0810, 0.9190],
        [0.1577, 0.8423],
        [0.1546, 0.8454],
        [0.0771, 0.9229],
        [0.1983, 0.8017],
        [0.0829, 0.9171],
        [0.1081, 0.8919],
        [0.4759, 0.5241],
        [0.2804, 0.7196],
        [0.0982, 0.9018],
        [0.0867, 0.9133],
        [0.0592, 0.9408],
        [0.0703, 0.9297],
        [0.1366, 0.8634],
        [0.0596, 0.9404],
        [0.1907, 0.8093],
        [0.0644, 0.9356],
        [0.2264, 0.7736],
        [0.0828, 0.9172],
        [0.1696, 0.8304],
        [0.2507, 0.7493],
        [0.1463, 0.8537],
        [0.3353, 0.6647],
        [0.0575, 0.9425],
        [0.1177, 0.8823],
        [0.2815, 0.7185],
        [0.0915, 0.9085],
        [0.0896, 0.9104],
        [0.4449, 0.5551],
        [0.1475, 0.8525],
        [0.1359, 0.8641],
        [0.3020, 0.6980],
        [0.0665, 0.9335]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1200, 0.8800],
        [0.0599, 0.9401],
        [0.1112, 0.8888],
        [0.3850, 0.6150],
        [0.4242, 0.5758],
        [0.3444, 0.6556],
        [0.0782, 0.9218],
        [0.0927, 0.9073],
        [0.2530, 0.7470],
        [0.1358, 0.8642],
        [0.3617, 0.6383],
        [0.0828, 0.9172],
        [0.0841, 0.9159],
        [0.0688, 0.9312],
        [0.0766, 0.9234],
        [0.0616, 0.9384],
        [0.0573, 0.9427],
        [0.0717, 0.9283],
        [0.3781, 0.6219],
        [0.8341, 0.1659],
        [0.1033, 0.8967],
        [0.4435, 0.5565],
        [0.1282, 0.8718],
        [0.1883, 0.8117],
        [0.0941, 0.9059],
        [0.0567, 0.9433],
        [0.0838, 0.9162],
        [0.3158, 0.6842],
        [0.4674, 0.5326],
        [0.0970, 0.9030],
        [0.3730, 0.6270],
        [0.3088, 0.6912],
        [0.0645, 0.9355],
        [0.0956, 0.9044],
        [0.1261, 0.8739],
        [0.0869, 0.9131],
        [0.1040, 0.8960],
        [0.0704, 0.9296],
        [0.0572, 0.9428],
        [0.5047, 0.4953],
        [0.3110, 0.6890],
        [0.1370, 0.8630],
        [0.1328, 0.8672],
        [0.0404, 0.9596],
        [0.1319, 0.8681],
        [0.0782, 0.9218],
        [0.0858, 0.9142],
        [0.0596, 0.9404],
        [0.0685, 0.9315],
        [0.2282, 0.7718],
        [0.1297, 0.8703],
        [0.0918, 0.9082],
        [0.1916, 0.8084],
        [0.2996, 0.7004],
        [0.1402, 0.8598],
        [0.2539, 0.7461],
        [0.4209, 0.5791],
        [0.2122, 0.7878],
        [0.1074, 0.8926],
        [0.3785, 0.6215],
        [0.0868, 0.9132],
        [0.0933, 0.9067],
        [0.0671, 0.9329],
        [0.0775, 0.9225]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1205, 0.8795],
        [0.3082, 0.6918],
        [0.1145, 0.8855],
        [0.1644, 0.8356],
        [0.0940, 0.9060],
        [0.3978, 0.6022],
        [0.1463, 0.8537],
        [0.0358, 0.9642],
        [0.2606, 0.7394],
        [0.0399, 0.9601],
        [0.1193, 0.8807],
        [0.1439, 0.8561],
        [0.4552, 0.5448],
        [0.1119, 0.8881],
        [0.1423, 0.8577],
        [0.1211, 0.8789],
        [0.1395, 0.8605],
        [0.0739, 0.9261],
        [0.1768, 0.8232],
        [0.3676, 0.6324],
        [0.1133, 0.8867],
        [0.1045, 0.8955],
        [0.0651, 0.9349],
        [0.1119, 0.8881],
        [0.0753, 0.9247],
        [0.0787, 0.9213],
        [0.5959, 0.4041],
        [0.1098, 0.8902],
        [0.3370, 0.6630],
        [0.0829, 0.9171],
        [0.2073, 0.7927],
        [0.6190, 0.3810],
        [0.1426, 0.8574],
        [0.2300, 0.7700],
        [0.0838, 0.9162],
        [0.0453, 0.9547],
        [0.0955, 0.9045],
        [0.1626, 0.8374],
        [0.1112, 0.8888],
        [0.0983, 0.9017],
        [0.2456, 0.7544],
        [0.0833, 0.9167],
        [0.0591, 0.9409],
        [0.0564, 0.9436],
        [0.3614, 0.6386],
        [0.1139, 0.8861],
        [0.0649, 0.9351],
        [0.0792, 0.9208],
        [0.0916, 0.9084],
        [0.0958, 0.9042],
        [0.1343, 0.8657],
        [0.0727, 0.9273],
        [0.0433, 0.9567],
        [0.0521, 0.9479],
        [0.2917, 0.7083],
        [0.1100, 0.8900],
        [0.1780, 0.8220],
        [0.1474, 0.8526],
        [0.0809, 0.9191],
        [0.1330, 0.8670],
        [0.0765, 0.9235],
        [0.4314, 0.5686],
        [0.1289, 0.8711],
        [0.1390, 0.8610]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0391, 0.9609],
        [0.3412, 0.6588],
        [0.0557, 0.9443],
        [0.1293, 0.8707],
        [0.4145, 0.5855],
        [0.0960, 0.9040],
        [0.1200, 0.8800],
        [0.0698, 0.9302],
        [0.1803, 0.8197],
        [0.0506, 0.9494],
        [0.1544, 0.8456],
        [0.1775, 0.8225],
        [0.0575, 0.9425],
        [0.0578, 0.9422],
        [0.1058, 0.8942],
        [0.3960, 0.6040],
        [0.1170, 0.8830],
        [0.5888, 0.4112],
        [0.1397, 0.8603],
        [0.0495, 0.9505],
        [0.1229, 0.8771],
        [0.1420, 0.8580],
        [0.0628, 0.9372],
        [0.0764, 0.9236],
        [0.0746, 0.9254],
        [0.1312, 0.8688],
        [0.0913, 0.9087],
        [0.1041, 0.8959],
        [0.0563, 0.9437],
        [0.0920, 0.9080],
        [0.0392, 0.9608],
        [0.0741, 0.9259],
        [0.1861, 0.8139],
        [0.0827, 0.9173],
        [0.1081, 0.8919],
        [0.3481, 0.6519],
        [0.0573, 0.9427],
        [0.1779, 0.8221],
        [0.0605, 0.9395],
        [0.1422, 0.8578],
        [0.4253, 0.5747],
        [0.1360, 0.8640],
        [0.1783, 0.8217],
        [0.0556, 0.9444],
        [0.1033, 0.8967],
        [0.4499, 0.5501],
        [0.1223, 0.8777],
        [0.0954, 0.9046],
        [0.0573, 0.9427],
        [0.1205, 0.8795],
        [0.1663, 0.8337],
        [0.0675, 0.9325],
        [0.1444, 0.8556],
        [0.0887, 0.9113],
        [0.1006, 0.8994],
        [0.0541, 0.9459],
        [0.3859, 0.6141],
        [0.2217, 0.7783],
        [0.1174, 0.8826],
        [0.2347, 0.7653],
        [0.0906, 0.9094],
        [0.1058, 0.8942],
        [0.2453, 0.7547],
        [0.5822, 0.4178]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1810, 0.8190],
        [0.3412, 0.6588],
        [0.0580, 0.9420],
        [0.0999, 0.9001],
        [0.0962, 0.9038],
        [0.0528, 0.9472],
        [0.4930, 0.5070],
        [0.0440, 0.9560],
        [0.3485, 0.6515],
        [0.1445, 0.8555],
        [0.0725, 0.9275],
        [0.1200, 0.8800],
        [0.3787, 0.6213],
        [0.0733, 0.9267],
        [0.0671, 0.9329],
        [0.2866, 0.7134],
        [0.0606, 0.9394],
        [0.0791, 0.9209],
        [0.6682, 0.3318],
        [0.3410, 0.6590],
        [0.5052, 0.4948],
        [0.0383, 0.9617],
        [0.1178, 0.8822],
        [0.6333, 0.3667],
        [0.0872, 0.9128],
        [0.2445, 0.7555],
        [0.0739, 0.9261],
        [0.0711, 0.9289],
        [0.1184, 0.8816],
        [0.1870, 0.8130],
        [0.0795, 0.9205],
        [0.5645, 0.4355],
        [0.1748, 0.8252],
        [0.4112, 0.5888],
        [0.5223, 0.4777],
        [0.4124, 0.5876],
        [0.0722, 0.9278],
        [0.0492, 0.9508],
        [0.0753, 0.9247],
        [0.1079, 0.8921],
        [0.0775, 0.9225],
        [0.1455, 0.8545],
        [0.1372, 0.8628],
        [0.1398, 0.8602],
        [0.1873, 0.8127],
        [0.3224, 0.6776],
        [0.4290, 0.5710],
        [0.2670, 0.7330],
        [0.0714, 0.9286],
        [0.4799, 0.5201],
        [0.0577, 0.9423],
        [0.0831, 0.9169],
        [0.1524, 0.8476],
        [0.5871, 0.4129],
        [0.0569, 0.9431],
        [0.0699, 0.9301],
        [0.0715, 0.9285],
        [0.1050, 0.8950],
        [0.0970, 0.9030],
        [0.1297, 0.8703],
        [0.1713, 0.8287],
        [0.1280, 0.8720],
        [0.1271, 0.8729],
        [0.0467, 0.9533]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1530, 0.8470],
        [0.1004, 0.8996],
        [0.1033, 0.8967],
        [0.3044, 0.6956],
        [0.0767, 0.9233],
        [0.6204, 0.3796],
        [0.1229, 0.8771],
        [0.1699, 0.8301],
        [0.0541, 0.9459],
        [0.1526, 0.8474],
        [0.2817, 0.7183],
        [0.6425, 0.3575],
        [0.0821, 0.9179],
        [0.0695, 0.9305],
        [0.0599, 0.9401],
        [0.0897, 0.9103],
        [0.1149, 0.8851],
        [0.4480, 0.5520],
        [0.0768, 0.9232],
        [0.1256, 0.8744],
        [0.5704, 0.4296],
        [0.3595, 0.6405],
        [0.4473, 0.5527],
        [0.0547, 0.9453],
        [0.4077, 0.5923],
        [0.0579, 0.9421],
        [0.0703, 0.9297],
        [0.1481, 0.8519],
        [0.2677, 0.7323],
        [0.0840, 0.9160],
        [0.1367, 0.8633],
        [0.4489, 0.5511],
        [0.0938, 0.9062],
        [0.4596, 0.5404],
        [0.0589, 0.9411],
        [0.0962, 0.9038],
        [0.0642, 0.9358],
        [0.1240, 0.8760],
        [0.3911, 0.6089],
        [0.1241, 0.8759],
        [0.4618, 0.5382],
        [0.1027, 0.8973],
        [0.2013, 0.7987],
        [0.1295, 0.8705],
        [0.1268, 0.8732],
        [0.3749, 0.6251],
        [0.3976, 0.6024],
        [0.0950, 0.9050],
        [0.5435, 0.4565],
        [0.0607, 0.9393],
        [0.1465, 0.8535],
        [0.0553, 0.9447],
        [0.1160, 0.8840],
        [0.4608, 0.5392],
        [0.1284, 0.8716],
        [0.2200, 0.7800],
        [0.2897, 0.7103],
        [0.1464, 0.8536],
        [0.0573, 0.9427],
        [0.0735, 0.9265],
        [0.0315, 0.9685],
        [0.0981, 0.9019],
        [0.5507, 0.4493],
        [0.1716, 0.8284]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0947, 0.9053],
        [0.1272, 0.8728],
        [0.1842, 0.8158],
        [0.6325, 0.3675],
        [0.1083, 0.8917],
        [0.1580, 0.8420],
        [0.6182, 0.3818],
        [0.2069, 0.7931],
        [0.1024, 0.8976],
        [0.2009, 0.7991],
        [0.4992, 0.5008],
        [0.4332, 0.5668],
        [0.1000, 0.9000],
        [0.2277, 0.7723],
        [0.1718, 0.8282],
        [0.2072, 0.7928],
        [0.0968, 0.9032],
        [0.6195, 0.3805],
        [0.5616, 0.4384],
        [0.0906, 0.9094],
        [0.1696, 0.8304],
        [0.0773, 0.9227],
        [0.1619, 0.8381],
        [0.1316, 0.8684],
        [0.6966, 0.3034],
        [0.1644, 0.8356],
        [0.5080, 0.4920],
        [0.0737, 0.9263],
        [0.6243, 0.3757],
        [0.6357, 0.3643],
        [0.8622, 0.1378],
        [0.0858, 0.9142],
        [0.0777, 0.9223],
        [0.3766, 0.6234],
        [0.1150, 0.8850],
        [0.6412, 0.3588],
        [0.0896, 0.9104],
        [0.1497, 0.8503],
        [0.1075, 0.8925],
        [0.0704, 0.9296],
        [0.0807, 0.9193],
        [0.1258, 0.8742],
        [0.6740, 0.3260],
        [0.1490, 0.8510],
        [0.1299, 0.8701],
        [0.1335, 0.8665],
        [0.2105, 0.7895],
        [0.3934, 0.6066],
        [0.0919, 0.9081],
        [0.2009, 0.7991],
        [0.1009, 0.8991],
        [0.1465, 0.8535],
        [0.1947, 0.8053],
        [0.5005, 0.4995],
        [0.6079, 0.3921],
        [0.0584, 0.9416],
        [0.6697, 0.3303],
        [0.1299, 0.8701],
        [0.2281, 0.7719],
        [0.1000, 0.9000],
        [0.1405, 0.8595],
        [0.2507, 0.7493],
        [0.5026, 0.4974],
        [0.1446, 0.8554]], grad_fn=<SoftmaxBackward0>)
tensor([[0.2713, 0.7287],
        [0.1496, 0.8504],
        [0.2949, 0.7051],
        [0.2513, 0.7487],
        [0.2353, 0.7647],
        [0.2096, 0.7904],
        [0.0761, 0.9239],
        [0.0708, 0.9292],
        [0.0808, 0.9192],
        [0.1423, 0.8577],
        [0.1491, 0.8509],
        [0.0757, 0.9243],
        [0.1546, 0.8454],
        [0.3373, 0.6627],
        [0.5631, 0.4369],
        [0.1370, 0.8630],
        [0.1754, 0.8246],
        [0.0914, 0.9086],
        [0.2024, 0.7976],
        [0.1390, 0.8610],
        [0.5963, 0.4037],
        [0.3979, 0.6021],
        [0.6778, 0.3222],
        [0.1504, 0.8496],
        [0.1655, 0.8345],
        [0.3205, 0.6795],
        [0.1057, 0.8943],
        [0.1805, 0.8195],
        [0.1199, 0.8801],
        [0.1466, 0.8534],
        [0.0782, 0.9218],
        [0.0654, 0.9346],
        [0.0758, 0.9242],
        [0.6233, 0.3767],
        [0.1409, 0.8591],
        [0.5064, 0.4936],
        [0.1580, 0.8420],
        [0.8269, 0.1731],
        [0.1851, 0.8149],
        [0.0662, 0.9338],
        [0.6594, 0.3406],
        [0.1600, 0.8400],
        [0.0548, 0.9452],
        [0.1203, 0.8797],
        [0.2864, 0.7136],
        [0.1679, 0.8321],
        [0.2271, 0.7729],
        [0.1555, 0.8445],
        [0.4292, 0.5708],
        [0.0832, 0.9168],
        [0.6962, 0.3038],
        [0.0537, 0.9463],
        [0.0758, 0.9242],
        [0.4758, 0.5242],
        [0.1937, 0.8063],
        [0.2404, 0.7596],
        [0.7574, 0.2426],
        [0.2201, 0.7799],
        [0.1024, 0.8976],
        [0.3806, 0.6194],
        [0.5789, 0.4211],
        [0.0763, 0.9237],
        [0.1888, 0.8112],
        [0.1649, 0.8351]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1410, 0.8590],
        [0.0806, 0.9194],
        [0.6458, 0.3542],
        [0.6596, 0.3404],
        [0.0872, 0.9128],
        [0.6748, 0.3252],
        [0.1962, 0.8038],
        [0.4230, 0.5770],
        [0.1643, 0.8357],
        [0.0927, 0.9073],
        [0.1776, 0.8224],
        [0.2012, 0.7988],
        [0.0776, 0.9224],
        [0.6833, 0.3167],
        [0.2496, 0.7504],
        [0.0873, 0.9127],
        [0.4894, 0.5106],
        [0.0931, 0.9069],
        [0.1851, 0.8149],
        [0.1812, 0.8188],
        [0.0717, 0.9283],
        [0.3122, 0.6878],
        [0.1812, 0.8188],
        [0.5898, 0.4102],
        [0.2148, 0.7852],
        [0.0884, 0.9116],
        [0.7992, 0.2008],
        [0.1474, 0.8526],
        [0.1563, 0.8437],
        [0.0597, 0.9403],
        [0.0700, 0.9300],
        [0.6665, 0.3335],
        [0.1859, 0.8141],
        [0.0696, 0.9304],
        [0.5980, 0.4020],
        [0.6069, 0.3931],
        [0.5362, 0.4638],
        [0.1822, 0.8178],
        [0.7090, 0.2910],
        [0.1536, 0.8464],
        [0.2553, 0.7447],
        [0.0908, 0.9092],
        [0.5763, 0.4237],
        [0.2287, 0.7713],
        [0.7695, 0.2305],
        [0.0687, 0.9313],
        [0.7762, 0.2238],
        [0.1050, 0.8950],
        [0.5601, 0.4399],
        [0.6964, 0.3036],
        [0.2560, 0.7440],
        [0.0544, 0.9456],
        [0.1409, 0.8591],
        [0.1796, 0.8204],
        [0.0769, 0.9231],
        [0.1983, 0.8017],
        [0.1273, 0.8727],
        [0.5813, 0.4187],
        [0.1101, 0.8899],
        [0.1740, 0.8260],
        [0.1744, 0.8256],
        [0.0560, 0.9440],
        [0.7967, 0.2033],
        [0.4942, 0.5058]], grad_fn=<SoftmaxBackward0>)
tensor([[0.3450, 0.6550],
        [0.2408, 0.7592],
        [0.1573, 0.8427],
        [0.6314, 0.3686],
        [0.1005, 0.8995],
        [0.4967, 0.5033],
        [0.1024, 0.8976],
        [0.8589, 0.1411],
        [0.2316, 0.7684],
        [0.0597, 0.9403],
        [0.6078, 0.3922],
        [0.1568, 0.8432],
        [0.0437, 0.9563],
        [0.1081, 0.8919],
        [0.0323, 0.9677],
        [0.1545, 0.8455],
        [0.2834, 0.7166],
        [0.4348, 0.5652],
        [0.1138, 0.8862],
        [0.2355, 0.7645],
        [0.2979, 0.7021],
        [0.1160, 0.8840],
        [0.3932, 0.6068],
        [0.2296, 0.7704],
        [0.2018, 0.7982],
        [0.1817, 0.8183],
        [0.0596, 0.9404],
        [0.1093, 0.8907],
        [0.1376, 0.8624],
        [0.0771, 0.9229],
        [0.0564, 0.9436],
        [0.5829, 0.4171],
        [0.2107, 0.7893],
        [0.0857, 0.9143],
        [0.4587, 0.5413],
        [0.4451, 0.5549],
        [0.1013, 0.8987],
        [0.0539, 0.9461],
        [0.7209, 0.2791],
        [0.0770, 0.9230],
        [0.0758, 0.9242],
        [0.3763, 0.6237],
        [0.6278, 0.3722],
        [0.2442, 0.7558],
        [0.8314, 0.1686],
        [0.1493, 0.8507],
        [0.1093, 0.8907],
        [0.0817, 0.9183],
        [0.0656, 0.9344],
        [0.1214, 0.8786],
        [0.5438, 0.4562],
        [0.1178, 0.8822],
        [0.0177, 0.9823],
        [0.1358, 0.8642],
        [0.8091, 0.1909],
        [0.5689, 0.4311],
        [0.1855, 0.8145],
        [0.1523, 0.8477],
        [0.1140, 0.8860],
        [0.1340, 0.8660],
        [0.0766, 0.9234],
        [0.6434, 0.3566],
        [0.1269, 0.8731],
        [0.0697, 0.9303]], grad_fn=<SoftmaxBackward0>)
tensor([[0.2750, 0.7250],
        [0.2564, 0.7436],
        [0.0208, 0.9792],
        [0.2788, 0.7212],
        [0.1345, 0.8655],
        [0.1542, 0.8458],
        [0.0666, 0.9334],
        [0.2147, 0.7853],
        [0.3633, 0.6367],
        [0.1382, 0.8618],
        [0.8434, 0.1566],
        [0.6598, 0.3402],
        [0.8240, 0.1760],
        [0.0511, 0.9489],
        [0.1410, 0.8590],
        [0.0571, 0.9429],
        [0.0834, 0.9166],
        [0.1182, 0.8818],
        [0.4898, 0.5102],
        [0.5186, 0.4814],
        [0.1574, 0.8426],
        [0.5986, 0.4014],
        [0.0360, 0.9640],
        [0.8599, 0.1401],
        [0.0518, 0.9482],
        [0.1328, 0.8672],
        [0.2269, 0.7731],
        [0.0791, 0.9209],
        [0.1397, 0.8603],
        [0.1960, 0.8040],
        [0.0866, 0.9134],
        [0.0458, 0.9542],
        [0.1145, 0.8855],
        [0.2651, 0.7349],
        [0.0890, 0.9110],
        [0.0679, 0.9321],
        [0.1603, 0.8397],
        [0.0717, 0.9283],
        [0.1214, 0.8786],
        [0.2547, 0.7453],
        [0.1501, 0.8499],
        [0.3076, 0.6924],
        [0.1298, 0.8702],
        [0.0921, 0.9079],
        [0.4379, 0.5621],
        [0.0456, 0.9544],
        [0.0548, 0.9452],
        [0.6966, 0.3034],
        [0.7656, 0.2344],
        [0.1553, 0.8447],
        [0.3162, 0.6838],
        [0.1181, 0.8819],
        [0.1083, 0.8917],
        [0.0838, 0.9162],
        [0.4440, 0.5560],
        [0.1388, 0.8612],
        [0.2388, 0.7612],
        [0.7199, 0.2801],
        [0.3679, 0.6321],
        [0.1313, 0.8687],
        [0.1780, 0.8220],
        [0.1301, 0.8699],
        [0.5472, 0.4528],
        [0.5816, 0.4184]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1174, 0.8826],
        [0.0859, 0.9141],
        [0.0968, 0.9032],
        [0.5570, 0.4430],
        [0.0160, 0.9840],
        [0.1888, 0.8112],
        [0.0702, 0.9298],
        [0.0766, 0.9234],
        [0.5497, 0.4503],
        [0.0625, 0.9375],
        [0.1105, 0.8895],
        [0.1539, 0.8461],
        [0.1077, 0.8923],
        [0.3845, 0.6155],
        [0.2068, 0.7932],
        [0.2408, 0.7592],
        [0.0886, 0.9114],
        [0.0522, 0.9478],
        [0.0610, 0.9390],
        [0.6149, 0.3851],
        [0.1036, 0.8964],
        [0.0367, 0.9633],
        [0.1134, 0.8866],
        [0.0920, 0.9080],
        [0.1580, 0.8420],
        [0.1084, 0.8916],
        [0.0386, 0.9614],
        [0.1490, 0.8510],
        [0.1656, 0.8344],
        [0.0270, 0.9730],
        [0.4452, 0.5548],
        [0.0608, 0.9392],
        [0.1310, 0.8690],
        [0.7956, 0.2044],
        [0.1072, 0.8928],
        [0.1990, 0.8010],
        [0.1787, 0.8213],
        [0.0364, 0.9636],
        [0.1300, 0.8700],
        [0.7841, 0.2159],
        [0.1482, 0.8518],
        [0.4138, 0.5862],
        [0.3371, 0.6629],
        [0.7499, 0.2501],
        [0.0687, 0.9313],
        [0.1392, 0.8608],
        [0.4929, 0.5071],
        [0.1963, 0.8037],
        [0.3419, 0.6581],
        [0.0357, 0.9643],
        [0.0978, 0.9022],
        [0.0973, 0.9027],
        [0.3947, 0.6053],
        [0.5324, 0.4676],
        [0.0708, 0.9292],
        [0.0255, 0.9745],
        [0.4962, 0.5038],
        [0.7112, 0.2888],
        [0.2909, 0.7091],
        [0.0327, 0.9673],
        [0.1032, 0.8968],
        [0.6657, 0.3343],
        [0.1094, 0.8906],
        [0.1057, 0.8943]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0225, 0.9775],
        [0.6827, 0.3173],
        [0.3103, 0.6897],
        [0.2384, 0.7616],
        [0.7935, 0.2065],
        [0.0628, 0.9372],
        [0.0984, 0.9016],
        [0.0204, 0.9796],
        [0.0887, 0.9113],
        [0.0210, 0.9790],
        [0.1952, 0.8048],
        [0.0212, 0.9788],
        [0.1637, 0.8363],
        [0.0854, 0.9146],
        [0.1138, 0.8862],
        [0.0349, 0.9651],
        [0.0268, 0.9732],
        [0.0281, 0.9719],
        [0.0244, 0.9756],
        [0.4183, 0.5817],
        [0.0240, 0.9760],
        [0.4741, 0.5259],
        [0.1021, 0.8979],
        [0.3564, 0.6436],
        [0.0301, 0.9699],
        [0.0259, 0.9741],
        [0.0976, 0.9024],
        [0.0543, 0.9457],
        [0.0278, 0.9722],
        [0.1355, 0.8645],
        [0.2736, 0.7264],
        [0.0169, 0.9831],
        [0.0606, 0.9394],
        [0.1037, 0.8963],
        [0.1188, 0.8812],
        [0.0179, 0.9821],
        [0.0486, 0.9514],
        [0.0773, 0.9227],
        [0.1019, 0.8981],
        [0.1154, 0.8846],
        [0.8543, 0.1457],
        [0.0403, 0.9597],
        [0.0472, 0.9528],
        [0.3143, 0.6857],
        [0.6560, 0.3440],
        [0.5990, 0.4010],
        [0.4492, 0.5508],
        [0.0501, 0.9499],
        [0.0934, 0.9066],
        [0.4670, 0.5330],
        [0.1460, 0.8540],
        [0.6403, 0.3597],
        [0.0816, 0.9184],
        [0.0150, 0.9850],
        [0.7222, 0.2778],
        [0.0974, 0.9026],
        [0.1839, 0.8161],
        [0.0617, 0.9383],
        [0.2598, 0.7402],
        [0.0218, 0.9782],
        [0.2073, 0.7927],
        [0.2637, 0.7363],
        [0.0185, 0.9815],
        [0.1178, 0.8822]], grad_fn=<SoftmaxBackward0>)
tensor([[0.3982, 0.6018],
        [0.0920, 0.9080],
        [0.1586, 0.8414],
        [0.0269, 0.9731],
        [0.3684, 0.6316],
        [0.1187, 0.8813],
        [0.4950, 0.5050],
        [0.0645, 0.9355],
        [0.0030, 0.9970],
        [0.0072, 0.9928],
        [0.6418, 0.3582],
        [0.0315, 0.9685],
        [0.1861, 0.8139],
        [0.0570, 0.9430],
        [0.6939, 0.3061],
        [0.0545, 0.9455],
        [0.0177, 0.9823],
        [0.0309, 0.9691],
        [0.0271, 0.9729],
        [0.0500, 0.9500],
        [0.0166, 0.9834],
        [0.4717, 0.5283],
        [0.0283, 0.9717],
        [0.0095, 0.9905],
        [0.0450, 0.9550],
        [0.1241, 0.8759],
        [0.0136, 0.9864],
        [0.1215, 0.8785],
        [0.0729, 0.9271],
        [0.0190, 0.9810],
        [0.0280, 0.9720],
        [0.0530, 0.9470],
        [0.0160, 0.9840],
        [0.0211, 0.9789],
        [0.0495, 0.9505],
        [0.6212, 0.3788],
        [0.0752, 0.9248],
        [0.1441, 0.8559],
        [0.0822, 0.9178],
        [0.0148, 0.9852],
        [0.1069, 0.8931],
        [0.0447, 0.9553],
        [0.4043, 0.5957],
        [0.0205, 0.9795],
        [0.0241, 0.9759],
        [0.0219, 0.9781],
        [0.0505, 0.9495],
        [0.0145, 0.9855],
        [0.6957, 0.3043],
        [0.0203, 0.9797],
        [0.0400, 0.9600],
        [0.0372, 0.9628],
        [0.5123, 0.4877],
        [0.0491, 0.9509],
        [0.0326, 0.9674],
        [0.0286, 0.9714],
        [0.0505, 0.9495],
        [0.0491, 0.9509],
        [0.0798, 0.9202],
        [0.1493, 0.8507],
        [0.0609, 0.9391],
        [0.6212, 0.3788],
        [0.0640, 0.9360],
        [0.6427, 0.3573]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0874, 0.9126],
        [0.0211, 0.9789],
        [0.0564, 0.9436],
        [0.0349, 0.9651],
        [0.0610, 0.9390],
        [0.0635, 0.9365],
        [0.0348, 0.9652],
        [0.0240, 0.9760],
        [0.5150, 0.4850],
        [0.0108, 0.9892],
        [0.0061, 0.9939],
        [0.0587, 0.9413],
        [0.0303, 0.9697],
        [0.8466, 0.1534],
        [0.1284, 0.8716],
        [0.0071, 0.9929],
        [0.0097, 0.9903],
        [0.0718, 0.9282],
        [0.0703, 0.9297],
        [0.0413, 0.9587],
        [0.0147, 0.9853],
        [0.0196, 0.9804],
        [0.0260, 0.9740],
        [0.0387, 0.9613],
        [0.0487, 0.9513],
        [0.0258, 0.9742],
        [0.8715, 0.1285],
        [0.0385, 0.9615],
        [0.1209, 0.8791],
        [0.8084, 0.1916],
        [0.0601, 0.9399],
        [0.0311, 0.9689],
        [0.0229, 0.9771],
        [0.0338, 0.9662],
        [0.0115, 0.9885],
        [0.0150, 0.9850],
        [0.2370, 0.7630],
        [0.3094, 0.6906],
        [0.0788, 0.9212],
        [0.3823, 0.6177],
        [0.5863, 0.4137],
        [0.0157, 0.9843],
        [0.5659, 0.4341],
        [0.1269, 0.8731],
        [0.0229, 0.9771],
        [0.1471, 0.8529],
        [0.0239, 0.9761],
        [0.4759, 0.5241],
        [0.0242, 0.9758],
        [0.0115, 0.9885],
        [0.0574, 0.9426],
        [0.0866, 0.9134],
        [0.0106, 0.9894],
        [0.0416, 0.9584],
        [0.0638, 0.9362],
        [0.0085, 0.9915],
        [0.1905, 0.8095],
        [0.0285, 0.9715],
        [0.0759, 0.9241],
        [0.0090, 0.9910],
        [0.0538, 0.9462],
        [0.0754, 0.9246],
        [0.0774, 0.9226],
        [0.0194, 0.9806]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0129, 0.9871],
        [0.0207, 0.9793],
        [0.0116, 0.9884],
        [0.3228, 0.6772],
        [0.8423, 0.1577],
        [0.0473, 0.9527],
        [0.6015, 0.3985],
        [0.0196, 0.9804],
        [0.0107, 0.9893],
        [0.5698, 0.4302],
        [0.1051, 0.8949],
        [0.0139, 0.9861],
        [0.0085, 0.9915],
        [0.6645, 0.3355],
        [0.1746, 0.8254],
        [0.0273, 0.9727],
        [0.0173, 0.9827],
        [0.0353, 0.9647],
        [0.0155, 0.9845],
        [0.0072, 0.9928],
        [0.5322, 0.4678],
        [0.0297, 0.9703],
        [0.6809, 0.3191],
        [0.0159, 0.9841],
        [0.4686, 0.5314],
        [0.0431, 0.9569],
        [0.4751, 0.5249],
        [0.0096, 0.9904],
        [0.0749, 0.9251],
        [0.0177, 0.9823],
        [0.0248, 0.9752],
        [0.0598, 0.9402],
        [0.0160, 0.9840],
        [0.9125, 0.0875],
        [0.5978, 0.4022],
        [0.0121, 0.9879],
        [0.0072, 0.9928],
        [0.0214, 0.9786],
        [0.0090, 0.9910],
        [0.1195, 0.8805],
        [0.2728, 0.7272],
        [0.0113, 0.9887],
        [0.0037, 0.9963],
        [0.3309, 0.6691],
        [0.0590, 0.9410],
        [0.0245, 0.9755],
        [0.8123, 0.1877],
        [0.1126, 0.8874],
        [0.0757, 0.9243],
        [0.1781, 0.8219],
        [0.0355, 0.9645],
        [0.0207, 0.9793],
        [0.0853, 0.9147],
        [0.1228, 0.8772],
        [0.0035, 0.9965],
        [0.0460, 0.9540],
        [0.0405, 0.9595],
        [0.6598, 0.3402],
        [0.6911, 0.3089],
        [0.0544, 0.9456],
        [0.2451, 0.7549],
        [0.0264, 0.9736],
        [0.0288, 0.9712],
        [0.0309, 0.9691]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0469, 0.9531],
        [0.0118, 0.9882],
        [0.0372, 0.9628],
        [0.1566, 0.8434],
        [0.0405, 0.9595],
        [0.7284, 0.2716],
        [0.0718, 0.9282],
        [0.0051, 0.9949],
        [0.7214, 0.2786],
        [0.6675, 0.3325],
        [0.0102, 0.9898],
        [0.0971, 0.9029],
        [0.0385, 0.9615],
        [0.0319, 0.9681],
        [0.0428, 0.9572],
        [0.0274, 0.9726],
        [0.7916, 0.2084],
        [0.0716, 0.9284],
        [0.0549, 0.9451],
        [0.1020, 0.8980],
        [0.0980, 0.9020],
        [0.0844, 0.9156],
        [0.0440, 0.9560],
        [0.0222, 0.9778],
        [0.4882, 0.5118],
        [0.2899, 0.7101],
        [0.0224, 0.9776],
        [0.8141, 0.1859],
        [0.0185, 0.9815],
        [0.0264, 0.9736],
        [0.0894, 0.9106],
        [0.0910, 0.9090],
        [0.0117, 0.9883],
        [0.5668, 0.4332],
        [0.0868, 0.9132],
        [0.0517, 0.9483],
        [0.1006, 0.8994],
        [0.7893, 0.2107],
        [0.9307, 0.0693],
        [0.0766, 0.9234],
        [0.2877, 0.7123],
        [0.0202, 0.9798],
        [0.1294, 0.8706],
        [0.0527, 0.9473],
        [0.7574, 0.2426],
        [0.0833, 0.9167],
        [0.0560, 0.9440],
        [0.0533, 0.9467],
        [0.5295, 0.4705],
        [0.0111, 0.9889],
        [0.0184, 0.9816],
        [0.0503, 0.9497],
        [0.0342, 0.9658],
        [0.0046, 0.9954],
        [0.7831, 0.2169],
        [0.0194, 0.9806],
        [0.6988, 0.3012],
        [0.0934, 0.9066],
        [0.3759, 0.6241],
        [0.8102, 0.1898],
        [0.0519, 0.9481],
        [0.0140, 0.9860],
        [0.8969, 0.1031],
        [0.0387, 0.9613]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0115, 0.9885],
        [0.7271, 0.2729],
        [0.1095, 0.8905],
        [0.0057, 0.9943],
        [0.0381, 0.9619],
        [0.5546, 0.4454],
        [0.0343, 0.9657],
        [0.1139, 0.8861],
        [0.5458, 0.4542],
        [0.2800, 0.7200],
        [0.0224, 0.9776],
        [0.0249, 0.9751],
        [0.0162, 0.9838],
        [0.0298, 0.9702],
        [0.0718, 0.9282],
        [0.0702, 0.9298],
        [0.4292, 0.5708],
        [0.6544, 0.3456],
        [0.1436, 0.8564],
        [0.6378, 0.3622],
        [0.6699, 0.3301],
        [0.1649, 0.8351],
        [0.6623, 0.3377],
        [0.0120, 0.9880],
        [0.6638, 0.3362],
        [0.0548, 0.9452],
        [0.0080, 0.9920],
        [0.0355, 0.9645],
        [0.3905, 0.6095],
        [0.0164, 0.9836],
        [0.0680, 0.9320],
        [0.2823, 0.7177],
        [0.8457, 0.1543],
        [0.0101, 0.9899],
        [0.0164, 0.9836],
        [0.9421, 0.0579],
        [0.0229, 0.9771],
        [0.0080, 0.9920],
        [0.0385, 0.9615],
        [0.0632, 0.9368],
        [0.1535, 0.8465],
        [0.0293, 0.9707],
        [0.4881, 0.5119],
        [0.0188, 0.9812],
        [0.1670, 0.8330],
        [0.0132, 0.9868],
        [0.2292, 0.7708],
        [0.9219, 0.0781],
        [0.0449, 0.9551],
        [0.7050, 0.2950],
        [0.0624, 0.9376],
        [0.3484, 0.6516],
        [0.8002, 0.1998],
        [0.8456, 0.1544],
        [0.0506, 0.9494],
        [0.8758, 0.1242],
        [0.1353, 0.8647],
        [0.0580, 0.9420],
        [0.0162, 0.9838],
        [0.0313, 0.9687],
        [0.0229, 0.9771],
        [0.0242, 0.9758],
        [0.0079, 0.9921],
        [0.0211, 0.9789]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0599, 0.9401],
        [0.0037, 0.9963],
        [0.9760, 0.0240],
        [0.1127, 0.8873],
        [0.0703, 0.9297],
        [0.4036, 0.5964],
        [0.8856, 0.1144],
        [0.0023, 0.9977],
        [0.1168, 0.8832],
        [0.7637, 0.2363],
        [0.0171, 0.9829],
        [0.0891, 0.9109],
        [0.9050, 0.0950],
        [0.0658, 0.9342],
        [0.0098, 0.9902],
        [0.1197, 0.8803],
        [0.0512, 0.9488],
        [0.0166, 0.9834],
        [0.0894, 0.9106],
        [0.0861, 0.9139],
        [0.0446, 0.9554],
        [0.2260, 0.7740],
        [0.0636, 0.9364],
        [0.0213, 0.9787],
        [0.0696, 0.9304],
        [0.1657, 0.8343],
        [0.1317, 0.8683],
        [0.1989, 0.8011],
        [0.0506, 0.9494],
        [0.0434, 0.9566],
        [0.2484, 0.7516],
        [0.0365, 0.9635],
        [0.0854, 0.9146],
        [0.9150, 0.0850],
        [0.6457, 0.3543],
        [0.0564, 0.9436],
        [0.1075, 0.8925],
        [0.5353, 0.4647],
        [0.0998, 0.9002],
        [0.4289, 0.5711],
        [0.7598, 0.2402],
        [0.7620, 0.2380],
        [0.0403, 0.9597],
        [0.1040, 0.8960],
        [0.0379, 0.9621],
        [0.1101, 0.8899],
        [0.6639, 0.3361],
        [0.0448, 0.9552],
        [0.0229, 0.9771],
        [0.0287, 0.9713],
        [0.1517, 0.8483],
        [0.1793, 0.8207],
        [0.7332, 0.2668],
        [0.8057, 0.1943],
        [0.0298, 0.9702],
        [0.0915, 0.9085],
        [0.0295, 0.9705],
        [0.1011, 0.8989],
        [0.0155, 0.9845],
        [0.9453, 0.0547],
        [0.0581, 0.9419],
        [0.0217, 0.9783],
        [0.8806, 0.1194],
        [0.1042, 0.8958]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1191, 0.8809],
        [0.1335, 0.8665],
        [0.2169, 0.7831],
        [0.0135, 0.9865],
        [0.1426, 0.8574],
        [0.0775, 0.9225],
        [0.0100, 0.9900],
        [0.5558, 0.4442],
        [0.0148, 0.9852],
        [0.7207, 0.2793],
        [0.8064, 0.1936],
        [0.8663, 0.1337],
        [0.0696, 0.9304],
        [0.0280, 0.9720],
        [0.6143, 0.3857],
        [0.1850, 0.8150],
        [0.4301, 0.5699],
        [0.9337, 0.0663],
        [0.0277, 0.9723],
        [0.9590, 0.0410],
        [0.0206, 0.9794],
        [0.1306, 0.8694],
        [0.0091, 0.9909],
        [0.2343, 0.7657],
        [0.1734, 0.8266],
        [0.2579, 0.7421],
        [0.0430, 0.9570],
        [0.9264, 0.0736],
        [0.4447, 0.5553],
        [0.6831, 0.3169],
        [0.8429, 0.1571],
        [0.1040, 0.8960],
        [0.0031, 0.9969],
        [0.0792, 0.9208],
        [0.0932, 0.9068],
        [0.1897, 0.8103],
        [0.0450, 0.9550],
        [0.0117, 0.9883],
        [0.0078, 0.9922],
        [0.6236, 0.3764],
        [0.0828, 0.9172],
        [0.1028, 0.8972],
        [0.4162, 0.5838],
        [0.0657, 0.9343],
        [0.9227, 0.0773],
        [0.2551, 0.7449],
        [0.2060, 0.7940],
        [0.9335, 0.0665],
        [0.0410, 0.9590],
        [0.1687, 0.8313],
        [0.3811, 0.6189],
        [0.0162, 0.9838],
        [0.4179, 0.5821],
        [0.7152, 0.2848],
        [0.0627, 0.9373],
        [0.1114, 0.8886],
        [0.9712, 0.0288],
        [0.7730, 0.2270],
        [0.1321, 0.8679],
        [0.0302, 0.9698],
        [0.1827, 0.8173],
        [0.0058, 0.9942],
        [0.8607, 0.1393],
        [0.2390, 0.7610]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0200, 0.9800],
        [0.0258, 0.9742],
        [0.1124, 0.8876],
        [0.3110, 0.6890],
        [0.0247, 0.9753],
        [0.0582, 0.9418],
        [0.9146, 0.0854],
        [0.1318, 0.8682],
        [0.0637, 0.9363],
        [0.0133, 0.9867],
        [0.3204, 0.6796],
        [0.0529, 0.9471],
        [0.5895, 0.4105],
        [0.0745, 0.9255],
        [0.0411, 0.9589],
        [0.2605, 0.7395],
        [0.0542, 0.9458],
        [0.8517, 0.1483],
        [0.0202, 0.9798],
        [0.1812, 0.8188],
        [0.4061, 0.5939],
        [0.5494, 0.4506],
        [0.2205, 0.7795],
        [0.0958, 0.9042],
        [0.1029, 0.8971],
        [0.7972, 0.2028],
        [0.0393, 0.9607],
        [0.7861, 0.2139],
        [0.9289, 0.0711],
        [0.8247, 0.1753],
        [0.0464, 0.9536],
        [0.8404, 0.1596],
        [0.0246, 0.9754],
        [0.0230, 0.9770],
        [0.0581, 0.9419],
        [0.8352, 0.1648],
        [0.8579, 0.1421],
        [0.5153, 0.4847],
        [0.0697, 0.9303],
        [0.0856, 0.9144],
        [0.1305, 0.8695],
        [0.7786, 0.2214],
        [0.9569, 0.0431],
        [0.2715, 0.7285],
        [0.8345, 0.1655],
        [0.0133, 0.9867],
        [0.0437, 0.9563],
        [0.6340, 0.3660],
        [0.0802, 0.9198],
        [0.3466, 0.6534],
        [0.1629, 0.8371],
        [0.9634, 0.0366],
        [0.0638, 0.9362],
        [0.9759, 0.0241],
        [0.0463, 0.9537],
        [0.0500, 0.9500],
        [0.0437, 0.9563],
        [0.8671, 0.1329],
        [0.0960, 0.9040],
        [0.2970, 0.7030],
        [0.0273, 0.9727],
        [0.1911, 0.8089],
        [0.9359, 0.0641],
        [0.0449, 0.9551]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0833, 0.9167],
        [0.1841, 0.8159],
        [0.2794, 0.7206],
        [0.0430, 0.9570],
        [0.7704, 0.2296],
        [0.0199, 0.9801],
        [0.0464, 0.9536],
        [0.1405, 0.8595],
        [0.0389, 0.9611],
        [0.2330, 0.7670],
        [0.5741, 0.4259],
        [0.2093, 0.7907],
        [0.6508, 0.3492],
        [0.0728, 0.9272],
        [0.1275, 0.8725],
        [0.6998, 0.3002],
        [0.4293, 0.5707],
        [0.3101, 0.6899],
        [0.8060, 0.1940],
        [0.0285, 0.9715],
        [0.2789, 0.7211],
        [0.0644, 0.9356],
        [0.0288, 0.9712],
        [0.0214, 0.9786],
        [0.4292, 0.5708],
        [0.0438, 0.9562],
        [0.0157, 0.9843],
        [0.0111, 0.9889],
        [0.1056, 0.8944],
        [0.3220, 0.6780],
        [0.1604, 0.8396],
        [0.0217, 0.9783],
        [0.0541, 0.9459],
        [0.0262, 0.9738],
        [0.0481, 0.9519],
        [0.0054, 0.9946],
        [0.1734, 0.8266],
        [0.1394, 0.8606],
        [0.0319, 0.9681],
        [0.0173, 0.9827],
        [0.1198, 0.8802],
        [0.0375, 0.9625],
        [0.3660, 0.6340],
        [0.0548, 0.9452],
        [0.1306, 0.8694],
        [0.9726, 0.0274],
        [0.9121, 0.0879],
        [0.0468, 0.9532],
        [0.8861, 0.1139],
        [0.0921, 0.9079],
        [0.0043, 0.9957],
        [0.0956, 0.9044],
        [0.9279, 0.0721],
        [0.0096, 0.9904],
        [0.6590, 0.3410],
        [0.0279, 0.9721],
        [0.9141, 0.0859],
        [0.0741, 0.9259],
        [0.2242, 0.7758],
        [0.1228, 0.8772],
        [0.7361, 0.2639],
        [0.0112, 0.9888],
        [0.1201, 0.8799],
        [0.7482, 0.2518]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1013, 0.8987],
        [0.3923, 0.6077],
        [0.0159, 0.9841],
        [0.7845, 0.2155],
        [0.1410, 0.8590],
        [0.0535, 0.9465],
        [0.0905, 0.9095],
        [0.0218, 0.9782],
        [0.0691, 0.9309],
        [0.0832, 0.9168],
        [0.9005, 0.0995],
        [0.0337, 0.9663],
        [0.9355, 0.0645],
        [0.1055, 0.8945],
        [0.1195, 0.8805],
        [0.0198, 0.9802],
        [0.9320, 0.0680],
        [0.8616, 0.1384],
        [0.8481, 0.1519],
        [0.3306, 0.6694],
        [0.2701, 0.7299],
        [0.8837, 0.1163],
        [0.1225, 0.8775],
        [0.7457, 0.2543],
        [0.0542, 0.9458],
        [0.1331, 0.8669],
        [0.0056, 0.9944],
        [0.5125, 0.4875],
        [0.7133, 0.2867],
        [0.0779, 0.9221],
        [0.0033, 0.9967],
        [0.9506, 0.0494],
        [0.0044, 0.9956],
        [0.0996, 0.9004],
        [0.8231, 0.1769],
        [0.1676, 0.8324],
        [0.1921, 0.8079],
        [0.0711, 0.9289],
        [0.0733, 0.9267],
        [0.0080, 0.9920],
        [0.0150, 0.9850],
        [0.0364, 0.9636],
        [0.0107, 0.9893],
        [0.0588, 0.9412],
        [0.0150, 0.9850],
        [0.0934, 0.9066],
        [0.8115, 0.1885],
        [0.8335, 0.1665],
        [0.0089, 0.9911],
        [0.0709, 0.9291],
        [0.9643, 0.0357],
        [0.0616, 0.9384],
        [0.2231, 0.7769],
        [0.4342, 0.5658],
        [0.8875, 0.1125],
        [0.0245, 0.9755],
        [0.8405, 0.1595],
        [0.0276, 0.9724],
        [0.0749, 0.9251],
        [0.0530, 0.9470],
        [0.2088, 0.7912],
        [0.0195, 0.9805],
        [0.0730, 0.9270],
        [0.0052, 0.9948]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0221, 0.9779],
        [0.0086, 0.9914],
        [0.0100, 0.9900],
        [0.0344, 0.9656],
        [0.0540, 0.9460],
        [0.1121, 0.8879],
        [0.0062, 0.9938],
        [0.0251, 0.9749],
        [0.0616, 0.9384],
        [0.4550, 0.5450],
        [0.8416, 0.1584],
        [0.0127, 0.9873],
        [0.0430, 0.9570],
        [0.0142, 0.9858],
        [0.0373, 0.9627],
        [0.0053, 0.9947],
        [0.0209, 0.9791],
        [0.8108, 0.1892],
        [0.0616, 0.9384],
        [0.0136, 0.9864],
        [0.4912, 0.5088],
        [0.5593, 0.4407],
        [0.1551, 0.8449],
        [0.6351, 0.3649],
        [0.0313, 0.9687],
        [0.0348, 0.9652],
        [0.9213, 0.0787],
        [0.8137, 0.1863],
        [0.0910, 0.9090],
        [0.0226, 0.9774],
        [0.0162, 0.9838],
        [0.0287, 0.9713],
        [0.0567, 0.9433],
        [0.1838, 0.8162],
        [0.0697, 0.9303],
        [0.9319, 0.0681],
        [0.9283, 0.0717],
        [0.9708, 0.0292],
        [0.0474, 0.9526],
        [0.7656, 0.2344],
        [0.4790, 0.5210],
        [0.0162, 0.9838],
        [0.7779, 0.2221],
        [0.0186, 0.9814],
        [0.0193, 0.9807],
        [0.8502, 0.1498],
        [0.0362, 0.9638],
        [0.0870, 0.9130],
        [0.0270, 0.9730],
        [0.0821, 0.9179],
        [0.3639, 0.6361],
        [0.1510, 0.8490],
        [0.0298, 0.9702],
        [0.0973, 0.9027],
        [0.6687, 0.3313],
        [0.2342, 0.7658],
        [0.3670, 0.6330],
        [0.0699, 0.9301],
        [0.0050, 0.9950],
        [0.0146, 0.9854],
        [0.0064, 0.9936],
        [0.0032, 0.9968],
        [0.0253, 0.9747],
        [0.1243, 0.8757]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0129, 0.9871],
        [0.9564, 0.0436],
        [0.0036, 0.9964],
        [0.0065, 0.9935],
        [0.0362, 0.9638],
        [0.0136, 0.9864],
        [0.0537, 0.9463],
        [0.0336, 0.9664],
        [0.2110, 0.7890],
        [0.9246, 0.0754],
        [0.0045, 0.9955],
        [0.0857, 0.9143],
        [0.0885, 0.9115],
        [0.0042, 0.9958],
        [0.7266, 0.2734],
        [0.0112, 0.9888],
        [0.0424, 0.9576],
        [0.9096, 0.0904],
        [0.0139, 0.9861],
        [0.1116, 0.8884],
        [0.3303, 0.6697],
        [0.3823, 0.6177],
        [0.0448, 0.9552],
        [0.0714, 0.9286],
        [0.3948, 0.6052],
        [0.9488, 0.0512],
        [0.0896, 0.9104],
        [0.0503, 0.9497],
        [0.7893, 0.2107],
        [0.0052, 0.9948],
        [0.2352, 0.7648],
        [0.9202, 0.0798],
        [0.0345, 0.9655],
        [0.9266, 0.0734],
        [0.9886, 0.0114],
        [0.5060, 0.4940],
        [0.6560, 0.3440],
        [0.0362, 0.9638],
        [0.6448, 0.3552],
        [0.0034, 0.9966],
        [0.3559, 0.6441],
        [0.0420, 0.9580],
        [0.3399, 0.6601],
        [0.0057, 0.9943],
        [0.3401, 0.6599],
        [0.9444, 0.0556],
        [0.2290, 0.7710],
        [0.0010, 0.9990],
        [0.0256, 0.9744],
        [0.8644, 0.1356],
        [0.0354, 0.9646],
        [0.1132, 0.8868],
        [0.8920, 0.1080],
        [0.5505, 0.4495],
        [0.0253, 0.9747],
        [0.6900, 0.3100],
        [0.0230, 0.9770],
        [0.6274, 0.3726],
        [0.0764, 0.9236],
        [0.3454, 0.6546],
        [0.0406, 0.9594],
        [0.0490, 0.9510],
        [0.0311, 0.9689],
        [0.0277, 0.9723]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2288e-02, 9.7771e-01],
        [1.6773e-02, 9.8323e-01],
        [9.3652e-04, 9.9906e-01],
        [9.5157e-01, 4.8425e-02],
        [1.9598e-02, 9.8040e-01],
        [1.2632e-02, 9.8737e-01],
        [1.3899e-01, 8.6101e-01],
        [1.4758e-01, 8.5242e-01],
        [9.8029e-01, 1.9714e-02],
        [6.8330e-02, 9.3167e-01],
        [6.3501e-03, 9.9365e-01],
        [2.1684e-03, 9.9783e-01],
        [1.0506e-02, 9.8949e-01],
        [1.8050e-02, 9.8195e-01],
        [2.5903e-03, 9.9741e-01],
        [5.9267e-02, 9.4073e-01],
        [6.3506e-01, 3.6494e-01],
        [8.5123e-03, 9.9149e-01],
        [6.5709e-02, 9.3429e-01],
        [2.0484e-02, 9.7952e-01],
        [1.8804e-01, 8.1196e-01],
        [5.7572e-02, 9.4243e-01],
        [6.1117e-01, 3.8883e-01],
        [5.2331e-02, 9.4767e-01],
        [5.0571e-02, 9.4943e-01],
        [1.8684e-02, 9.8132e-01],
        [7.7666e-01, 2.2334e-01],
        [2.8373e-03, 9.9716e-01],
        [1.0352e-02, 9.8965e-01],
        [8.9950e-02, 9.1005e-01],
        [7.2279e-01, 2.7721e-01],
        [7.2393e-02, 9.2761e-01],
        [4.4165e-02, 9.5584e-01],
        [1.1135e-02, 9.8887e-01],
        [5.3088e-01, 4.6912e-01],
        [1.1719e-02, 9.8828e-01],
        [8.4067e-01, 1.5933e-01],
        [9.0592e-02, 9.0941e-01],
        [6.4290e-02, 9.3571e-01],
        [1.3783e-03, 9.9862e-01],
        [4.5103e-01, 5.4897e-01],
        [3.8698e-02, 9.6130e-01],
        [7.4239e-01, 2.5761e-01],
        [9.6430e-03, 9.9036e-01],
        [5.0346e-02, 9.4965e-01],
        [3.5085e-02, 9.6491e-01],
        [8.4538e-03, 9.9155e-01],
        [1.1930e-02, 9.8807e-01],
        [5.5846e-01, 4.4154e-01],
        [4.8971e-02, 9.5103e-01],
        [4.3815e-03, 9.9562e-01],
        [9.6364e-03, 9.9036e-01],
        [4.6708e-01, 5.3292e-01],
        [9.3539e-01, 6.4606e-02],
        [8.2977e-03, 9.9170e-01],
        [8.3824e-01, 1.6176e-01],
        [9.4701e-01, 5.2991e-02],
        [1.9521e-03, 9.9805e-01],
        [4.8129e-02, 9.5187e-01],
        [6.3956e-03, 9.9360e-01],
        [1.4821e-02, 9.8518e-01],
        [3.7443e-01, 6.2557e-01],
        [2.6525e-02, 9.7348e-01],
        [7.1870e-01, 2.8130e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6575e-02, 9.8342e-01],
        [4.4836e-02, 9.5516e-01],
        [3.3169e-02, 9.6683e-01],
        [3.3291e-02, 9.6671e-01],
        [2.1964e-03, 9.9780e-01],
        [9.3258e-01, 6.7416e-02],
        [9.6500e-01, 3.5005e-02],
        [1.0213e-02, 9.8979e-01],
        [2.3335e-03, 9.9767e-01],
        [5.5948e-01, 4.4052e-01],
        [3.3894e-03, 9.9661e-01],
        [3.1274e-04, 9.9969e-01],
        [1.4383e-02, 9.8562e-01],
        [1.6361e-02, 9.8364e-01],
        [9.3378e-01, 6.6223e-02],
        [2.2668e-03, 9.9773e-01],
        [4.3351e-02, 9.5665e-01],
        [7.5725e-03, 9.9243e-01],
        [2.0166e-01, 7.9834e-01],
        [9.2910e-01, 7.0902e-02],
        [5.9487e-02, 9.4051e-01],
        [1.4848e-02, 9.8515e-01],
        [1.0385e-01, 8.9615e-01],
        [9.5015e-01, 4.9845e-02],
        [2.6906e-03, 9.9731e-01],
        [8.7687e-01, 1.2313e-01],
        [7.7924e-01, 2.2076e-01],
        [8.4517e-01, 1.5483e-01],
        [9.2057e-01, 7.9432e-02],
        [5.1419e-03, 9.9486e-01],
        [1.5418e-02, 9.8458e-01],
        [1.3147e-01, 8.6853e-01],
        [8.8754e-03, 9.9112e-01],
        [2.9266e-02, 9.7073e-01],
        [2.8900e-02, 9.7110e-01],
        [2.1342e-02, 9.7866e-01],
        [5.7665e-03, 9.9423e-01],
        [5.0104e-01, 4.9896e-01],
        [3.3895e-03, 9.9661e-01],
        [6.6930e-03, 9.9331e-01],
        [9.3884e-01, 6.1164e-02],
        [4.1189e-03, 9.9588e-01],
        [9.8287e-01, 1.7135e-02],
        [8.0989e-01, 1.9011e-01],
        [3.7286e-02, 9.6271e-01],
        [1.1918e-01, 8.8082e-01],
        [1.8222e-02, 9.8178e-01],
        [3.7578e-03, 9.9624e-01],
        [3.9983e-02, 9.6002e-01],
        [9.6948e-01, 3.0524e-02],
        [2.5624e-02, 9.7438e-01],
        [5.9650e-03, 9.9404e-01],
        [1.1165e-02, 9.8884e-01],
        [2.0868e-01, 7.9132e-01],
        [2.0025e-02, 9.7998e-01],
        [3.3707e-03, 9.9663e-01],
        [6.1705e-03, 9.9383e-01],
        [5.1616e-02, 9.4838e-01],
        [8.3433e-02, 9.1657e-01],
        [1.0432e-01, 8.9568e-01],
        [7.8835e-01, 2.1165e-01],
        [8.6795e-01, 1.3205e-01],
        [3.5400e-02, 9.6460e-01],
        [5.8265e-01, 4.1735e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3528e-02, 9.7647e-01],
        [1.2285e-02, 9.8772e-01],
        [3.6320e-03, 9.9637e-01],
        [4.6644e-03, 9.9534e-01],
        [1.8982e-02, 9.8102e-01],
        [2.7873e-01, 7.2127e-01],
        [1.1869e-03, 9.9881e-01],
        [2.1428e-02, 9.7857e-01],
        [2.4461e-02, 9.7554e-01],
        [7.1863e-03, 9.9281e-01],
        [4.2148e-01, 5.7852e-01],
        [9.4625e-01, 5.3755e-02],
        [3.7543e-02, 9.6246e-01],
        [1.5018e-02, 9.8498e-01],
        [8.1928e-03, 9.9181e-01],
        [3.3030e-03, 9.9670e-01],
        [5.7556e-03, 9.9424e-01],
        [2.8372e-02, 9.7163e-01],
        [3.6692e-01, 6.3308e-01],
        [9.9084e-04, 9.9901e-01],
        [4.2340e-02, 9.5766e-01],
        [2.2048e-02, 9.7795e-01],
        [4.4263e-03, 9.9557e-01],
        [5.3247e-01, 4.6753e-01],
        [3.6599e-02, 9.6340e-01],
        [7.6138e-02, 9.2386e-01],
        [2.7098e-02, 9.7290e-01],
        [6.6628e-02, 9.3337e-01],
        [3.7486e-02, 9.6251e-01],
        [3.3500e-02, 9.6650e-01],
        [2.6148e-02, 9.7385e-01],
        [2.7495e-03, 9.9725e-01],
        [6.3960e-02, 9.3604e-01],
        [1.7993e-01, 8.2007e-01],
        [7.9993e-03, 9.9200e-01],
        [8.6418e-01, 1.3582e-01],
        [4.1190e-02, 9.5881e-01],
        [1.2767e-02, 9.8723e-01],
        [8.1182e-01, 1.8818e-01],
        [1.0132e-01, 8.9868e-01],
        [1.0885e-01, 8.9115e-01],
        [1.1216e-01, 8.8784e-01],
        [3.1691e-03, 9.9683e-01],
        [9.2516e-03, 9.9075e-01],
        [3.7565e-02, 9.6244e-01],
        [1.1082e-02, 9.8892e-01],
        [9.3368e-01, 6.6319e-02],
        [8.3907e-03, 9.9161e-01],
        [1.1929e-02, 9.8807e-01],
        [1.8637e-02, 9.8136e-01],
        [1.8162e-02, 9.8184e-01],
        [2.0766e-02, 9.7923e-01],
        [4.7188e-02, 9.5281e-01],
        [2.1755e-02, 9.7825e-01],
        [3.1002e-02, 9.6900e-01],
        [1.8381e-02, 9.8162e-01],
        [1.3763e-02, 9.8624e-01],
        [6.3802e-03, 9.9362e-01],
        [2.0811e-01, 7.9189e-01],
        [8.5218e-03, 9.9148e-01],
        [5.3615e-02, 9.4638e-01],
        [7.4298e-02, 9.2570e-01],
        [8.7580e-02, 9.1242e-01],
        [2.8367e-03, 9.9716e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[0.8123, 0.1877],
        [0.8383, 0.1617],
        [0.0147, 0.9853],
        [0.0619, 0.9381],
        [0.0653, 0.9347],
        [0.6505, 0.3495],
        [0.0118, 0.9882],
        [0.0169, 0.9831],
        [0.5756, 0.4244],
        [0.8824, 0.1176],
        [0.0319, 0.9681],
        [0.0885, 0.9115],
        [0.0133, 0.9867],
        [0.0020, 0.9980],
        [0.0049, 0.9951],
        [0.0226, 0.9774],
        [0.0010, 0.9990],
        [0.0175, 0.9825],
        [0.0075, 0.9925],
        [0.2895, 0.7105],
        [0.0544, 0.9456],
        [0.6959, 0.3041],
        [0.1232, 0.8768],
        [0.0040, 0.9960],
        [0.0470, 0.9530],
        [0.0131, 0.9869],
        [0.0938, 0.9062],
        [0.0068, 0.9932],
        [0.0084, 0.9916],
        [0.0234, 0.9766],
        [0.7688, 0.2312],
        [0.8644, 0.1356],
        [0.0055, 0.9945],
        [0.7874, 0.2126],
        [0.7769, 0.2231],
        [0.0028, 0.9972],
        [0.8915, 0.1085],
        [0.5232, 0.4768],
        [0.0358, 0.9642],
        [0.3249, 0.6751],
        [0.0192, 0.9808],
        [0.0019, 0.9981],
        [0.0080, 0.9920],
        [0.0303, 0.9697],
        [0.2928, 0.7072],
        [0.1987, 0.8013],
        [0.0047, 0.9953],
        [0.0177, 0.9823],
        [0.3141, 0.6859],
        [0.0259, 0.9741],
        [0.0671, 0.9329],
        [0.4677, 0.5323],
        [0.6895, 0.3105],
        [0.0204, 0.9796],
        [0.0785, 0.9215],
        [0.0128, 0.9872],
        [0.8153, 0.1847],
        [0.0270, 0.9730],
        [0.0100, 0.9900],
        [0.0031, 0.9969],
        [0.0062, 0.9938],
        [0.2658, 0.7342],
        [0.0839, 0.9161],
        [0.3191, 0.6809]], grad_fn=<SoftmaxBackward0>)
tensor([[0.0033, 0.9967],
        [0.0101, 0.9899],
        [0.0091, 0.9909],
        [0.0282, 0.9718],
        [0.0401, 0.9599],
        [0.0294, 0.9706],
        [0.8998, 0.1002],
        [0.0651, 0.9349],
        [0.3627, 0.6373],
        [0.0370, 0.9630],
        [0.0142, 0.9858],
        [0.1039, 0.8961],
        [0.0542, 0.9458],
        [0.8078, 0.1922],
        [0.9523, 0.0477],
        [0.6594, 0.3406],
        [0.0393, 0.9607],
        [0.0349, 0.9651],
        [0.1860, 0.8140],
        [0.0030, 0.9970],
        [0.0490, 0.9510],
        [0.9218, 0.0782],
        [0.0037, 0.9963],
        [0.0634, 0.9366],
        [0.7412, 0.2588],
        [0.0573, 0.9427],
        [0.0268, 0.9732],
        [0.0545, 0.9455],
        [0.1680, 0.8320],
        [0.8317, 0.1683],
        [0.0362, 0.9638],
        [0.0180, 0.9820],
        [0.6289, 0.3711],
        [0.0547, 0.9453],
        [0.0109, 0.9891],
        [0.0132, 0.9868],
        [0.5319, 0.4681],
        [0.9534, 0.0466],
        [0.0012, 0.9988],
        [0.9412, 0.0588],
        [0.0130, 0.9870],
        [0.0274, 0.9726],
        [0.0860, 0.9140],
        [0.0026, 0.9974],
        [0.0325, 0.9675],
        [0.6859, 0.3141],
        [0.0502, 0.9498],
        [0.8982, 0.1018],
        [0.1022, 0.8978],
        [0.0046, 0.9954],
        [0.0122, 0.9878],
        [0.7740, 0.2260],
        [0.0576, 0.9424],
        [0.0881, 0.9119],
        [0.8317, 0.1683],
        [0.6219, 0.3781],
        [0.1294, 0.8706],
        [0.0049, 0.9951],
        [0.0056, 0.9944],
        [0.0056, 0.9944],
        [0.0449, 0.9551],
        [0.0490, 0.9510],
        [0.0759, 0.9241],
        [0.0367, 0.9633]], grad_fn=<SoftmaxBackward0>)
tensor([[5.8347e-02, 9.4165e-01],
        [1.5171e-03, 9.9848e-01],
        [5.7289e-04, 9.9943e-01],
        [2.9186e-03, 9.9708e-01],
        [5.0627e-01, 4.9373e-01],
        [9.1616e-01, 8.3838e-02],
        [2.7130e-02, 9.7287e-01],
        [8.5784e-01, 1.4216e-01],
        [7.1783e-01, 2.8217e-01],
        [7.8860e-02, 9.2114e-01],
        [1.3918e-03, 9.9861e-01],
        [7.4920e-02, 9.2508e-01],
        [8.0498e-01, 1.9502e-01],
        [1.1734e-01, 8.8266e-01],
        [1.3094e-02, 9.8691e-01],
        [1.4003e-03, 9.9860e-01],
        [2.1399e-02, 9.7860e-01],
        [5.9847e-02, 9.4015e-01],
        [6.9052e-03, 9.9309e-01],
        [1.3026e-02, 9.8697e-01],
        [2.7984e-03, 9.9720e-01],
        [9.3612e-03, 9.9064e-01],
        [9.8212e-01, 1.7882e-02],
        [2.0131e-03, 9.9799e-01],
        [1.1186e-02, 9.8881e-01],
        [1.3374e-02, 9.8663e-01],
        [7.5433e-01, 2.4567e-01],
        [2.6347e-01, 7.3653e-01],
        [8.6375e-03, 9.9136e-01],
        [5.1679e-02, 9.4832e-01],
        [2.5093e-02, 9.7491e-01],
        [9.3356e-03, 9.9066e-01],
        [5.0534e-02, 9.4947e-01],
        [7.6633e-03, 9.9234e-01],
        [2.4521e-02, 9.7548e-01],
        [5.9336e-01, 4.0664e-01],
        [1.6586e-02, 9.8341e-01],
        [7.2777e-03, 9.9272e-01],
        [5.2361e-02, 9.4764e-01],
        [2.5511e-02, 9.7449e-01],
        [9.5569e-01, 4.4314e-02],
        [5.2349e-03, 9.9477e-01],
        [2.9630e-02, 9.7037e-01],
        [8.5328e-03, 9.9147e-01],
        [1.7712e-03, 9.9823e-01],
        [4.6137e-02, 9.5386e-01],
        [3.3476e-03, 9.9665e-01],
        [1.4626e-01, 8.5374e-01],
        [7.4991e-02, 9.2501e-01],
        [3.8955e-02, 9.6104e-01],
        [2.0204e-02, 9.7980e-01],
        [1.2410e-01, 8.7590e-01],
        [8.0547e-02, 9.1945e-01],
        [1.4907e-01, 8.5093e-01],
        [9.8568e-01, 1.4323e-02],
        [2.4840e-03, 9.9752e-01],
        [3.5596e-02, 9.6440e-01],
        [8.5935e-03, 9.9141e-01],
        [8.3702e-02, 9.1630e-01],
        [8.3228e-01, 1.6772e-01],
        [2.2793e-02, 9.7721e-01],
        [5.5654e-02, 9.4435e-01],
        [6.2919e-02, 9.3708e-01],
        [8.3810e-01, 1.6190e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.6017e-02, 9.5398e-01],
        [2.4463e-02, 9.7554e-01],
        [2.8123e-02, 9.7188e-01],
        [8.3348e-03, 9.9167e-01],
        [1.5645e-01, 8.4355e-01],
        [9.7737e-01, 2.2631e-02],
        [7.7896e-01, 2.2104e-01],
        [9.4172e-03, 9.9058e-01],
        [7.2578e-02, 9.2742e-01],
        [6.5170e-02, 9.3483e-01],
        [1.2539e-03, 9.9875e-01],
        [7.3103e-03, 9.9269e-01],
        [9.6982e-01, 3.0183e-02],
        [1.0604e-01, 8.9396e-01],
        [7.4842e-03, 9.9252e-01],
        [1.7467e-01, 8.2533e-01],
        [1.0170e-01, 8.9830e-01],
        [2.7255e-01, 7.2745e-01],
        [6.2205e-02, 9.3779e-01],
        [8.7635e-01, 1.2365e-01],
        [2.7027e-02, 9.7297e-01],
        [7.1849e-01, 2.8151e-01],
        [3.8838e-03, 9.9612e-01],
        [8.5029e-01, 1.4971e-01],
        [6.2315e-01, 3.7685e-01],
        [1.0353e-01, 8.9647e-01],
        [6.1572e-03, 9.9384e-01],
        [2.4377e-01, 7.5623e-01],
        [1.4378e-01, 8.5622e-01],
        [1.0543e-01, 8.9457e-01],
        [5.0705e-02, 9.4930e-01],
        [9.8544e-01, 1.4561e-02],
        [5.0950e-02, 9.4905e-01],
        [1.2419e-02, 9.8758e-01],
        [2.0349e-01, 7.9651e-01],
        [8.0291e-04, 9.9920e-01],
        [7.4037e-01, 2.5963e-01],
        [4.0261e-02, 9.5974e-01],
        [9.8094e-01, 1.9055e-02],
        [9.4926e-03, 9.9051e-01],
        [8.6893e-01, 1.3107e-01],
        [4.2023e-03, 9.9580e-01],
        [8.5334e-01, 1.4666e-01],
        [3.4772e-03, 9.9652e-01],
        [1.0888e-03, 9.9891e-01],
        [2.3877e-02, 9.7612e-01],
        [7.2966e-02, 9.2703e-01],
        [2.0490e-02, 9.7951e-01],
        [1.9954e-02, 9.8005e-01],
        [3.4309e-03, 9.9657e-01],
        [9.5862e-01, 4.1382e-02],
        [3.9361e-02, 9.6064e-01],
        [6.2816e-02, 9.3718e-01],
        [3.8165e-03, 9.9618e-01],
        [9.4277e-01, 5.7233e-02],
        [9.8923e-01, 1.0766e-02],
        [2.0879e-02, 9.7912e-01],
        [8.8404e-01, 1.1596e-01],
        [4.1285e-03, 9.9587e-01],
        [6.4956e-02, 9.3504e-01],
        [1.1860e-01, 8.8140e-01],
        [9.0401e-01, 9.5994e-02],
        [7.8664e-03, 9.9213e-01],
        [9.5285e-01, 4.7146e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8279e-01, 1.7206e-02],
        [3.9161e-03, 9.9608e-01],
        [2.3433e-03, 9.9766e-01],
        [6.9383e-02, 9.3062e-01],
        [4.0711e-02, 9.5929e-01],
        [3.4144e-03, 9.9659e-01],
        [8.0897e-01, 1.9103e-01],
        [1.9722e-02, 9.8028e-01],
        [1.4390e-02, 9.8561e-01],
        [3.6070e-02, 9.6393e-01],
        [9.7033e-01, 2.9673e-02],
        [2.5346e-02, 9.7465e-01],
        [1.1908e-01, 8.8092e-01],
        [3.0186e-03, 9.9698e-01],
        [6.3378e-02, 9.3662e-01],
        [7.3633e-04, 9.9926e-01],
        [6.6365e-01, 3.3635e-01],
        [2.0141e-02, 9.7986e-01],
        [5.5857e-02, 9.4414e-01],
        [9.5955e-01, 4.0452e-02],
        [4.7801e-02, 9.5220e-01],
        [3.9092e-03, 9.9609e-01],
        [2.0521e-02, 9.7948e-01],
        [6.9470e-02, 9.3053e-01],
        [8.7725e-01, 1.2275e-01],
        [8.2578e-03, 9.9174e-01],
        [2.0119e-02, 9.7988e-01],
        [3.2408e-02, 9.6759e-01],
        [2.3009e-02, 9.7699e-01],
        [4.0614e-02, 9.5939e-01],
        [6.3451e-02, 9.3655e-01],
        [4.2198e-02, 9.5780e-01],
        [1.1245e-01, 8.8755e-01],
        [1.2052e-01, 8.7948e-01],
        [5.7999e-02, 9.4200e-01],
        [6.8408e-03, 9.9316e-01],
        [7.3127e-02, 9.2687e-01],
        [1.1822e-02, 9.8818e-01],
        [9.0805e-02, 9.0920e-01],
        [2.4858e-03, 9.9751e-01],
        [2.1018e-02, 9.7898e-01],
        [2.0820e-03, 9.9792e-01],
        [9.0567e-01, 9.4326e-02],
        [7.6795e-02, 9.2320e-01],
        [1.9587e-03, 9.9804e-01],
        [9.4298e-01, 5.7023e-02],
        [9.4235e-01, 5.7651e-02],
        [3.6358e-03, 9.9636e-01],
        [9.0801e-01, 9.1992e-02],
        [1.8473e-02, 9.8153e-01],
        [1.3429e-02, 9.8657e-01],
        [7.7135e-02, 9.2287e-01],
        [1.5495e-02, 9.8450e-01],
        [1.5654e-03, 9.9843e-01],
        [1.0677e-02, 9.8932e-01],
        [9.2938e-01, 7.0625e-02],
        [2.0810e-02, 9.7919e-01],
        [1.3074e-02, 9.8693e-01],
        [2.0476e-01, 7.9524e-01],
        [7.7063e-01, 2.2937e-01],
        [1.2436e-01, 8.7564e-01],
        [9.3740e-01, 6.2598e-02],
        [1.7585e-03, 9.9824e-01],
        [2.2339e-03, 9.9777e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.0741e-01, 9.2594e-02],
        [4.0157e-03, 9.9598e-01],
        [2.8179e-01, 7.1821e-01],
        [7.2458e-03, 9.9275e-01],
        [8.7813e-01, 1.2187e-01],
        [1.4854e-02, 9.8515e-01],
        [8.8887e-02, 9.1111e-01],
        [9.5322e-01, 4.6776e-02],
        [7.6686e-02, 9.2331e-01],
        [9.4248e-02, 9.0575e-01],
        [8.4811e-01, 1.5189e-01],
        [9.9335e-03, 9.9007e-01],
        [2.9243e-01, 7.0757e-01],
        [6.2953e-01, 3.7047e-01],
        [4.2373e-02, 9.5763e-01],
        [8.4558e-02, 9.1544e-01],
        [3.8689e-03, 9.9613e-01],
        [9.7439e-01, 2.5614e-02],
        [9.0745e-01, 9.2548e-02],
        [4.0551e-01, 5.9449e-01],
        [1.1441e-01, 8.8559e-01],
        [5.0663e-02, 9.4934e-01],
        [4.3854e-03, 9.9561e-01],
        [2.7032e-02, 9.7297e-01],
        [1.0877e-01, 8.9123e-01],
        [5.1295e-03, 9.9487e-01],
        [3.8241e-02, 9.6176e-01],
        [8.3743e-01, 1.6257e-01],
        [9.2199e-01, 7.8009e-02],
        [7.6185e-02, 9.2382e-01],
        [1.3638e-03, 9.9864e-01],
        [6.8080e-02, 9.3192e-01],
        [9.3071e-02, 9.0693e-01],
        [9.4984e-01, 5.0155e-02],
        [9.5692e-01, 4.3078e-02],
        [5.6759e-03, 9.9432e-01],
        [9.0593e-03, 9.9094e-01],
        [5.5574e-02, 9.4443e-01],
        [4.7990e-01, 5.2010e-01],
        [1.7854e-01, 8.2146e-01],
        [6.0589e-01, 3.9411e-01],
        [8.9836e-04, 9.9910e-01],
        [2.0427e-02, 9.7957e-01],
        [4.2484e-03, 9.9575e-01],
        [1.9577e-01, 8.0423e-01],
        [5.9786e-03, 9.9402e-01],
        [9.9134e-01, 8.6597e-03],
        [4.3493e-02, 9.5651e-01],
        [9.8230e-01, 1.7696e-02],
        [2.5594e-02, 9.7441e-01],
        [2.7519e-02, 9.7248e-01],
        [7.5234e-01, 2.4766e-01],
        [9.8464e-01, 1.5361e-02],
        [7.2469e-03, 9.9275e-01],
        [8.9673e-01, 1.0327e-01],
        [2.7752e-02, 9.7225e-01],
        [9.9576e-01, 4.2415e-03],
        [9.2912e-01, 7.0882e-02],
        [3.9785e-02, 9.6022e-01],
        [2.6153e-02, 9.7385e-01],
        [7.4118e-03, 9.9259e-01],
        [2.6600e-02, 9.7340e-01],
        [2.8113e-02, 9.7189e-01],
        [6.9515e-04, 9.9930e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.6667e-02, 9.6333e-01],
        [5.3949e-02, 9.4605e-01],
        [1.6522e-01, 8.3478e-01],
        [8.7154e-01, 1.2846e-01],
        [1.9500e-01, 8.0500e-01],
        [8.4610e-01, 1.5390e-01],
        [1.0176e-02, 9.8982e-01],
        [5.2451e-02, 9.4755e-01],
        [1.2280e-02, 9.8772e-01],
        [1.1756e-02, 9.8824e-01],
        [1.5166e-02, 9.8483e-01],
        [8.8973e-01, 1.1027e-01],
        [1.4827e-03, 9.9852e-01],
        [9.9233e-01, 7.6715e-03],
        [9.3631e-03, 9.9064e-01],
        [2.7407e-02, 9.7259e-01],
        [9.9436e-02, 9.0056e-01],
        [1.0773e-03, 9.9892e-01],
        [1.2326e-01, 8.7674e-01],
        [2.5454e-03, 9.9745e-01],
        [1.2539e-03, 9.9875e-01],
        [3.2579e-02, 9.6742e-01],
        [9.7539e-01, 2.4605e-02],
        [8.9346e-02, 9.1065e-01],
        [1.0611e-02, 9.8939e-01],
        [4.6738e-02, 9.5326e-01],
        [2.4312e-02, 9.7569e-01],
        [9.0491e-01, 9.5091e-02],
        [1.1236e-02, 9.8876e-01],
        [6.8299e-02, 9.3170e-01],
        [1.9426e-02, 9.8057e-01],
        [6.2326e-04, 9.9938e-01],
        [9.6176e-03, 9.9038e-01],
        [2.2604e-01, 7.7396e-01],
        [1.3328e-02, 9.8667e-01],
        [7.7197e-01, 2.2803e-01],
        [1.3815e-01, 8.6185e-01],
        [9.1003e-02, 9.0900e-01],
        [2.1616e-02, 9.7838e-01],
        [1.4443e-03, 9.9856e-01],
        [4.2563e-02, 9.5744e-01],
        [9.6333e-01, 3.6674e-02],
        [1.8916e-01, 8.1084e-01],
        [1.8963e-01, 8.1037e-01],
        [9.2863e-01, 7.1375e-02],
        [7.1964e-01, 2.8036e-01],
        [2.3396e-02, 9.7660e-01],
        [3.3790e-02, 9.6621e-01],
        [3.4669e-02, 9.6533e-01],
        [1.0330e-03, 9.9897e-01],
        [1.3008e-02, 9.8699e-01],
        [4.5926e-01, 5.4074e-01],
        [2.8517e-02, 9.7148e-01],
        [1.0084e-01, 8.9916e-01],
        [8.7029e-01, 1.2971e-01],
        [5.3211e-02, 9.4679e-01],
        [9.8645e-01, 1.3555e-02],
        [3.2667e-02, 9.6733e-01],
        [7.2656e-03, 9.9273e-01],
        [9.6924e-01, 3.0764e-02],
        [9.3602e-01, 6.3976e-02],
        [1.0541e-01, 8.9459e-01],
        [1.1759e-01, 8.8241e-01],
        [6.0586e-04, 9.9939e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[0.2651, 0.7349],
        [0.6117, 0.3883],
        [0.9823, 0.0177],
        [0.0053, 0.9947],
        [0.0343, 0.9657],
        [0.6543, 0.3457],
        [0.0397, 0.9603],
        [0.7495, 0.2505],
        [0.0129, 0.9871],
        [0.2115, 0.7885],
        [0.9015, 0.0985],
        [0.0776, 0.9224],
        [0.0172, 0.9828],
        [0.0427, 0.9573],
        [0.0018, 0.9982],
        [0.0793, 0.9207],
        [0.0251, 0.9749],
        [0.0051, 0.9949],
        [0.0059, 0.9941],
        [0.0169, 0.9831],
        [0.0061, 0.9939],
        [0.0179, 0.9821],
        [0.0816, 0.9184],
        [0.9805, 0.0195],
        [0.5258, 0.4742],
        [0.0272, 0.9728],
        [0.9175, 0.0825],
        [0.0391, 0.9609],
        [0.6972, 0.3028],
        [0.0066, 0.9934],
        [0.0600, 0.9400],
        [0.1022, 0.8978],
        [0.0015, 0.9985],
        [0.0011, 0.9989],
        [0.1799, 0.8201],
        [0.8983, 0.1017],
        [0.1217, 0.8783],
        [0.0134, 0.9866],
        [0.4957, 0.5043],
        [0.4172, 0.5828],
        [0.0599, 0.9401],
        [0.0277, 0.9723],
        [0.0128, 0.9872],
        [0.0044, 0.9956],
        [0.3757, 0.6243],
        [0.9369, 0.0631],
        [0.0046, 0.9954],
        [0.2322, 0.7678],
        [0.9404, 0.0596],
        [0.2021, 0.7979],
        [0.0095, 0.9905],
        [0.1700, 0.8300],
        [0.7524, 0.2476],
        [0.0050, 0.9950],
        [0.0526, 0.9474],
        [0.1160, 0.8840],
        [0.6313, 0.3687],
        [0.0612, 0.9388],
        [0.0902, 0.9098],
        [0.0374, 0.9626],
        [0.0060, 0.9940],
        [0.0859, 0.9141],
        [0.2268, 0.7732],
        [0.0060, 0.9940]], grad_fn=<SoftmaxBackward0>)
tensor([[0.1820, 0.8180],
        [0.9707, 0.0293],
        [0.0959, 0.9041],
        [0.0037, 0.9963],
        [0.0025, 0.9975],
        [0.0079, 0.9921],
        [0.0021, 0.9979],
        [0.0068, 0.9932],
        [0.0830, 0.9170],
        [0.0102, 0.9898],
        [0.8905, 0.1095],
        [0.0348, 0.9652],
        [0.0719, 0.9281],
        [0.0022, 0.9978],
        [0.0081, 0.9919],
        [0.0393, 0.9607],
        [0.7414, 0.2586],
        [0.1311, 0.8689],
        [0.0043, 0.9957],
        [0.0019, 0.9981],
        [0.0048, 0.9952],
        [0.9767, 0.0233],
        [0.0059, 0.9941],
        [0.0028, 0.9972],
        [0.1185, 0.8815],
        [0.0373, 0.9627],
        [0.9750, 0.0250],
        [0.0348, 0.9652],
        [0.0096, 0.9904],
        [0.0032, 0.9968],
        [0.6095, 0.3905],
        [0.5339, 0.4661],
        [0.0197, 0.9803],
        [0.0330, 0.9670],
        [0.0314, 0.9686],
        [0.0091, 0.9909],
        [0.9565, 0.0435],
        [0.0125, 0.9875],
        [0.7067, 0.2933],
        [0.8561, 0.1439],
        [0.0027, 0.9973],
        [0.0023, 0.9977],
        [0.4457, 0.5543],
        [0.0241, 0.9759],
        [0.0339, 0.9661],
        [0.9597, 0.0403],
        [0.0011, 0.9989],
        [0.9798, 0.0202],
        [0.0188, 0.9812],
        [0.0612, 0.9388],
        [0.0331, 0.9669],
        [0.0897, 0.9103],
        [0.0374, 0.9626],
        [0.9313, 0.0687],
        [0.1166, 0.8834],
        [0.3766, 0.6234],
        [0.0013, 0.9987],
        [0.0018, 0.9982],
        [0.9387, 0.0613],
        [0.1588, 0.8412],
        [0.9551, 0.0449],
        [0.0078, 0.9922],
        [0.0134, 0.9866],
        [0.0519, 0.9481]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9935e-02, 9.8007e-01],
        [9.6511e-01, 3.4894e-02],
        [4.4446e-03, 9.9556e-01],
        [1.8973e-02, 9.8103e-01],
        [2.3504e-02, 9.7650e-01],
        [5.6665e-04, 9.9943e-01],
        [1.2614e-01, 8.7386e-01],
        [6.0930e-03, 9.9391e-01],
        [4.7935e-01, 5.2065e-01],
        [7.4303e-03, 9.9257e-01],
        [9.7533e-01, 2.4674e-02],
        [8.9918e-01, 1.0082e-01],
        [9.4113e-01, 5.8874e-02],
        [3.9546e-04, 9.9960e-01],
        [1.3654e-01, 8.6346e-01],
        [9.9887e-01, 1.1275e-03],
        [1.6660e-01, 8.3340e-01],
        [9.5088e-01, 4.9124e-02],
        [1.3962e-01, 8.6038e-01],
        [8.2573e-01, 1.7427e-01],
        [2.1701e-03, 9.9783e-01],
        [6.6004e-02, 9.3400e-01],
        [1.9264e-03, 9.9807e-01],
        [9.6754e-01, 3.2460e-02],
        [9.0250e-01, 9.7502e-02],
        [3.6461e-03, 9.9635e-01],
        [9.9782e-01, 2.1774e-03],
        [1.7418e-01, 8.2582e-01],
        [1.2774e-01, 8.7226e-01],
        [5.1449e-02, 9.4855e-01],
        [7.5044e-01, 2.4956e-01],
        [9.8039e-01, 1.9606e-02],
        [5.2785e-02, 9.4722e-01],
        [3.6700e-02, 9.6330e-01],
        [3.7756e-03, 9.9622e-01],
        [1.0419e-03, 9.9896e-01],
        [1.9610e-02, 9.8039e-01],
        [3.0327e-03, 9.9697e-01],
        [1.4926e-04, 9.9985e-01],
        [8.6119e-03, 9.9139e-01],
        [6.5830e-03, 9.9342e-01],
        [4.8138e-03, 9.9519e-01],
        [9.4515e-01, 5.4849e-02],
        [7.5949e-03, 9.9241e-01],
        [9.7986e-01, 2.0136e-02],
        [5.4186e-02, 9.4581e-01],
        [2.9006e-03, 9.9710e-01],
        [2.8415e-04, 9.9972e-01],
        [6.1838e-02, 9.3816e-01],
        [2.0063e-02, 9.7994e-01],
        [1.1480e-01, 8.8520e-01],
        [1.7752e-03, 9.9822e-01],
        [6.0633e-01, 3.9367e-01],
        [2.9498e-03, 9.9705e-01],
        [6.4727e-03, 9.9353e-01],
        [6.3653e-02, 9.3635e-01],
        [9.6308e-01, 3.6922e-02],
        [6.8142e-01, 3.1858e-01],
        [2.2539e-02, 9.7746e-01],
        [6.2523e-02, 9.3748e-01],
        [9.7818e-01, 2.1819e-02],
        [1.5080e-03, 9.9849e-01],
        [3.9822e-03, 9.9602e-01],
        [9.2083e-01, 7.9174e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[4.8047e-01, 5.1953e-01],
        [4.8950e-04, 9.9951e-01],
        [9.7194e-01, 2.8062e-02],
        [6.6738e-03, 9.9333e-01],
        [5.8622e-01, 4.1378e-01],
        [4.3460e-03, 9.9565e-01],
        [3.1027e-02, 9.6897e-01],
        [1.3306e-03, 9.9867e-01],
        [9.8501e-01, 1.4992e-02],
        [2.3430e-02, 9.7657e-01],
        [1.1825e-02, 9.8818e-01],
        [2.9476e-02, 9.7052e-01],
        [1.8116e-02, 9.8188e-01],
        [5.7578e-02, 9.4242e-01],
        [1.3410e-03, 9.9866e-01],
        [5.2075e-04, 9.9948e-01],
        [4.8795e-04, 9.9951e-01],
        [1.5964e-02, 9.8404e-01],
        [8.9337e-01, 1.0663e-01],
        [3.9581e-04, 9.9960e-01],
        [2.6658e-01, 7.3342e-01],
        [3.7529e-03, 9.9625e-01],
        [1.1352e-01, 8.8648e-01],
        [1.2226e-02, 9.8777e-01],
        [1.5261e-01, 8.4739e-01],
        [1.4943e-02, 9.8506e-01],
        [1.9011e-02, 9.8099e-01],
        [9.8982e-01, 1.0183e-02],
        [3.7760e-02, 9.6224e-01],
        [5.5409e-03, 9.9446e-01],
        [2.6164e-01, 7.3836e-01],
        [2.6018e-01, 7.3982e-01],
        [1.0466e-02, 9.8953e-01],
        [7.2240e-04, 9.9928e-01],
        [8.0686e-02, 9.1931e-01],
        [9.9307e-01, 6.9261e-03],
        [1.1400e-03, 9.9886e-01],
        [1.9896e-02, 9.8010e-01],
        [3.2539e-02, 9.6746e-01],
        [7.9367e-03, 9.9206e-01],
        [3.1758e-01, 6.8242e-01],
        [9.3852e-01, 6.1484e-02],
        [7.3522e-04, 9.9926e-01],
        [4.0579e-03, 9.9594e-01],
        [1.8344e-02, 9.8166e-01],
        [3.9094e-02, 9.6091e-01],
        [9.8141e-01, 1.8594e-02],
        [1.8584e-03, 9.9814e-01],
        [9.1226e-03, 9.9088e-01],
        [1.9593e-02, 9.8041e-01],
        [1.1230e-03, 9.9888e-01],
        [8.4425e-01, 1.5575e-01],
        [5.9627e-03, 9.9404e-01],
        [2.0608e-03, 9.9794e-01],
        [4.3653e-02, 9.5635e-01],
        [2.9773e-02, 9.7023e-01],
        [1.9898e-01, 8.0102e-01],
        [8.6054e-04, 9.9914e-01],
        [8.7457e-03, 9.9125e-01],
        [8.0742e-04, 9.9919e-01],
        [1.5225e-02, 9.8478e-01],
        [9.7361e-01, 2.6393e-02],
        [3.8187e-02, 9.6181e-01],
        [4.9715e-03, 9.9503e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8231e-01, 1.7692e-02],
        [8.7039e-01, 1.2961e-01],
        [5.5985e-03, 9.9440e-01],
        [8.9997e-01, 1.0003e-01],
        [7.3062e-02, 9.2694e-01],
        [6.2233e-02, 9.3777e-01],
        [1.9052e-01, 8.0948e-01],
        [6.6055e-01, 3.3945e-01],
        [2.4844e-03, 9.9752e-01],
        [4.1181e-02, 9.5882e-01],
        [9.5347e-03, 9.9047e-01],
        [7.7731e-01, 2.2269e-01],
        [2.2905e-01, 7.7095e-01],
        [1.3072e-01, 8.6928e-01],
        [6.6928e-01, 3.3072e-01],
        [9.6332e-01, 3.6685e-02],
        [2.0570e-02, 9.7943e-01],
        [6.7652e-03, 9.9323e-01],
        [2.7850e-03, 9.9721e-01],
        [6.8086e-03, 9.9319e-01],
        [2.6460e-04, 9.9974e-01],
        [8.3166e-01, 1.6834e-01],
        [1.3080e-02, 9.8692e-01],
        [2.5611e-03, 9.9744e-01],
        [5.5098e-03, 9.9449e-01],
        [4.3194e-02, 9.5681e-01],
        [7.1219e-02, 9.2878e-01],
        [1.7840e-02, 9.8216e-01],
        [7.3410e-02, 9.2659e-01],
        [9.9153e-01, 8.4725e-03],
        [8.7526e-01, 1.2474e-01],
        [9.4478e-01, 5.5218e-02],
        [2.8353e-02, 9.7165e-01],
        [2.4097e-02, 9.7590e-01],
        [1.4875e-03, 9.9851e-01],
        [6.1248e-03, 9.9388e-01],
        [1.7542e-01, 8.2458e-01],
        [7.3473e-04, 9.9927e-01],
        [4.5435e-02, 9.5456e-01],
        [1.2120e-03, 9.9879e-01],
        [3.7293e-03, 9.9627e-01],
        [6.1145e-02, 9.3886e-01],
        [1.0519e-03, 9.9895e-01],
        [3.2214e-02, 9.6779e-01],
        [6.4320e-03, 9.9357e-01],
        [2.8828e-02, 9.7117e-01],
        [2.4802e-02, 9.7520e-01],
        [9.9035e-01, 9.6451e-03],
        [1.5125e-02, 9.8488e-01],
        [1.2784e-03, 9.9872e-01],
        [9.8079e-01, 1.9214e-02],
        [1.6898e-01, 8.3102e-01],
        [7.9208e-01, 2.0792e-01],
        [9.7355e-01, 2.6450e-02],
        [2.3205e-02, 9.7679e-01],
        [2.5744e-02, 9.7426e-01],
        [3.8884e-02, 9.6112e-01],
        [4.4202e-02, 9.5580e-01],
        [8.4617e-01, 1.5383e-01],
        [9.3549e-03, 9.9065e-01],
        [5.9802e-02, 9.4020e-01],
        [4.1051e-03, 9.9589e-01],
        [3.2705e-02, 9.6729e-01],
        [5.8259e-01, 4.1741e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.5110e-02, 9.4489e-01],
        [9.8615e-03, 9.9014e-01],
        [2.2224e-03, 9.9778e-01],
        [1.3548e-03, 9.9865e-01],
        [2.7161e-03, 9.9728e-01],
        [1.9745e-01, 8.0255e-01],
        [8.2240e-01, 1.7760e-01],
        [9.7055e-01, 2.9453e-02],
        [4.4340e-01, 5.5660e-01],
        [5.3649e-01, 4.6351e-01],
        [1.5802e-04, 9.9984e-01],
        [3.7700e-04, 9.9962e-01],
        [9.8637e-01, 1.3626e-02],
        [7.7174e-01, 2.2826e-01],
        [1.1291e-02, 9.8871e-01],
        [1.6795e-02, 9.8320e-01],
        [1.5243e-02, 9.8476e-01],
        [5.8066e-03, 9.9419e-01],
        [9.7289e-01, 2.7105e-02],
        [9.9069e-01, 9.3108e-03],
        [5.1645e-02, 9.4835e-01],
        [1.9508e-03, 9.9805e-01],
        [1.2007e-02, 9.8799e-01],
        [3.7633e-03, 9.9624e-01],
        [8.1353e-04, 9.9919e-01],
        [9.5280e-01, 4.7199e-02],
        [2.5729e-02, 9.7427e-01],
        [4.9539e-02, 9.5046e-01],
        [6.5643e-04, 9.9934e-01],
        [9.3903e-01, 6.0968e-02],
        [3.0147e-04, 9.9970e-01],
        [1.4668e-02, 9.8533e-01],
        [2.6199e-01, 7.3801e-01],
        [1.2439e-01, 8.7561e-01],
        [5.3623e-04, 9.9946e-01],
        [5.1321e-03, 9.9487e-01],
        [4.7309e-01, 5.2691e-01],
        [1.1615e-03, 9.9884e-01],
        [2.6062e-04, 9.9974e-01],
        [2.6705e-03, 9.9733e-01],
        [2.9411e-02, 9.7059e-01],
        [7.9760e-04, 9.9920e-01],
        [6.9606e-02, 9.3039e-01],
        [4.3065e-01, 5.6935e-01],
        [5.4549e-04, 9.9945e-01],
        [9.9570e-01, 4.2952e-03],
        [9.3571e-03, 9.9064e-01],
        [2.3702e-02, 9.7630e-01],
        [1.4343e-02, 9.8566e-01],
        [8.8320e-03, 9.9117e-01],
        [6.8039e-02, 9.3196e-01],
        [1.3451e-02, 9.8655e-01],
        [9.4842e-01, 5.1579e-02],
        [3.1840e-02, 9.6816e-01],
        [1.6409e-01, 8.3591e-01],
        [7.3883e-02, 9.2612e-01],
        [5.7278e-04, 9.9943e-01],
        [2.2591e-01, 7.7409e-01],
        [9.0054e-01, 9.9456e-02],
        [6.1021e-01, 3.8979e-01],
        [6.8012e-03, 9.9320e-01],
        [2.9822e-02, 9.7018e-01],
        [6.0169e-02, 9.3983e-01],
        [6.4709e-02, 9.3529e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3925e-02, 9.8608e-01],
        [5.0422e-01, 4.9578e-01],
        [2.1081e-02, 9.7892e-01],
        [4.1125e-01, 5.8875e-01],
        [8.1354e-03, 9.9186e-01],
        [1.9386e-03, 9.9806e-01],
        [1.2949e-04, 9.9987e-01],
        [3.2542e-03, 9.9675e-01],
        [8.9229e-04, 9.9911e-01],
        [9.5575e-02, 9.0442e-01],
        [2.0629e-03, 9.9794e-01],
        [7.3532e-04, 9.9926e-01],
        [8.1743e-02, 9.1826e-01],
        [3.5571e-03, 9.9644e-01],
        [1.8484e-03, 9.9815e-01],
        [4.5853e-02, 9.5415e-01],
        [2.3962e-02, 9.7604e-01],
        [2.2508e-02, 9.7749e-01],
        [7.9575e-05, 9.9992e-01],
        [7.7401e-01, 2.2599e-01],
        [5.3908e-02, 9.4609e-01],
        [2.6705e-02, 9.7330e-01],
        [1.7355e-02, 9.8264e-01],
        [7.7009e-01, 2.2991e-01],
        [2.7007e-03, 9.9730e-01],
        [2.4503e-03, 9.9755e-01],
        [9.9828e-01, 1.7229e-03],
        [1.8815e-03, 9.9812e-01],
        [9.9117e-01, 8.8303e-03],
        [8.3373e-02, 9.1663e-01],
        [2.1183e-02, 9.7882e-01],
        [2.1567e-02, 9.7843e-01],
        [7.7512e-03, 9.9225e-01],
        [4.7843e-03, 9.9522e-01],
        [9.8903e-01, 1.0969e-02],
        [8.0553e-01, 1.9447e-01],
        [2.7032e-03, 9.9730e-01],
        [1.3407e-01, 8.6593e-01],
        [8.6682e-01, 1.3318e-01],
        [3.9412e-03, 9.9606e-01],
        [3.2189e-04, 9.9968e-01],
        [1.5492e-03, 9.9845e-01],
        [6.8655e-03, 9.9313e-01],
        [2.9427e-03, 9.9706e-01],
        [8.2279e-02, 9.1772e-01],
        [6.0315e-01, 3.9685e-01],
        [1.4386e-03, 9.9856e-01],
        [5.3766e-02, 9.4623e-01],
        [2.8838e-04, 9.9971e-01],
        [3.5407e-01, 6.4593e-01],
        [9.7500e-04, 9.9902e-01],
        [1.1891e-01, 8.8109e-01],
        [8.6720e-01, 1.3280e-01],
        [6.1925e-04, 9.9938e-01],
        [9.4736e-01, 5.2640e-02],
        [9.1971e-01, 8.0286e-02],
        [7.7280e-04, 9.9923e-01],
        [1.6809e-01, 8.3191e-01],
        [8.8001e-02, 9.1200e-01],
        [2.3365e-04, 9.9977e-01],
        [3.6101e-01, 6.3899e-01],
        [1.8244e-02, 9.8176e-01],
        [9.8468e-01, 1.5320e-02],
        [1.4214e-01, 8.5786e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6852e-03, 9.9831e-01],
        [1.4401e-03, 9.9856e-01],
        [4.4253e-03, 9.9557e-01],
        [2.5472e-04, 9.9975e-01],
        [3.7083e-03, 9.9629e-01],
        [5.9278e-02, 9.4072e-01],
        [7.7782e-04, 9.9922e-01],
        [9.1432e-01, 8.5681e-02],
        [1.7981e-02, 9.8202e-01],
        [2.4664e-04, 9.9975e-01],
        [6.2856e-03, 9.9371e-01],
        [7.9477e-02, 9.2052e-01],
        [1.1551e-02, 9.8845e-01],
        [4.8050e-02, 9.5195e-01],
        [9.9397e-01, 6.0344e-03],
        [9.2567e-01, 7.4333e-02],
        [8.2151e-01, 1.7849e-01],
        [9.4051e-01, 5.9489e-02],
        [9.1714e-01, 8.2858e-02],
        [4.9114e-01, 5.0886e-01],
        [3.2413e-01, 6.7587e-01],
        [1.1832e-02, 9.8817e-01],
        [9.7972e-01, 2.0283e-02],
        [4.4346e-03, 9.9557e-01],
        [1.4027e-03, 9.9860e-01],
        [9.8739e-01, 1.2613e-02],
        [3.0399e-02, 9.6960e-01],
        [7.1416e-01, 2.8584e-01],
        [4.0306e-01, 5.9694e-01],
        [1.8908e-02, 9.8109e-01],
        [8.4407e-01, 1.5593e-01],
        [2.1854e-02, 9.7815e-01],
        [4.5785e-02, 9.5421e-01],
        [9.4499e-04, 9.9905e-01],
        [8.6762e-01, 1.3238e-01],
        [7.2918e-01, 2.7082e-01],
        [3.7146e-04, 9.9963e-01],
        [2.6050e-02, 9.7395e-01],
        [3.6322e-02, 9.6368e-01],
        [4.7587e-03, 9.9524e-01],
        [1.3678e-02, 9.8632e-01],
        [4.9248e-04, 9.9951e-01],
        [9.1857e-03, 9.9081e-01],
        [4.0214e-04, 9.9960e-01],
        [1.4933e-01, 8.5067e-01],
        [8.9338e-01, 1.0662e-01],
        [3.8931e-02, 9.6107e-01],
        [4.6956e-01, 5.3044e-01],
        [1.6095e-02, 9.8391e-01],
        [7.4755e-01, 2.5245e-01],
        [3.0656e-03, 9.9693e-01],
        [9.4829e-01, 5.1708e-02],
        [6.0660e-04, 9.9939e-01],
        [9.9188e-01, 8.1211e-03],
        [3.4959e-02, 9.6504e-01],
        [1.2530e-03, 9.9875e-01],
        [9.6706e-01, 3.2938e-02],
        [9.8125e-01, 1.8751e-02],
        [2.0475e-03, 9.9795e-01],
        [9.4571e-01, 5.4292e-02],
        [9.8625e-01, 1.3749e-02],
        [2.4071e-03, 9.9759e-01],
        [2.9242e-04, 9.9971e-01],
        [8.4031e-04, 9.9916e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.9756e-03, 9.9202e-01],
        [2.4742e-01, 7.5258e-01],
        [9.3613e-01, 6.3871e-02],
        [2.4281e-01, 7.5719e-01],
        [7.0191e-03, 9.9298e-01],
        [6.3428e-04, 9.9937e-01],
        [1.4381e-03, 9.9856e-01],
        [9.5815e-01, 4.1846e-02],
        [1.0835e-03, 9.9892e-01],
        [9.8892e-01, 1.1082e-02],
        [9.6716e-03, 9.9033e-01],
        [5.2320e-03, 9.9477e-01],
        [7.9469e-02, 9.2053e-01],
        [9.4273e-01, 5.7274e-02],
        [2.4387e-02, 9.7561e-01],
        [1.9346e-02, 9.8065e-01],
        [3.0774e-02, 9.6923e-01],
        [6.2400e-02, 9.3760e-01],
        [1.2375e-03, 9.9876e-01],
        [2.8315e-03, 9.9717e-01],
        [1.4705e-03, 9.9853e-01],
        [1.2776e-02, 9.8722e-01],
        [5.7156e-01, 4.2844e-01],
        [1.3529e-01, 8.6471e-01],
        [3.8278e-04, 9.9962e-01],
        [8.9940e-03, 9.9101e-01],
        [2.4732e-04, 9.9975e-01],
        [8.9129e-01, 1.0871e-01],
        [4.8446e-02, 9.5155e-01],
        [3.3193e-02, 9.6681e-01],
        [1.2503e-03, 9.9875e-01],
        [5.4058e-02, 9.4594e-01],
        [4.6229e-04, 9.9954e-01],
        [4.6854e-03, 9.9531e-01],
        [2.9196e-03, 9.9708e-01],
        [5.6955e-04, 9.9943e-01],
        [1.2711e-01, 8.7289e-01],
        [1.3817e-03, 9.9862e-01],
        [1.4233e-03, 9.9858e-01],
        [5.4267e-02, 9.4573e-01],
        [8.1295e-03, 9.9187e-01],
        [1.4852e-02, 9.8515e-01],
        [6.7291e-02, 9.3271e-01],
        [8.8212e-03, 9.9118e-01],
        [2.2033e-01, 7.7967e-01],
        [3.5207e-03, 9.9648e-01],
        [2.2315e-03, 9.9777e-01],
        [8.7621e-01, 1.2379e-01],
        [1.7688e-03, 9.9823e-01],
        [7.6265e-03, 9.9237e-01],
        [9.0503e-01, 9.4975e-02],
        [2.6147e-03, 9.9739e-01],
        [3.4136e-04, 9.9966e-01],
        [5.2452e-03, 9.9475e-01],
        [2.2881e-01, 7.7119e-01],
        [8.0859e-03, 9.9191e-01],
        [2.1347e-02, 9.7865e-01],
        [6.1434e-03, 9.9386e-01],
        [4.4370e-03, 9.9556e-01],
        [6.5155e-03, 9.9348e-01],
        [3.7111e-01, 6.2889e-01],
        [1.5335e-01, 8.4665e-01],
        [2.5865e-02, 9.7413e-01],
        [5.2210e-03, 9.9478e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7375e-02, 9.6262e-01],
        [3.8210e-04, 9.9962e-01],
        [1.0528e-02, 9.8947e-01],
        [4.6823e-03, 9.9532e-01],
        [1.1562e-01, 8.8438e-01],
        [2.8345e-02, 9.7166e-01],
        [8.7786e-01, 1.2214e-01],
        [9.5340e-03, 9.9047e-01],
        [9.8922e-01, 1.0776e-02],
        [5.5176e-01, 4.4824e-01],
        [9.8887e-01, 1.1134e-02],
        [1.7468e-02, 9.8253e-01],
        [2.3716e-03, 9.9763e-01],
        [3.3621e-03, 9.9664e-01],
        [2.4977e-01, 7.5023e-01],
        [9.5139e-01, 4.8607e-02],
        [9.9109e-01, 8.9135e-03],
        [8.1773e-01, 1.8227e-01],
        [1.1481e-02, 9.8852e-01],
        [5.0769e-04, 9.9949e-01],
        [8.5274e-01, 1.4726e-01],
        [4.3302e-04, 9.9957e-01],
        [9.8415e-01, 1.5850e-02],
        [9.3122e-01, 6.8783e-02],
        [7.3391e-01, 2.6609e-01],
        [7.6787e-05, 9.9992e-01],
        [1.5815e-02, 9.8419e-01],
        [3.5114e-02, 9.6489e-01],
        [1.7710e-02, 9.8229e-01],
        [5.2044e-03, 9.9480e-01],
        [9.7058e-01, 2.9418e-02],
        [3.8376e-03, 9.9616e-01],
        [2.4481e-03, 9.9755e-01],
        [3.3844e-02, 9.6616e-01],
        [5.7105e-01, 4.2895e-01],
        [6.4293e-01, 3.5707e-01],
        [1.2121e-02, 9.8788e-01],
        [4.7737e-02, 9.5226e-01],
        [8.2551e-01, 1.7449e-01],
        [9.2275e-03, 9.9077e-01],
        [5.3859e-03, 9.9461e-01],
        [5.7638e-02, 9.4236e-01],
        [8.4040e-01, 1.5960e-01],
        [2.7849e-02, 9.7215e-01],
        [3.4437e-03, 9.9656e-01],
        [1.3418e-02, 9.8658e-01],
        [2.7334e-02, 9.7267e-01],
        [8.2065e-03, 9.9179e-01],
        [4.7989e-03, 9.9520e-01],
        [3.3260e-02, 9.6674e-01],
        [8.9593e-01, 1.0407e-01],
        [9.6090e-01, 3.9102e-02],
        [7.7644e-03, 9.9224e-01],
        [2.9182e-02, 9.7082e-01],
        [3.1012e-04, 9.9969e-01],
        [1.7163e-02, 9.8284e-01],
        [7.3848e-03, 9.9262e-01],
        [1.1621e-02, 9.8838e-01],
        [2.9390e-03, 9.9706e-01],
        [1.9194e-02, 9.8081e-01],
        [1.3057e-02, 9.8694e-01],
        [2.2966e-03, 9.9770e-01],
        [9.4895e-01, 5.1047e-02],
        [1.3418e-03, 9.9866e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.2992e-03, 9.9470e-01],
        [4.5140e-04, 9.9955e-01],
        [9.3058e-01, 6.9425e-02],
        [5.7447e-04, 9.9943e-01],
        [3.1258e-03, 9.9687e-01],
        [1.7307e-02, 9.8269e-01],
        [1.5393e-02, 9.8461e-01],
        [8.0242e-03, 9.9198e-01],
        [1.0462e-04, 9.9990e-01],
        [3.6736e-03, 9.9633e-01],
        [1.2815e-03, 9.9872e-01],
        [1.1680e-01, 8.8320e-01],
        [9.9146e-01, 8.5411e-03],
        [1.3310e-03, 9.9867e-01],
        [3.0916e-03, 9.9691e-01],
        [8.8989e-01, 1.1011e-01],
        [8.3608e-01, 1.6392e-01],
        [3.5899e-01, 6.4101e-01],
        [5.7073e-03, 9.9429e-01],
        [4.6997e-02, 9.5300e-01],
        [2.9225e-03, 9.9708e-01],
        [1.0374e-03, 9.9896e-01],
        [9.3021e-01, 6.9791e-02],
        [7.9536e-01, 2.0464e-01],
        [5.0633e-03, 9.9494e-01],
        [5.7995e-01, 4.2005e-01],
        [5.5305e-02, 9.4469e-01],
        [5.7852e-01, 4.2148e-01],
        [7.8869e-03, 9.9211e-01],
        [4.1065e-04, 9.9959e-01],
        [7.5415e-03, 9.9246e-01],
        [1.3848e-02, 9.8615e-01],
        [1.5644e-02, 9.8436e-01],
        [3.5328e-02, 9.6467e-01],
        [2.7752e-02, 9.7225e-01],
        [1.4971e-03, 9.9850e-01],
        [7.1712e-03, 9.9283e-01],
        [4.8549e-02, 9.5145e-01],
        [4.2184e-04, 9.9958e-01],
        [9.2000e-05, 9.9991e-01],
        [6.7018e-02, 9.3298e-01],
        [1.2826e-03, 9.9872e-01],
        [9.0237e-03, 9.9098e-01],
        [9.7964e-01, 2.0362e-02],
        [1.3703e-04, 9.9986e-01],
        [5.9139e-04, 9.9941e-01],
        [2.5385e-02, 9.7461e-01],
        [5.1628e-02, 9.4837e-01],
        [4.4697e-03, 9.9553e-01],
        [1.3772e-04, 9.9986e-01],
        [2.5890e-02, 9.7411e-01],
        [3.2883e-03, 9.9671e-01],
        [9.3598e-01, 6.4019e-02],
        [3.9452e-04, 9.9961e-01],
        [2.0106e-03, 9.9799e-01],
        [1.7510e-04, 9.9982e-01],
        [7.6062e-01, 2.3938e-01],
        [5.6744e-03, 9.9433e-01],
        [1.9324e-02, 9.8068e-01],
        [8.7976e-04, 9.9912e-01],
        [8.2186e-03, 9.9178e-01],
        [8.3177e-03, 9.9168e-01],
        [2.3663e-03, 9.9763e-01],
        [4.0976e-03, 9.9590e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9955e-03, 9.9800e-01],
        [1.4448e-02, 9.8555e-01],
        [1.0834e-02, 9.8917e-01],
        [1.0576e-01, 8.9424e-01],
        [1.2192e-03, 9.9878e-01],
        [2.2983e-02, 9.7702e-01],
        [1.8506e-02, 9.8149e-01],
        [2.0796e-03, 9.9792e-01],
        [9.7111e-01, 2.8887e-02],
        [7.4818e-01, 2.5182e-01],
        [9.6437e-01, 3.5627e-02],
        [4.0719e-02, 9.5928e-01],
        [1.7086e-03, 9.9829e-01],
        [1.1773e-01, 8.8227e-01],
        [3.9280e-04, 9.9961e-01],
        [2.1338e-02, 9.7866e-01],
        [9.2688e-01, 7.3124e-02],
        [9.6286e-01, 3.7138e-02],
        [2.2044e-02, 9.7796e-01],
        [4.2972e-04, 9.9957e-01],
        [3.6220e-02, 9.6378e-01],
        [2.1662e-03, 9.9783e-01],
        [6.6210e-04, 9.9934e-01],
        [2.5226e-03, 9.9748e-01],
        [8.2995e-01, 1.7005e-01],
        [9.9535e-01, 4.6465e-03],
        [1.2186e-01, 8.7814e-01],
        [4.0434e-02, 9.5957e-01],
        [5.8084e-03, 9.9419e-01],
        [5.5206e-04, 9.9945e-01],
        [7.0645e-03, 9.9294e-01],
        [3.6474e-03, 9.9635e-01],
        [9.6897e-01, 3.1033e-02],
        [8.6727e-01, 1.3273e-01],
        [2.2684e-03, 9.9773e-01],
        [2.1844e-03, 9.9782e-01],
        [5.1334e-02, 9.4867e-01],
        [2.1196e-02, 9.7880e-01],
        [1.8350e-03, 9.9816e-01],
        [9.9214e-01, 7.8631e-03],
        [9.7328e-01, 2.6717e-02],
        [2.7639e-02, 9.7236e-01],
        [9.6974e-03, 9.9030e-01],
        [1.0904e-01, 8.9096e-01],
        [6.7752e-02, 9.3225e-01],
        [1.2214e-03, 9.9878e-01],
        [4.5445e-03, 9.9546e-01],
        [7.8195e-04, 9.9922e-01],
        [4.2300e-02, 9.5770e-01],
        [9.9860e-01, 1.4024e-03],
        [6.0263e-04, 9.9940e-01],
        [3.7490e-02, 9.6251e-01],
        [4.7405e-01, 5.2595e-01],
        [9.7574e-01, 2.4260e-02],
        [8.8885e-03, 9.9111e-01],
        [3.5101e-04, 9.9965e-01],
        [3.0653e-04, 9.9969e-01],
        [2.7268e-02, 9.7273e-01],
        [6.9707e-03, 9.9303e-01],
        [9.1405e-01, 8.5953e-02],
        [1.5957e-03, 9.9840e-01],
        [5.3251e-02, 9.4675e-01],
        [3.1773e-04, 9.9968e-01],
        [1.7514e-03, 9.9825e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8078e-03, 9.9819e-01],
        [1.3970e-02, 9.8603e-01],
        [3.9187e-05, 9.9996e-01],
        [2.7925e-05, 9.9997e-01],
        [1.6159e-02, 9.8384e-01],
        [4.4443e-03, 9.9556e-01],
        [2.4083e-01, 7.5917e-01],
        [3.5543e-04, 9.9964e-01],
        [9.9273e-01, 7.2713e-03],
        [9.9195e-01, 8.0485e-03],
        [3.2814e-02, 9.6719e-01],
        [9.7690e-01, 2.3104e-02],
        [3.0278e-03, 9.9697e-01],
        [1.2034e-03, 9.9880e-01],
        [1.9893e-04, 9.9980e-01],
        [1.5572e-02, 9.8443e-01],
        [9.8768e-01, 1.2320e-02],
        [3.3644e-04, 9.9966e-01],
        [9.7556e-03, 9.9024e-01],
        [8.9655e-01, 1.0345e-01],
        [5.4941e-04, 9.9945e-01],
        [3.1582e-03, 9.9684e-01],
        [9.9578e-01, 4.2237e-03],
        [2.7075e-01, 7.2925e-01],
        [1.2509e-03, 9.9875e-01],
        [1.6055e-04, 9.9984e-01],
        [1.5807e-02, 9.8419e-01],
        [9.9740e-01, 2.5963e-03],
        [9.0772e-01, 9.2279e-02],
        [1.0457e-02, 9.8954e-01],
        [4.1983e-01, 5.8017e-01],
        [1.2109e-02, 9.8789e-01],
        [1.8491e-02, 9.8151e-01],
        [9.7272e-01, 2.7276e-02],
        [9.9318e-01, 6.8201e-03],
        [8.7952e-05, 9.9991e-01],
        [4.9351e-03, 9.9506e-01],
        [8.4098e-01, 1.5902e-01],
        [5.6601e-03, 9.9434e-01],
        [6.4643e-03, 9.9354e-01],
        [2.8402e-01, 7.1598e-01],
        [1.7058e-02, 9.8294e-01],
        [9.9200e-01, 8.0021e-03],
        [8.0175e-01, 1.9825e-01],
        [3.8259e-01, 6.1741e-01],
        [7.4255e-02, 9.2575e-01],
        [1.0153e-01, 8.9847e-01],
        [1.0261e-03, 9.9897e-01],
        [2.0230e-03, 9.9798e-01],
        [2.4469e-03, 9.9755e-01],
        [9.5641e-01, 4.3591e-02],
        [1.1293e-01, 8.8707e-01],
        [9.8749e-01, 1.2511e-02],
        [9.9170e-01, 8.2965e-03],
        [6.1464e-04, 9.9939e-01],
        [1.7737e-01, 8.2263e-01],
        [1.4438e-03, 9.9856e-01],
        [5.4640e-03, 9.9454e-01],
        [6.3050e-02, 9.3695e-01],
        [1.5250e-02, 9.8475e-01],
        [2.7017e-04, 9.9973e-01],
        [7.4408e-04, 9.9926e-01],
        [1.6199e-02, 9.8380e-01],
        [6.8063e-03, 9.9319e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8892e-02, 9.8111e-01],
        [5.9444e-04, 9.9941e-01],
        [1.5072e-02, 9.8493e-01],
        [1.8568e-01, 8.1432e-01],
        [3.3990e-02, 9.6601e-01],
        [9.5957e-01, 4.0434e-02],
        [9.9941e-01, 5.8669e-04],
        [9.6935e-01, 3.0653e-02],
        [4.4560e-02, 9.5544e-01],
        [4.4599e-02, 9.5540e-01],
        [1.2650e-04, 9.9987e-01],
        [2.6140e-04, 9.9974e-01],
        [9.8405e-01, 1.5953e-02],
        [1.2448e-02, 9.8755e-01],
        [6.5883e-01, 3.4117e-01],
        [9.3644e-04, 9.9906e-01],
        [3.7780e-01, 6.2220e-01],
        [2.3700e-03, 9.9763e-01],
        [3.6640e-01, 6.3360e-01],
        [2.1266e-03, 9.9787e-01],
        [1.1071e-01, 8.8929e-01],
        [9.3074e-01, 6.9264e-02],
        [1.2032e-03, 9.9880e-01],
        [3.3919e-02, 9.6608e-01],
        [6.9204e-03, 9.9308e-01],
        [9.8938e-01, 1.0620e-02],
        [2.1318e-01, 7.8682e-01],
        [5.8703e-03, 9.9413e-01],
        [1.7756e-03, 9.9822e-01],
        [6.7189e-01, 3.2811e-01],
        [1.1669e-01, 8.8331e-01],
        [2.7109e-05, 9.9997e-01],
        [9.8304e-01, 1.6956e-02],
        [3.9064e-02, 9.6094e-01],
        [1.9559e-03, 9.9804e-01],
        [8.9974e-01, 1.0026e-01],
        [7.5822e-04, 9.9924e-01],
        [2.5264e-01, 7.4736e-01],
        [8.1751e-01, 1.8249e-01],
        [2.2084e-01, 7.7916e-01],
        [7.2815e-03, 9.9272e-01],
        [1.7655e-02, 9.8234e-01],
        [6.8148e-01, 3.1852e-01],
        [5.7812e-02, 9.4219e-01],
        [1.1703e-02, 9.8830e-01],
        [9.1368e-01, 8.6315e-02],
        [1.5049e-03, 9.9850e-01],
        [6.3998e-01, 3.6002e-01],
        [2.5123e-02, 9.7488e-01],
        [4.4708e-02, 9.5529e-01],
        [8.8356e-02, 9.1164e-01],
        [9.0702e-01, 9.2975e-02],
        [9.5640e-01, 4.3600e-02],
        [5.5088e-01, 4.4912e-01],
        [2.4354e-01, 7.5646e-01],
        [1.1283e-01, 8.8717e-01],
        [6.6336e-01, 3.3664e-01],
        [4.3896e-04, 9.9956e-01],
        [1.6089e-03, 9.9839e-01],
        [9.0823e-01, 9.1768e-02],
        [2.4651e-04, 9.9975e-01],
        [8.3056e-01, 1.6944e-01],
        [9.9015e-01, 9.8490e-03],
        [9.9331e-01, 6.6882e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8404e-03, 9.9016e-01],
        [9.8835e-01, 1.1650e-02],
        [4.8021e-04, 9.9952e-01],
        [8.8140e-01, 1.1860e-01],
        [9.2285e-01, 7.7149e-02],
        [8.9987e-01, 1.0013e-01],
        [9.9739e-01, 2.6055e-03],
        [8.2740e-02, 9.1726e-01],
        [9.5292e-01, 4.7084e-02],
        [8.8293e-01, 1.1707e-01],
        [3.0565e-02, 9.6943e-01],
        [2.5490e-02, 9.7451e-01],
        [8.3143e-01, 1.6857e-01],
        [9.2440e-01, 7.5597e-02],
        [1.6989e-03, 9.9830e-01],
        [1.2243e-02, 9.8776e-01],
        [2.5262e-02, 9.7474e-01],
        [1.3294e-02, 9.8671e-01],
        [1.0644e-02, 9.8936e-01],
        [4.2577e-02, 9.5742e-01],
        [9.9368e-01, 6.3239e-03],
        [2.6607e-04, 9.9973e-01],
        [9.8267e-01, 1.7332e-02],
        [4.6252e-02, 9.5375e-01],
        [1.6419e-03, 9.9836e-01],
        [1.8076e-02, 9.8192e-01],
        [2.7339e-01, 7.2661e-01],
        [4.6640e-03, 9.9534e-01],
        [5.8379e-01, 4.1621e-01],
        [7.9504e-01, 2.0496e-01],
        [6.4108e-02, 9.3589e-01],
        [3.9633e-02, 9.6037e-01],
        [1.8637e-01, 8.1363e-01],
        [1.7266e-04, 9.9983e-01],
        [1.7592e-02, 9.8241e-01],
        [9.8404e-03, 9.9016e-01],
        [2.1212e-02, 9.7879e-01],
        [5.5082e-02, 9.4492e-01],
        [4.1318e-02, 9.5868e-01],
        [1.2000e-02, 9.8800e-01],
        [5.9695e-04, 9.9940e-01],
        [1.3391e-01, 8.6609e-01],
        [8.8023e-01, 1.1977e-01],
        [5.3222e-02, 9.4678e-01],
        [3.0176e-01, 6.9824e-01],
        [1.9692e-03, 9.9803e-01],
        [3.0383e-02, 9.6962e-01],
        [1.5767e-04, 9.9984e-01],
        [9.9146e-01, 8.5449e-03],
        [5.8217e-01, 4.1783e-01],
        [8.2396e-03, 9.9176e-01],
        [3.5148e-04, 9.9965e-01],
        [1.8864e-03, 9.9811e-01],
        [9.7901e-01, 2.0986e-02],
        [5.1200e-04, 9.9949e-01],
        [8.6397e-01, 1.3603e-01],
        [9.8242e-01, 1.7579e-02],
        [2.3635e-05, 9.9998e-01],
        [1.8005e-04, 9.9982e-01],
        [2.1604e-03, 9.9784e-01],
        [2.0567e-02, 9.7943e-01],
        [1.4498e-04, 9.9986e-01],
        [1.6999e-03, 9.9830e-01],
        [2.2954e-03, 9.9770e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.7137e-01, 2.8628e-02],
        [3.0171e-03, 9.9698e-01],
        [8.0342e-01, 1.9658e-01],
        [8.0339e-02, 9.1966e-01],
        [5.2845e-03, 9.9472e-01],
        [3.1485e-04, 9.9969e-01],
        [9.8762e-01, 1.2378e-02],
        [1.8484e-04, 9.9982e-01],
        [1.4544e-02, 9.8546e-01],
        [7.8528e-01, 2.1472e-01],
        [9.7805e-01, 2.1954e-02],
        [9.3066e-04, 9.9907e-01],
        [2.1780e-03, 9.9782e-01],
        [8.9625e-04, 9.9910e-01],
        [5.1792e-02, 9.4821e-01],
        [2.0987e-02, 9.7901e-01],
        [9.0806e-01, 9.1944e-02],
        [2.4038e-03, 9.9760e-01],
        [9.3509e-02, 9.0649e-01],
        [6.1144e-04, 9.9939e-01],
        [3.6922e-02, 9.6308e-01],
        [1.7803e-02, 9.8220e-01],
        [9.9916e-01, 8.4378e-04],
        [2.9616e-02, 9.7038e-01],
        [1.8840e-01, 8.1160e-01],
        [8.4119e-01, 1.5881e-01],
        [8.3673e-05, 9.9992e-01],
        [1.7454e-02, 9.8255e-01],
        [4.9019e-01, 5.0981e-01],
        [9.8366e-01, 1.6337e-02],
        [1.1435e-04, 9.9989e-01],
        [4.4968e-02, 9.5503e-01],
        [1.2403e-03, 9.9876e-01],
        [5.0016e-01, 4.9984e-01],
        [9.4222e-03, 9.9058e-01],
        [5.4082e-02, 9.4592e-01],
        [8.8979e-01, 1.1021e-01],
        [1.5922e-01, 8.4078e-01],
        [3.2607e-03, 9.9674e-01],
        [1.0301e-02, 9.8970e-01],
        [9.8682e-04, 9.9901e-01],
        [6.8374e-04, 9.9932e-01],
        [9.0681e-01, 9.3189e-02],
        [1.7828e-03, 9.9822e-01],
        [6.5370e-04, 9.9935e-01],
        [2.2880e-06, 1.0000e+00],
        [1.9897e-03, 9.9801e-01],
        [2.7337e-01, 7.2663e-01],
        [1.8254e-02, 9.8175e-01],
        [1.1982e-02, 9.8802e-01],
        [9.9969e-01, 3.1127e-04],
        [8.9533e-03, 9.9105e-01],
        [8.0803e-05, 9.9992e-01],
        [1.9889e-02, 9.8011e-01],
        [2.6604e-02, 9.7340e-01],
        [3.0453e-04, 9.9970e-01],
        [9.5713e-01, 4.2865e-02],
        [1.9573e-02, 9.8043e-01],
        [9.9338e-01, 6.6215e-03],
        [6.2623e-01, 3.7377e-01],
        [1.3093e-02, 9.8691e-01],
        [7.0692e-02, 9.2931e-01],
        [8.4904e-03, 9.9151e-01],
        [7.6764e-02, 9.2324e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.0236e-04, 9.9950e-01],
        [1.1179e-02, 9.8882e-01],
        [7.5730e-03, 9.9243e-01],
        [4.6304e-04, 9.9954e-01],
        [6.7421e-03, 9.9326e-01],
        [6.6238e-05, 9.9993e-01],
        [8.3187e-01, 1.6813e-01],
        [9.9767e-01, 2.3291e-03],
        [9.8133e-01, 1.8674e-02],
        [9.8925e-01, 1.0752e-02],
        [1.7837e-03, 9.9822e-01],
        [4.7948e-03, 9.9521e-01],
        [7.7440e-01, 2.2560e-01],
        [5.4988e-03, 9.9450e-01],
        [2.9507e-05, 9.9997e-01],
        [4.9902e-02, 9.5010e-01],
        [7.3190e-01, 2.6810e-01],
        [6.2653e-03, 9.9373e-01],
        [3.8938e-02, 9.6106e-01],
        [3.6434e-04, 9.9964e-01],
        [2.1340e-01, 7.8660e-01],
        [9.0440e-01, 9.5596e-02],
        [9.8850e-01, 1.1496e-02],
        [5.2863e-03, 9.9471e-01],
        [8.0282e-01, 1.9718e-01],
        [9.3191e-01, 6.8086e-02],
        [4.8754e-03, 9.9512e-01],
        [1.5539e-02, 9.8446e-01],
        [4.8581e-03, 9.9514e-01],
        [9.8683e-01, 1.3174e-02],
        [2.7597e-02, 9.7240e-01],
        [1.7448e-02, 9.8255e-01],
        [1.0336e-04, 9.9990e-01],
        [3.1432e-01, 6.8568e-01],
        [8.6113e-03, 9.9139e-01],
        [1.2287e-02, 9.8771e-01],
        [8.3807e-03, 9.9162e-01],
        [7.2614e-04, 9.9927e-01],
        [4.9962e-04, 9.9950e-01],
        [1.0375e-01, 8.9625e-01],
        [3.2252e-04, 9.9968e-01],
        [9.1993e-01, 8.0073e-02],
        [9.2735e-01, 7.2651e-02],
        [5.4077e-04, 9.9946e-01],
        [4.1289e-02, 9.5871e-01],
        [9.8070e-01, 1.9300e-02],
        [4.1694e-03, 9.9583e-01],
        [8.2273e-01, 1.7727e-01],
        [2.9083e-02, 9.7092e-01],
        [2.4222e-01, 7.5778e-01],
        [9.4579e-01, 5.4207e-02],
        [3.3179e-04, 9.9967e-01],
        [3.2519e-04, 9.9967e-01],
        [4.2748e-03, 9.9573e-01],
        [3.9791e-02, 9.6021e-01],
        [9.4981e-01, 5.0185e-02],
        [7.7752e-04, 9.9922e-01],
        [9.5460e-01, 4.5400e-02],
        [5.0580e-01, 4.9420e-01],
        [3.7876e-02, 9.6212e-01],
        [8.5915e-02, 9.1408e-01],
        [2.1464e-02, 9.7854e-01],
        [2.4205e-03, 9.9758e-01],
        [7.5228e-01, 2.4772e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9024e-04, 9.9981e-01],
        [2.6574e-02, 9.7343e-01],
        [1.7206e-03, 9.9828e-01],
        [4.8757e-02, 9.5124e-01],
        [1.0376e-03, 9.9896e-01],
        [7.5599e-03, 9.9244e-01],
        [6.6164e-02, 9.3384e-01],
        [7.7236e-03, 9.9228e-01],
        [6.9448e-03, 9.9306e-01],
        [1.2331e-01, 8.7669e-01],
        [1.1920e-02, 9.8808e-01],
        [3.4320e-04, 9.9966e-01],
        [9.6633e-01, 3.3674e-02],
        [8.4781e-01, 1.5219e-01],
        [2.8293e-02, 9.7171e-01],
        [6.7016e-03, 9.9330e-01],
        [5.5438e-02, 9.4456e-01],
        [3.1746e-02, 9.6825e-01],
        [5.5644e-03, 9.9444e-01],
        [8.7167e-04, 9.9913e-01],
        [1.2402e-02, 9.8760e-01],
        [5.0328e-03, 9.9497e-01],
        [8.7264e-01, 1.2736e-01],
        [1.6389e-02, 9.8361e-01],
        [3.3843e-03, 9.9662e-01],
        [2.8472e-03, 9.9715e-01],
        [9.7537e-01, 2.4627e-02],
        [7.4932e-03, 9.9251e-01],
        [6.5544e-02, 9.3446e-01],
        [3.8566e-01, 6.1434e-01],
        [1.0072e-03, 9.9899e-01],
        [7.9116e-04, 9.9921e-01],
        [9.8265e-01, 1.7347e-02],
        [3.4100e-03, 9.9659e-01],
        [6.8103e-04, 9.9932e-01],
        [1.0795e-02, 9.8920e-01],
        [1.1149e-03, 9.9889e-01],
        [1.4671e-04, 9.9985e-01],
        [3.2110e-04, 9.9968e-01],
        [1.1969e-02, 9.8803e-01],
        [1.0104e-01, 8.9896e-01],
        [4.2165e-04, 9.9958e-01],
        [9.8631e-01, 1.3694e-02],
        [3.3052e-02, 9.6695e-01],
        [5.0976e-01, 4.9024e-01],
        [5.1256e-03, 9.9487e-01],
        [1.0217e-03, 9.9898e-01],
        [2.7577e-03, 9.9724e-01],
        [1.1756e-02, 9.8824e-01],
        [1.1294e-03, 9.9887e-01],
        [9.8344e-01, 1.6557e-02],
        [3.9551e-02, 9.6045e-01],
        [1.4198e-01, 8.5802e-01],
        [1.7690e-02, 9.8231e-01],
        [1.9582e-02, 9.8042e-01],
        [1.1564e-02, 9.8844e-01],
        [7.0242e-02, 9.2976e-01],
        [1.1837e-01, 8.8163e-01],
        [2.0897e-03, 9.9791e-01],
        [2.8711e-03, 9.9713e-01],
        [3.0313e-03, 9.9697e-01],
        [1.1276e-03, 9.9887e-01],
        [8.9485e-03, 9.9105e-01],
        [6.0042e-04, 9.9940e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.9828e-03, 9.9702e-01],
        [2.4647e-02, 9.7535e-01],
        [9.0396e-01, 9.6042e-02],
        [9.6100e-04, 9.9904e-01],
        [7.6166e-01, 2.3834e-01],
        [7.5425e-01, 2.4575e-01],
        [2.5406e-03, 9.9746e-01],
        [9.3469e-01, 6.5309e-02],
        [1.7926e-02, 9.8207e-01],
        [1.4065e-03, 9.9859e-01],
        [1.9792e-03, 9.9802e-01],
        [1.0515e-02, 9.8948e-01],
        [2.6592e-03, 9.9734e-01],
        [2.2626e-02, 9.7737e-01],
        [4.4920e-03, 9.9551e-01],
        [7.1231e-04, 9.9929e-01],
        [2.6951e-03, 9.9730e-01],
        [2.5391e-04, 9.9975e-01],
        [9.4885e-01, 5.1154e-02],
        [3.4699e-01, 6.5301e-01],
        [1.8463e-04, 9.9982e-01],
        [4.0702e-04, 9.9959e-01],
        [9.1048e-01, 8.9517e-02],
        [9.0834e-01, 9.1662e-02],
        [4.6357e-05, 9.9995e-01],
        [9.4109e-01, 5.8914e-02],
        [2.9596e-04, 9.9970e-01],
        [4.8847e-02, 9.5115e-01],
        [1.6268e-01, 8.3732e-01],
        [1.8308e-02, 9.8169e-01],
        [4.6302e-05, 9.9995e-01],
        [2.7464e-04, 9.9973e-01],
        [8.9858e-01, 1.0142e-01],
        [3.0793e-03, 9.9692e-01],
        [9.5375e-04, 9.9905e-01],
        [1.1080e-02, 9.8892e-01],
        [9.7246e-01, 2.7538e-02],
        [2.3762e-02, 9.7624e-01],
        [2.0340e-02, 9.7966e-01],
        [1.2409e-02, 9.8759e-01],
        [1.6470e-03, 9.9835e-01],
        [5.5534e-02, 9.4447e-01],
        [9.3854e-01, 6.1460e-02],
        [6.3540e-03, 9.9365e-01],
        [9.5078e-01, 4.9215e-02],
        [1.0963e-02, 9.8904e-01],
        [2.8368e-05, 9.9997e-01],
        [5.4578e-02, 9.4542e-01],
        [2.3386e-01, 7.6614e-01],
        [1.7255e-02, 9.8274e-01],
        [4.7672e-05, 9.9995e-01],
        [6.4578e-03, 9.9354e-01],
        [1.1905e-02, 9.8809e-01],
        [1.1681e-02, 9.8832e-01],
        [1.6657e-03, 9.9833e-01],
        [9.9297e-01, 7.0250e-03],
        [3.4855e-02, 9.6515e-01],
        [8.2560e-01, 1.7440e-01],
        [6.2824e-02, 9.3718e-01],
        [7.5146e-04, 9.9925e-01],
        [9.5245e-01, 4.7548e-02],
        [5.6766e-01, 4.3234e-01],
        [6.5487e-04, 9.9935e-01],
        [2.4023e-03, 9.9760e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3450e-02, 9.7655e-01],
        [9.7269e-01, 2.7309e-02],
        [1.1460e-03, 9.9885e-01],
        [3.8761e-02, 9.6124e-01],
        [8.1478e-01, 1.8522e-01],
        [9.2840e-03, 9.9072e-01],
        [9.6781e-05, 9.9990e-01],
        [2.0857e-04, 9.9979e-01],
        [9.6740e-01, 3.2601e-02],
        [9.8382e-01, 1.6176e-02],
        [2.9499e-04, 9.9971e-01],
        [5.4554e-02, 9.4545e-01],
        [2.7996e-02, 9.7200e-01],
        [5.7024e-04, 9.9943e-01],
        [9.1228e-04, 9.9909e-01],
        [8.2117e-04, 9.9918e-01],
        [1.5810e-02, 9.8419e-01],
        [4.9241e-04, 9.9951e-01],
        [9.8638e-01, 1.3616e-02],
        [5.0174e-03, 9.9498e-01],
        [9.1709e-04, 9.9908e-01],
        [2.2375e-03, 9.9776e-01],
        [5.0092e-03, 9.9499e-01],
        [1.0484e-02, 9.8952e-01],
        [2.6024e-03, 9.9740e-01],
        [2.1558e-01, 7.8442e-01],
        [4.4136e-04, 9.9956e-01],
        [1.7507e-03, 9.9825e-01],
        [9.9123e-01, 8.7714e-03],
        [8.0129e-02, 9.1987e-01],
        [7.9171e-01, 2.0829e-01],
        [5.7221e-02, 9.4278e-01],
        [2.2191e-02, 9.7781e-01],
        [7.0689e-02, 9.2931e-01],
        [6.1904e-01, 3.8096e-01],
        [1.2233e-03, 9.9878e-01],
        [9.0822e-04, 9.9909e-01],
        [8.4345e-01, 1.5655e-01],
        [6.8170e-04, 9.9932e-01],
        [2.4767e-04, 9.9975e-01],
        [3.6108e-02, 9.6389e-01],
        [1.2294e-02, 9.8771e-01],
        [3.0221e-04, 9.9970e-01],
        [7.1293e-03, 9.9287e-01],
        [1.0568e-02, 9.8943e-01],
        [7.6739e-03, 9.9233e-01],
        [4.9035e-04, 9.9951e-01],
        [1.5437e-02, 9.8456e-01],
        [9.8999e-01, 1.0013e-02],
        [4.4560e-03, 9.9554e-01],
        [9.8590e-02, 9.0141e-01],
        [9.2634e-02, 9.0737e-01],
        [7.2615e-02, 9.2738e-01],
        [2.0537e-03, 9.9795e-01],
        [1.6693e-03, 9.9833e-01],
        [5.7723e-02, 9.4228e-01],
        [1.1279e-02, 9.8872e-01],
        [2.1424e-02, 9.7858e-01],
        [3.2752e-05, 9.9997e-01],
        [3.8064e-03, 9.9619e-01],
        [4.1294e-03, 9.9587e-01],
        [1.1022e-01, 8.8978e-01],
        [7.1474e-03, 9.9285e-01],
        [1.3342e-03, 9.9867e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1385e-03, 9.9786e-01],
        [1.8716e-01, 8.1284e-01],
        [9.8429e-01, 1.5710e-02],
        [1.1212e-01, 8.8788e-01],
        [4.7249e-04, 9.9953e-01],
        [9.9017e-01, 9.8341e-03],
        [3.1294e-04, 9.9969e-01],
        [2.7774e-03, 9.9722e-01],
        [8.3674e-01, 1.6326e-01],
        [9.9277e-01, 7.2277e-03],
        [9.4418e-01, 5.5817e-02],
        [2.0501e-03, 9.9795e-01],
        [3.8677e-01, 6.1323e-01],
        [9.8167e-01, 1.8333e-02],
        [7.5099e-02, 9.2490e-01],
        [1.1059e-03, 9.9889e-01],
        [8.3429e-03, 9.9166e-01],
        [2.9337e-03, 9.9707e-01],
        [6.6069e-03, 9.9339e-01],
        [1.7438e-01, 8.2562e-01],
        [9.6946e-01, 3.0540e-02],
        [1.0090e-02, 9.8991e-01],
        [1.0194e-02, 9.8981e-01],
        [8.5513e-04, 9.9914e-01],
        [1.5804e-03, 9.9842e-01],
        [5.7506e-03, 9.9425e-01],
        [9.5903e-01, 4.0974e-02],
        [1.4114e-03, 9.9859e-01],
        [5.7282e-04, 9.9943e-01],
        [9.6398e-01, 3.6019e-02],
        [6.9104e-04, 9.9931e-01],
        [9.6567e-02, 9.0343e-01],
        [9.8681e-01, 1.3187e-02],
        [1.8371e-02, 9.8163e-01],
        [1.3521e-04, 9.9986e-01],
        [1.3065e-03, 9.9869e-01],
        [8.8914e-02, 9.1109e-01],
        [2.3789e-04, 9.9976e-01],
        [9.8327e-01, 1.6725e-02],
        [9.9326e-01, 6.7371e-03],
        [2.0799e-02, 9.7920e-01],
        [5.6864e-05, 9.9994e-01],
        [9.1865e-01, 8.1353e-02],
        [2.3531e-02, 9.7647e-01],
        [3.2873e-03, 9.9671e-01],
        [4.8934e-01, 5.1066e-01],
        [8.1929e-01, 1.8071e-01],
        [5.5861e-03, 9.9441e-01],
        [9.8857e-01, 1.1429e-02],
        [1.6954e-02, 9.8305e-01],
        [2.5413e-03, 9.9746e-01],
        [2.3377e-03, 9.9766e-01],
        [3.8428e-03, 9.9616e-01],
        [1.9229e-03, 9.9808e-01],
        [1.8774e-02, 9.8123e-01],
        [2.0269e-01, 7.9731e-01],
        [9.8207e-01, 1.7931e-02],
        [7.7781e-04, 9.9922e-01],
        [1.3312e-03, 9.9867e-01],
        [1.2533e-03, 9.9875e-01],
        [1.2066e-02, 9.8793e-01],
        [6.5052e-05, 9.9993e-01],
        [7.6339e-04, 9.9924e-01],
        [6.2896e-04, 9.9937e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.9896e-01, 5.0104e-01],
        [3.0167e-02, 9.6983e-01],
        [8.8497e-03, 9.9115e-01],
        [9.4067e-01, 5.9330e-02],
        [1.2488e-02, 9.8751e-01],
        [1.7525e-02, 9.8248e-01],
        [4.9588e-04, 9.9950e-01],
        [9.3386e-01, 6.6136e-02],
        [5.9843e-03, 9.9402e-01],
        [5.6577e-03, 9.9434e-01],
        [3.7517e-03, 9.9625e-01],
        [1.1854e-02, 9.8815e-01],
        [9.2274e-01, 7.7260e-02],
        [9.9955e-01, 4.4562e-04],
        [2.0031e-04, 9.9980e-01],
        [9.8181e-01, 1.8193e-02],
        [3.0789e-02, 9.6921e-01],
        [8.8818e-01, 1.1182e-01],
        [3.1680e-03, 9.9683e-01],
        [1.8806e-03, 9.9812e-01],
        [6.1033e-03, 9.9390e-01],
        [1.0989e-04, 9.9989e-01],
        [3.9600e-01, 6.0400e-01],
        [5.0276e-01, 4.9724e-01],
        [4.5297e-03, 9.9547e-01],
        [1.5824e-03, 9.9842e-01],
        [9.9240e-02, 9.0076e-01],
        [9.9015e-01, 9.8478e-03],
        [1.1042e-01, 8.8958e-01],
        [3.8810e-03, 9.9612e-01],
        [9.5605e-01, 4.3952e-02],
        [1.7263e-03, 9.9827e-01],
        [6.5262e-02, 9.3474e-01],
        [1.3158e-01, 8.6842e-01],
        [1.1142e-04, 9.9989e-01],
        [2.0992e-04, 9.9979e-01],
        [2.0589e-04, 9.9979e-01],
        [8.8736e-03, 9.9113e-01],
        [2.0411e-03, 9.9796e-01],
        [1.4126e-04, 9.9986e-01],
        [8.0700e-01, 1.9300e-01],
        [9.0525e-05, 9.9991e-01],
        [1.0808e-01, 8.9192e-01],
        [7.7417e-03, 9.9226e-01],
        [8.9390e-03, 9.9106e-01],
        [9.9026e-01, 9.7384e-03],
        [8.7328e-04, 9.9913e-01],
        [9.1053e-04, 9.9909e-01],
        [7.5373e-02, 9.2463e-01],
        [9.9040e-01, 9.6022e-03],
        [3.3594e-03, 9.9664e-01],
        [9.1885e-01, 8.1153e-02],
        [8.6158e-04, 9.9914e-01],
        [4.0386e-02, 9.5961e-01],
        [1.3166e-02, 9.8683e-01],
        [2.0126e-04, 9.9980e-01],
        [2.5116e-02, 9.7488e-01],
        [9.1427e-03, 9.9086e-01],
        [7.4635e-03, 9.9254e-01],
        [2.7210e-02, 9.7279e-01],
        [1.4970e-03, 9.9850e-01],
        [9.5034e-01, 4.9658e-02],
        [3.0446e-02, 9.6955e-01],
        [4.2323e-02, 9.5768e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.4228e-01, 1.5772e-01],
        [5.8277e-02, 9.4172e-01],
        [9.0746e-01, 9.2537e-02],
        [5.2684e-04, 9.9947e-01],
        [5.7544e-03, 9.9425e-01],
        [1.1084e-02, 9.8892e-01],
        [5.1496e-04, 9.9949e-01],
        [2.8393e-02, 9.7161e-01],
        [1.5522e-04, 9.9984e-01],
        [9.8984e-01, 1.0164e-02],
        [1.2151e-02, 9.8785e-01],
        [1.0285e-03, 9.9897e-01],
        [1.0008e-02, 9.8999e-01],
        [2.0706e-04, 9.9979e-01],
        [2.0011e-01, 7.9989e-01],
        [1.6111e-02, 9.8389e-01],
        [1.0693e-02, 9.8931e-01],
        [8.0859e-01, 1.9141e-01],
        [5.7606e-02, 9.4239e-01],
        [9.9542e-01, 4.5763e-03],
        [9.9722e-01, 2.7760e-03],
        [1.0894e-04, 9.9989e-01],
        [9.5000e-01, 5.0003e-02],
        [1.6336e-02, 9.8366e-01],
        [1.5799e-03, 9.9842e-01],
        [9.9743e-01, 2.5676e-03],
        [5.7091e-03, 9.9429e-01],
        [4.1233e-03, 9.9588e-01],
        [1.1068e-02, 9.8893e-01],
        [5.4199e-02, 9.4580e-01],
        [4.4272e-02, 9.5573e-01],
        [9.0290e-02, 9.0971e-01],
        [3.4857e-03, 9.9651e-01],
        [1.3364e-02, 9.8664e-01],
        [2.3283e-02, 9.7672e-01],
        [9.0419e-01, 9.5813e-02],
        [2.8497e-03, 9.9715e-01],
        [1.0000e-02, 9.9000e-01],
        [3.0203e-03, 9.9698e-01],
        [6.1916e-02, 9.3808e-01],
        [9.7478e-01, 2.5221e-02],
        [2.8080e-04, 9.9972e-01],
        [5.4106e-01, 4.5894e-01],
        [2.4447e-04, 9.9976e-01],
        [3.5607e-03, 9.9644e-01],
        [8.9317e-03, 9.9107e-01],
        [9.5095e-01, 4.9048e-02],
        [6.7053e-03, 9.9329e-01],
        [1.5162e-03, 9.9848e-01],
        [7.3951e-01, 2.6049e-01],
        [6.6891e-03, 9.9331e-01],
        [8.4077e-04, 9.9916e-01],
        [1.8557e-02, 9.8144e-01],
        [9.3771e-05, 9.9991e-01],
        [6.6130e-01, 3.3870e-01],
        [1.0921e-02, 9.8908e-01],
        [8.5536e-05, 9.9991e-01],
        [5.8490e-04, 9.9942e-01],
        [1.0493e-03, 9.9895e-01],
        [4.0343e-01, 5.9657e-01],
        [2.9297e-04, 9.9971e-01],
        [3.3381e-03, 9.9666e-01],
        [7.9945e-01, 2.0055e-01],
        [2.1321e-01, 7.8679e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.4332e-04, 9.9966e-01],
        [3.3373e-03, 9.9666e-01],
        [3.9402e-02, 9.6060e-01],
        [2.6828e-02, 9.7317e-01],
        [3.2236e-04, 9.9968e-01],
        [8.8168e-01, 1.1832e-01],
        [1.2029e-02, 9.8797e-01],
        [2.9836e-03, 9.9702e-01],
        [5.3369e-02, 9.4663e-01],
        [5.9474e-03, 9.9405e-01],
        [8.5837e-02, 9.1416e-01],
        [8.7097e-04, 9.9913e-01],
        [2.3509e-01, 7.6491e-01],
        [1.0779e-02, 9.8922e-01],
        [9.8422e-01, 1.5784e-02],
        [7.4676e-03, 9.9253e-01],
        [2.7935e-03, 9.9721e-01],
        [9.9661e-01, 3.3927e-03],
        [8.2390e-01, 1.7610e-01],
        [2.0590e-03, 9.9794e-01],
        [9.8256e-01, 1.7443e-02],
        [1.0105e-03, 9.9899e-01],
        [2.5651e-02, 9.7435e-01],
        [9.9586e-01, 4.1434e-03],
        [7.2932e-04, 9.9927e-01],
        [3.9133e-02, 9.6087e-01],
        [5.5186e-02, 9.4481e-01],
        [1.0829e-02, 9.8917e-01],
        [1.5241e-02, 9.8476e-01],
        [9.1813e-04, 9.9908e-01],
        [2.3163e-02, 9.7684e-01],
        [1.5935e-02, 9.8406e-01],
        [4.9380e-03, 9.9506e-01],
        [1.0267e-02, 9.8973e-01],
        [1.0336e-01, 8.9664e-01],
        [9.9434e-01, 5.6556e-03],
        [2.8113e-04, 9.9972e-01],
        [3.2697e-03, 9.9673e-01],
        [6.0451e-01, 3.9549e-01],
        [1.8754e-03, 9.9812e-01],
        [9.7657e-01, 2.3434e-02],
        [7.4726e-03, 9.9253e-01],
        [3.2198e-04, 9.9968e-01],
        [1.3191e-03, 9.9868e-01],
        [1.7850e-04, 9.9982e-01],
        [6.7098e-03, 9.9329e-01],
        [3.6353e-03, 9.9636e-01],
        [1.4318e-02, 9.8568e-01],
        [7.1695e-01, 2.8305e-01],
        [9.9254e-01, 7.4574e-03],
        [9.8960e-01, 1.0398e-02],
        [1.2337e-03, 9.9877e-01],
        [1.1218e-02, 9.8878e-01],
        [5.0091e-02, 9.4991e-01],
        [3.9105e-05, 9.9996e-01],
        [3.1812e-01, 6.8188e-01],
        [3.9106e-02, 9.6089e-01],
        [9.1531e-05, 9.9991e-01],
        [1.9515e-04, 9.9980e-01],
        [9.9424e-01, 5.7564e-03],
        [6.7004e-02, 9.3300e-01],
        [3.5493e-02, 9.6451e-01],
        [2.2863e-03, 9.9771e-01],
        [1.3695e-04, 9.9986e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9362e-02, 9.8064e-01],
        [9.9205e-01, 7.9530e-03],
        [8.0825e-04, 9.9919e-01],
        [8.5049e-03, 9.9150e-01],
        [1.1979e-03, 9.9880e-01],
        [3.1986e-02, 9.6801e-01],
        [9.9231e-01, 7.6921e-03],
        [8.0937e-01, 1.9063e-01],
        [9.3215e-04, 9.9907e-01],
        [3.0220e-04, 9.9970e-01],
        [5.4365e-03, 9.9456e-01],
        [9.9869e-01, 1.3090e-03],
        [4.8709e-02, 9.5129e-01],
        [5.8759e-04, 9.9941e-01],
        [1.0527e-01, 8.9473e-01],
        [9.8536e-01, 1.4644e-02],
        [8.4315e-03, 9.9157e-01],
        [2.9089e-03, 9.9709e-01],
        [3.2014e-02, 9.6799e-01],
        [4.0358e-02, 9.5964e-01],
        [9.9424e-01, 5.7626e-03],
        [9.9448e-01, 5.5203e-03],
        [1.5313e-01, 8.4687e-01],
        [3.6045e-03, 9.9640e-01],
        [8.0455e-05, 9.9992e-01],
        [5.1105e-01, 4.8895e-01],
        [1.3226e-03, 9.9868e-01],
        [6.8892e-03, 9.9311e-01],
        [1.7025e-02, 9.8297e-01],
        [2.5275e-03, 9.9747e-01],
        [1.1621e-02, 9.8838e-01],
        [1.3699e-02, 9.8630e-01],
        [1.1903e-04, 9.9988e-01],
        [1.8650e-03, 9.9813e-01],
        [9.4698e-02, 9.0530e-01],
        [6.9580e-03, 9.9304e-01],
        [2.1727e-03, 9.9783e-01],
        [9.5178e-01, 4.8220e-02],
        [9.8832e-01, 1.1681e-02],
        [1.2540e-02, 9.8746e-01],
        [3.1734e-02, 9.6827e-01],
        [9.9962e-01, 3.8161e-04],
        [7.7606e-03, 9.9224e-01],
        [2.3703e-03, 9.9763e-01],
        [5.4800e-01, 4.5200e-01],
        [8.3094e-04, 9.9917e-01],
        [1.3351e-02, 9.8665e-01],
        [4.9575e-03, 9.9504e-01],
        [1.2257e-02, 9.8774e-01],
        [9.9476e-01, 5.2358e-03],
        [3.5235e-01, 6.4765e-01],
        [9.9846e-01, 1.5440e-03],
        [4.4826e-05, 9.9996e-01],
        [1.3585e-01, 8.6414e-01],
        [9.7389e-01, 2.6110e-02],
        [2.7740e-02, 9.7226e-01],
        [9.7975e-01, 2.0247e-02],
        [2.9120e-03, 9.9709e-01],
        [7.4531e-04, 9.9925e-01],
        [9.4166e-04, 9.9906e-01],
        [8.9482e-04, 9.9911e-01],
        [4.2551e-02, 9.5745e-01],
        [3.7871e-02, 9.6213e-01],
        [1.5150e-02, 9.8485e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3902e-01, 7.6098e-01],
        [7.9377e-03, 9.9206e-01],
        [5.7187e-03, 9.9428e-01],
        [2.7108e-01, 7.2892e-01],
        [4.3695e-03, 9.9563e-01],
        [1.1408e-02, 9.8859e-01],
        [7.8582e-01, 2.1418e-01],
        [7.1553e-05, 9.9993e-01],
        [1.3927e-02, 9.8607e-01],
        [9.5643e-01, 4.3568e-02],
        [1.9710e-02, 9.8029e-01],
        [2.8322e-03, 9.9717e-01],
        [1.4672e-01, 8.5328e-01],
        [1.0129e-03, 9.9899e-01],
        [2.0421e-03, 9.9796e-01],
        [2.3416e-03, 9.9766e-01],
        [9.9975e-01, 2.4637e-04],
        [1.6223e-02, 9.8378e-01],
        [2.1511e-02, 9.7849e-01],
        [6.0179e-03, 9.9398e-01],
        [4.2450e-02, 9.5755e-01],
        [5.0695e-01, 4.9305e-01],
        [5.5106e-03, 9.9449e-01]], grad_fn=<SoftmaxBackward0>)
Epoch 1, loss: 0.4508
tensor([[8.7008e-01, 1.2992e-01],
        [7.0212e-02, 9.2979e-01],
        [4.5035e-04, 9.9955e-01],
        [9.7316e-01, 2.6838e-02],
        [9.8779e-01, 1.2210e-02],
        [9.8779e-01, 1.2210e-02],
        [1.7848e-03, 9.9822e-01],
        [2.3148e-03, 9.9769e-01],
        [9.9596e-01, 4.0384e-03],
        [4.4047e-03, 9.9560e-01],
        [3.0841e-03, 9.9692e-01],
        [3.5615e-04, 9.9964e-01],
        [9.5472e-01, 4.5277e-02],
        [4.5747e-03, 9.9543e-01],
        [5.8169e-03, 9.9418e-01],
        [4.4226e-04, 9.9956e-01],
        [1.0649e-01, 8.9351e-01],
        [3.1887e-04, 9.9968e-01],
        [9.5619e-01, 4.3805e-02],
        [1.4280e-03, 9.9857e-01],
        [2.1977e-04, 9.9978e-01],
        [1.0070e-03, 9.9899e-01],
        [8.0497e-02, 9.1950e-01],
        [1.5383e-03, 9.9846e-01],
        [9.9666e-01, 3.3424e-03],
        [1.4943e-05, 9.9999e-01],
        [1.0271e-02, 9.8973e-01],
        [9.9250e-01, 7.4986e-03],
        [3.6976e-05, 9.9996e-01],
        [2.4828e-03, 9.9752e-01],
        [9.6655e-04, 9.9903e-01],
        [9.0903e-01, 9.0970e-02],
        [9.9146e-01, 8.5450e-03],
        [1.4143e-02, 9.8586e-01],
        [1.9733e-03, 9.9803e-01],
        [6.1057e-03, 9.9389e-01],
        [2.4914e-01, 7.5086e-01],
        [1.7626e-02, 9.8237e-01],
        [4.7076e-04, 9.9953e-01],
        [2.5157e-04, 9.9975e-01],
        [8.8862e-01, 1.1138e-01],
        [6.7045e-03, 9.9330e-01],
        [4.0782e-03, 9.9592e-01],
        [2.1458e-03, 9.9785e-01],
        [2.8065e-04, 9.9972e-01],
        [9.1160e-05, 9.9991e-01],
        [8.3275e-02, 9.1672e-01],
        [5.9047e-03, 9.9410e-01],
        [1.9583e-01, 8.0417e-01],
        [1.2647e-04, 9.9987e-01],
        [6.1681e-03, 9.9383e-01],
        [1.1079e-02, 9.8892e-01],
        [9.9610e-01, 3.9036e-03],
        [6.7882e-03, 9.9321e-01],
        [8.4967e-01, 1.5033e-01],
        [9.7952e-01, 2.0480e-02],
        [6.3102e-05, 9.9994e-01],
        [5.1083e-01, 4.8917e-01],
        [1.1721e-01, 8.8279e-01],
        [2.3085e-02, 9.7692e-01],
        [4.9637e-03, 9.9504e-01],
        [9.7536e-01, 2.4635e-02],
        [2.2152e-04, 9.9978e-01],
        [8.7806e-01, 1.2194e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5699e-04, 9.9984e-01],
        [9.5260e-01, 4.7403e-02],
        [9.5130e-01, 4.8703e-02],
        [1.1945e-02, 9.8806e-01],
        [1.4019e-02, 9.8598e-01],
        [9.9238e-01, 7.6207e-03],
        [1.4789e-02, 9.8521e-01],
        [6.9941e-02, 9.3006e-01],
        [6.8783e-03, 9.9312e-01],
        [5.0701e-06, 9.9999e-01],
        [3.4426e-03, 9.9656e-01],
        [9.8699e-01, 1.3013e-02],
        [4.6028e-03, 9.9540e-01],
        [6.1758e-01, 3.8242e-01],
        [3.2897e-04, 9.9967e-01],
        [1.2473e-02, 9.8753e-01],
        [3.2386e-04, 9.9968e-01],
        [1.6477e-02, 9.8352e-01],
        [5.7578e-02, 9.4242e-01],
        [9.8994e-01, 1.0056e-02],
        [1.5862e-03, 9.9841e-01],
        [1.4393e-03, 9.9856e-01],
        [5.7192e-03, 9.9428e-01],
        [2.2251e-01, 7.7749e-01],
        [1.0748e-02, 9.8925e-01],
        [3.6950e-04, 9.9963e-01],
        [9.5739e-04, 9.9904e-01],
        [7.7392e-01, 2.2608e-01],
        [2.9255e-02, 9.7074e-01],
        [2.0320e-01, 7.9680e-01],
        [2.4417e-03, 9.9756e-01],
        [3.8694e-01, 6.1306e-01],
        [3.0477e-03, 9.9695e-01],
        [4.9710e-04, 9.9950e-01],
        [2.8925e-02, 9.7108e-01],
        [8.1849e-03, 9.9182e-01],
        [9.3595e-01, 6.4051e-02],
        [1.0782e-04, 9.9989e-01],
        [1.8306e-03, 9.9817e-01],
        [2.6303e-04, 9.9974e-01],
        [1.1609e-05, 9.9999e-01],
        [1.5690e-02, 9.8431e-01],
        [3.1423e-04, 9.9969e-01],
        [2.2809e-05, 9.9998e-01],
        [1.2180e-03, 9.9878e-01],
        [5.8085e-03, 9.9419e-01],
        [8.4853e-03, 9.9151e-01],
        [9.8805e-03, 9.9012e-01],
        [1.4834e-04, 9.9985e-01],
        [5.0382e-04, 9.9950e-01],
        [9.1392e-01, 8.6080e-02],
        [1.4736e-05, 9.9999e-01],
        [1.8235e-01, 8.1765e-01],
        [1.1305e-04, 9.9989e-01],
        [5.7610e-05, 9.9994e-01],
        [3.4891e-04, 9.9965e-01],
        [8.9074e-01, 1.0926e-01],
        [9.5577e-01, 4.4228e-02],
        [7.2272e-03, 9.9277e-01],
        [6.3560e-03, 9.9364e-01],
        [7.7340e-01, 2.2660e-01],
        [4.3973e-03, 9.9560e-01],
        [5.7562e-04, 9.9942e-01],
        [1.0653e-03, 9.9893e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.9765e-04, 9.9960e-01],
        [1.9934e-02, 9.8007e-01],
        [2.0218e-03, 9.9798e-01],
        [1.9548e-01, 8.0452e-01],
        [9.9246e-01, 7.5374e-03],
        [1.9328e-04, 9.9981e-01],
        [2.6057e-05, 9.9997e-01],
        [9.9893e-01, 1.0651e-03],
        [1.8925e-05, 9.9998e-01],
        [9.9493e-01, 5.0678e-03],
        [3.8248e-05, 9.9996e-01],
        [7.5054e-02, 9.2495e-01],
        [9.2986e-01, 7.0138e-02],
        [9.8775e-01, 1.2245e-02],
        [4.4341e-05, 9.9996e-01],
        [1.7898e-04, 9.9982e-01],
        [5.0375e-05, 9.9995e-01],
        [1.9581e-05, 9.9998e-01],
        [3.7752e-06, 1.0000e+00],
        [1.0914e-01, 8.9086e-01],
        [6.7283e-04, 9.9933e-01],
        [1.4165e-02, 9.8584e-01],
        [1.2685e-02, 9.8732e-01],
        [8.1057e-01, 1.8943e-01],
        [2.9266e-04, 9.9971e-01],
        [9.7084e-01, 2.9164e-02],
        [6.4713e-01, 3.5287e-01],
        [1.6835e-02, 9.8316e-01],
        [5.5069e-03, 9.9449e-01],
        [6.2140e-03, 9.9379e-01],
        [9.9944e-01, 5.5875e-04],
        [1.6322e-03, 9.9837e-01],
        [3.6066e-02, 9.6393e-01],
        [1.7951e-05, 9.9998e-01],
        [4.5985e-05, 9.9995e-01],
        [4.0199e-04, 9.9960e-01],
        [1.1609e-03, 9.9884e-01],
        [3.5283e-01, 6.4717e-01],
        [3.5143e-04, 9.9965e-01],
        [4.3488e-04, 9.9957e-01],
        [8.6607e-04, 9.9913e-01],
        [8.7424e-02, 9.1258e-01],
        [4.0401e-04, 9.9960e-01],
        [1.2473e-03, 9.9875e-01],
        [9.6101e-01, 3.8991e-02],
        [2.9291e-04, 9.9971e-01],
        [4.3166e-04, 9.9957e-01],
        [6.8010e-05, 9.9993e-01],
        [1.4204e-02, 9.8580e-01],
        [1.1778e-05, 9.9999e-01],
        [3.2299e-02, 9.6770e-01],
        [2.0114e-05, 9.9998e-01],
        [1.6645e-02, 9.8335e-01],
        [7.3054e-06, 9.9999e-01],
        [4.4060e-04, 9.9956e-01],
        [7.2220e-02, 9.2778e-01],
        [9.5596e-01, 4.4040e-02],
        [9.7816e-01, 2.1839e-02],
        [3.8416e-05, 9.9996e-01],
        [8.8962e-03, 9.9110e-01],
        [4.1197e-05, 9.9996e-01],
        [7.3845e-05, 9.9993e-01],
        [9.8794e-01, 1.2056e-02],
        [4.8612e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.1346e-04, 9.9969e-01],
        [6.3213e-05, 9.9994e-01],
        [1.7668e-04, 9.9982e-01],
        [9.7439e-04, 9.9903e-01],
        [3.1466e-05, 9.9997e-01],
        [1.8650e-05, 9.9998e-01],
        [3.9154e-05, 9.9996e-01],
        [2.2027e-04, 9.9978e-01],
        [1.7519e-04, 9.9982e-01],
        [4.4127e-04, 9.9956e-01],
        [5.7149e-01, 4.2851e-01],
        [6.6522e-03, 9.9335e-01],
        [2.7554e-02, 9.7245e-01],
        [2.4487e-05, 9.9998e-01],
        [8.7236e-04, 9.9913e-01],
        [4.5281e-01, 5.4719e-01],
        [2.8531e-04, 9.9971e-01],
        [7.2232e-04, 9.9928e-01],
        [2.2836e-03, 9.9772e-01],
        [6.6374e-06, 9.9999e-01],
        [2.1253e-04, 9.9979e-01],
        [9.4065e-03, 9.9059e-01],
        [1.0236e-04, 9.9990e-01],
        [9.6777e-04, 9.9903e-01],
        [2.9713e-05, 9.9997e-01],
        [9.0643e-04, 9.9909e-01],
        [6.4900e-02, 9.3510e-01],
        [2.3340e-04, 9.9977e-01],
        [2.0183e-05, 9.9998e-01],
        [8.6673e-01, 1.3327e-01],
        [2.0125e-01, 7.9875e-01],
        [3.8035e-04, 9.9962e-01],
        [1.0837e-04, 9.9989e-01],
        [1.6865e-05, 9.9998e-01],
        [3.4929e-01, 6.5071e-01],
        [6.6381e-04, 9.9934e-01],
        [2.9058e-04, 9.9971e-01],
        [2.3278e-04, 9.9977e-01],
        [9.7810e-01, 2.1896e-02],
        [2.6529e-03, 9.9735e-01],
        [9.9234e-01, 7.6638e-03],
        [4.3319e-03, 9.9567e-01],
        [2.3775e-05, 9.9998e-01],
        [3.5066e-05, 9.9996e-01],
        [1.6859e-02, 9.8314e-01],
        [9.9209e-01, 7.9060e-03],
        [2.5553e-03, 9.9744e-01],
        [1.2994e-03, 9.9870e-01],
        [7.1432e-03, 9.9286e-01],
        [1.1550e-03, 9.9884e-01],
        [4.0789e-03, 9.9592e-01],
        [9.6122e-05, 9.9990e-01],
        [2.1912e-05, 9.9998e-01],
        [9.9762e-01, 2.3760e-03],
        [2.4065e-04, 9.9976e-01],
        [7.3100e-03, 9.9269e-01],
        [3.5777e-03, 9.9642e-01],
        [5.0285e-05, 9.9995e-01],
        [7.1399e-01, 2.8601e-01],
        [3.9980e-05, 9.9996e-01],
        [1.9447e-04, 9.9981e-01],
        [8.9967e-04, 9.9910e-01],
        [6.2149e-04, 9.9938e-01],
        [1.8145e-04, 9.9982e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.4836e-01, 6.5164e-01],
        [8.8719e-04, 9.9911e-01],
        [1.0995e-01, 8.9005e-01],
        [8.2643e-03, 9.9174e-01],
        [1.9482e-04, 9.9981e-01],
        [3.9855e-06, 1.0000e+00],
        [7.4063e-06, 9.9999e-01],
        [4.7067e-02, 9.5293e-01],
        [6.3998e-03, 9.9360e-01],
        [6.5963e-04, 9.9934e-01],
        [1.1607e-02, 9.8839e-01],
        [1.5126e-05, 9.9998e-01],
        [4.0952e-05, 9.9996e-01],
        [2.3563e-04, 9.9976e-01],
        [9.6291e-03, 9.9037e-01],
        [7.4352e-04, 9.9926e-01],
        [5.8303e-03, 9.9417e-01],
        [4.8371e-03, 9.9516e-01],
        [9.7771e-05, 9.9990e-01],
        [7.5288e-04, 9.9925e-01],
        [6.6946e-03, 9.9331e-01],
        [3.4408e-04, 9.9966e-01],
        [8.6318e-01, 1.3682e-01],
        [4.1488e-04, 9.9959e-01],
        [4.9256e-03, 9.9507e-01],
        [1.3615e-02, 9.8638e-01],
        [4.7054e-04, 9.9953e-01],
        [3.0271e-03, 9.9697e-01],
        [2.0295e-03, 9.9797e-01],
        [3.2792e-04, 9.9967e-01],
        [9.5141e-01, 4.8591e-02],
        [9.7314e-01, 2.6860e-02],
        [4.6240e-03, 9.9538e-01],
        [3.1586e-03, 9.9684e-01],
        [1.4897e-01, 8.5103e-01],
        [6.7809e-01, 3.2191e-01],
        [1.3182e-03, 9.9868e-01],
        [2.4837e-03, 9.9752e-01],
        [3.6352e-05, 9.9996e-01],
        [1.7994e-04, 9.9982e-01],
        [2.8423e-04, 9.9972e-01],
        [2.3009e-01, 7.6991e-01],
        [1.2073e-03, 9.9879e-01],
        [1.3044e-03, 9.9870e-01],
        [3.2769e-06, 1.0000e+00],
        [1.5660e-04, 9.9984e-01],
        [2.8971e-04, 9.9971e-01],
        [2.0126e-01, 7.9874e-01],
        [7.0099e-04, 9.9930e-01],
        [4.7137e-03, 9.9529e-01],
        [9.1644e-06, 9.9999e-01],
        [7.1375e-03, 9.9286e-01],
        [6.7962e-04, 9.9932e-01],
        [7.4246e-01, 2.5754e-01],
        [1.7726e-05, 9.9998e-01],
        [9.8416e-01, 1.5844e-02],
        [1.6815e-03, 9.9832e-01],
        [2.1023e-04, 9.9979e-01],
        [2.7688e-03, 9.9723e-01],
        [4.2885e-05, 9.9996e-01],
        [4.8471e-05, 9.9995e-01],
        [6.4918e-03, 9.9351e-01],
        [2.0809e-04, 9.9979e-01],
        [2.1037e-01, 7.8963e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.9432e-05, 9.9994e-01],
        [1.0189e-03, 9.9898e-01],
        [1.4822e-04, 9.9985e-01],
        [9.3866e-01, 6.1343e-02],
        [9.0978e-01, 9.0224e-02],
        [6.8784e-04, 9.9931e-01],
        [7.9980e-04, 9.9920e-01],
        [9.9708e-01, 2.9228e-03],
        [5.9988e-01, 4.0012e-01],
        [6.2137e-01, 3.7863e-01],
        [3.1083e-03, 9.9689e-01],
        [2.2274e-06, 1.0000e+00],
        [9.9544e-01, 4.5598e-03],
        [8.3309e-01, 1.6691e-01],
        [8.2739e-01, 1.7261e-01],
        [2.2926e-05, 9.9998e-01],
        [6.8995e-04, 9.9931e-01],
        [4.7098e-04, 9.9953e-01],
        [1.2361e-04, 9.9988e-01],
        [6.1908e-01, 3.8092e-01],
        [1.0609e-02, 9.8939e-01],
        [8.0859e-05, 9.9992e-01],
        [9.9579e-01, 4.2147e-03],
        [3.9063e-01, 6.0937e-01],
        [9.6504e-01, 3.4963e-02],
        [1.1606e-02, 9.8839e-01],
        [9.7025e-01, 2.9751e-02],
        [2.8473e-03, 9.9715e-01],
        [2.1352e-03, 9.9786e-01],
        [6.9553e-04, 9.9930e-01],
        [9.9520e-01, 4.7994e-03],
        [2.1131e-02, 9.7887e-01],
        [9.2942e-01, 7.0582e-02],
        [2.6015e-02, 9.7398e-01],
        [4.9185e-01, 5.0815e-01],
        [6.7362e-04, 9.9933e-01],
        [9.3909e-01, 6.0905e-02],
        [9.9957e-01, 4.3398e-04],
        [9.9782e-04, 9.9900e-01],
        [7.5824e-04, 9.9924e-01],
        [2.6822e-03, 9.9732e-01],
        [1.4750e-03, 9.9853e-01],
        [3.0943e-04, 9.9969e-01],
        [7.0235e-01, 2.9765e-01],
        [4.1912e-05, 9.9996e-01],
        [1.1994e-05, 9.9999e-01],
        [2.9777e-05, 9.9997e-01],
        [2.5485e-01, 7.4515e-01],
        [4.0766e-04, 9.9959e-01],
        [4.6025e-04, 9.9954e-01],
        [9.8689e-01, 1.3110e-02],
        [2.2898e-05, 9.9998e-01],
        [9.4359e-01, 5.6405e-02],
        [7.2923e-01, 2.7077e-01],
        [8.3914e-07, 1.0000e+00],
        [5.4817e-03, 9.9452e-01],
        [8.3439e-01, 1.6560e-01],
        [8.2539e-01, 1.7461e-01],
        [8.1191e-04, 9.9919e-01],
        [3.5871e-04, 9.9964e-01],
        [1.6875e-01, 8.3125e-01],
        [3.8626e-01, 6.1374e-01],
        [5.8618e-04, 9.9941e-01],
        [1.1795e-04, 9.9988e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.4182e-01, 5.5818e-01],
        [2.7907e-03, 9.9721e-01],
        [4.6918e-03, 9.9531e-01],
        [6.8437e-05, 9.9993e-01],
        [1.3422e-03, 9.9866e-01],
        [6.2987e-05, 9.9994e-01],
        [1.3993e-02, 9.8601e-01],
        [9.9475e-01, 5.2532e-03],
        [3.7295e-05, 9.9996e-01],
        [5.3140e-02, 9.4686e-01],
        [8.9335e-04, 9.9911e-01],
        [2.5742e-01, 7.4258e-01],
        [7.2529e-05, 9.9993e-01],
        [1.0343e-02, 9.8966e-01],
        [5.6525e-05, 9.9994e-01],
        [1.3170e-02, 9.8683e-01],
        [8.4538e-04, 9.9915e-01],
        [2.8569e-03, 9.9714e-01],
        [4.9122e-02, 9.5088e-01],
        [7.6764e-06, 9.9999e-01],
        [3.9638e-01, 6.0362e-01],
        [7.3049e-04, 9.9927e-01],
        [9.7916e-01, 2.0837e-02],
        [1.2291e-02, 9.8771e-01],
        [1.8106e-04, 9.9982e-01],
        [2.5551e-02, 9.7445e-01],
        [1.5839e-03, 9.9842e-01],
        [8.4940e-05, 9.9992e-01],
        [2.1982e-02, 9.7802e-01],
        [4.0660e-05, 9.9996e-01],
        [9.5718e-01, 4.2822e-02],
        [2.3267e-03, 9.9767e-01],
        [3.5592e-04, 9.9964e-01],
        [9.9800e-01, 1.9998e-03],
        [7.8705e-04, 9.9921e-01],
        [1.1819e-01, 8.8181e-01],
        [1.0857e-03, 9.9891e-01],
        [6.5529e-03, 9.9345e-01],
        [1.3266e-02, 9.8673e-01],
        [1.1112e-02, 9.8889e-01],
        [1.3787e-03, 9.9862e-01],
        [9.9981e-01, 1.9235e-04],
        [9.8760e-01, 1.2400e-02],
        [4.4060e-05, 9.9996e-01],
        [9.2227e-01, 7.7729e-02],
        [8.9579e-05, 9.9991e-01],
        [6.0193e-04, 9.9940e-01],
        [1.1771e-02, 9.8823e-01],
        [5.3611e-05, 9.9995e-01],
        [5.9055e-03, 9.9409e-01],
        [1.4267e-02, 9.8573e-01],
        [1.4515e-04, 9.9985e-01],
        [1.4578e-01, 8.5422e-01],
        [9.6292e-01, 3.7079e-02],
        [9.9279e-01, 7.2150e-03],
        [4.9529e-04, 9.9950e-01],
        [9.8752e-01, 1.2483e-02],
        [1.8610e-04, 9.9981e-01],
        [3.1768e-03, 9.9682e-01],
        [3.1462e-05, 9.9997e-01],
        [6.3975e-03, 9.9360e-01],
        [3.8816e-03, 9.9612e-01],
        [2.1020e-05, 9.9998e-01],
        [1.2844e-03, 9.9872e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.9894e-03, 9.9501e-01],
        [6.5279e-03, 9.9347e-01],
        [1.7698e-04, 9.9982e-01],
        [9.5983e-01, 4.0170e-02],
        [2.5581e-03, 9.9744e-01],
        [9.9947e-01, 5.3247e-04],
        [7.4102e-06, 9.9999e-01],
        [6.0015e-05, 9.9994e-01],
        [4.5241e-03, 9.9548e-01],
        [8.6419e-02, 9.1358e-01],
        [9.1224e-04, 9.9909e-01],
        [3.0299e-01, 6.9701e-01],
        [2.0114e-03, 9.9799e-01],
        [1.5956e-03, 9.9840e-01],
        [1.9130e-03, 9.9809e-01],
        [1.3660e-01, 8.6340e-01],
        [1.6193e-02, 9.8381e-01],
        [7.9867e-04, 9.9920e-01],
        [2.1371e-02, 9.7863e-01],
        [8.5117e-03, 9.9149e-01],
        [6.6816e-02, 9.3318e-01],
        [3.3183e-01, 6.6817e-01],
        [9.8367e-05, 9.9990e-01],
        [1.5883e-02, 9.8412e-01],
        [9.5605e-01, 4.3955e-02],
        [9.7326e-01, 2.6735e-02],
        [1.6851e-02, 9.8315e-01],
        [6.7879e-03, 9.9321e-01],
        [6.1201e-01, 3.8799e-01],
        [5.6000e-03, 9.9440e-01],
        [1.7163e-04, 9.9983e-01],
        [5.4814e-04, 9.9945e-01],
        [8.8093e-05, 9.9991e-01],
        [8.8631e-05, 9.9991e-01],
        [2.1381e-03, 9.9786e-01],
        [9.9369e-01, 6.3114e-03],
        [3.8788e-05, 9.9996e-01],
        [1.3606e-03, 9.9864e-01],
        [9.6495e-05, 9.9990e-01],
        [9.7053e-01, 2.9468e-02],
        [6.1637e-05, 9.9994e-01],
        [9.0792e-01, 9.2081e-02],
        [7.1737e-01, 2.8263e-01],
        [5.8963e-03, 9.9410e-01],
        [2.7633e-04, 9.9972e-01],
        [2.1679e-02, 9.7832e-01],
        [7.0043e-04, 9.9930e-01],
        [2.9791e-03, 9.9702e-01],
        [9.1758e-01, 8.2416e-02],
        [5.7087e-02, 9.4291e-01],
        [6.4216e-01, 3.5784e-01],
        [8.9644e-01, 1.0356e-01],
        [8.8413e-01, 1.1587e-01],
        [6.5508e-02, 9.3449e-01],
        [6.4335e-02, 9.3567e-01],
        [1.6320e-01, 8.3680e-01],
        [1.5799e-05, 9.9998e-01],
        [9.8027e-02, 9.0197e-01],
        [3.6818e-05, 9.9996e-01],
        [2.3477e-04, 9.9977e-01],
        [1.7474e-03, 9.9825e-01],
        [7.9913e-02, 9.2009e-01],
        [4.4088e-04, 9.9956e-01],
        [3.4443e-01, 6.5557e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9726e-01, 2.7414e-03],
        [9.0615e-04, 9.9909e-01],
        [7.5057e-03, 9.9249e-01],
        [7.0324e-03, 9.9297e-01],
        [9.7831e-01, 2.1686e-02],
        [1.8681e-01, 8.1319e-01],
        [9.8026e-01, 1.9740e-02],
        [3.1534e-03, 9.9685e-01],
        [4.7666e-04, 9.9952e-01],
        [9.8949e-01, 1.0512e-02],
        [5.0567e-03, 9.9494e-01],
        [9.8318e-01, 1.6823e-02],
        [9.8320e-01, 1.6801e-02],
        [1.0428e-02, 9.8957e-01],
        [4.8017e-02, 9.5198e-01],
        [3.3990e-03, 9.9660e-01],
        [2.7611e-04, 9.9972e-01],
        [2.4163e-03, 9.9758e-01],
        [8.5364e-02, 9.1464e-01],
        [3.1640e-03, 9.9684e-01],
        [6.0661e-04, 9.9939e-01],
        [9.9018e-01, 9.8204e-03],
        [9.0458e-03, 9.9095e-01],
        [2.3462e-04, 9.9977e-01],
        [2.5117e-02, 9.7488e-01],
        [1.1691e-04, 9.9988e-01],
        [1.2363e-01, 8.7637e-01],
        [9.8320e-01, 1.6800e-02],
        [9.7301e-01, 2.6991e-02],
        [8.0069e-05, 9.9992e-01],
        [9.8460e-01, 1.5395e-02],
        [1.4205e-03, 9.9858e-01],
        [4.4874e-04, 9.9955e-01],
        [9.6549e-01, 3.4514e-02],
        [6.1763e-02, 9.3824e-01],
        [8.0599e-01, 1.9401e-01],
        [9.9393e-03, 9.9006e-01],
        [1.3689e-02, 9.8631e-01],
        [3.8567e-04, 9.9961e-01],
        [9.0615e-04, 9.9909e-01],
        [1.3967e-03, 9.9860e-01],
        [1.0815e-01, 8.9185e-01],
        [6.6672e-04, 9.9933e-01],
        [1.0218e-01, 8.9782e-01],
        [4.3724e-02, 9.5628e-01],
        [2.7989e-03, 9.9720e-01],
        [9.5171e-01, 4.8295e-02],
        [3.2382e-01, 6.7618e-01],
        [7.8633e-05, 9.9992e-01],
        [8.8720e-06, 9.9999e-01],
        [1.5317e-03, 9.9847e-01],
        [9.9162e-01, 8.3766e-03],
        [6.5032e-02, 9.3497e-01],
        [4.4333e-01, 5.5667e-01],
        [1.4185e-04, 9.9986e-01],
        [8.1100e-01, 1.8900e-01],
        [3.1635e-04, 9.9968e-01],
        [1.1373e-03, 9.9886e-01],
        [2.5985e-04, 9.9974e-01],
        [7.2510e-03, 9.9275e-01],
        [5.7394e-03, 9.9426e-01],
        [1.6625e-02, 9.8338e-01],
        [2.0104e-02, 9.7990e-01],
        [8.6588e-03, 9.9134e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.6417e-04, 9.9934e-01],
        [9.9905e-01, 9.5424e-04],
        [1.4754e-02, 9.8525e-01],
        [9.6270e-01, 3.7298e-02],
        [8.2425e-03, 9.9176e-01],
        [5.3366e-03, 9.9466e-01],
        [4.2867e-04, 9.9957e-01],
        [9.9593e-01, 4.0656e-03],
        [6.9363e-04, 9.9931e-01],
        [9.7253e-01, 2.7470e-02],
        [1.1053e-02, 9.8895e-01],
        [4.5633e-03, 9.9544e-01],
        [9.3499e-01, 6.5010e-02],
        [4.0418e-01, 5.9582e-01],
        [9.9624e-01, 3.7631e-03],
        [3.4114e-02, 9.6589e-01],
        [7.5589e-03, 9.9244e-01],
        [3.2491e-01, 6.7509e-01],
        [1.1856e-04, 9.9988e-01],
        [2.1478e-02, 9.7852e-01],
        [9.9331e-01, 6.6941e-03],
        [3.7211e-03, 9.9628e-01],
        [2.1583e-03, 9.9784e-01],
        [1.5343e-02, 9.8466e-01],
        [8.3702e-01, 1.6298e-01],
        [4.3005e-04, 9.9957e-01],
        [3.4046e-02, 9.6595e-01],
        [2.7550e-03, 9.9724e-01],
        [9.3904e-01, 6.0964e-02],
        [5.6029e-02, 9.4397e-01],
        [3.1278e-04, 9.9969e-01],
        [8.6911e-04, 9.9913e-01],
        [8.5719e-01, 1.4281e-01],
        [3.9276e-02, 9.6072e-01],
        [1.4482e-04, 9.9986e-01],
        [9.5415e-01, 4.5846e-02],
        [9.9768e-01, 2.3237e-03],
        [1.4419e-03, 9.9856e-01],
        [9.9325e-01, 6.7507e-03],
        [2.1991e-01, 7.8009e-01],
        [9.6569e-03, 9.9034e-01],
        [3.1317e-02, 9.6868e-01],
        [9.5450e-01, 4.5498e-02],
        [4.7749e-05, 9.9995e-01],
        [1.1185e-03, 9.9888e-01],
        [9.5575e-04, 9.9904e-01],
        [2.0633e-04, 9.9979e-01],
        [9.9406e-01, 5.9415e-03],
        [6.5822e-02, 9.3418e-01],
        [5.2743e-03, 9.9473e-01],
        [4.7082e-03, 9.9529e-01],
        [9.7059e-01, 2.9410e-02],
        [9.9354e-01, 6.4591e-03],
        [7.5951e-06, 9.9999e-01],
        [9.9473e-01, 5.2683e-03],
        [7.8953e-04, 9.9921e-01],
        [7.4379e-03, 9.9256e-01],
        [4.6584e-02, 9.5342e-01],
        [9.9897e-01, 1.0321e-03],
        [2.0785e-04, 9.9979e-01],
        [1.6466e-04, 9.9984e-01],
        [1.0473e-02, 9.8953e-01],
        [3.4409e-02, 9.6559e-01],
        [3.4226e-03, 9.9658e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.7152e-04, 9.9973e-01],
        [1.6805e-04, 9.9983e-01],
        [8.5858e-01, 1.4142e-01],
        [2.2670e-03, 9.9773e-01],
        [9.4912e-03, 9.9051e-01],
        [1.0391e-02, 9.8961e-01],
        [1.9496e-03, 9.9805e-01],
        [2.6855e-02, 9.7314e-01],
        [2.1633e-02, 9.7837e-01],
        [1.3435e-02, 9.8656e-01],
        [6.4694e-02, 9.3531e-01],
        [1.3374e-02, 9.8663e-01],
        [5.5876e-01, 4.4124e-01],
        [1.6349e-02, 9.8365e-01],
        [1.3254e-04, 9.9987e-01],
        [5.7367e-03, 9.9426e-01],
        [9.6335e-01, 3.6645e-02],
        [2.7171e-04, 9.9973e-01],
        [8.9891e-02, 9.1011e-01],
        [1.6819e-01, 8.3181e-01],
        [9.9850e-01, 1.5007e-03],
        [1.1421e-02, 9.8858e-01],
        [1.7203e-01, 8.2797e-01],
        [1.0293e-03, 9.9897e-01],
        [9.8504e-01, 1.4956e-02],
        [4.8513e-01, 5.1487e-01],
        [9.8995e-03, 9.9010e-01],
        [6.8206e-02, 9.3179e-01],
        [4.8780e-03, 9.9512e-01],
        [5.5452e-03, 9.9445e-01],
        [2.0023e-03, 9.9800e-01],
        [9.9027e-01, 9.7299e-03],
        [9.8714e-01, 1.2864e-02],
        [9.0530e-04, 9.9909e-01],
        [2.2363e-04, 9.9978e-01],
        [5.1050e-02, 9.4895e-01],
        [9.2881e-02, 9.0712e-01],
        [1.2382e-02, 9.8762e-01],
        [1.5507e-01, 8.4493e-01],
        [1.2548e-03, 9.9875e-01],
        [1.6002e-03, 9.9840e-01],
        [9.7952e-02, 9.0205e-01],
        [5.7496e-04, 9.9943e-01],
        [9.8738e-01, 1.2623e-02],
        [5.0072e-04, 9.9950e-01],
        [3.1252e-05, 9.9997e-01],
        [5.6517e-01, 4.3483e-01],
        [1.6284e-03, 9.9837e-01],
        [9.9780e-01, 2.2026e-03],
        [1.6991e-02, 9.8301e-01],
        [1.0220e-04, 9.9990e-01],
        [1.1731e-02, 9.8827e-01],
        [5.1707e-03, 9.9483e-01],
        [1.5138e-01, 8.4862e-01],
        [1.3025e-03, 9.9870e-01],
        [3.2643e-05, 9.9997e-01],
        [1.0692e-03, 9.9893e-01],
        [2.1831e-02, 9.7817e-01],
        [4.5852e-04, 9.9954e-01],
        [9.7747e-01, 2.2527e-02],
        [3.0509e-03, 9.9695e-01],
        [9.9970e-01, 3.0114e-04],
        [2.1442e-02, 9.7856e-01],
        [9.9648e-01, 3.5216e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.5188e-04, 9.9905e-01],
        [4.9918e-02, 9.5008e-01],
        [7.0280e-03, 9.9297e-01],
        [9.9388e-01, 6.1200e-03],
        [9.7418e-01, 2.5823e-02],
        [1.5430e-03, 9.9846e-01],
        [1.8539e-04, 9.9981e-01],
        [9.9491e-01, 5.0907e-03],
        [3.4645e-05, 9.9997e-01],
        [1.9371e-04, 9.9981e-01],
        [1.2160e-02, 9.8784e-01],
        [4.5281e-04, 9.9955e-01],
        [9.9374e-01, 6.2641e-03],
        [6.7160e-03, 9.9328e-01],
        [5.9170e-01, 4.0830e-01],
        [4.4221e-04, 9.9956e-01],
        [1.6952e-02, 9.8305e-01],
        [3.5350e-04, 9.9965e-01],
        [4.8116e-05, 9.9995e-01],
        [1.4016e-02, 9.8598e-01],
        [2.3896e-03, 9.9761e-01],
        [6.5184e-02, 9.3482e-01],
        [2.2719e-03, 9.9773e-01],
        [3.8921e-01, 6.1079e-01],
        [4.1401e-03, 9.9586e-01],
        [9.9141e-01, 8.5948e-03],
        [9.5949e-01, 4.0505e-02],
        [9.9474e-01, 5.2595e-03],
        [1.4398e-04, 9.9986e-01],
        [1.8376e-03, 9.9816e-01],
        [4.2303e-03, 9.9577e-01],
        [6.2259e-02, 9.3774e-01],
        [7.9091e-05, 9.9992e-01],
        [6.6947e-01, 3.3053e-01],
        [2.2906e-02, 9.7709e-01],
        [6.0807e-03, 9.9392e-01],
        [2.5664e-04, 9.9974e-01],
        [8.3678e-02, 9.1632e-01],
        [9.8183e-01, 1.8174e-02],
        [1.7386e-04, 9.9983e-01],
        [3.2370e-03, 9.9676e-01],
        [2.2598e-03, 9.9774e-01],
        [9.1973e-01, 8.0272e-02],
        [1.1123e-02, 9.8888e-01],
        [3.6589e-03, 9.9634e-01],
        [1.8219e-04, 9.9982e-01],
        [3.9845e-04, 9.9960e-01],
        [2.1135e-02, 9.7886e-01],
        [2.9977e-02, 9.7002e-01],
        [6.2635e-02, 9.3737e-01],
        [9.9603e-01, 3.9717e-03],
        [1.5531e-02, 9.8447e-01],
        [7.2379e-01, 2.7621e-01],
        [5.5097e-04, 9.9945e-01],
        [9.9924e-01, 7.5781e-04],
        [5.4391e-05, 9.9995e-01],
        [2.1102e-01, 7.8898e-01],
        [1.0101e-03, 9.9899e-01],
        [3.6872e-03, 9.9631e-01],
        [3.3652e-02, 9.6635e-01],
        [6.0555e-01, 3.9445e-01],
        [2.3131e-05, 9.9998e-01],
        [1.4730e-04, 9.9985e-01],
        [7.6775e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.2312e-05, 9.9992e-01],
        [1.1975e-02, 9.8803e-01],
        [1.3374e-02, 9.8663e-01],
        [8.4614e-04, 9.9915e-01],
        [9.8741e-01, 1.2593e-02],
        [3.8117e-04, 9.9962e-01],
        [5.0147e-05, 9.9995e-01],
        [9.9904e-01, 9.6408e-04],
        [9.8906e-01, 1.0937e-02],
        [9.9266e-01, 7.3363e-03],
        [9.7990e-01, 2.0099e-02],
        [1.8540e-02, 9.8146e-01],
        [1.7116e-02, 9.8288e-01],
        [7.8152e-01, 2.1848e-01],
        [3.4130e-03, 9.9659e-01],
        [1.5791e-01, 8.4209e-01],
        [2.2087e-04, 9.9978e-01],
        [9.8180e-01, 1.8203e-02],
        [7.7639e-04, 9.9922e-01],
        [9.9596e-01, 4.0389e-03],
        [9.7494e-01, 2.5058e-02],
        [1.7621e-02, 9.8238e-01],
        [3.2276e-03, 9.9677e-01],
        [9.9946e-01, 5.3692e-04],
        [3.4140e-02, 9.6586e-01],
        [1.5468e-04, 9.9985e-01],
        [9.9060e-01, 9.3956e-03],
        [2.2233e-03, 9.9778e-01],
        [1.5832e-05, 9.9998e-01],
        [9.8050e-01, 1.9495e-02],
        [9.4585e-03, 9.9054e-01],
        [9.4434e-01, 5.5664e-02],
        [1.0658e-03, 9.9893e-01],
        [4.9263e-02, 9.5074e-01],
        [4.4702e-01, 5.5299e-01],
        [2.6230e-04, 9.9974e-01],
        [1.3897e-03, 9.9861e-01],
        [5.5518e-03, 9.9445e-01],
        [6.6081e-01, 3.3919e-01],
        [2.0181e-01, 7.9819e-01],
        [1.4432e-03, 9.9856e-01],
        [3.0138e-02, 9.6986e-01],
        [1.4791e-03, 9.9852e-01],
        [3.3378e-02, 9.6662e-01],
        [1.0052e-03, 9.9899e-01],
        [2.5552e-04, 9.9974e-01],
        [9.9733e-01, 2.6749e-03],
        [4.1906e-03, 9.9581e-01],
        [6.3389e-01, 3.6611e-01],
        [5.2936e-02, 9.4706e-01],
        [9.7261e-01, 2.7391e-02],
        [7.3525e-03, 9.9265e-01],
        [1.4452e-04, 9.9986e-01],
        [1.1013e-03, 9.9890e-01],
        [1.8057e-04, 9.9982e-01],
        [9.1342e-03, 9.9087e-01],
        [2.6964e-01, 7.3036e-01],
        [2.5228e-03, 9.9748e-01],
        [2.4024e-03, 9.9760e-01],
        [9.8335e-01, 1.6648e-02],
        [4.6361e-03, 9.9536e-01],
        [3.3268e-02, 9.6673e-01],
        [6.7380e-03, 9.9326e-01],
        [6.7397e-02, 9.3260e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.5841e-05, 9.9993e-01],
        [9.8431e-01, 1.5693e-02],
        [5.0600e-03, 9.9494e-01],
        [8.8434e-03, 9.9116e-01],
        [4.7320e-03, 9.9527e-01],
        [9.9709e-01, 2.9144e-03],
        [9.9458e-01, 5.4158e-03],
        [9.9259e-01, 7.4116e-03],
        [9.9021e-01, 9.7917e-03],
        [6.7316e-01, 3.2684e-01],
        [1.1501e-02, 9.8850e-01],
        [9.2246e-01, 7.7540e-02],
        [5.1392e-04, 9.9949e-01],
        [1.5887e-01, 8.4113e-01],
        [1.3188e-02, 9.8681e-01],
        [1.6116e-01, 8.3884e-01],
        [4.7143e-01, 5.2857e-01],
        [1.1604e-03, 9.9884e-01],
        [4.7316e-03, 9.9527e-01],
        [5.0016e-05, 9.9995e-01],
        [2.9440e-04, 9.9971e-01],
        [2.2221e-01, 7.7779e-01],
        [1.0186e-03, 9.9898e-01],
        [9.1904e-03, 9.9081e-01],
        [1.8412e-02, 9.8159e-01],
        [1.7059e-05, 9.9998e-01],
        [2.1218e-04, 9.9979e-01],
        [2.8155e-03, 9.9718e-01],
        [9.9741e-01, 2.5905e-03],
        [6.4620e-03, 9.9354e-01],
        [4.4773e-04, 9.9955e-01],
        [9.9522e-01, 4.7794e-03],
        [1.0134e-01, 8.9866e-01],
        [8.2676e-03, 9.9173e-01],
        [2.8005e-02, 9.7200e-01],
        [3.2537e-04, 9.9967e-01],
        [1.1178e-02, 9.8882e-01],
        [5.2821e-03, 9.9472e-01],
        [1.1821e-02, 9.8818e-01],
        [9.5439e-01, 4.5615e-02],
        [9.9236e-01, 7.6405e-03],
        [3.7403e-04, 9.9963e-01],
        [2.9644e-03, 9.9704e-01],
        [3.6295e-03, 9.9637e-01],
        [1.2981e-03, 9.9870e-01],
        [4.5812e-04, 9.9954e-01],
        [6.6755e-03, 9.9332e-01],
        [1.6741e-02, 9.8326e-01],
        [7.2737e-01, 2.7263e-01],
        [7.4165e-01, 2.5835e-01],
        [9.6734e-01, 3.2655e-02],
        [1.7374e-03, 9.9826e-01],
        [1.2439e-01, 8.7561e-01],
        [9.6634e-02, 9.0337e-01],
        [1.4869e-02, 9.8513e-01],
        [3.3185e-01, 6.6815e-01],
        [2.4936e-02, 9.7506e-01],
        [9.2589e-01, 7.4113e-02],
        [1.9447e-03, 9.9806e-01],
        [2.7619e-01, 7.2381e-01],
        [9.0605e-01, 9.3955e-02],
        [1.0813e-03, 9.9892e-01],
        [9.5979e-01, 4.0209e-02],
        [1.0038e-02, 9.8996e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9681e-01, 3.1923e-03],
        [5.0019e-01, 4.9981e-01],
        [1.9063e-02, 9.8094e-01],
        [1.5614e-04, 9.9984e-01],
        [9.9838e-01, 1.6192e-03],
        [9.9123e-01, 8.7666e-03],
        [1.5927e-03, 9.9841e-01],
        [9.8883e-01, 1.1170e-02],
        [4.2840e-02, 9.5716e-01],
        [2.0124e-04, 9.9980e-01],
        [6.9084e-05, 9.9993e-01],
        [9.9811e-01, 1.8888e-03],
        [1.5700e-03, 9.9843e-01],
        [9.8905e-01, 1.0953e-02],
        [2.2178e-04, 9.9978e-01],
        [9.9887e-01, 1.1255e-03],
        [6.7483e-07, 1.0000e+00],
        [4.1209e-04, 9.9959e-01],
        [8.8869e-01, 1.1131e-01],
        [8.7171e-02, 9.1283e-01],
        [2.1780e-03, 9.9782e-01],
        [3.3321e-04, 9.9967e-01],
        [2.0434e-04, 9.9980e-01],
        [3.1662e-02, 9.6834e-01],
        [4.2504e-02, 9.5750e-01],
        [2.2971e-02, 9.7703e-01],
        [7.9917e-05, 9.9992e-01],
        [1.4429e-03, 9.9856e-01],
        [9.5927e-01, 4.0727e-02],
        [4.2080e-02, 9.5792e-01],
        [3.3872e-03, 9.9661e-01],
        [5.9633e-01, 4.0367e-01],
        [4.3156e-02, 9.5684e-01],
        [6.2831e-03, 9.9372e-01],
        [1.2301e-02, 9.8770e-01],
        [1.8990e-01, 8.1010e-01],
        [2.7653e-03, 9.9723e-01],
        [9.1925e-04, 9.9908e-01],
        [6.2744e-04, 9.9937e-01],
        [9.7819e-01, 2.1813e-02],
        [1.2007e-02, 9.8799e-01],
        [2.3262e-03, 9.9767e-01],
        [5.3381e-03, 9.9466e-01],
        [4.9329e-04, 9.9951e-01],
        [2.3349e-03, 9.9767e-01],
        [9.9813e-01, 1.8658e-03],
        [2.2087e-03, 9.9779e-01],
        [9.9555e-01, 4.4488e-03],
        [9.9553e-01, 4.4720e-03],
        [9.5418e-01, 4.5820e-02],
        [1.2228e-03, 9.9878e-01],
        [2.0119e-02, 9.7988e-01],
        [9.9780e-01, 2.2014e-03],
        [1.8542e-02, 9.8146e-01],
        [1.7714e-03, 9.9823e-01],
        [2.4123e-04, 9.9976e-01],
        [9.9290e-01, 7.1031e-03],
        [1.7489e-03, 9.9825e-01],
        [9.7819e-01, 2.1807e-02],
        [6.6110e-04, 9.9934e-01],
        [6.0173e-03, 9.9398e-01],
        [6.4072e-03, 9.9359e-01],
        [9.9767e-01, 2.3341e-03],
        [9.7755e-01, 2.2450e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[3.9044e-04, 9.9961e-01],
        [1.1290e-03, 9.9887e-01],
        [3.2246e-03, 9.9678e-01],
        [8.2168e-04, 9.9918e-01],
        [4.4906e-01, 5.5094e-01],
        [4.2180e-03, 9.9578e-01],
        [2.1913e-03, 9.9781e-01],
        [3.6470e-05, 9.9996e-01],
        [3.7545e-03, 9.9625e-01],
        [6.5289e-01, 3.4711e-01],
        [2.9367e-02, 9.7063e-01],
        [9.0773e-01, 9.2271e-02],
        [1.7537e-03, 9.9825e-01],
        [4.4509e-05, 9.9996e-01],
        [1.1887e-04, 9.9988e-01],
        [1.0123e-02, 9.8988e-01],
        [1.4518e-02, 9.8548e-01],
        [1.5433e-04, 9.9985e-01],
        [7.9805e-01, 2.0195e-01],
        [4.4924e-05, 9.9996e-01],
        [3.1943e-04, 9.9968e-01],
        [9.9864e-02, 9.0014e-01],
        [1.3148e-01, 8.6852e-01],
        [6.6614e-05, 9.9993e-01],
        [1.1454e-02, 9.8855e-01],
        [9.7374e-01, 2.6258e-02],
        [9.6882e-01, 3.1182e-02],
        [5.5106e-02, 9.4489e-01],
        [1.9748e-05, 9.9998e-01],
        [4.5851e-05, 9.9995e-01],
        [1.9796e-03, 9.9802e-01],
        [2.6774e-05, 9.9997e-01],
        [6.3727e-03, 9.9363e-01],
        [1.4488e-04, 9.9986e-01],
        [1.0830e-01, 8.9170e-01],
        [5.9245e-02, 9.4075e-01],
        [9.6429e-01, 3.5706e-02],
        [4.6868e-01, 5.3132e-01],
        [6.3406e-05, 9.9994e-01],
        [2.1915e-03, 9.9781e-01],
        [1.1439e-02, 9.8856e-01],
        [3.8347e-06, 1.0000e+00],
        [2.6614e-02, 9.7339e-01],
        [1.3239e-03, 9.9868e-01],
        [9.9985e-01, 1.4844e-04],
        [6.8216e-03, 9.9318e-01],
        [5.5993e-03, 9.9440e-01],
        [2.9606e-02, 9.7039e-01],
        [9.4875e-01, 5.1254e-02],
        [9.8844e-01, 1.1558e-02],
        [4.2037e-04, 9.9958e-01],
        [3.5589e-02, 9.6441e-01],
        [3.1878e-05, 9.9997e-01],
        [3.2926e-04, 9.9967e-01],
        [1.1243e-02, 9.8876e-01],
        [4.5376e-04, 9.9955e-01],
        [1.7569e-01, 8.2431e-01],
        [9.9687e-01, 3.1313e-03],
        [1.5412e-01, 8.4588e-01],
        [1.8380e-04, 9.9982e-01],
        [8.7023e-06, 9.9999e-01],
        [2.4890e-04, 9.9975e-01],
        [9.7858e-01, 2.1418e-02],
        [3.2641e-02, 9.6736e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.2505e-05, 9.9993e-01],
        [1.4473e-03, 9.9855e-01],
        [5.8356e-03, 9.9416e-01],
        [1.3127e-04, 9.9987e-01],
        [9.9659e-01, 3.4087e-03],
        [8.5602e-03, 9.9144e-01],
        [2.1338e-04, 9.9979e-01],
        [9.8824e-01, 1.1758e-02],
        [9.4614e-03, 9.9054e-01],
        [9.4730e-02, 9.0527e-01],
        [9.9092e-01, 9.0767e-03],
        [1.6905e-04, 9.9983e-01],
        [2.3197e-02, 9.7680e-01],
        [9.9470e-01, 5.3021e-03],
        [4.9126e-01, 5.0874e-01],
        [8.1087e-04, 9.9919e-01],
        [9.3698e-03, 9.9063e-01],
        [3.1051e-02, 9.6895e-01],
        [4.5334e-04, 9.9955e-01],
        [4.7816e-03, 9.9522e-01],
        [4.7119e-04, 9.9953e-01],
        [5.5999e-04, 9.9944e-01],
        [8.1782e-03, 9.9182e-01],
        [4.3329e-01, 5.6671e-01],
        [9.7607e-01, 2.3928e-02],
        [1.5316e-04, 9.9985e-01],
        [9.5464e-01, 4.5363e-02],
        [7.9322e-04, 9.9921e-01],
        [2.8455e-05, 9.9997e-01],
        [3.5797e-01, 6.4203e-01],
        [2.9507e-05, 9.9997e-01],
        [6.3200e-03, 9.9368e-01],
        [1.5331e-03, 9.9847e-01],
        [9.9358e-01, 6.4161e-03],
        [7.5435e-03, 9.9246e-01],
        [9.7544e-01, 2.4565e-02],
        [2.4861e-04, 9.9975e-01],
        [9.9683e-01, 3.1699e-03],
        [1.2098e-02, 9.8790e-01],
        [4.6064e-02, 9.5394e-01],
        [9.6451e-01, 3.5486e-02],
        [4.7463e-03, 9.9525e-01],
        [5.0093e-05, 9.9995e-01],
        [9.5601e-01, 4.3988e-02],
        [9.6080e-01, 3.9195e-02],
        [1.4725e-03, 9.9853e-01],
        [8.4475e-05, 9.9992e-01],
        [9.4588e-01, 5.4115e-02],
        [5.1644e-05, 9.9995e-01],
        [2.9714e-03, 9.9703e-01],
        [1.1721e-04, 9.9988e-01],
        [4.2346e-02, 9.5765e-01],
        [9.9302e-01, 6.9778e-03],
        [2.4770e-02, 9.7523e-01],
        [9.0663e-02, 9.0934e-01],
        [6.2818e-05, 9.9994e-01],
        [9.4282e-01, 5.7176e-02],
        [9.9222e-01, 7.7769e-03],
        [2.2683e-03, 9.9773e-01],
        [3.0750e-03, 9.9692e-01],
        [9.8188e-01, 1.8125e-02],
        [9.9801e-01, 1.9923e-03],
        [4.4920e-02, 9.5508e-01],
        [2.0388e-01, 7.9612e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.8935e-03, 9.9611e-01],
        [4.0971e-04, 9.9959e-01],
        [3.9145e-02, 9.6086e-01],
        [3.9553e-03, 9.9604e-01],
        [9.5593e-05, 9.9990e-01],
        [2.6915e-03, 9.9731e-01],
        [9.0363e-01, 9.6371e-02],
        [6.3674e-02, 9.3633e-01],
        [3.4035e-03, 9.9660e-01],
        [9.6140e-01, 3.8599e-02],
        [2.1390e-05, 9.9998e-01],
        [9.9720e-01, 2.7961e-03],
        [9.1957e-01, 8.0427e-02],
        [1.8360e-02, 9.8164e-01],
        [2.2262e-04, 9.9978e-01],
        [7.0592e-05, 9.9993e-01],
        [4.1475e-04, 9.9959e-01],
        [9.9047e-01, 9.5340e-03],
        [9.4131e-03, 9.9059e-01],
        [5.5583e-02, 9.4442e-01],
        [1.7307e-03, 9.9827e-01],
        [1.7596e-04, 9.9982e-01],
        [1.3987e-01, 8.6013e-01],
        [8.8428e-05, 9.9991e-01],
        [6.8309e-03, 9.9317e-01],
        [6.4634e-01, 3.5366e-01],
        [6.7891e-02, 9.3211e-01],
        [4.3427e-04, 9.9957e-01],
        [9.9141e-01, 8.5868e-03],
        [9.9985e-01, 1.5217e-04],
        [3.5323e-02, 9.6468e-01],
        [2.4393e-02, 9.7561e-01],
        [1.3987e-01, 8.6013e-01],
        [9.8705e-01, 1.2945e-02],
        [8.8138e-03, 9.9119e-01],
        [3.8750e-05, 9.9996e-01],
        [2.1556e-03, 9.9784e-01],
        [9.9471e-01, 5.2921e-03],
        [3.4350e-02, 9.6565e-01],
        [7.1666e-01, 2.8334e-01],
        [1.1710e-03, 9.9883e-01],
        [1.5372e-04, 9.9985e-01],
        [3.7190e-03, 9.9628e-01],
        [3.0928e-04, 9.9969e-01],
        [4.2103e-02, 9.5790e-01],
        [9.0613e-01, 9.3871e-02],
        [2.2331e-04, 9.9978e-01],
        [9.6514e-01, 3.4857e-02],
        [3.5102e-04, 9.9965e-01],
        [1.6145e-03, 9.9839e-01],
        [1.3743e-04, 9.9986e-01],
        [1.5864e-01, 8.4136e-01],
        [2.1449e-03, 9.9786e-01],
        [1.5605e-03, 9.9844e-01],
        [7.7394e-05, 9.9992e-01],
        [3.1537e-02, 9.6846e-01],
        [2.9582e-03, 9.9704e-01],
        [1.0403e-02, 9.8960e-01],
        [1.7178e-03, 9.9828e-01],
        [7.2249e-03, 9.9278e-01],
        [5.8174e-02, 9.4183e-01],
        [2.7968e-03, 9.9720e-01],
        [4.2134e-04, 9.9958e-01],
        [2.3020e-03, 9.9770e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9539e-03, 9.9005e-01],
        [3.7171e-04, 9.9963e-01],
        [3.2534e-05, 9.9997e-01],
        [4.5910e-05, 9.9995e-01],
        [1.7514e-05, 9.9998e-01],
        [6.2714e-04, 9.9937e-01],
        [1.8023e-03, 9.9820e-01],
        [1.7127e-02, 9.8287e-01],
        [7.9992e-01, 2.0008e-01],
        [5.5323e-04, 9.9945e-01],
        [9.3804e-01, 6.1963e-02],
        [4.5680e-05, 9.9995e-01],
        [6.4262e-03, 9.9357e-01],
        [9.1169e-05, 9.9991e-01],
        [8.2646e-01, 1.7354e-01],
        [8.8655e-01, 1.1345e-01],
        [2.8158e-04, 9.9972e-01],
        [2.2671e-04, 9.9977e-01],
        [2.1352e-04, 9.9979e-01],
        [9.7522e-01, 2.4781e-02],
        [2.3464e-04, 9.9977e-01],
        [7.8260e-03, 9.9217e-01],
        [8.5694e-01, 1.4306e-01],
        [5.9203e-05, 9.9994e-01],
        [5.2455e-01, 4.7545e-01],
        [4.8850e-03, 9.9512e-01],
        [1.7716e-02, 9.8228e-01],
        [1.4412e-02, 9.8559e-01],
        [1.3690e-05, 9.9999e-01],
        [1.2906e-01, 8.7094e-01],
        [5.2210e-03, 9.9478e-01],
        [3.5070e-04, 9.9965e-01],
        [4.1454e-01, 5.8546e-01],
        [9.8868e-01, 1.1322e-02],
        [1.3914e-03, 9.9861e-01],
        [5.4355e-03, 9.9456e-01],
        [9.8331e-01, 1.6694e-02],
        [9.9411e-01, 5.8891e-03],
        [6.8322e-02, 9.3168e-01],
        [2.2669e-03, 9.9773e-01],
        [9.5685e-01, 4.3152e-02],
        [9.8396e-01, 1.6039e-02],
        [4.5946e-03, 9.9541e-01],
        [9.9095e-01, 9.0539e-03],
        [1.1159e-06, 1.0000e+00],
        [7.1358e-04, 9.9929e-01],
        [6.2225e-05, 9.9994e-01],
        [8.0728e-01, 1.9272e-01],
        [9.9657e-01, 3.4281e-03],
        [9.3210e-04, 9.9907e-01],
        [9.8603e-01, 1.3971e-02],
        [6.8087e-04, 9.9932e-01],
        [1.3827e-02, 9.8617e-01],
        [1.5005e-04, 9.9985e-01],
        [2.0582e-03, 9.9794e-01],
        [9.5826e-06, 9.9999e-01],
        [1.4483e-03, 9.9855e-01],
        [1.9496e-03, 9.9805e-01],
        [5.5269e-06, 9.9999e-01],
        [4.9455e-06, 1.0000e+00],
        [3.9529e-01, 6.0471e-01],
        [2.5296e-03, 9.9747e-01],
        [9.5939e-04, 9.9904e-01],
        [1.8946e-02, 9.8105e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.8109e-03, 9.9319e-01],
        [1.1088e-03, 9.9889e-01],
        [2.4468e-01, 7.5532e-01],
        [9.9431e-01, 5.6890e-03],
        [3.7990e-03, 9.9620e-01],
        [2.7042e-02, 9.7296e-01],
        [3.1698e-02, 9.6830e-01],
        [9.7115e-02, 9.0288e-01],
        [1.0564e-05, 9.9999e-01],
        [8.7011e-01, 1.2989e-01],
        [9.9577e-01, 4.2328e-03],
        [2.3829e-03, 9.9762e-01],
        [7.0452e-01, 2.9548e-01],
        [1.7513e-02, 9.8249e-01],
        [5.5543e-05, 9.9994e-01],
        [6.5480e-04, 9.9935e-01],
        [9.9514e-01, 4.8581e-03],
        [1.1490e-04, 9.9989e-01],
        [9.3374e-01, 6.6255e-02],
        [5.9920e-02, 9.4008e-01],
        [1.0633e-02, 9.8937e-01],
        [5.2998e-02, 9.4700e-01],
        [8.3810e-03, 9.9162e-01],
        [2.2728e-05, 9.9998e-01],
        [2.3207e-03, 9.9768e-01],
        [1.9238e-05, 9.9998e-01],
        [9.9356e-01, 6.4395e-03],
        [3.4245e-05, 9.9997e-01],
        [1.4255e-05, 9.9999e-01],
        [2.7327e-01, 7.2673e-01],
        [6.0574e-05, 9.9994e-01],
        [8.7635e-01, 1.2365e-01],
        [9.0850e-01, 9.1496e-02],
        [9.1569e-02, 9.0843e-01],
        [2.5001e-05, 9.9997e-01],
        [3.6236e-05, 9.9996e-01],
        [6.0577e-04, 9.9939e-01],
        [1.1544e-02, 9.8846e-01],
        [9.1422e-01, 8.5785e-02],
        [3.0409e-02, 9.6959e-01],
        [2.2232e-02, 9.7777e-01],
        [9.9539e-01, 4.6080e-03],
        [3.1706e-06, 1.0000e+00],
        [9.3912e-01, 6.0881e-02],
        [2.2331e-04, 9.9978e-01],
        [3.3808e-03, 9.9662e-01],
        [4.3790e-03, 9.9562e-01],
        [4.0773e-04, 9.9959e-01],
        [9.5615e-01, 4.3845e-02],
        [3.3873e-04, 9.9966e-01],
        [2.2905e-02, 9.7710e-01],
        [9.2444e-01, 7.5562e-02],
        [9.4343e-01, 5.6573e-02],
        [2.1142e-03, 9.9789e-01],
        [5.8126e-05, 9.9994e-01],
        [7.3333e-01, 2.6667e-01],
        [5.4460e-03, 9.9455e-01],
        [7.3311e-06, 9.9999e-01],
        [6.6827e-03, 9.9332e-01],
        [9.9563e-01, 4.3684e-03],
        [3.8194e-04, 9.9962e-01],
        [2.1016e-01, 7.8984e-01],
        [5.7773e-04, 9.9942e-01],
        [6.4010e-04, 9.9936e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.8905e-03, 9.9111e-01],
        [9.4592e-01, 5.4081e-02],
        [1.9982e-06, 1.0000e+00],
        [4.8125e-03, 9.9519e-01],
        [9.5824e-01, 4.1762e-02],
        [2.0551e-04, 9.9979e-01],
        [4.8541e-04, 9.9951e-01],
        [7.9474e-05, 9.9992e-01],
        [7.9785e-01, 2.0215e-01],
        [9.0831e-01, 9.1687e-02],
        [3.9284e-04, 9.9961e-01],
        [9.8197e-01, 1.8034e-02],
        [9.9165e-01, 8.3490e-03],
        [5.1754e-04, 9.9948e-01],
        [9.3862e-01, 6.1383e-02],
        [5.8469e-04, 9.9942e-01],
        [1.3650e-01, 8.6350e-01],
        [2.3080e-03, 9.9769e-01],
        [1.3428e-03, 9.9866e-01],
        [1.4189e-05, 9.9999e-01],
        [1.1486e-01, 8.8514e-01],
        [5.6826e-02, 9.4317e-01],
        [1.9484e-03, 9.9805e-01],
        [2.7828e-04, 9.9972e-01],
        [9.9196e-01, 8.0426e-03],
        [9.9739e-01, 2.6123e-03],
        [9.1078e-04, 9.9909e-01],
        [1.1175e-04, 9.9989e-01],
        [4.6716e-05, 9.9995e-01],
        [4.0585e-03, 9.9594e-01],
        [9.9725e-01, 2.7498e-03],
        [1.2119e-01, 8.7881e-01],
        [1.8219e-05, 9.9998e-01],
        [8.4102e-01, 1.5898e-01],
        [1.8596e-04, 9.9981e-01],
        [3.1193e-02, 9.6881e-01],
        [3.9939e-03, 9.9601e-01],
        [2.8641e-01, 7.1359e-01],
        [9.9885e-01, 1.1497e-03],
        [9.7540e-01, 2.4597e-02],
        [7.1969e-05, 9.9993e-01],
        [5.5850e-03, 9.9442e-01],
        [1.5461e-02, 9.8454e-01],
        [2.4875e-05, 9.9998e-01],
        [7.4755e-01, 2.5245e-01],
        [5.5219e-04, 9.9945e-01],
        [9.9890e-01, 1.0954e-03],
        [9.8034e-01, 1.9665e-02],
        [1.3935e-04, 9.9986e-01],
        [9.0517e-01, 9.4832e-02],
        [6.8302e-01, 3.1698e-01],
        [8.6736e-01, 1.3264e-01],
        [5.3721e-03, 9.9463e-01],
        [1.7786e-05, 9.9998e-01],
        [3.5418e-03, 9.9646e-01],
        [9.4278e-03, 9.9057e-01],
        [4.1790e-03, 9.9582e-01],
        [1.8219e-05, 9.9998e-01],
        [7.2620e-03, 9.9274e-01],
        [9.1677e-01, 8.3225e-02],
        [9.4529e-03, 9.9055e-01],
        [5.1848e-02, 9.4815e-01],
        [9.8951e-01, 1.0494e-02],
        [1.2089e-03, 9.9879e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9099e-01, 9.0097e-03],
        [7.4273e-05, 9.9993e-01],
        [7.0250e-01, 2.9750e-01],
        [4.6799e-03, 9.9532e-01],
        [4.0190e-03, 9.9598e-01],
        [6.2794e-04, 9.9937e-01],
        [8.5044e-03, 9.9150e-01],
        [7.5016e-04, 9.9925e-01],
        [6.0124e-04, 9.9940e-01],
        [9.5981e-01, 4.0193e-02],
        [1.4027e-03, 9.9860e-01],
        [8.0101e-04, 9.9920e-01],
        [9.8775e-01, 1.2252e-02],
        [9.9181e-01, 8.1859e-03],
        [9.9231e-01, 7.6863e-03],
        [5.1853e-03, 9.9481e-01],
        [9.9718e-01, 2.8220e-03],
        [9.9108e-01, 8.9172e-03],
        [9.9399e-01, 6.0055e-03],
        [2.5157e-02, 9.7484e-01],
        [6.2525e-04, 9.9937e-01],
        [1.9676e-04, 9.9980e-01],
        [2.2894e-02, 9.7711e-01],
        [9.8630e-01, 1.3701e-02],
        [8.8148e-01, 1.1852e-01],
        [5.8407e-05, 9.9994e-01],
        [1.5043e-03, 9.9850e-01],
        [3.2957e-05, 9.9997e-01],
        [1.0185e-02, 9.8981e-01],
        [9.9489e-01, 5.1137e-03],
        [9.5417e-01, 4.5829e-02],
        [1.2276e-03, 9.9877e-01],
        [9.9385e-01, 6.1504e-03],
        [1.2740e-03, 9.9873e-01],
        [9.9237e-01, 7.6313e-03],
        [1.4251e-03, 9.9857e-01],
        [1.0928e-03, 9.9891e-01],
        [2.7369e-01, 7.2631e-01],
        [3.1389e-04, 9.9969e-01],
        [7.6919e-06, 9.9999e-01],
        [9.4165e-01, 5.8351e-02],
        [8.0062e-03, 9.9199e-01],
        [3.5666e-03, 9.9643e-01],
        [1.6572e-04, 9.9983e-01],
        [9.9626e-01, 3.7436e-03],
        [3.4938e-03, 9.9651e-01],
        [9.2193e-01, 7.8067e-02],
        [4.7905e-03, 9.9521e-01],
        [4.2663e-04, 9.9957e-01],
        [2.3766e-03, 9.9762e-01],
        [7.2967e-03, 9.9270e-01],
        [1.4745e-04, 9.9985e-01],
        [2.8247e-04, 9.9972e-01],
        [5.5206e-01, 4.4794e-01],
        [4.0711e-02, 9.5929e-01],
        [9.9171e-01, 8.2909e-03],
        [9.8934e-01, 1.0660e-02],
        [4.0390e-04, 9.9960e-01],
        [4.0230e-03, 9.9598e-01],
        [6.5015e-01, 3.4985e-01],
        [6.1871e-02, 9.3813e-01],
        [6.3577e-03, 9.9364e-01],
        [9.7828e-01, 2.1722e-02],
        [4.5647e-04, 9.9954e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5932e-02, 9.8407e-01],
        [9.2794e-02, 9.0721e-01],
        [8.9671e-01, 1.0329e-01],
        [1.2201e-05, 9.9999e-01],
        [1.5096e-03, 9.9849e-01],
        [1.8388e-02, 9.8161e-01],
        [9.9467e-01, 5.3346e-03],
        [1.1986e-02, 9.8801e-01],
        [9.0359e-01, 9.6407e-02],
        [9.6120e-03, 9.9039e-01],
        [5.4604e-05, 9.9995e-01],
        [6.4597e-02, 9.3540e-01],
        [4.3079e-03, 9.9569e-01],
        [3.8769e-03, 9.9612e-01],
        [1.3632e-02, 9.8637e-01],
        [9.9462e-01, 5.3757e-03],
        [1.1485e-04, 9.9989e-01],
        [1.0783e-02, 9.8922e-01],
        [4.9790e-05, 9.9995e-01],
        [1.0656e-02, 9.8934e-01],
        [3.5289e-04, 9.9965e-01],
        [1.0643e-02, 9.8936e-01],
        [2.6363e-03, 9.9736e-01],
        [2.3728e-04, 9.9976e-01],
        [9.9105e-01, 8.9518e-03],
        [2.9413e-04, 9.9971e-01],
        [1.6154e-02, 9.8385e-01],
        [4.2486e-03, 9.9575e-01],
        [9.6878e-01, 3.1224e-02],
        [1.4620e-05, 9.9999e-01],
        [1.0863e-03, 9.9891e-01],
        [4.7112e-01, 5.2888e-01],
        [5.8914e-03, 9.9411e-01],
        [1.8070e-04, 9.9982e-01],
        [4.7920e-03, 9.9521e-01],
        [2.7585e-03, 9.9724e-01],
        [7.2167e-04, 9.9928e-01],
        [5.5589e-02, 9.4441e-01],
        [1.0064e-02, 9.8994e-01],
        [9.9244e-01, 7.5633e-03],
        [1.8499e-03, 9.9815e-01],
        [4.3280e-03, 9.9567e-01],
        [1.4339e-05, 9.9999e-01],
        [9.9628e-01, 3.7221e-03],
        [4.6970e-03, 9.9530e-01],
        [1.6876e-02, 9.8312e-01],
        [2.0578e-04, 9.9979e-01],
        [6.6173e-05, 9.9993e-01],
        [8.6263e-01, 1.3737e-01],
        [9.7152e-01, 2.8475e-02],
        [4.6368e-02, 9.5363e-01],
        [9.8900e-01, 1.0997e-02],
        [9.7426e-01, 2.5736e-02],
        [9.8337e-01, 1.6633e-02],
        [9.9794e-01, 2.0594e-03],
        [1.7196e-03, 9.9828e-01],
        [9.2563e-01, 7.4368e-02],
        [6.1485e-02, 9.3851e-01],
        [3.9383e-04, 9.9961e-01],
        [9.9927e-01, 7.2823e-04],
        [7.7167e-05, 9.9992e-01],
        [5.4936e-05, 9.9995e-01],
        [2.8559e-04, 9.9971e-01],
        [8.2618e-02, 9.1738e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8401e-01, 1.5987e-02],
        [7.4723e-01, 2.5277e-01],
        [3.0443e-05, 9.9997e-01],
        [9.9038e-01, 9.6218e-03],
        [2.5730e-03, 9.9743e-01],
        [4.6112e-03, 9.9539e-01],
        [1.0661e-03, 9.9893e-01],
        [9.4341e-02, 9.0566e-01],
        [3.3862e-04, 9.9966e-01],
        [2.8341e-05, 9.9997e-01],
        [4.2820e-03, 9.9572e-01],
        [1.0821e-04, 9.9989e-01],
        [1.5252e-06, 1.0000e+00],
        [8.6256e-05, 9.9991e-01],
        [8.4546e-01, 1.5454e-01],
        [3.6067e-03, 9.9639e-01],
        [1.7941e-04, 9.9982e-01],
        [6.2467e-01, 3.7533e-01],
        [2.3134e-03, 9.9769e-01],
        [1.5722e-01, 8.4278e-01],
        [1.7679e-06, 1.0000e+00],
        [9.9280e-01, 7.1989e-03],
        [1.0335e-03, 9.9897e-01],
        [4.0421e-02, 9.5958e-01],
        [8.3213e-04, 9.9917e-01],
        [1.6990e-05, 9.9998e-01],
        [7.5423e-05, 9.9992e-01],
        [9.5725e-01, 4.2754e-02],
        [1.7553e-04, 9.9982e-01],
        [4.5517e-02, 9.5448e-01],
        [9.8160e-01, 1.8404e-02],
        [2.8765e-01, 7.1235e-01],
        [1.5544e-03, 9.9845e-01],
        [9.3014e-01, 6.9864e-02],
        [2.0822e-04, 9.9979e-01],
        [4.0578e-01, 5.9422e-01],
        [1.1696e-02, 9.8830e-01],
        [3.0092e-04, 9.9970e-01],
        [1.6491e-05, 9.9998e-01],
        [9.4790e-01, 5.2105e-02],
        [3.0522e-02, 9.6948e-01],
        [9.0914e-01, 9.0856e-02],
        [1.7808e-03, 9.9822e-01],
        [9.8861e-01, 1.1386e-02],
        [5.6210e-06, 9.9999e-01],
        [9.9445e-01, 5.5514e-03],
        [2.2267e-02, 9.7773e-01],
        [1.9363e-03, 9.9806e-01],
        [7.0941e-04, 9.9929e-01],
        [8.0676e-01, 1.9324e-01],
        [1.0898e-02, 9.8910e-01],
        [1.0444e-03, 9.9896e-01],
        [7.5848e-06, 9.9999e-01],
        [3.1213e-01, 6.8787e-01],
        [5.1655e-02, 9.4834e-01],
        [8.6027e-03, 9.9140e-01],
        [5.5644e-03, 9.9444e-01],
        [1.1625e-02, 9.8838e-01],
        [1.4558e-01, 8.5442e-01],
        [8.7954e-03, 9.9120e-01],
        [9.8803e-01, 1.1972e-02],
        [7.0450e-02, 9.2955e-01],
        [7.6792e-05, 9.9992e-01],
        [7.3128e-02, 9.2687e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2799e-03, 9.9672e-01],
        [6.5440e-03, 9.9346e-01],
        [9.9309e-01, 6.9127e-03],
        [9.9254e-01, 7.4596e-03],
        [4.7983e-03, 9.9520e-01],
        [9.6528e-01, 3.4724e-02],
        [7.0044e-02, 9.2996e-01],
        [2.9292e-03, 9.9707e-01],
        [1.4418e-02, 9.8558e-01],
        [5.2041e-05, 9.9995e-01],
        [7.3510e-04, 9.9926e-01],
        [1.6393e-04, 9.9984e-01],
        [3.5215e-02, 9.6478e-01],
        [5.1551e-04, 9.9948e-01],
        [3.1351e-04, 9.9969e-01],
        [5.6909e-05, 9.9994e-01],
        [2.5026e-03, 9.9750e-01],
        [1.0716e-01, 8.9284e-01],
        [9.9594e-01, 4.0624e-03],
        [3.7492e-04, 9.9963e-01],
        [1.4678e-02, 9.8532e-01],
        [1.1915e-03, 9.9881e-01],
        [1.4650e-02, 9.8535e-01],
        [6.6904e-02, 9.3310e-01],
        [1.8715e-03, 9.9813e-01],
        [6.1205e-02, 9.3880e-01],
        [4.7874e-03, 9.9521e-01],
        [1.5038e-04, 9.9985e-01],
        [1.3407e-03, 9.9866e-01],
        [3.8861e-04, 9.9961e-01],
        [2.8215e-01, 7.1785e-01],
        [1.1239e-04, 9.9989e-01],
        [9.7449e-01, 2.5512e-02],
        [2.2417e-03, 9.9776e-01],
        [1.4551e-02, 9.8545e-01],
        [2.5793e-05, 9.9997e-01],
        [1.0844e-03, 9.9892e-01],
        [3.1636e-01, 6.8364e-01],
        [8.4395e-03, 9.9156e-01],
        [2.9088e-02, 9.7091e-01],
        [1.1273e-01, 8.8727e-01],
        [3.8014e-02, 9.6199e-01],
        [7.3780e-04, 9.9926e-01],
        [9.9738e-01, 2.6158e-03],
        [1.4940e-03, 9.9851e-01],
        [8.1664e-03, 9.9183e-01],
        [9.9278e-01, 7.2219e-03],
        [2.1322e-02, 9.7868e-01],
        [6.7645e-01, 3.2355e-01],
        [1.2923e-04, 9.9987e-01],
        [9.9148e-03, 9.9009e-01],
        [2.2417e-03, 9.9776e-01],
        [7.3475e-03, 9.9265e-01],
        [9.9730e-01, 2.6995e-03],
        [1.0514e-03, 9.9895e-01],
        [3.4238e-03, 9.9658e-01],
        [7.8856e-03, 9.9211e-01],
        [1.1461e-04, 9.9989e-01],
        [9.9933e-01, 6.6662e-04],
        [3.0910e-03, 9.9691e-01],
        [6.9936e-04, 9.9930e-01],
        [7.3018e-01, 2.6982e-01],
        [3.7079e-04, 9.9963e-01],
        [7.1734e-04, 9.9928e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.7168e-01, 7.2832e-01],
        [1.9616e-03, 9.9804e-01],
        [9.9999e-01, 1.0162e-05],
        [9.9555e-01, 4.4548e-03],
        [4.2318e-03, 9.9577e-01],
        [9.8878e-01, 1.1224e-02],
        [9.9952e-01, 4.7774e-04],
        [9.9339e-01, 6.6111e-03],
        [9.8085e-01, 1.9148e-02],
        [6.4000e-03, 9.9360e-01],
        [2.8368e-03, 9.9716e-01],
        [8.9085e-06, 9.9999e-01],
        [1.0389e-03, 9.9896e-01],
        [9.4422e-01, 5.5784e-02],
        [1.3757e-01, 8.6243e-01],
        [5.7773e-01, 4.2227e-01],
        [9.3212e-03, 9.9068e-01],
        [1.5209e-02, 9.8479e-01],
        [3.3518e-02, 9.6648e-01],
        [1.0572e-03, 9.9894e-01],
        [9.9746e-01, 2.5389e-03],
        [1.6716e-02, 9.8328e-01],
        [2.8629e-05, 9.9997e-01],
        [6.5937e-01, 3.4063e-01],
        [3.7964e-02, 9.6204e-01],
        [9.9396e-01, 6.0445e-03],
        [6.4558e-02, 9.3544e-01],
        [9.9818e-01, 1.8160e-03],
        [8.8258e-01, 1.1742e-01],
        [9.9010e-01, 9.9010e-03],
        [3.5913e-02, 9.6409e-01],
        [9.7830e-01, 2.1701e-02],
        [2.5415e-04, 9.9975e-01],
        [7.8353e-04, 9.9922e-01],
        [2.9386e-02, 9.7061e-01],
        [3.2900e-01, 6.7100e-01],
        [9.2964e-03, 9.9070e-01],
        [5.8390e-02, 9.4161e-01],
        [2.6850e-03, 9.9731e-01],
        [9.9367e-01, 6.3267e-03],
        [9.9644e-01, 3.5570e-03],
        [1.9616e-03, 9.9804e-01],
        [1.0246e-02, 9.8975e-01],
        [1.2203e-01, 8.7797e-01],
        [9.8273e-01, 1.7274e-02],
        [3.3913e-01, 6.6087e-01],
        [2.2095e-02, 9.7791e-01],
        [9.9799e-01, 2.0069e-03],
        [9.8746e-01, 1.2542e-02],
        [2.4317e-02, 9.7568e-01],
        [9.0713e-01, 9.2869e-02],
        [2.6809e-05, 9.9997e-01],
        [1.4034e-02, 9.8597e-01],
        [4.5005e-04, 9.9955e-01],
        [9.7755e-05, 9.9990e-01],
        [2.5360e-03, 9.9746e-01],
        [3.4138e-03, 9.9659e-01],
        [5.1969e-01, 4.8031e-01],
        [3.7673e-03, 9.9623e-01],
        [9.9699e-01, 3.0084e-03],
        [9.9935e-01, 6.4979e-04],
        [1.5749e-03, 9.9843e-01],
        [9.8701e-01, 1.2985e-02],
        [1.1499e-01, 8.8501e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.5990e-04, 9.9944e-01],
        [2.8412e-04, 9.9972e-01],
        [4.4117e-01, 5.5883e-01],
        [7.8142e-01, 2.1858e-01],
        [1.5213e-02, 9.8479e-01],
        [4.3172e-03, 9.9568e-01],
        [9.9533e-03, 9.9005e-01],
        [2.0957e-03, 9.9790e-01],
        [1.4846e-03, 9.9852e-01],
        [3.8232e-02, 9.6177e-01],
        [2.4002e-04, 9.9976e-01],
        [9.8888e-01, 1.1118e-02],
        [7.8474e-04, 9.9922e-01],
        [9.9845e-01, 1.5523e-03],
        [3.1390e-01, 6.8610e-01],
        [7.7014e-01, 2.2986e-01],
        [2.2756e-04, 9.9977e-01],
        [6.2892e-01, 3.7108e-01],
        [4.9817e-06, 9.9999e-01],
        [1.5481e-03, 9.9845e-01],
        [1.2538e-02, 9.8746e-01],
        [7.4484e-04, 9.9926e-01],
        [2.0383e-03, 9.9796e-01],
        [4.9928e-03, 9.9501e-01],
        [5.3046e-03, 9.9470e-01],
        [9.9556e-01, 4.4366e-03],
        [6.4645e-03, 9.9354e-01],
        [5.1548e-05, 9.9995e-01],
        [1.3156e-05, 9.9999e-01],
        [1.8703e-03, 9.9813e-01],
        [1.6287e-02, 9.8371e-01],
        [3.0910e-02, 9.6909e-01],
        [9.6794e-01, 3.2057e-02],
        [2.0383e-03, 9.9796e-01],
        [9.9282e-01, 7.1812e-03],
        [4.5963e-04, 9.9954e-01],
        [3.8329e-04, 9.9962e-01],
        [2.3813e-01, 7.6187e-01],
        [1.4604e-02, 9.8540e-01],
        [2.2300e-06, 1.0000e+00],
        [9.8041e-01, 1.9587e-02],
        [5.5148e-04, 9.9945e-01],
        [2.4165e-02, 9.7584e-01],
        [1.2767e-06, 1.0000e+00],
        [1.1304e-04, 9.9989e-01],
        [9.9819e-01, 1.8076e-03],
        [1.0477e-01, 8.9523e-01],
        [7.6122e-02, 9.2388e-01],
        [6.5223e-02, 9.3478e-01],
        [8.3831e-05, 9.9992e-01],
        [2.7742e-03, 9.9723e-01],
        [1.8676e-06, 1.0000e+00],
        [9.9036e-01, 9.6366e-03],
        [2.1715e-02, 9.7828e-01],
        [5.4694e-05, 9.9995e-01],
        [9.4801e-01, 5.1989e-02],
        [9.8437e-01, 1.5632e-02],
        [9.9675e-01, 3.2505e-03],
        [3.9037e-04, 9.9961e-01],
        [9.9818e-01, 1.8212e-03],
        [4.5801e-02, 9.5420e-01],
        [9.5191e-03, 9.9048e-01],
        [9.8903e-01, 1.0971e-02],
        [5.8688e-03, 9.9413e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.1326e-05, 9.9994e-01],
        [1.0695e-04, 9.9989e-01],
        [2.1058e-05, 9.9998e-01],
        [9.8799e-01, 1.2007e-02],
        [6.6380e-03, 9.9336e-01],
        [6.9893e-04, 9.9930e-01],
        [3.4254e-05, 9.9997e-01],
        [8.6042e-04, 9.9914e-01],
        [3.5220e-01, 6.4780e-01],
        [3.6394e-03, 9.9636e-01],
        [1.4193e-02, 9.8581e-01],
        [2.7609e-02, 9.7239e-01],
        [3.1658e-04, 9.9968e-01],
        [9.9677e-01, 3.2324e-03],
        [3.6889e-05, 9.9996e-01],
        [2.1270e-04, 9.9979e-01],
        [3.9220e-02, 9.6078e-01],
        [2.0321e-02, 9.7968e-01],
        [2.9297e-03, 9.9707e-01],
        [2.5591e-04, 9.9974e-01],
        [9.1127e-05, 9.9991e-01],
        [7.2023e-01, 2.7977e-01],
        [3.9269e-05, 9.9996e-01],
        [2.3505e-03, 9.9765e-01],
        [6.9232e-01, 3.0768e-01],
        [9.5486e-01, 4.5139e-02],
        [1.7916e-02, 9.8208e-01],
        [9.5295e-01, 4.7052e-02],
        [2.7730e-04, 9.9972e-01],
        [1.0491e-05, 9.9999e-01],
        [6.7570e-03, 9.9324e-01],
        [2.7902e-05, 9.9997e-01],
        [3.2856e-03, 9.9671e-01],
        [8.7737e-01, 1.2263e-01],
        [3.5629e-05, 9.9996e-01],
        [1.6369e-02, 9.8363e-01],
        [3.8461e-03, 9.9615e-01],
        [1.6810e-01, 8.3190e-01],
        [9.9572e-01, 4.2848e-03],
        [3.9990e-02, 9.6001e-01],
        [9.9257e-01, 7.4310e-03],
        [6.2239e-01, 3.7761e-01],
        [2.5095e-05, 9.9997e-01],
        [3.3767e-02, 9.6623e-01],
        [3.3007e-03, 9.9670e-01],
        [5.3156e-04, 9.9947e-01],
        [9.1344e-03, 9.9087e-01],
        [1.4159e-03, 9.9858e-01],
        [9.8280e-04, 9.9902e-01],
        [1.4026e-03, 9.9860e-01],
        [7.5837e-02, 9.2416e-01],
        [5.7237e-04, 9.9943e-01],
        [9.9749e-01, 2.5067e-03],
        [2.1752e-01, 7.8248e-01],
        [3.1364e-04, 9.9969e-01],
        [1.3357e-04, 9.9987e-01],
        [9.9830e-01, 1.6965e-03],
        [3.2724e-02, 9.6728e-01],
        [9.9448e-01, 5.5227e-03],
        [2.5230e-03, 9.9748e-01],
        [9.6617e-01, 3.3826e-02],
        [2.2507e-03, 9.9775e-01],
        [9.9449e-01, 5.5105e-03],
        [2.9192e-04, 9.9971e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.5887e-01, 4.1130e-02],
        [9.9922e-01, 7.8058e-04],
        [2.5250e-04, 9.9975e-01],
        [1.4187e-04, 9.9986e-01],
        [1.5653e-02, 9.8435e-01],
        [1.0466e-03, 9.9895e-01],
        [3.2375e-04, 9.9968e-01],
        [9.6720e-01, 3.2797e-02],
        [9.3341e-01, 6.6587e-02],
        [1.0426e-01, 8.9574e-01],
        [4.7470e-04, 9.9953e-01],
        [9.8093e-01, 1.9070e-02],
        [3.0911e-04, 9.9969e-01],
        [1.2557e-04, 9.9987e-01],
        [9.2411e-01, 7.5887e-02],
        [1.1321e-03, 9.9887e-01],
        [9.8863e-01, 1.1374e-02],
        [9.9592e-01, 4.0800e-03],
        [1.9729e-03, 9.9803e-01],
        [9.3439e-01, 6.5615e-02],
        [6.1130e-01, 3.8870e-01],
        [8.7221e-03, 9.9128e-01],
        [5.9135e-01, 4.0865e-01],
        [8.7930e-04, 9.9912e-01],
        [1.3734e-02, 9.8627e-01],
        [1.3847e-02, 9.8615e-01],
        [1.6011e-03, 9.9840e-01],
        [2.7459e-03, 9.9725e-01],
        [3.2915e-06, 1.0000e+00],
        [9.8305e-01, 1.6948e-02],
        [3.0821e-06, 1.0000e+00],
        [5.2432e-01, 4.7568e-01],
        [9.4395e-01, 5.6052e-02],
        [2.1520e-03, 9.9785e-01],
        [6.4055e-03, 9.9359e-01],
        [3.1244e-05, 9.9997e-01],
        [5.3176e-02, 9.4682e-01],
        [2.7122e-05, 9.9997e-01],
        [8.0544e-01, 1.9456e-01],
        [4.1144e-03, 9.9589e-01],
        [4.7142e-02, 9.5286e-01],
        [3.1177e-03, 9.9688e-01],
        [2.3391e-06, 1.0000e+00],
        [3.3207e-02, 9.6679e-01],
        [8.4410e-01, 1.5590e-01],
        [9.9356e-01, 6.4438e-03],
        [1.1204e-04, 9.9989e-01],
        [4.0418e-02, 9.5958e-01],
        [9.9784e-01, 2.1621e-03],
        [7.5612e-05, 9.9992e-01],
        [9.9384e-01, 6.1602e-03],
        [3.1981e-03, 9.9680e-01],
        [9.9766e-01, 2.3383e-03],
        [3.3335e-04, 9.9967e-01],
        [4.4140e-05, 9.9996e-01],
        [6.6317e-04, 9.9934e-01],
        [1.1541e-03, 9.9885e-01],
        [1.4120e-03, 9.9859e-01],
        [2.1443e-05, 9.9998e-01],
        [3.6479e-03, 9.9635e-01],
        [7.8295e-03, 9.9217e-01],
        [3.2096e-03, 9.9679e-01],
        [4.0333e-06, 1.0000e+00],
        [9.9104e-01, 8.9595e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0844e-02, 9.8916e-01],
        [7.8079e-01, 2.1921e-01],
        [6.6441e-02, 9.3356e-01],
        [1.0102e-04, 9.9990e-01],
        [8.7501e-03, 9.9125e-01],
        [9.9740e-01, 2.6031e-03],
        [7.4373e-03, 9.9256e-01],
        [5.0617e-03, 9.9494e-01],
        [3.8226e-05, 9.9996e-01],
        [1.5902e-05, 9.9998e-01],
        [9.5135e-01, 4.8648e-02],
        [6.0173e-06, 9.9999e-01],
        [5.0908e-03, 9.9491e-01],
        [1.3695e-03, 9.9863e-01],
        [1.4832e-03, 9.9852e-01],
        [6.7012e-05, 9.9993e-01],
        [2.8388e-04, 9.9972e-01],
        [4.7551e-03, 9.9524e-01],
        [2.2026e-04, 9.9978e-01],
        [1.6229e-02, 9.8377e-01],
        [8.0314e-06, 9.9999e-01],
        [2.5982e-02, 9.7402e-01],
        [9.9451e-02, 9.0055e-01],
        [7.7279e-05, 9.9992e-01],
        [3.6542e-02, 9.6346e-01],
        [1.1916e-04, 9.9988e-01],
        [9.0830e-01, 9.1700e-02],
        [3.2853e-03, 9.9671e-01],
        [4.4964e-02, 9.5504e-01],
        [1.8252e-03, 9.9817e-01],
        [1.0883e-02, 9.8912e-01],
        [3.7598e-02, 9.6240e-01],
        [2.0932e-04, 9.9979e-01],
        [3.1575e-03, 9.9684e-01],
        [5.2472e-05, 9.9995e-01],
        [1.6632e-02, 9.8337e-01],
        [1.4625e-02, 9.8537e-01],
        [1.2647e-05, 9.9999e-01],
        [4.0268e-03, 9.9597e-01],
        [9.5893e-01, 4.1075e-02],
        [2.8007e-04, 9.9972e-01],
        [9.3564e-01, 6.4361e-02],
        [4.5400e-02, 9.5460e-01],
        [3.9732e-04, 9.9960e-01],
        [2.0959e-04, 9.9979e-01],
        [3.6497e-04, 9.9963e-01],
        [1.4503e-03, 9.9855e-01],
        [4.9376e-01, 5.0624e-01],
        [9.9315e-03, 9.9007e-01],
        [1.3991e-03, 9.9860e-01],
        [3.2606e-05, 9.9997e-01],
        [9.0856e-06, 9.9999e-01],
        [2.0632e-02, 9.7937e-01],
        [8.6383e-03, 9.9136e-01],
        [1.2877e-03, 9.9871e-01],
        [2.8846e-04, 9.9971e-01],
        [7.5487e-01, 2.4513e-01],
        [2.0255e-02, 9.7975e-01],
        [1.3043e-03, 9.9870e-01],
        [1.3640e-02, 9.8636e-01],
        [9.9646e-01, 3.5390e-03],
        [9.4865e-01, 5.1355e-02],
        [2.3396e-05, 9.9998e-01],
        [6.7012e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3489e-04, 9.9987e-01],
        [2.8226e-01, 7.1774e-01],
        [6.9509e-05, 9.9993e-01],
        [3.0037e-05, 9.9997e-01],
        [9.5107e-01, 4.8933e-02],
        [9.7025e-01, 2.9753e-02],
        [1.6502e-03, 9.9835e-01],
        [9.9170e-01, 8.2982e-03],
        [9.3909e-01, 6.0913e-02],
        [3.6315e-03, 9.9637e-01],
        [3.0012e-05, 9.9997e-01],
        [3.8877e-02, 9.6112e-01],
        [9.9985e-01, 1.4587e-04],
        [7.6575e-01, 2.3425e-01],
        [8.3348e-03, 9.9167e-01],
        [9.9633e-01, 3.6725e-03],
        [9.3994e-01, 6.0062e-02],
        [9.1723e-01, 8.2767e-02],
        [3.7498e-03, 9.9625e-01],
        [9.2481e-03, 9.9075e-01],
        [1.2431e-01, 8.7569e-01],
        [3.6859e-06, 1.0000e+00],
        [9.5038e-01, 4.9617e-02],
        [3.4580e-05, 9.9997e-01],
        [6.2338e-06, 9.9999e-01],
        [1.0742e-02, 9.8926e-01],
        [9.9889e-01, 1.1095e-03],
        [3.7323e-03, 9.9627e-01],
        [9.9126e-03, 9.9009e-01],
        [6.9885e-03, 9.9301e-01],
        [5.5968e-05, 9.9994e-01],
        [2.6364e-02, 9.7364e-01],
        [9.9598e-01, 4.0190e-03],
        [9.7056e-02, 9.0294e-01],
        [1.5984e-02, 9.8402e-01],
        [1.5672e-03, 9.9843e-01],
        [1.4446e-04, 9.9986e-01],
        [1.0384e-04, 9.9990e-01],
        [2.4748e-04, 9.9975e-01],
        [3.9745e-05, 9.9996e-01],
        [9.8321e-01, 1.6792e-02],
        [9.8275e-01, 1.7247e-02],
        [9.4769e-01, 5.2306e-02],
        [5.2120e-04, 9.9948e-01],
        [2.3510e-01, 7.6490e-01],
        [1.0399e-02, 9.8960e-01],
        [1.4266e-02, 9.8573e-01],
        [7.5767e-04, 9.9924e-01],
        [2.8935e-04, 9.9971e-01],
        [4.1214e-01, 5.8786e-01],
        [2.8861e-04, 9.9971e-01],
        [2.1378e-05, 9.9998e-01],
        [8.4644e-05, 9.9992e-01],
        [1.0864e-02, 9.8914e-01],
        [9.9483e-05, 9.9990e-01],
        [2.7703e-03, 9.9723e-01],
        [1.6036e-02, 9.8396e-01],
        [9.9550e-01, 4.5047e-03],
        [6.1007e-04, 9.9939e-01],
        [1.9088e-02, 9.8091e-01],
        [9.7738e-01, 2.2623e-02],
        [1.4631e-04, 9.9985e-01],
        [9.9760e-01, 2.3952e-03],
        [9.1760e-04, 9.9908e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.7589e-01, 2.4107e-02],
        [1.9075e-03, 9.9809e-01],
        [9.4672e-01, 5.3280e-02],
        [9.5594e-01, 4.4061e-02],
        [5.3006e-05, 9.9995e-01],
        [4.6052e-05, 9.9995e-01],
        [1.1001e-02, 9.8900e-01],
        [9.7363e-01, 2.6367e-02],
        [9.8954e-01, 1.0464e-02],
        [3.5165e-06, 1.0000e+00],
        [1.4584e-03, 9.9854e-01],
        [7.8822e-05, 9.9992e-01],
        [9.9384e-01, 6.1570e-03],
        [2.8869e-05, 9.9997e-01],
        [9.5814e-01, 4.1857e-02],
        [1.6831e-05, 9.9998e-01],
        [9.7934e-01, 2.0663e-02],
        [9.8287e-01, 1.7126e-02],
        [3.7987e-05, 9.9996e-01],
        [2.9765e-03, 9.9702e-01],
        [2.8696e-05, 9.9997e-01],
        [1.1078e-03, 9.9889e-01],
        [1.6834e-05, 9.9998e-01],
        [9.9859e-01, 1.4128e-03],
        [1.0861e-01, 8.9139e-01],
        [5.1890e-03, 9.9481e-01],
        [9.9958e-04, 9.9900e-01],
        [1.3542e-05, 9.9999e-01],
        [1.2638e-03, 9.9874e-01],
        [1.2721e-02, 9.8728e-01],
        [4.8907e-05, 9.9995e-01],
        [2.7229e-04, 9.9973e-01],
        [2.1762e-04, 9.9978e-01],
        [1.2585e-05, 9.9999e-01],
        [1.5893e-02, 9.8411e-01],
        [4.5663e-03, 9.9543e-01],
        [2.4725e-05, 9.9998e-01],
        [1.0148e-04, 9.9990e-01],
        [2.2446e-03, 9.9776e-01],
        [2.4727e-03, 9.9753e-01],
        [1.7830e-02, 9.8217e-01],
        [6.3667e-05, 9.9994e-01],
        [2.7355e-05, 9.9997e-01],
        [4.7172e-02, 9.5283e-01],
        [1.3184e-04, 9.9987e-01],
        [9.9988e-01, 1.1909e-04],
        [9.7621e-01, 2.3792e-02],
        [9.9800e-01, 2.0042e-03],
        [1.2360e-01, 8.7640e-01],
        [9.9810e-01, 1.8955e-03],
        [6.1235e-04, 9.9939e-01],
        [9.1958e-01, 8.0419e-02],
        [9.5232e-01, 4.7684e-02],
        [8.7170e-03, 9.9128e-01],
        [3.0490e-06, 1.0000e+00],
        [1.5612e-03, 9.9844e-01],
        [4.3455e-04, 9.9957e-01],
        [1.0374e-03, 9.9896e-01],
        [1.2761e-05, 9.9999e-01],
        [1.0499e-05, 9.9999e-01],
        [7.3085e-06, 9.9999e-01],
        [2.5194e-04, 9.9975e-01],
        [9.9808e-01, 1.9198e-03],
        [8.7057e-05, 9.9991e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.2933e-03, 9.9271e-01],
        [1.8823e-07, 1.0000e+00],
        [1.9804e-02, 9.8020e-01],
        [6.3458e-02, 9.3654e-01],
        [1.8782e-04, 9.9981e-01],
        [9.9646e-01, 3.5365e-03],
        [1.0672e-03, 9.9893e-01],
        [8.4518e-05, 9.9992e-01],
        [4.0972e-01, 5.9028e-01],
        [2.6968e-03, 9.9730e-01],
        [5.8999e-03, 9.9410e-01],
        [8.0815e-01, 1.9185e-01],
        [7.4801e-03, 9.9252e-01],
        [1.0095e-04, 9.9990e-01],
        [2.0626e-03, 9.9794e-01],
        [6.8120e-05, 9.9993e-01],
        [6.3217e-01, 3.6783e-01],
        [8.0973e-04, 9.9919e-01],
        [6.8480e-05, 9.9993e-01],
        [8.5885e-05, 9.9991e-01],
        [9.9357e-01, 6.4339e-03],
        [2.0932e-05, 9.9998e-01],
        [5.3689e-03, 9.9463e-01],
        [1.8562e-02, 9.8144e-01],
        [5.5267e-01, 4.4733e-01],
        [1.1744e-03, 9.9883e-01],
        [4.5640e-03, 9.9544e-01],
        [4.1509e-04, 9.9958e-01],
        [2.3507e-02, 9.7649e-01],
        [2.3820e-04, 9.9976e-01],
        [9.7246e-01, 2.7537e-02],
        [1.0859e-05, 9.9999e-01],
        [9.9753e-01, 2.4668e-03],
        [2.7087e-03, 9.9729e-01],
        [4.4539e-01, 5.5461e-01],
        [4.2670e-04, 9.9957e-01],
        [9.4210e-01, 5.7902e-02],
        [1.7442e-04, 9.9983e-01],
        [7.0062e-01, 2.9938e-01],
        [4.1249e-06, 1.0000e+00],
        [5.1387e-03, 9.9486e-01],
        [9.4947e-01, 5.0530e-02],
        [8.6965e-04, 9.9913e-01],
        [9.1573e-03, 9.9084e-01],
        [9.9832e-01, 1.6761e-03],
        [7.2357e-01, 2.7643e-01],
        [8.8130e-02, 9.1187e-01],
        [9.9087e-01, 9.1349e-03],
        [2.7230e-05, 9.9997e-01],
        [9.6875e-01, 3.1252e-02],
        [2.2760e-04, 9.9977e-01],
        [5.9003e-03, 9.9410e-01],
        [9.8169e-01, 1.8314e-02],
        [4.6805e-03, 9.9532e-01],
        [3.1347e-03, 9.9687e-01],
        [1.6474e-01, 8.3526e-01],
        [4.0545e-04, 9.9959e-01],
        [9.8932e-01, 1.0684e-02],
        [1.3287e-04, 9.9987e-01],
        [1.0360e-02, 9.8964e-01],
        [3.4476e-04, 9.9966e-01],
        [9.9637e-01, 3.6326e-03],
        [5.5835e-06, 9.9999e-01],
        [3.5552e-04, 9.9964e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.4336e-01, 5.6638e-02],
        [3.6662e-05, 9.9996e-01],
        [1.2404e-03, 9.9876e-01],
        [9.8724e-01, 1.2757e-02],
        [6.0377e-03, 9.9396e-01],
        [5.7162e-01, 4.2838e-01],
        [6.4052e-05, 9.9994e-01],
        [9.7384e-02, 9.0262e-01],
        [2.5391e-04, 9.9975e-01],
        [3.8863e-04, 9.9961e-01],
        [9.3439e-01, 6.5609e-02],
        [2.3858e-04, 9.9976e-01],
        [9.9683e-01, 3.1654e-03],
        [1.7673e-04, 9.9982e-01],
        [5.1996e-06, 9.9999e-01],
        [9.0650e-03, 9.9093e-01],
        [5.2370e-03, 9.9476e-01],
        [2.3462e-02, 9.7654e-01],
        [1.1554e-04, 9.9988e-01],
        [2.2266e-04, 9.9978e-01],
        [9.8373e-01, 1.6274e-02],
        [5.7010e-03, 9.9430e-01],
        [4.3542e-06, 1.0000e+00],
        [1.1836e-04, 9.9988e-01],
        [1.1808e-03, 9.9882e-01],
        [7.4124e-01, 2.5876e-01],
        [1.5242e-05, 9.9998e-01],
        [7.4186e-03, 9.9258e-01],
        [2.4698e-06, 1.0000e+00],
        [1.4137e-03, 9.9859e-01],
        [7.6633e-04, 9.9923e-01],
        [1.9581e-03, 9.9804e-01],
        [6.6790e-05, 9.9993e-01],
        [6.6931e-04, 9.9933e-01],
        [1.5983e-04, 9.9984e-01],
        [4.1856e-03, 9.9581e-01],
        [3.6814e-03, 9.9632e-01],
        [3.8590e-03, 9.9614e-01],
        [9.9557e-01, 4.4286e-03],
        [9.4588e-01, 5.4117e-02],
        [2.1329e-02, 9.7867e-01],
        [9.6223e-01, 3.7771e-02],
        [2.6912e-02, 9.7309e-01],
        [2.4809e-02, 9.7519e-01],
        [5.5152e-05, 9.9994e-01],
        [2.1238e-02, 9.7876e-01],
        [6.2261e-03, 9.9377e-01],
        [9.4619e-04, 9.9905e-01],
        [3.9885e-03, 9.9601e-01],
        [1.4308e-04, 9.9986e-01],
        [1.4773e-04, 9.9985e-01],
        [7.1056e-06, 9.9999e-01],
        [8.8524e-03, 9.9115e-01],
        [2.5412e-02, 9.7459e-01],
        [2.0483e-04, 9.9980e-01],
        [9.9631e-01, 3.6948e-03],
        [9.8444e-01, 1.5558e-02],
        [1.1938e-03, 9.9881e-01],
        [2.7526e-04, 9.9972e-01],
        [3.4510e-04, 9.9965e-01],
        [5.2112e-04, 9.9948e-01],
        [2.3984e-03, 9.9760e-01],
        [8.5110e-02, 9.1489e-01],
        [8.4536e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9682e-01, 3.1799e-03],
        [9.9337e-01, 6.6349e-03],
        [8.2075e-02, 9.1793e-01],
        [3.3010e-03, 9.9670e-01],
        [1.1430e-03, 9.9886e-01],
        [9.9609e-01, 3.9125e-03],
        [6.0227e-04, 9.9940e-01],
        [9.8764e-01, 1.2358e-02],
        [1.2156e-02, 9.8784e-01],
        [6.4154e-03, 9.9358e-01],
        [2.3521e-04, 9.9976e-01],
        [1.9083e-03, 9.9809e-01],
        [1.4169e-05, 9.9999e-01],
        [2.2997e-05, 9.9998e-01],
        [5.7492e-05, 9.9994e-01],
        [1.5166e-04, 9.9985e-01],
        [5.6408e-04, 9.9944e-01],
        [5.1085e-05, 9.9995e-01],
        [6.6181e-02, 9.3382e-01],
        [9.5571e-01, 4.4290e-02],
        [8.0834e-04, 9.9919e-01],
        [7.1073e-04, 9.9929e-01],
        [9.5091e-01, 4.9087e-02],
        [1.4489e-01, 8.5511e-01],
        [4.4468e-04, 9.9956e-01],
        [1.7131e-05, 9.9998e-01],
        [7.8293e-04, 9.9922e-01],
        [1.4025e-02, 9.8597e-01],
        [6.7178e-03, 9.9328e-01],
        [8.8360e-01, 1.1640e-01],
        [7.6429e-05, 9.9992e-01],
        [5.5169e-03, 9.9448e-01],
        [5.8256e-03, 9.9417e-01],
        [1.7915e-01, 8.2085e-01],
        [1.7605e-01, 8.2395e-01],
        [1.1646e-03, 9.9884e-01],
        [9.9525e-01, 4.7523e-03],
        [2.7668e-04, 9.9972e-01],
        [6.7973e-01, 3.2027e-01],
        [3.0863e-03, 9.9691e-01],
        [1.2865e-03, 9.9871e-01],
        [4.4974e-03, 9.9550e-01],
        [2.7525e-06, 1.0000e+00],
        [6.3987e-03, 9.9360e-01],
        [2.1823e-03, 9.9782e-01],
        [8.4273e-04, 9.9916e-01],
        [6.1380e-01, 3.8620e-01],
        [3.6694e-04, 9.9963e-01],
        [9.6547e-01, 3.4531e-02],
        [9.6669e-04, 9.9903e-01],
        [2.9958e-05, 9.9997e-01],
        [9.9858e-01, 1.4241e-03],
        [4.5115e-03, 9.9549e-01],
        [7.9744e-05, 9.9992e-01],
        [1.3394e-04, 9.9987e-01],
        [1.6680e-05, 9.9998e-01],
        [2.8132e-05, 9.9997e-01],
        [3.5473e-04, 9.9965e-01],
        [5.5047e-05, 9.9994e-01],
        [1.0300e-05, 9.9999e-01],
        [1.3852e-04, 9.9986e-01],
        [1.2490e-04, 9.9988e-01],
        [4.5042e-07, 1.0000e+00],
        [9.9096e-01, 9.0374e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[5.7309e-02, 9.4269e-01],
        [8.3544e-03, 9.9165e-01],
        [1.1946e-04, 9.9988e-01],
        [1.0339e-03, 9.9897e-01],
        [9.8922e-01, 1.0779e-02],
        [9.9043e-01, 9.5667e-03],
        [7.4366e-06, 9.9999e-01],
        [9.0397e-01, 9.6027e-02],
        [1.6808e-04, 9.9983e-01],
        [9.9808e-01, 1.9170e-03],
        [1.7459e-04, 9.9983e-01],
        [4.2177e-03, 9.9578e-01],
        [1.7702e-02, 9.8230e-01],
        [9.9571e-01, 4.2881e-03],
        [4.2953e-03, 9.9570e-01],
        [7.8585e-01, 2.1415e-01],
        [2.5755e-06, 1.0000e+00],
        [1.7959e-04, 9.9982e-01],
        [9.9077e-01, 9.2254e-03],
        [2.4555e-05, 9.9998e-01],
        [1.7283e-02, 9.8272e-01],
        [3.6494e-05, 9.9996e-01],
        [9.5326e-05, 9.9990e-01],
        [9.9288e-01, 7.1204e-03],
        [9.4864e-01, 5.1360e-02],
        [9.8647e-01, 1.3533e-02],
        [9.6811e-01, 3.1891e-02],
        [3.3723e-02, 9.6628e-01],
        [9.9808e-01, 1.9248e-03],
        [3.9547e-03, 9.9605e-01],
        [4.3489e-06, 1.0000e+00],
        [3.0116e-02, 9.6988e-01],
        [6.5559e-04, 9.9934e-01],
        [9.9152e-01, 8.4843e-03],
        [1.3952e-03, 9.9860e-01],
        [1.0262e-03, 9.9897e-01],
        [2.5989e-04, 9.9974e-01],
        [1.8956e-01, 8.1044e-01],
        [2.0596e-02, 9.7940e-01],
        [3.7326e-03, 9.9627e-01],
        [1.5461e-03, 9.9845e-01],
        [1.7165e-04, 9.9983e-01],
        [9.8655e-01, 1.3453e-02],
        [9.6681e-01, 3.3189e-02],
        [4.2631e-02, 9.5737e-01],
        [9.8910e-01, 1.0901e-02],
        [1.9327e-05, 9.9998e-01],
        [2.5841e-02, 9.7416e-01],
        [1.1792e-03, 9.9882e-01],
        [9.8708e-01, 1.2924e-02],
        [2.5197e-05, 9.9997e-01],
        [3.6216e-02, 9.6378e-01],
        [1.0494e-03, 9.9895e-01],
        [8.7445e-03, 9.9126e-01],
        [9.9464e-01, 5.3624e-03],
        [1.7419e-04, 9.9983e-01],
        [2.0663e-03, 9.9793e-01],
        [9.9560e-01, 4.3971e-03],
        [9.9730e-01, 2.6958e-03],
        [7.3882e-04, 9.9926e-01],
        [8.5775e-03, 9.9142e-01],
        [2.5081e-01, 7.4919e-01],
        [9.4654e-04, 9.9905e-01],
        [2.3620e-03, 9.9764e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2552e-04, 9.9967e-01],
        [4.0405e-05, 9.9996e-01],
        [7.3297e-01, 2.6703e-01],
        [9.9753e-01, 2.4691e-03],
        [1.9223e-03, 9.9808e-01],
        [1.4879e-02, 9.8512e-01],
        [3.9781e-03, 9.9602e-01],
        [2.3338e-03, 9.9767e-01],
        [6.4225e-03, 9.9358e-01],
        [2.2176e-03, 9.9778e-01],
        [9.9802e-01, 1.9772e-03],
        [5.9281e-03, 9.9407e-01],
        [2.8263e-04, 9.9972e-01],
        [7.1789e-05, 9.9993e-01],
        [3.9241e-03, 9.9608e-01],
        [7.4933e-04, 9.9925e-01],
        [8.9383e-01, 1.0617e-01],
        [3.4592e-03, 9.9654e-01],
        [9.9976e-01, 2.4371e-04],
        [5.9954e-06, 9.9999e-01],
        [8.1124e-04, 9.9919e-01],
        [3.7106e-03, 9.9629e-01],
        [2.9170e-03, 9.9708e-01],
        [4.7114e-04, 9.9953e-01],
        [9.9753e-05, 9.9990e-01],
        [9.7796e-03, 9.9022e-01],
        [9.9679e-01, 3.2069e-03],
        [3.0443e-01, 6.9557e-01],
        [9.9756e-01, 2.4443e-03],
        [1.3895e-03, 9.9861e-01],
        [3.7062e-05, 9.9996e-01],
        [9.4540e-01, 5.4600e-02],
        [1.3543e-05, 9.9999e-01],
        [2.0620e-04, 9.9979e-01],
        [4.3210e-04, 9.9957e-01],
        [1.5825e-03, 9.9842e-01],
        [9.7703e-04, 9.9902e-01],
        [2.0133e-03, 9.9799e-01],
        [7.4972e-06, 9.9999e-01],
        [2.8836e-04, 9.9971e-01],
        [7.3849e-03, 9.9262e-01],
        [9.9123e-01, 8.7731e-03],
        [2.2644e-03, 9.9774e-01],
        [1.1231e-02, 9.8877e-01],
        [3.4647e-04, 9.9965e-01],
        [1.2273e-03, 9.9877e-01],
        [9.9668e-01, 3.3245e-03],
        [9.7516e-01, 2.4841e-02],
        [1.0498e-01, 8.9502e-01],
        [3.0125e-04, 9.9970e-01],
        [3.1855e-02, 9.6815e-01],
        [9.7062e-01, 2.9376e-02],
        [8.6337e-05, 9.9991e-01],
        [3.9676e-04, 9.9960e-01],
        [1.8962e-05, 9.9998e-01],
        [8.0946e-04, 9.9919e-01],
        [1.6831e-05, 9.9998e-01],
        [5.5803e-07, 1.0000e+00],
        [4.6383e-05, 9.9995e-01],
        [4.5774e-04, 9.9954e-01],
        [7.0259e-03, 9.9297e-01],
        [2.0442e-03, 9.9796e-01],
        [1.6997e-02, 9.8300e-01],
        [2.0033e-02, 9.7997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7091e-03, 9.9629e-01],
        [1.0005e-01, 8.9995e-01],
        [3.7614e-03, 9.9624e-01],
        [9.9609e-01, 3.9067e-03],
        [9.8592e-01, 1.4078e-02],
        [1.1278e-04, 9.9989e-01],
        [1.7592e-02, 9.8241e-01],
        [6.5267e-04, 9.9935e-01],
        [9.9359e-01, 6.4142e-03],
        [7.6149e-01, 2.3851e-01],
        [2.1220e-04, 9.9979e-01],
        [1.8027e-02, 9.8197e-01],
        [3.2492e-04, 9.9968e-01],
        [6.2829e-03, 9.9372e-01],
        [6.7788e-05, 9.9993e-01],
        [3.1351e-05, 9.9997e-01],
        [3.2431e-02, 9.6757e-01],
        [2.5934e-03, 9.9741e-01],
        [2.6852e-05, 9.9997e-01],
        [7.8871e-01, 2.1129e-01],
        [7.1275e-04, 9.9929e-01],
        [8.1676e-02, 9.1832e-01],
        [6.0365e-07, 1.0000e+00],
        [3.2620e-03, 9.9674e-01],
        [1.3186e-04, 9.9987e-01],
        [2.2642e-03, 9.9774e-01],
        [2.8559e-03, 9.9714e-01],
        [9.8674e-01, 1.3260e-02],
        [9.9775e-01, 2.2545e-03],
        [3.2858e-03, 9.9671e-01],
        [7.7204e-01, 2.2796e-01],
        [9.9941e-03, 9.9001e-01],
        [3.1095e-05, 9.9997e-01],
        [3.4612e-02, 9.6539e-01],
        [4.3592e-01, 5.6408e-01],
        [8.9966e-01, 1.0034e-01],
        [7.8242e-01, 2.1758e-01],
        [1.5043e-04, 9.9985e-01],
        [9.9750e-01, 2.4996e-03],
        [9.7800e-01, 2.2002e-02],
        [5.0978e-01, 4.9022e-01],
        [1.2907e-05, 9.9999e-01],
        [7.6503e-01, 2.3497e-01],
        [4.1150e-03, 9.9588e-01],
        [2.6361e-05, 9.9997e-01],
        [8.9121e-04, 9.9911e-01],
        [1.8021e-02, 9.8198e-01],
        [9.3369e-01, 6.6308e-02],
        [9.5744e-01, 4.2564e-02],
        [1.0675e-02, 9.8932e-01],
        [9.7938e-01, 2.0622e-02],
        [8.7346e-03, 9.9127e-01],
        [9.9936e-01, 6.3848e-04],
        [7.5961e-05, 9.9992e-01],
        [2.9286e-03, 9.9707e-01],
        [8.0894e-03, 9.9191e-01],
        [5.6769e-03, 9.9432e-01],
        [1.6717e-04, 9.9983e-01],
        [1.7374e-03, 9.9826e-01],
        [1.5632e-03, 9.9844e-01],
        [9.9638e-01, 3.6199e-03],
        [9.9868e-01, 1.3177e-03],
        [1.4682e-02, 9.8532e-01],
        [9.0379e-04, 9.9910e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.4686e-01, 5.3142e-02],
        [8.8899e-01, 1.1101e-01],
        [2.2749e-05, 9.9998e-01],
        [1.9299e-01, 8.0701e-01],
        [1.4347e-01, 8.5653e-01],
        [2.1460e-04, 9.9979e-01],
        [9.8384e-04, 9.9902e-01],
        [4.7947e-06, 1.0000e+00],
        [8.9574e-01, 1.0426e-01],
        [1.5652e-04, 9.9984e-01],
        [2.2866e-04, 9.9977e-01],
        [1.9983e-03, 9.9800e-01],
        [8.4821e-01, 1.5179e-01],
        [4.4457e-06, 1.0000e+00],
        [9.9913e-01, 8.6535e-04],
        [1.6415e-01, 8.3585e-01],
        [5.1896e-06, 9.9999e-01],
        [2.1948e-01, 7.8052e-01],
        [2.5638e-03, 9.9744e-01],
        [3.1394e-02, 9.6861e-01],
        [1.0262e-03, 9.9897e-01],
        [3.7274e-01, 6.2726e-01],
        [9.9550e-01, 4.5029e-03],
        [8.1560e-03, 9.9184e-01],
        [5.7251e-01, 4.2749e-01],
        [8.7676e-01, 1.2324e-01],
        [5.7062e-05, 9.9994e-01],
        [1.9733e-03, 9.9803e-01],
        [2.9949e-03, 9.9701e-01],
        [9.9545e-01, 4.5507e-03],
        [6.9653e-05, 9.9993e-01],
        [5.3495e-03, 9.9465e-01],
        [5.3991e-03, 9.9460e-01],
        [1.1296e-04, 9.9989e-01],
        [6.6872e-02, 9.3313e-01],
        [3.4663e-04, 9.9965e-01],
        [4.7433e-02, 9.5257e-01],
        [9.9757e-01, 2.4258e-03],
        [8.7248e-03, 9.9128e-01],
        [4.3576e-02, 9.5642e-01],
        [2.6912e-03, 9.9731e-01],
        [3.6445e-03, 9.9636e-01],
        [4.6758e-03, 9.9532e-01],
        [3.3704e-02, 9.6630e-01],
        [1.6429e-04, 9.9984e-01],
        [2.6203e-04, 9.9974e-01],
        [5.0591e-04, 9.9949e-01],
        [4.2816e-04, 9.9957e-01],
        [9.8821e-01, 1.1788e-02],
        [3.1528e-02, 9.6847e-01],
        [1.8857e-05, 9.9998e-01],
        [1.2742e-02, 9.8726e-01],
        [9.2835e-01, 7.1648e-02],
        [9.9467e-01, 5.3279e-03],
        [4.2701e-02, 9.5730e-01],
        [9.9714e-01, 2.8571e-03],
        [2.8903e-03, 9.9711e-01],
        [1.4340e-03, 9.9857e-01],
        [1.0213e-02, 9.8979e-01],
        [1.0759e-03, 9.9892e-01],
        [9.9210e-01, 7.8972e-03],
        [9.5662e-01, 4.3377e-02],
        [1.1412e-05, 9.9999e-01],
        [1.3265e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.7832e-05, 9.9993e-01],
        [1.6075e-03, 9.9839e-01],
        [9.9992e-01, 8.4493e-05],
        [4.0972e-04, 9.9959e-01],
        [1.0260e-01, 8.9740e-01],
        [2.3702e-03, 9.9763e-01],
        [8.3503e-02, 9.1650e-01],
        [2.2794e-02, 9.7721e-01],
        [3.0652e-03, 9.9693e-01],
        [9.8684e-01, 1.3162e-02],
        [9.8647e-01, 1.3529e-02],
        [3.8697e-04, 9.9961e-01],
        [4.0008e-05, 9.9996e-01],
        [9.9511e-01, 4.8916e-03],
        [9.4121e-01, 5.8793e-02],
        [1.2495e-04, 9.9988e-01],
        [2.1320e-06, 1.0000e+00],
        [3.6945e-05, 9.9996e-01],
        [1.0144e-02, 9.8986e-01],
        [2.3768e-03, 9.9762e-01],
        [6.6405e-05, 9.9993e-01],
        [2.3591e-01, 7.6409e-01],
        [9.0728e-05, 9.9991e-01],
        [4.8601e-03, 9.9514e-01],
        [3.0041e-03, 9.9700e-01],
        [8.5637e-05, 9.9991e-01],
        [4.7597e-05, 9.9995e-01],
        [6.1412e-03, 9.9386e-01],
        [1.9625e-05, 9.9998e-01],
        [9.2295e-01, 7.7050e-02],
        [1.5581e-04, 9.9984e-01],
        [1.5242e-02, 9.8476e-01],
        [3.0125e-03, 9.9699e-01],
        [1.4773e-01, 8.5227e-01],
        [1.8493e-04, 9.9982e-01],
        [4.0253e-05, 9.9996e-01],
        [3.4804e-04, 9.9965e-01],
        [1.2314e-03, 9.9877e-01],
        [1.2125e-01, 8.7875e-01],
        [1.6544e-05, 9.9998e-01],
        [4.6469e-03, 9.9535e-01],
        [9.9711e-01, 2.8917e-03],
        [3.3050e-04, 9.9967e-01],
        [2.6267e-04, 9.9974e-01],
        [3.4752e-03, 9.9652e-01],
        [9.9542e-01, 4.5815e-03],
        [9.9762e-01, 2.3829e-03],
        [2.0453e-03, 9.9795e-01],
        [2.2057e-04, 9.9978e-01],
        [1.4744e-04, 9.9985e-01],
        [5.6888e-02, 9.4311e-01],
        [5.7583e-02, 9.4242e-01],
        [1.6350e-02, 9.8365e-01],
        [4.7663e-03, 9.9523e-01],
        [6.5190e-01, 3.4810e-01],
        [6.3712e-03, 9.9363e-01],
        [4.3230e-04, 9.9957e-01],
        [7.7346e-01, 2.2654e-01],
        [3.0001e-03, 9.9700e-01],
        [2.9588e-05, 9.9997e-01],
        [2.9323e-05, 9.9997e-01],
        [5.9450e-05, 9.9994e-01],
        [6.1262e-05, 9.9994e-01],
        [1.5885e-02, 9.8411e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.9632e-03, 9.9604e-01],
        [6.1387e-03, 9.9386e-01],
        [9.5522e-03, 9.9045e-01],
        [3.2415e-02, 9.6758e-01],
        [3.0326e-02, 9.6967e-01],
        [3.8018e-03, 9.9620e-01],
        [1.5035e-03, 9.9850e-01],
        [1.0548e-03, 9.9895e-01],
        [9.3781e-01, 6.2192e-02],
        [1.2390e-01, 8.7610e-01],
        [8.1242e-06, 9.9999e-01],
        [8.8563e-01, 1.1437e-01],
        [7.5532e-01, 2.4468e-01],
        [1.7318e-03, 9.9827e-01],
        [1.6661e-02, 9.8334e-01],
        [1.2107e-04, 9.9988e-01],
        [9.9792e-01, 2.0823e-03],
        [2.4038e-01, 7.5962e-01],
        [1.7203e-02, 9.8280e-01],
        [3.5206e-03, 9.9648e-01],
        [9.0599e-01, 9.4014e-02],
        [5.5381e-01, 4.4619e-01],
        [3.8019e-02, 9.6198e-01],
        [6.5675e-06, 9.9999e-01],
        [6.6341e-02, 9.3366e-01],
        [4.7869e-05, 9.9995e-01],
        [1.4413e-02, 9.8559e-01],
        [1.2461e-02, 9.8754e-01],
        [9.1098e-06, 9.9999e-01],
        [3.4672e-03, 9.9653e-01],
        [2.1719e-03, 9.9783e-01],
        [8.0064e-01, 1.9936e-01],
        [9.9428e-01, 5.7173e-03],
        [3.2235e-02, 9.6776e-01],
        [1.4063e-05, 9.9999e-01],
        [2.7728e-03, 9.9723e-01],
        [5.0963e-02, 9.4904e-01],
        [6.4023e-02, 9.3598e-01],
        [7.6508e-03, 9.9235e-01],
        [2.0583e-05, 9.9998e-01],
        [9.3788e-01, 6.2119e-02],
        [6.3867e-03, 9.9361e-01],
        [6.2562e-01, 3.7438e-01],
        [9.9919e-01, 8.0536e-04],
        [9.9841e-01, 1.5945e-03],
        [9.5478e-01, 4.5224e-02],
        [1.0108e-05, 9.9999e-01],
        [9.9074e-01, 9.2610e-03],
        [1.0350e-03, 9.9897e-01],
        [3.2263e-05, 9.9997e-01],
        [9.5560e-02, 9.0444e-01],
        [9.9775e-01, 2.2492e-03],
        [8.6621e-05, 9.9991e-01],
        [9.6819e-05, 9.9990e-01],
        [9.9336e-01, 6.6365e-03],
        [3.5184e-03, 9.9648e-01],
        [7.3789e-03, 9.9262e-01],
        [3.1159e-01, 6.8841e-01],
        [4.0743e-03, 9.9593e-01],
        [2.5179e-02, 9.7482e-01],
        [1.4091e-03, 9.9859e-01],
        [4.4329e-05, 9.9996e-01],
        [2.6982e-03, 9.9730e-01],
        [4.9814e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.7047e-02, 9.5295e-01],
        [4.2712e-06, 1.0000e+00],
        [2.5652e-04, 9.9974e-01],
        [1.2801e-06, 1.0000e+00],
        [5.1062e-04, 9.9949e-01],
        [1.5539e-02, 9.8446e-01],
        [9.9804e-01, 1.9605e-03],
        [9.9330e-01, 6.7029e-03],
        [1.6948e-04, 9.9983e-01],
        [1.3289e-04, 9.9987e-01],
        [5.9142e-03, 9.9409e-01],
        [7.5263e-03, 9.9247e-01],
        [4.1355e-05, 9.9996e-01],
        [5.6026e-03, 9.9440e-01],
        [1.5433e-04, 9.9985e-01],
        [4.6837e-02, 9.5316e-01],
        [9.9933e-01, 6.7243e-04],
        [3.9166e-03, 9.9608e-01],
        [1.0527e-02, 9.8947e-01],
        [7.7732e-04, 9.9922e-01],
        [3.2398e-06, 1.0000e+00],
        [5.3743e-05, 9.9995e-01],
        [3.5065e-03, 9.9649e-01],
        [9.9802e-01, 1.9800e-03],
        [8.7806e-01, 1.2194e-01],
        [9.7436e-01, 2.5644e-02],
        [8.1904e-03, 9.9181e-01],
        [2.6548e-05, 9.9997e-01],
        [7.0581e-04, 9.9929e-01],
        [2.2189e-03, 9.9778e-01],
        [2.7220e-03, 9.9728e-01],
        [2.7201e-04, 9.9973e-01],
        [3.1100e-05, 9.9997e-01],
        [1.3352e-02, 9.8665e-01],
        [9.9881e-01, 1.1889e-03],
        [2.5074e-05, 9.9997e-01],
        [3.0499e-05, 9.9997e-01],
        [9.2499e-01, 7.5008e-02],
        [2.1740e-04, 9.9978e-01],
        [1.2198e-04, 9.9988e-01],
        [4.0429e-05, 9.9996e-01],
        [3.6332e-02, 9.6367e-01],
        [4.0210e-02, 9.5979e-01],
        [9.9587e-01, 4.1281e-03],
        [1.1785e-03, 9.9882e-01],
        [4.1976e-01, 5.8024e-01],
        [1.7783e-03, 9.9822e-01],
        [8.8179e-04, 9.9912e-01],
        [9.6842e-01, 3.1575e-02],
        [9.9494e-01, 5.0595e-03],
        [3.3285e-03, 9.9667e-01],
        [4.7098e-03, 9.9529e-01],
        [2.3535e-02, 9.7646e-01],
        [9.7597e-04, 9.9902e-01],
        [1.8955e-01, 8.1045e-01],
        [1.6340e-03, 9.9837e-01],
        [9.8636e-01, 1.3639e-02],
        [2.3492e-03, 9.9765e-01],
        [4.8253e-04, 9.9952e-01],
        [1.7711e-02, 9.8229e-01],
        [3.6735e-03, 9.9633e-01],
        [1.4308e-02, 9.8569e-01],
        [2.8716e-03, 9.9713e-01],
        [9.9545e-01, 4.5517e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.6295e-01, 3.7049e-02],
        [3.7447e-06, 1.0000e+00],
        [1.6634e-03, 9.9834e-01],
        [4.7773e-05, 9.9995e-01],
        [5.2655e-02, 9.4735e-01],
        [8.2717e-04, 9.9917e-01],
        [6.4770e-05, 9.9994e-01],
        [8.7373e-03, 9.9126e-01],
        [1.0423e-05, 9.9999e-01],
        [2.0994e-05, 9.9998e-01],
        [9.6449e-01, 3.5512e-02],
        [5.3677e-05, 9.9995e-01],
        [3.3530e-05, 9.9997e-01],
        [2.0196e-06, 1.0000e+00],
        [9.5575e-01, 4.4251e-02],
        [1.2479e-05, 9.9999e-01],
        [2.5196e-02, 9.7480e-01],
        [1.4356e-02, 9.8564e-01],
        [2.4282e-02, 9.7572e-01],
        [5.4582e-04, 9.9945e-01],
        [3.7434e-05, 9.9996e-01],
        [3.5830e-05, 9.9996e-01],
        [1.3621e-03, 9.9864e-01],
        [1.5728e-02, 9.8427e-01],
        [2.1818e-04, 9.9978e-01],
        [1.3844e-07, 1.0000e+00],
        [9.4628e-01, 5.3724e-02],
        [1.9504e-03, 9.9805e-01],
        [2.7065e-01, 7.2935e-01],
        [9.9457e-01, 5.4261e-03],
        [5.5641e-04, 9.9944e-01],
        [2.1374e-04, 9.9979e-01],
        [9.8692e-01, 1.3076e-02],
        [3.8882e-04, 9.9961e-01],
        [2.5032e-02, 9.7497e-01],
        [5.7086e-04, 9.9943e-01],
        [1.4237e-03, 9.9858e-01],
        [1.1876e-04, 9.9988e-01],
        [3.0474e-02, 9.6953e-01],
        [9.2135e-01, 7.8648e-02],
        [5.9472e-04, 9.9941e-01],
        [1.7706e-02, 9.8229e-01],
        [1.1543e-02, 9.8846e-01],
        [7.5891e-04, 9.9924e-01],
        [4.6690e-04, 9.9953e-01],
        [1.6000e-03, 9.9840e-01],
        [1.0719e-05, 9.9999e-01],
        [1.7036e-04, 9.9983e-01],
        [9.0178e-01, 9.8222e-02],
        [3.6486e-05, 9.9996e-01],
        [1.9642e-02, 9.8036e-01],
        [2.2179e-05, 9.9998e-01],
        [1.2648e-05, 9.9999e-01],
        [1.0643e-05, 9.9999e-01],
        [9.9774e-01, 2.2626e-03],
        [8.8949e-01, 1.1051e-01],
        [9.9252e-01, 7.4774e-03],
        [3.0680e-03, 9.9693e-01],
        [5.7504e-01, 4.2496e-01],
        [1.7650e-04, 9.9982e-01],
        [1.1708e-04, 9.9988e-01],
        [9.9848e-01, 1.5242e-03],
        [1.1438e-02, 9.8856e-01],
        [7.2397e-01, 2.7603e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2011e-03, 9.9780e-01],
        [2.7806e-02, 9.7219e-01],
        [9.2173e-01, 7.8273e-02],
        [1.4487e-03, 9.9855e-01],
        [6.5009e-04, 9.9935e-01],
        [9.9637e-01, 3.6340e-03],
        [4.8364e-04, 9.9952e-01],
        [3.1331e-05, 9.9997e-01],
        [4.1278e-05, 9.9996e-01],
        [4.3856e-07, 1.0000e+00],
        [7.0437e-01, 2.9563e-01],
        [1.1485e-05, 9.9999e-01],
        [2.3631e-02, 9.7637e-01],
        [4.7239e-05, 9.9995e-01],
        [9.8985e-01, 1.0149e-02],
        [6.5398e-03, 9.9346e-01],
        [9.7407e-01, 2.5933e-02],
        [2.7583e-04, 9.9972e-01],
        [3.1608e-04, 9.9968e-01],
        [9.9424e-01, 5.7637e-03],
        [3.2503e-04, 9.9968e-01],
        [1.1405e-03, 9.9886e-01],
        [1.2744e-02, 9.8726e-01],
        [4.2070e-04, 9.9958e-01],
        [4.7601e-03, 9.9524e-01],
        [1.1461e-05, 9.9999e-01],
        [1.5774e-01, 8.4226e-01],
        [5.1103e-04, 9.9949e-01],
        [7.9019e-01, 2.0981e-01],
        [3.9723e-02, 9.6028e-01],
        [6.2212e-04, 9.9938e-01],
        [1.5771e-03, 9.9842e-01],
        [1.8485e-05, 9.9998e-01],
        [3.3591e-01, 6.6409e-01],
        [7.5057e-02, 9.2494e-01],
        [1.5600e-05, 9.9998e-01],
        [4.6977e-03, 9.9530e-01],
        [5.7472e-05, 9.9994e-01],
        [1.8120e-01, 8.1880e-01],
        [8.9454e-04, 9.9911e-01],
        [2.5441e-01, 7.4559e-01],
        [1.3451e-05, 9.9999e-01],
        [6.0742e-02, 9.3926e-01],
        [4.8002e-01, 5.1998e-01],
        [1.6782e-03, 9.9832e-01],
        [2.3781e-04, 9.9976e-01],
        [1.5068e-03, 9.9849e-01],
        [9.5656e-06, 9.9999e-01],
        [9.9641e-01, 3.5889e-03],
        [9.9325e-01, 6.7528e-03],
        [9.9527e-01, 4.7261e-03],
        [3.7691e-05, 9.9996e-01],
        [2.3578e-06, 1.0000e+00],
        [9.3878e-01, 6.1225e-02],
        [9.6295e-01, 3.7045e-02],
        [2.2354e-02, 9.7765e-01],
        [6.8495e-03, 9.9315e-01],
        [2.5031e-04, 9.9975e-01],
        [1.4594e-03, 9.9854e-01],
        [9.9951e-01, 4.8522e-04],
        [1.2842e-03, 9.9872e-01],
        [5.2666e-01, 4.7334e-01],
        [9.9883e-01, 1.1671e-03],
        [9.9960e-01, 4.0183e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1482e-04, 9.9989e-01],
        [1.9178e-03, 9.9808e-01],
        [9.9811e-01, 1.8877e-03],
        [2.2099e-03, 9.9779e-01],
        [5.5830e-05, 9.9994e-01],
        [2.7088e-03, 9.9729e-01],
        [2.6693e-03, 9.9733e-01],
        [9.9888e-01, 1.1163e-03],
        [1.0717e-03, 9.9893e-01],
        [4.2508e-01, 5.7492e-01],
        [1.8176e-05, 9.9998e-01],
        [9.8368e-06, 9.9999e-01],
        [3.7599e-04, 9.9962e-01],
        [9.8210e-01, 1.7899e-02],
        [9.0488e-05, 9.9991e-01],
        [2.2448e-03, 9.9776e-01],
        [4.5244e-01, 5.4756e-01],
        [2.7563e-04, 9.9972e-01],
        [4.7057e-05, 9.9995e-01],
        [1.8180e-01, 8.1820e-01],
        [9.9375e-01, 6.2494e-03],
        [1.1218e-03, 9.9888e-01],
        [5.4878e-04, 9.9945e-01],
        [6.8420e-01, 3.1580e-01],
        [1.1014e-03, 9.9890e-01],
        [8.9785e-03, 9.9102e-01],
        [7.3846e-03, 9.9262e-01],
        [1.6861e-04, 9.9983e-01],
        [7.9971e-04, 9.9920e-01],
        [9.2933e-04, 9.9907e-01],
        [1.5177e-04, 9.9985e-01],
        [2.1098e-02, 9.7890e-01],
        [7.6862e-03, 9.9231e-01],
        [6.7208e-04, 9.9933e-01],
        [1.0812e-03, 9.9892e-01],
        [9.9466e-01, 5.3434e-03],
        [2.6801e-03, 9.9732e-01],
        [2.9039e-04, 9.9971e-01],
        [9.9854e-01, 1.4609e-03],
        [9.1816e-01, 8.1838e-02],
        [9.4468e-03, 9.9055e-01],
        [3.5195e-03, 9.9648e-01],
        [9.9668e-01, 3.3225e-03],
        [3.8821e-02, 9.6118e-01],
        [9.7089e-01, 2.9110e-02],
        [2.8608e-03, 9.9714e-01],
        [3.3407e-01, 6.6593e-01],
        [1.0885e-03, 9.9891e-01],
        [4.7035e-02, 9.5296e-01],
        [2.1199e-03, 9.9788e-01],
        [1.1857e-04, 9.9988e-01],
        [6.8764e-04, 9.9931e-01],
        [9.9879e-01, 1.2073e-03],
        [3.2535e-04, 9.9967e-01],
        [1.1224e-01, 8.8776e-01],
        [9.2110e-06, 9.9999e-01],
        [1.3614e-05, 9.9999e-01],
        [2.1291e-04, 9.9979e-01],
        [5.5479e-03, 9.9445e-01],
        [2.6968e-02, 9.7303e-01],
        [4.3260e-04, 9.9957e-01],
        [8.3490e-01, 1.6510e-01],
        [9.9585e-01, 4.1547e-03],
        [5.6169e-04, 9.9944e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.1971e-03, 9.9480e-01],
        [9.9653e-01, 3.4714e-03],
        [3.4914e-02, 9.6509e-01],
        [1.7728e-05, 9.9998e-01],
        [1.2363e-03, 9.9876e-01],
        [9.9183e-01, 8.1738e-03],
        [5.4932e-04, 9.9945e-01],
        [1.1975e-03, 9.9880e-01],
        [3.3604e-03, 9.9664e-01],
        [8.3896e-01, 1.6104e-01],
        [1.4329e-04, 9.9986e-01],
        [6.9605e-04, 9.9930e-01],
        [9.9571e-01, 4.2935e-03],
        [7.8149e-01, 2.1851e-01],
        [7.0674e-03, 9.9293e-01],
        [7.3275e-01, 2.6725e-01],
        [5.2266e-03, 9.9477e-01],
        [1.0986e-04, 9.9989e-01],
        [1.3776e-04, 9.9986e-01],
        [6.0402e-01, 3.9598e-01],
        [9.9917e-01, 8.3184e-04],
        [3.3749e-02, 9.6625e-01],
        [3.3598e-05, 9.9997e-01],
        [1.9319e-03, 9.9807e-01],
        [5.7002e-05, 9.9994e-01],
        [9.9675e-01, 3.2490e-03],
        [1.2879e-05, 9.9999e-01],
        [8.4368e-05, 9.9992e-01],
        [9.9979e-01, 2.0678e-04],
        [8.5130e-03, 9.9149e-01],
        [1.8413e-03, 9.9816e-01],
        [9.9479e-01, 5.2058e-03],
        [2.6718e-05, 9.9997e-01],
        [2.1393e-07, 1.0000e+00],
        [1.0164e-03, 9.9898e-01],
        [2.0662e-02, 9.7934e-01],
        [6.1640e-02, 9.3836e-01],
        [4.6727e-02, 9.5327e-01],
        [3.2419e-04, 9.9968e-01],
        [9.9443e-01, 5.5714e-03],
        [2.6435e-06, 1.0000e+00],
        [9.9681e-01, 3.1913e-03],
        [3.3678e-04, 9.9966e-01],
        [9.1096e-05, 9.9991e-01],
        [2.6498e-05, 9.9997e-01],
        [9.9969e-01, 3.1363e-04],
        [1.1487e-05, 9.9999e-01],
        [2.2547e-03, 9.9775e-01],
        [1.9285e-02, 9.8072e-01],
        [9.8491e-01, 1.5087e-02],
        [4.8292e-04, 9.9952e-01],
        [2.3685e-02, 9.7631e-01],
        [6.3416e-03, 9.9366e-01],
        [1.2056e-05, 9.9999e-01],
        [2.9246e-03, 9.9708e-01],
        [1.6531e-02, 9.8347e-01],
        [8.1195e-05, 9.9992e-01],
        [5.9423e-04, 9.9941e-01],
        [9.0033e-03, 9.9100e-01],
        [9.9743e-01, 2.5669e-03],
        [1.1230e-02, 9.8877e-01],
        [3.7489e-03, 9.9625e-01],
        [9.9676e-01, 3.2445e-03],
        [3.0985e-04, 9.9969e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2982e-02, 9.7702e-01],
        [3.7738e-03, 9.9623e-01],
        [9.7131e-04, 9.9903e-01],
        [1.1109e-03, 9.9889e-01],
        [3.2690e-01, 6.7310e-01],
        [2.8720e-02, 9.7128e-01],
        [3.4245e-06, 1.0000e+00],
        [2.2136e-01, 7.7864e-01],
        [3.3492e-03, 9.9665e-01],
        [2.0939e-04, 9.9979e-01],
        [1.1982e-04, 9.9988e-01],
        [3.5154e-06, 1.0000e+00],
        [4.6250e-02, 9.5375e-01],
        [9.6587e-01, 3.4129e-02],
        [1.0767e-03, 9.9892e-01],
        [2.0926e-04, 9.9979e-01],
        [8.6209e-06, 9.9999e-01],
        [4.8650e-03, 9.9514e-01],
        [9.7464e-01, 2.5360e-02],
        [9.7552e-05, 9.9990e-01],
        [9.9906e-01, 9.3915e-04],
        [9.2542e-01, 7.4576e-02],
        [1.0542e-04, 9.9989e-01],
        [5.9413e-04, 9.9941e-01],
        [3.4792e-03, 9.9652e-01],
        [9.2620e-05, 9.9991e-01],
        [2.6654e-04, 9.9973e-01],
        [5.1049e-03, 9.9490e-01],
        [1.1810e-03, 9.9882e-01],
        [1.7004e-04, 9.9983e-01],
        [3.4851e-05, 9.9997e-01],
        [9.0397e-01, 9.6033e-02],
        [2.2271e-06, 1.0000e+00],
        [8.4691e-01, 1.5309e-01],
        [9.7551e-07, 1.0000e+00],
        [1.9307e-05, 9.9998e-01],
        [2.7038e-05, 9.9997e-01],
        [5.0605e-03, 9.9494e-01],
        [9.9467e-01, 5.3301e-03],
        [9.9908e-01, 9.1725e-04],
        [5.4642e-03, 9.9454e-01],
        [8.6373e-04, 9.9914e-01],
        [3.9678e-03, 9.9603e-01],
        [9.9922e-01, 7.8140e-04],
        [1.8591e-07, 1.0000e+00],
        [3.7987e-05, 9.9996e-01],
        [1.5057e-05, 9.9998e-01],
        [9.6083e-01, 3.9170e-02],
        [9.9962e-01, 3.7883e-04],
        [5.1073e-03, 9.9489e-01],
        [2.8456e-04, 9.9972e-01],
        [5.7002e-03, 9.9430e-01],
        [1.6342e-02, 9.8366e-01],
        [8.8726e-01, 1.1274e-01],
        [8.4639e-03, 9.9154e-01],
        [1.8270e-06, 1.0000e+00],
        [6.6298e-04, 9.9934e-01],
        [2.9227e-04, 9.9971e-01],
        [9.4641e-01, 5.3591e-02],
        [2.0240e-03, 9.9798e-01],
        [9.6418e-01, 3.5822e-02],
        [3.3551e-03, 9.9664e-01],
        [1.6782e-04, 9.9983e-01],
        [3.5717e-05, 9.9996e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9737e-01, 2.6332e-03],
        [2.9723e-05, 9.9997e-01],
        [9.9883e-01, 1.1688e-03],
        [2.8016e-04, 9.9972e-01],
        [5.9349e-02, 9.4065e-01],
        [9.9996e-01, 4.1623e-05],
        [4.3965e-05, 9.9996e-01],
        [2.8679e-02, 9.7132e-01],
        [3.2556e-03, 9.9674e-01],
        [9.9825e-01, 1.7544e-03],
        [9.9331e-01, 6.6924e-03],
        [9.9328e-01, 6.7152e-03],
        [3.7900e-04, 9.9962e-01],
        [9.9622e-01, 3.7769e-03],
        [5.0268e-06, 9.9999e-01],
        [8.9401e-01, 1.0599e-01],
        [9.8680e-01, 1.3202e-02],
        [1.2329e-04, 9.9988e-01],
        [2.2292e-06, 1.0000e+00],
        [5.1351e-03, 9.9486e-01],
        [2.0876e-02, 9.7912e-01],
        [1.8502e-03, 9.9815e-01],
        [8.7828e-06, 9.9999e-01],
        [9.9059e-01, 9.4150e-03],
        [2.0483e-04, 9.9980e-01],
        [3.0804e-03, 9.9692e-01],
        [3.9601e-02, 9.6040e-01],
        [2.2743e-02, 9.7726e-01],
        [8.0536e-05, 9.9992e-01],
        [4.6426e-06, 1.0000e+00],
        [4.7773e-02, 9.5223e-01],
        [1.4958e-02, 9.8504e-01],
        [7.8437e-04, 9.9922e-01],
        [1.3838e-02, 9.8616e-01],
        [9.8065e-01, 1.9355e-02],
        [6.1969e-03, 9.9380e-01],
        [9.8741e-01, 1.2590e-02],
        [2.6385e-03, 9.9736e-01],
        [1.7316e-02, 9.8268e-01],
        [9.9808e-01, 1.9177e-03],
        [1.8184e-04, 9.9982e-01],
        [3.2198e-03, 9.9678e-01],
        [9.7661e-01, 2.3389e-02],
        [4.1606e-05, 9.9996e-01],
        [9.9828e-01, 1.7191e-03],
        [4.6392e-03, 9.9536e-01],
        [1.7613e-06, 1.0000e+00],
        [1.7914e-03, 9.9821e-01],
        [7.1211e-05, 9.9993e-01],
        [2.8052e-02, 9.7195e-01],
        [5.3584e-04, 9.9946e-01],
        [1.3983e-04, 9.9986e-01],
        [9.5192e-01, 4.8080e-02],
        [8.0586e-04, 9.9919e-01],
        [9.9845e-01, 1.5480e-03],
        [9.9039e-03, 9.9010e-01],
        [1.8359e-03, 9.9816e-01],
        [9.9049e-01, 9.5144e-03],
        [1.1619e-03, 9.9884e-01],
        [3.5909e-03, 9.9641e-01],
        [9.8200e-05, 9.9990e-01],
        [8.1123e-03, 9.9189e-01],
        [4.0792e-05, 9.9996e-01],
        [3.9601e-02, 9.6040e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.8631e-03, 9.9614e-01],
        [7.0743e-04, 9.9929e-01],
        [7.8661e-06, 9.9999e-01],
        [6.4724e-01, 3.5276e-01],
        [2.2312e-05, 9.9998e-01],
        [9.9295e-01, 7.0479e-03],
        [7.8179e-04, 9.9922e-01],
        [1.1969e-05, 9.9999e-01],
        [7.7321e-02, 9.2268e-01],
        [9.5807e-01, 4.1925e-02],
        [5.7627e-03, 9.9424e-01],
        [5.0261e-04, 9.9950e-01],
        [1.0641e-03, 9.9894e-01],
        [9.8206e-01, 1.7941e-02],
        [2.9470e-03, 9.9705e-01],
        [5.5337e-04, 9.9945e-01],
        [3.6482e-04, 9.9964e-01],
        [6.2902e-03, 9.9371e-01],
        [9.9910e-01, 9.0446e-04],
        [9.6356e-01, 3.6439e-02],
        [2.3153e-05, 9.9998e-01],
        [4.4233e-03, 9.9558e-01],
        [5.6196e-04, 9.9944e-01],
        [6.0748e-03, 9.9393e-01],
        [5.8550e-04, 9.9941e-01],
        [8.2204e-03, 9.9178e-01],
        [6.3580e-06, 9.9999e-01],
        [2.0921e-05, 9.9998e-01],
        [4.6981e-06, 1.0000e+00],
        [1.5726e-04, 9.9984e-01],
        [2.3327e-01, 7.6673e-01],
        [4.3945e-03, 9.9561e-01],
        [2.2260e-05, 9.9998e-01],
        [4.1103e-05, 9.9996e-01],
        [2.2626e-05, 9.9998e-01],
        [5.3276e-04, 9.9947e-01],
        [1.2775e-01, 8.7225e-01],
        [4.3620e-05, 9.9996e-01],
        [1.0567e-03, 9.9894e-01],
        [1.6974e-01, 8.3026e-01],
        [7.1044e-03, 9.9290e-01],
        [1.5066e-05, 9.9998e-01],
        [9.3567e-04, 9.9906e-01],
        [6.4407e-05, 9.9994e-01],
        [9.6935e-02, 9.0307e-01],
        [2.6881e-05, 9.9997e-01],
        [6.0731e-05, 9.9994e-01],
        [9.9829e-01, 1.7118e-03],
        [9.9638e-01, 3.6241e-03],
        [1.8859e-02, 9.8114e-01],
        [3.0088e-03, 9.9699e-01],
        [3.2582e-06, 1.0000e+00],
        [8.8122e-03, 9.9119e-01],
        [9.9646e-01, 3.5383e-03],
        [9.6406e-01, 3.5941e-02],
        [5.4507e-05, 9.9995e-01],
        [7.9790e-03, 9.9202e-01],
        [7.9552e-01, 2.0448e-01],
        [7.5840e-05, 9.9992e-01],
        [9.8713e-01, 1.2872e-02],
        [1.4992e-04, 9.9985e-01],
        [9.7508e-01, 2.4922e-02],
        [1.0812e-07, 1.0000e+00],
        [5.1464e-04, 9.9949e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9610e-01, 3.9028e-03],
        [9.9681e-01, 3.1909e-03],
        [1.3673e-04, 9.9986e-01],
        [1.0814e-03, 9.9892e-01],
        [9.9445e-01, 5.5549e-03],
        [9.8137e-06, 9.9999e-01],
        [9.9373e-01, 6.2746e-03],
        [2.5348e-04, 9.9975e-01],
        [9.3295e-01, 6.7046e-02],
        [6.0957e-04, 9.9939e-01],
        [1.1120e-04, 9.9989e-01],
        [8.2384e-01, 1.7616e-01],
        [1.5163e-03, 9.9848e-01],
        [4.6132e-06, 1.0000e+00],
        [1.8436e-05, 9.9998e-01],
        [2.7770e-04, 9.9972e-01],
        [7.8828e-04, 9.9921e-01],
        [9.9881e-01, 1.1923e-03],
        [9.9383e-01, 6.1719e-03],
        [6.9696e-04, 9.9930e-01],
        [3.4912e-04, 9.9965e-01],
        [2.3286e-03, 9.9767e-01],
        [7.1262e-02, 9.2874e-01],
        [9.9764e-01, 2.3561e-03],
        [5.4421e-04, 9.9946e-01],
        [5.8971e-04, 9.9941e-01],
        [1.5966e-04, 9.9984e-01],
        [7.1892e-04, 9.9928e-01],
        [9.9044e-01, 9.5574e-03],
        [6.2223e-03, 9.9378e-01],
        [5.2248e-03, 9.9478e-01],
        [9.9278e-01, 7.2238e-03],
        [2.6203e-04, 9.9974e-01],
        [3.3315e-04, 9.9967e-01],
        [7.0637e-04, 9.9929e-01],
        [4.2695e-06, 1.0000e+00],
        [8.8473e-03, 9.9115e-01],
        [3.8101e-03, 9.9619e-01],
        [6.1071e-03, 9.9389e-01],
        [4.0500e-06, 1.0000e+00],
        [6.1495e-08, 1.0000e+00],
        [9.9056e-05, 9.9990e-01],
        [2.3184e-04, 9.9977e-01],
        [1.8784e-01, 8.1216e-01],
        [7.7114e-04, 9.9923e-01],
        [4.0821e-03, 9.9592e-01],
        [1.3856e-04, 9.9986e-01],
        [5.5549e-03, 9.9445e-01],
        [1.9316e-05, 9.9998e-01],
        [7.8132e-05, 9.9992e-01],
        [9.1829e-04, 9.9908e-01],
        [9.8307e-02, 9.0169e-01],
        [1.3874e-03, 9.9861e-01],
        [7.0368e-05, 9.9993e-01],
        [1.0708e-04, 9.9989e-01],
        [3.5424e-05, 9.9996e-01],
        [1.8220e-05, 9.9998e-01],
        [5.8625e-03, 9.9414e-01],
        [6.6591e-01, 3.3409e-01],
        [4.4724e-06, 1.0000e+00],
        [9.7821e-04, 9.9902e-01],
        [9.9599e-01, 4.0111e-03],
        [1.8293e-02, 9.8171e-01],
        [7.4518e-01, 2.5482e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1924e-03, 9.9781e-01],
        [3.6699e-03, 9.9633e-01],
        [6.3601e-06, 9.9999e-01],
        [5.4821e-03, 9.9452e-01],
        [2.8164e-01, 7.1836e-01],
        [2.5105e-06, 1.0000e+00],
        [5.1364e-04, 9.9949e-01],
        [6.0044e-06, 9.9999e-01],
        [9.5153e-01, 4.8466e-02],
        [1.7208e-05, 9.9998e-01],
        [9.5762e-01, 4.2377e-02],
        [9.9750e-01, 2.4960e-03],
        [9.2342e-01, 7.6583e-02],
        [7.5102e-03, 9.9249e-01],
        [1.1588e-02, 9.8841e-01],
        [5.4811e-05, 9.9995e-01],
        [1.4425e-03, 9.9856e-01],
        [9.3468e-02, 9.0653e-01],
        [7.0150e-02, 9.2985e-01],
        [6.6836e-01, 3.3164e-01],
        [3.9737e-06, 1.0000e+00],
        [1.7684e-05, 9.9998e-01],
        [1.9446e-02, 9.8055e-01],
        [3.1186e-05, 9.9997e-01],
        [7.3339e-04, 9.9927e-01],
        [3.9608e-06, 1.0000e+00],
        [2.9813e-05, 9.9997e-01],
        [9.8194e-06, 9.9999e-01],
        [1.0348e-04, 9.9990e-01],
        [1.7099e-04, 9.9983e-01],
        [2.1396e-06, 1.0000e+00],
        [3.3365e-02, 9.6664e-01],
        [9.7678e-01, 2.3222e-02],
        [1.0386e-04, 9.9990e-01],
        [1.5338e-03, 9.9847e-01],
        [1.5868e-03, 9.9841e-01],
        [6.5839e-05, 9.9993e-01],
        [5.7633e-03, 9.9424e-01],
        [8.7232e-01, 1.2768e-01],
        [4.3333e-03, 9.9567e-01],
        [9.9433e-01, 5.6740e-03],
        [8.4490e-01, 1.5510e-01],
        [2.3505e-05, 9.9998e-01],
        [9.3501e-03, 9.9065e-01],
        [9.5825e-03, 9.9042e-01],
        [3.9676e-04, 9.9960e-01],
        [2.3476e-03, 9.9765e-01],
        [1.0961e-04, 9.9989e-01],
        [1.2772e-03, 9.9872e-01],
        [9.9686e-01, 3.1381e-03],
        [9.8500e-01, 1.5003e-02],
        [9.8624e-01, 1.3757e-02],
        [1.1399e-06, 1.0000e+00],
        [1.5923e-04, 9.9984e-01],
        [1.0950e-04, 9.9989e-01],
        [9.9778e-01, 2.2212e-03],
        [9.5199e-01, 4.8006e-02],
        [1.9442e-03, 9.9806e-01],
        [9.9865e-01, 1.3473e-03],
        [6.0564e-05, 9.9994e-01],
        [5.7394e-04, 9.9943e-01],
        [2.4624e-04, 9.9975e-01],
        [4.6983e-05, 9.9995e-01],
        [3.6582e-05, 9.9996e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4546e-04, 9.9985e-01],
        [1.0977e-04, 9.9989e-01],
        [9.1964e-01, 8.0362e-02],
        [1.3818e-05, 9.9999e-01],
        [6.4666e-02, 9.3533e-01],
        [7.2516e-01, 2.7484e-01],
        [1.6510e-06, 1.0000e+00],
        [7.0114e-04, 9.9930e-01],
        [2.0114e-04, 9.9980e-01],
        [5.7936e-06, 9.9999e-01],
        [2.9067e-05, 9.9997e-01],
        [8.6471e-01, 1.3529e-01],
        [9.8933e-01, 1.0667e-02],
        [9.9646e-01, 3.5395e-03],
        [9.9818e-01, 1.8191e-03],
        [3.5815e-04, 9.9964e-01],
        [3.7807e-02, 9.6219e-01],
        [5.1303e-02, 9.4870e-01],
        [1.2985e-05, 9.9999e-01],
        [1.7016e-03, 9.9830e-01],
        [4.5690e-05, 9.9995e-01],
        [5.9482e-03, 9.9405e-01],
        [1.6462e-03, 9.9835e-01],
        [9.7667e-07, 1.0000e+00],
        [9.7427e-01, 2.5731e-02],
        [7.4274e-05, 9.9993e-01],
        [9.9143e-01, 8.5735e-03],
        [2.8182e-03, 9.9718e-01],
        [1.7473e-02, 9.8253e-01],
        [3.7955e-06, 1.0000e+00],
        [3.1745e-05, 9.9997e-01],
        [2.0087e-04, 9.9980e-01],
        [1.2797e-03, 9.9872e-01],
        [4.5004e-02, 9.5500e-01],
        [9.7219e-02, 9.0278e-01],
        [2.8104e-06, 1.0000e+00],
        [5.9710e-03, 9.9403e-01],
        [3.3623e-04, 9.9966e-01],
        [6.6779e-01, 3.3221e-01],
        [2.0130e-03, 9.9799e-01],
        [2.2276e-04, 9.9978e-01],
        [1.9483e-06, 1.0000e+00],
        [1.2214e-02, 9.8779e-01],
        [1.3023e-05, 9.9999e-01],
        [2.2255e-02, 9.7774e-01],
        [6.4186e-03, 9.9358e-01],
        [9.7058e-01, 2.9417e-02],
        [2.2705e-03, 9.9773e-01],
        [1.1159e-03, 9.9888e-01],
        [1.0273e-02, 9.8973e-01],
        [2.0627e-05, 9.9998e-01],
        [2.4578e-01, 7.5422e-01],
        [2.4358e-04, 9.9976e-01],
        [3.9219e-03, 9.9608e-01],
        [8.9374e-05, 9.9991e-01],
        [1.7839e-01, 8.2161e-01],
        [9.9707e-01, 2.9350e-03],
        [1.7015e-04, 9.9983e-01],
        [4.6091e-02, 9.5391e-01],
        [1.0193e-04, 9.9990e-01],
        [4.4518e-01, 5.5482e-01],
        [4.9926e-02, 9.5007e-01],
        [2.4626e-03, 9.9754e-01],
        [1.7618e-04, 9.9982e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.3476e-04, 9.9937e-01],
        [6.0407e-05, 9.9994e-01],
        [5.6434e-04, 9.9944e-01],
        [1.5588e-02, 9.8441e-01],
        [3.5024e-03, 9.9650e-01],
        [5.9774e-01, 4.0226e-01],
        [1.2663e-05, 9.9999e-01],
        [9.9828e-01, 1.7244e-03],
        [9.7692e-01, 2.3080e-02],
        [6.9552e-05, 9.9993e-01],
        [4.2511e-02, 9.5749e-01],
        [1.6429e-04, 9.9984e-01],
        [3.1769e-03, 9.9682e-01],
        [3.7129e-03, 9.9629e-01],
        [1.3041e-05, 9.9999e-01],
        [6.5282e-01, 3.4718e-01],
        [9.4942e-06, 9.9999e-01],
        [5.7934e-04, 9.9942e-01],
        [9.3777e-01, 6.2233e-02],
        [5.1229e-03, 9.9488e-01],
        [6.5355e-05, 9.9993e-01],
        [2.0063e-04, 9.9980e-01],
        [5.2764e-05, 9.9995e-01],
        [9.6694e-01, 3.3064e-02],
        [2.4465e-02, 9.7553e-01],
        [1.5696e-04, 9.9984e-01],
        [8.5826e-01, 1.4174e-01],
        [9.0284e-04, 9.9910e-01],
        [1.5771e-03, 9.9842e-01],
        [2.3739e-05, 9.9998e-01],
        [2.7189e-01, 7.2811e-01],
        [1.1199e-03, 9.9888e-01],
        [8.8059e-01, 1.1941e-01],
        [1.1376e-05, 9.9999e-01],
        [2.8444e-02, 9.7156e-01],
        [9.9575e-01, 4.2480e-03],
        [1.7460e-03, 9.9825e-01],
        [9.9751e-01, 2.4939e-03],
        [9.2621e-03, 9.9074e-01],
        [9.9848e-01, 1.5183e-03],
        [3.0167e-03, 9.9698e-01],
        [6.6705e-05, 9.9993e-01],
        [1.6007e-03, 9.9840e-01],
        [9.8618e-01, 1.3816e-02],
        [4.8743e-04, 9.9951e-01],
        [2.8786e-03, 9.9712e-01],
        [2.0851e-03, 9.9791e-01],
        [2.4582e-04, 9.9975e-01],
        [3.5518e-04, 9.9964e-01],
        [9.4177e-06, 9.9999e-01],
        [5.8534e-05, 9.9994e-01],
        [8.8319e-01, 1.1681e-01],
        [2.6774e-02, 9.7323e-01],
        [1.0228e-05, 9.9999e-01],
        [6.5948e-05, 9.9993e-01],
        [2.6075e-04, 9.9974e-01],
        [2.3202e-04, 9.9977e-01],
        [2.2410e-02, 9.7759e-01],
        [1.2235e-02, 9.8777e-01],
        [2.8393e-04, 9.9972e-01],
        [9.9252e-01, 7.4775e-03],
        [5.9082e-05, 9.9994e-01],
        [2.9247e-04, 9.9971e-01],
        [2.6419e-04, 9.9974e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.4790e-03, 9.9452e-01],
        [5.8997e-06, 9.9999e-01],
        [1.0909e-04, 9.9989e-01],
        [9.6589e-03, 9.9034e-01],
        [9.7478e-01, 2.5223e-02],
        [6.3920e-05, 9.9994e-01],
        [6.0881e-06, 9.9999e-01],
        [2.4886e-03, 9.9751e-01],
        [4.4362e-04, 9.9956e-01],
        [8.5606e-07, 1.0000e+00],
        [4.5612e-04, 9.9954e-01],
        [1.6358e-03, 9.9836e-01],
        [4.6659e-04, 9.9953e-01],
        [9.9872e-03, 9.9001e-01],
        [9.8986e-01, 1.0140e-02],
        [2.2396e-03, 9.9776e-01],
        [2.7650e-03, 9.9724e-01],
        [1.3112e-03, 9.9869e-01],
        [6.4045e-03, 9.9360e-01],
        [8.9246e-05, 9.9991e-01],
        [1.1858e-03, 9.9881e-01],
        [9.1649e-01, 8.3510e-02],
        [4.0626e-06, 1.0000e+00],
        [1.9068e-03, 9.9809e-01],
        [1.0051e-03, 9.9899e-01],
        [9.8045e-01, 1.9545e-02],
        [9.7957e-01, 2.0429e-02],
        [8.4990e-06, 9.9999e-01],
        [9.9845e-01, 1.5549e-03],
        [7.3409e-05, 9.9993e-01],
        [7.5266e-02, 9.2473e-01],
        [1.0784e-03, 9.9892e-01],
        [1.7264e-03, 9.9827e-01],
        [9.9836e-01, 1.6367e-03],
        [9.9396e-01, 6.0370e-03],
        [5.7025e-04, 9.9943e-01],
        [1.0874e-02, 9.8913e-01],
        [1.0727e-02, 9.8927e-01],
        [8.0942e-03, 9.9191e-01],
        [4.7166e-04, 9.9953e-01],
        [2.0736e-03, 9.9793e-01],
        [1.8684e-03, 9.9813e-01],
        [4.2367e-05, 9.9996e-01],
        [9.8440e-01, 1.5597e-02],
        [5.3310e-03, 9.9467e-01],
        [9.8371e-01, 1.6291e-02],
        [9.9926e-01, 7.3539e-04],
        [8.4999e-01, 1.5001e-01],
        [2.5037e-04, 9.9975e-01],
        [9.4000e-01, 6.0004e-02],
        [9.9714e-01, 2.8575e-03],
        [1.1937e-02, 9.8806e-01],
        [7.0306e-03, 9.9297e-01],
        [2.2870e-02, 9.7713e-01],
        [1.4510e-05, 9.9999e-01],
        [1.7249e-02, 9.8275e-01],
        [4.2873e-05, 9.9996e-01],
        [9.9661e-01, 3.3938e-03],
        [2.0418e-04, 9.9980e-01],
        [6.9303e-01, 3.0697e-01],
        [6.2894e-01, 3.7106e-01],
        [9.8456e-01, 1.5443e-02],
        [2.0689e-04, 9.9979e-01],
        [2.6088e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.7186e-04, 9.9953e-01],
        [3.2984e-04, 9.9967e-01],
        [9.4756e-01, 5.2443e-02],
        [2.2389e-02, 9.7761e-01],
        [1.6777e-03, 9.9832e-01],
        [1.7024e-02, 9.8298e-01],
        [1.1769e-03, 9.9882e-01],
        [2.1547e-03, 9.9785e-01],
        [1.9381e-05, 9.9998e-01],
        [2.1350e-03, 9.9787e-01],
        [4.7272e-01, 5.2728e-01],
        [1.7980e-03, 9.9820e-01],
        [2.5621e-04, 9.9974e-01],
        [2.9242e-04, 9.9971e-01],
        [1.4845e-05, 9.9999e-01],
        [1.5907e-03, 9.9841e-01],
        [9.9882e-01, 1.1817e-03],
        [4.0586e-04, 9.9959e-01],
        [1.4294e-06, 1.0000e+00],
        [7.2373e-03, 9.9276e-01],
        [2.1836e-03, 9.9782e-01],
        [2.5998e-02, 9.7400e-01],
        [1.3949e-06, 1.0000e+00],
        [2.7349e-03, 9.9727e-01],
        [2.3160e-05, 9.9998e-01],
        [9.9946e-01, 5.3720e-04],
        [9.8107e-01, 1.8933e-02],
        [2.5581e-01, 7.4419e-01],
        [5.1764e-01, 4.8236e-01],
        [7.4150e-03, 9.9259e-01],
        [8.6219e-06, 9.9999e-01],
        [4.1860e-03, 9.9581e-01],
        [5.1846e-04, 9.9948e-01],
        [8.3671e-05, 9.9992e-01],
        [9.2604e-01, 7.3960e-02],
        [1.0211e-07, 1.0000e+00],
        [9.9944e-01, 5.6104e-04],
        [1.7636e-03, 9.9824e-01],
        [6.6325e-07, 1.0000e+00],
        [3.2452e-02, 9.6755e-01],
        [1.3901e-02, 9.8610e-01],
        [1.0184e-05, 9.9999e-01],
        [1.6153e-04, 9.9984e-01],
        [2.0955e-02, 9.7904e-01],
        [9.8752e-01, 1.2475e-02],
        [8.3568e-05, 9.9992e-01],
        [9.9901e-01, 9.9070e-04],
        [4.5665e-03, 9.9543e-01],
        [9.9747e-01, 2.5252e-03],
        [2.9037e-03, 9.9710e-01],
        [1.1560e-04, 9.9988e-01],
        [6.2493e-01, 3.7507e-01],
        [7.1289e-02, 9.2871e-01],
        [6.7142e-01, 3.2858e-01],
        [4.3336e-04, 9.9957e-01],
        [4.9815e-03, 9.9502e-01],
        [9.7862e-01, 2.1381e-02],
        [9.9819e-01, 1.8106e-03],
        [1.5440e-02, 9.8456e-01],
        [2.2191e-02, 9.7781e-01],
        [2.0945e-05, 9.9998e-01],
        [1.9141e-02, 9.8086e-01],
        [9.9155e-03, 9.9008e-01],
        [3.6208e-02, 9.6379e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9791e-01, 2.0877e-03],
        [1.4946e-05, 9.9999e-01],
        [9.9900e-01, 1.0005e-03],
        [9.9920e-01, 8.0285e-04],
        [7.5064e-03, 9.9249e-01],
        [3.6707e-03, 9.9633e-01],
        [9.9596e-01, 4.0360e-03],
        [2.4849e-03, 9.9752e-01],
        [1.5540e-02, 9.8446e-01],
        [1.0641e-03, 9.9894e-01],
        [1.8721e-02, 9.8128e-01],
        [9.9919e-01, 8.0966e-04],
        [9.9860e-01, 1.3955e-03],
        [2.3469e-04, 9.9977e-01],
        [9.8992e-01, 1.0078e-02],
        [9.2065e-05, 9.9991e-01],
        [5.3764e-02, 9.4624e-01],
        [3.2848e-02, 9.6715e-01],
        [1.5843e-02, 9.8416e-01],
        [1.2664e-02, 9.8734e-01],
        [6.7061e-06, 9.9999e-01],
        [9.9288e-01, 7.1245e-03],
        [9.9319e-01, 6.8115e-03],
        [9.9802e-01, 1.9813e-03],
        [7.4632e-01, 2.5368e-01],
        [3.9878e-05, 9.9996e-01],
        [4.9532e-01, 5.0468e-01],
        [2.7294e-04, 9.9973e-01],
        [2.9596e-05, 9.9997e-01],
        [5.3965e-04, 9.9946e-01],
        [5.2540e-01, 4.7460e-01],
        [9.9040e-01, 9.6001e-03],
        [2.8628e-04, 9.9971e-01],
        [8.3196e-01, 1.6804e-01],
        [2.0437e-04, 9.9980e-01],
        [3.2431e-03, 9.9676e-01],
        [5.2559e-04, 9.9947e-01],
        [6.1596e-05, 9.9994e-01],
        [1.6928e-04, 9.9983e-01],
        [6.6984e-03, 9.9330e-01],
        [9.9560e-01, 4.3957e-03],
        [2.9959e-03, 9.9700e-01],
        [9.9372e-01, 6.2787e-03],
        [7.8510e-02, 9.2149e-01],
        [4.0577e-05, 9.9996e-01],
        [3.8378e-03, 9.9616e-01],
        [3.7204e-03, 9.9628e-01],
        [2.2523e-03, 9.9775e-01],
        [9.1726e-01, 8.2738e-02],
        [4.3370e-05, 9.9996e-01],
        [9.8971e-01, 1.0290e-02],
        [1.1641e-05, 9.9999e-01],
        [7.1748e-03, 9.9283e-01],
        [5.2733e-07, 1.0000e+00],
        [2.0871e-05, 9.9998e-01],
        [2.9856e-05, 9.9997e-01],
        [1.6611e-04, 9.9983e-01],
        [2.3293e-04, 9.9977e-01],
        [8.5740e-03, 9.9143e-01],
        [4.6168e-05, 9.9995e-01],
        [9.7000e-01, 3.0004e-02],
        [2.4867e-03, 9.9751e-01],
        [3.2286e-05, 9.9997e-01],
        [7.0079e-01, 2.9921e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.6360e-05, 9.9997e-01],
        [9.9914e-01, 8.5984e-04],
        [2.4205e-03, 9.9758e-01],
        [2.3184e-02, 9.7682e-01],
        [4.5946e-05, 9.9995e-01],
        [4.2530e-03, 9.9575e-01],
        [2.2260e-03, 9.9777e-01],
        [3.4695e-03, 9.9653e-01],
        [8.5587e-03, 9.9144e-01],
        [5.4095e-04, 9.9946e-01],
        [2.9251e-04, 9.9971e-01],
        [6.1601e-05, 9.9994e-01],
        [7.2954e-04, 9.9927e-01],
        [6.7841e-05, 9.9993e-01],
        [3.3238e-01, 6.6762e-01],
        [5.4877e-03, 9.9451e-01],
        [1.2235e-04, 9.9988e-01],
        [3.0881e-01, 6.9119e-01],
        [1.3534e-05, 9.9999e-01],
        [1.2922e-02, 9.8708e-01],
        [7.6857e-05, 9.9992e-01],
        [1.1855e-04, 9.9988e-01],
        [8.8054e-06, 9.9999e-01],
        [9.9644e-01, 3.5589e-03],
        [9.4776e-01, 5.2244e-02],
        [3.2369e-04, 9.9968e-01],
        [8.1252e-05, 9.9992e-01],
        [1.3662e-05, 9.9999e-01],
        [1.7031e-02, 9.8297e-01],
        [9.9854e-01, 1.4634e-03],
        [4.8748e-02, 9.5125e-01],
        [8.1252e-05, 9.9992e-01],
        [2.7903e-02, 9.7210e-01],
        [3.1707e-01, 6.8293e-01],
        [7.0219e-01, 2.9781e-01],
        [4.9056e-01, 5.0944e-01],
        [9.8757e-01, 1.2426e-02],
        [3.2504e-06, 1.0000e+00],
        [1.2368e-02, 9.8763e-01],
        [4.7886e-04, 9.9952e-01],
        [9.9842e-01, 1.5799e-03],
        [2.5923e-04, 9.9974e-01],
        [5.3127e-05, 9.9995e-01],
        [1.4373e-03, 9.9856e-01],
        [9.9252e-01, 7.4839e-03],
        [9.9916e-01, 8.4316e-04],
        [3.6960e-02, 9.6304e-01],
        [8.7969e-04, 9.9912e-01],
        [1.1353e-03, 9.9886e-01],
        [3.1116e-02, 9.6888e-01],
        [1.0760e-03, 9.9892e-01],
        [9.8687e-01, 1.3125e-02],
        [9.9933e-01, 6.6657e-04],
        [1.6184e-03, 9.9838e-01],
        [4.6499e-03, 9.9535e-01],
        [2.0800e-05, 9.9998e-01],
        [1.7217e-02, 9.8278e-01],
        [9.9892e-01, 1.0786e-03],
        [1.4427e-02, 9.8557e-01],
        [2.7706e-04, 9.9972e-01],
        [1.8927e-04, 9.9981e-01],
        [5.3214e-02, 9.4679e-01],
        [2.1813e-03, 9.9782e-01],
        [6.7962e-02, 9.3204e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.2957e-03, 9.9470e-01],
        [9.1690e-01, 8.3098e-02],
        [9.7234e-02, 9.0277e-01],
        [7.8246e-05, 9.9992e-01],
        [5.3056e-01, 4.6944e-01],
        [7.0719e-03, 9.9293e-01],
        [6.2186e-03, 9.9378e-01],
        [3.3390e-03, 9.9666e-01],
        [3.3289e-03, 9.9667e-01],
        [3.1456e-05, 9.9997e-01],
        [1.4633e-05, 9.9999e-01],
        [1.8720e-04, 9.9981e-01],
        [4.4542e-04, 9.9955e-01],
        [1.6245e-02, 9.8376e-01],
        [1.8427e-01, 8.1573e-01],
        [7.7899e-03, 9.9221e-01],
        [9.9814e-01, 1.8561e-03],
        [5.3844e-04, 9.9946e-01],
        [1.5783e-02, 9.8422e-01],
        [1.2501e-04, 9.9987e-01],
        [3.8362e-02, 9.6164e-01],
        [4.6340e-01, 5.3660e-01],
        [9.9773e-01, 2.2691e-03],
        [1.9142e-04, 9.9981e-01],
        [9.8436e-03, 9.9016e-01],
        [5.9563e-03, 9.9404e-01],
        [1.4935e-02, 9.8506e-01],
        [9.1110e-05, 9.9991e-01],
        [7.0154e-02, 9.2985e-01],
        [3.4297e-03, 9.9657e-01],
        [9.4195e-03, 9.9058e-01],
        [1.8165e-01, 8.1835e-01],
        [2.2301e-05, 9.9998e-01],
        [2.3535e-02, 9.7646e-01],
        [1.8574e-05, 9.9998e-01],
        [1.8677e-02, 9.8132e-01],
        [9.9883e-01, 1.1702e-03],
        [3.1224e-05, 9.9997e-01],
        [9.8524e-01, 1.4756e-02],
        [2.9779e-06, 1.0000e+00],
        [1.4453e-02, 9.8555e-01],
        [9.9712e-01, 2.8829e-03],
        [9.9983e-01, 1.6794e-04],
        [1.5097e-03, 9.9849e-01],
        [1.4619e-02, 9.8538e-01],
        [7.4293e-02, 9.2571e-01],
        [6.3230e-03, 9.9368e-01],
        [4.6235e-01, 5.3765e-01],
        [2.0308e-03, 9.9797e-01],
        [4.3710e-05, 9.9996e-01],
        [1.2390e-02, 9.8761e-01],
        [9.9861e-01, 1.3943e-03],
        [8.9626e-02, 9.1037e-01],
        [2.2952e-04, 9.9977e-01],
        [2.0310e-05, 9.9998e-01],
        [9.4783e-01, 5.2174e-02],
        [9.8476e-01, 1.5238e-02],
        [1.9497e-02, 9.8050e-01],
        [2.3032e-03, 9.9770e-01],
        [2.0050e-03, 9.9800e-01],
        [6.2043e-03, 9.9380e-01],
        [5.7770e-03, 9.9422e-01],
        [6.9911e-05, 9.9993e-01],
        [5.0591e-03, 9.9494e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.3098e-01, 2.6902e-01],
        [9.6041e-01, 3.9595e-02],
        [2.7060e-03, 9.9729e-01],
        [1.7630e-03, 9.9824e-01],
        [1.5359e-01, 8.4641e-01],
        [4.3114e-03, 9.9569e-01],
        [2.7208e-06, 1.0000e+00],
        [1.1363e-04, 9.9989e-01],
        [9.5768e-02, 9.0423e-01],
        [3.5543e-02, 9.6446e-01],
        [1.8580e-06, 1.0000e+00],
        [2.6724e-03, 9.9733e-01],
        [1.9149e-05, 9.9998e-01],
        [9.5694e-03, 9.9043e-01],
        [3.9262e-06, 1.0000e+00],
        [3.3455e-05, 9.9997e-01],
        [1.8555e-06, 1.0000e+00],
        [1.0852e-04, 9.9989e-01],
        [9.6931e-01, 3.0692e-02],
        [8.9253e-01, 1.0747e-01],
        [9.9203e-01, 7.9678e-03],
        [5.8638e-05, 9.9994e-01],
        [2.0225e-04, 9.9980e-01],
        [2.0285e-04, 9.9980e-01],
        [1.9865e-04, 9.9980e-01],
        [1.5593e-03, 9.9844e-01],
        [5.4857e-06, 9.9999e-01],
        [1.0697e-05, 9.9999e-01],
        [8.5804e-05, 9.9991e-01],
        [2.6948e-03, 9.9731e-01],
        [9.9994e-01, 6.1462e-05],
        [9.6470e-01, 3.5302e-02],
        [1.5581e-02, 9.8442e-01],
        [9.9590e-01, 4.1044e-03],
        [9.4952e-07, 1.0000e+00],
        [5.7139e-04, 9.9943e-01],
        [1.4909e-01, 8.5091e-01],
        [8.6771e-04, 9.9913e-01],
        [4.2826e-04, 9.9957e-01],
        [9.7428e-01, 2.5720e-02],
        [2.1272e-03, 9.9787e-01],
        [9.9467e-01, 5.3306e-03],
        [3.8591e-04, 9.9961e-01],
        [7.2092e-02, 9.2791e-01],
        [4.4359e-05, 9.9996e-01],
        [1.1466e-02, 9.8853e-01],
        [9.9839e-01, 1.6094e-03],
        [8.5804e-05, 9.9991e-01],
        [1.1472e-03, 9.9885e-01],
        [2.0109e-05, 9.9998e-01],
        [5.4096e-02, 9.4590e-01],
        [1.1548e-03, 9.9885e-01],
        [1.4872e-04, 9.9985e-01],
        [1.7805e-03, 9.9822e-01],
        [1.7500e-04, 9.9983e-01],
        [2.7916e-04, 9.9972e-01],
        [8.6907e-06, 9.9999e-01],
        [2.5137e-04, 9.9975e-01],
        [1.7732e-04, 9.9982e-01],
        [6.0028e-04, 9.9940e-01],
        [1.3071e-05, 9.9999e-01],
        [2.8429e-02, 9.7157e-01],
        [3.7547e-04, 9.9962e-01],
        [6.1305e-03, 9.9387e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.5764e-01, 4.2362e-02],
        [1.5544e-04, 9.9984e-01],
        [9.5567e-01, 4.4328e-02],
        [2.0085e-04, 9.9980e-01],
        [8.3902e-04, 9.9916e-01],
        [9.9894e-01, 1.0617e-03],
        [9.5883e-05, 9.9990e-01],
        [1.1959e-04, 9.9988e-01],
        [5.9735e-03, 9.9403e-01],
        [9.8686e-02, 9.0131e-01],
        [6.1401e-04, 9.9939e-01],
        [8.5642e-04, 9.9914e-01],
        [9.9862e-01, 1.3833e-03],
        [5.8876e-03, 9.9411e-01],
        [1.4052e-02, 9.8595e-01],
        [1.9581e-02, 9.8042e-01],
        [1.2330e-05, 9.9999e-01],
        [6.8491e-03, 9.9315e-01],
        [2.8343e-03, 9.9717e-01],
        [2.4235e-04, 9.9976e-01],
        [6.2777e-03, 9.9372e-01],
        [2.2097e-04, 9.9978e-01],
        [4.5640e-05, 9.9995e-01],
        [5.7198e-04, 9.9943e-01],
        [9.9774e-01, 2.2597e-03],
        [7.6085e-01, 2.3915e-01],
        [1.3158e-03, 9.9868e-01],
        [9.4827e-03, 9.9052e-01],
        [1.4500e-04, 9.9985e-01],
        [1.8062e-06, 1.0000e+00],
        [5.6873e-05, 9.9994e-01],
        [9.1610e-06, 9.9999e-01],
        [5.6174e-01, 4.3826e-01],
        [1.2785e-04, 9.9987e-01],
        [9.9507e-01, 4.9298e-03],
        [9.9609e-01, 3.9064e-03],
        [1.3073e-02, 9.8693e-01],
        [1.1149e-04, 9.9989e-01],
        [1.8065e-01, 8.1935e-01],
        [4.7087e-03, 9.9529e-01],
        [9.4406e-03, 9.9056e-01],
        [1.8040e-04, 9.9982e-01],
        [3.2579e-04, 9.9967e-01],
        [2.0928e-01, 7.9072e-01],
        [4.5913e-05, 9.9995e-01],
        [1.4972e-01, 8.5028e-01],
        [2.5709e-04, 9.9974e-01],
        [1.4351e-02, 9.8565e-01],
        [9.4899e-06, 9.9999e-01],
        [1.8009e-04, 9.9982e-01],
        [6.3526e-03, 9.9365e-01],
        [1.6165e-03, 9.9838e-01],
        [1.1045e-03, 9.9890e-01],
        [1.0655e-02, 9.8935e-01],
        [9.8196e-02, 9.0180e-01],
        [5.5924e-06, 9.9999e-01],
        [5.4963e-05, 9.9995e-01],
        [7.5404e-05, 9.9992e-01],
        [9.0061e-06, 9.9999e-01],
        [9.9071e-01, 9.2946e-03],
        [9.9871e-01, 1.2938e-03],
        [3.6569e-02, 9.6343e-01],
        [5.6831e-04, 9.9943e-01],
        [1.5351e-03, 9.9846e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5152e-05, 9.9997e-01],
        [5.9985e-05, 9.9994e-01],
        [4.6936e-01, 5.3064e-01],
        [3.7516e-05, 9.9996e-01],
        [4.8080e-04, 9.9952e-01],
        [1.8988e-02, 9.8101e-01],
        [2.0142e-01, 7.9858e-01],
        [1.1953e-03, 9.9880e-01],
        [1.3339e-05, 9.9999e-01],
        [3.2763e-03, 9.9672e-01],
        [1.0027e-05, 9.9999e-01],
        [4.8297e-05, 9.9995e-01],
        [1.7465e-04, 9.9983e-01],
        [7.2770e-04, 9.9927e-01],
        [2.0516e-03, 9.9795e-01],
        [9.8969e-01, 1.0314e-02],
        [1.1903e-01, 8.8097e-01],
        [8.2645e-01, 1.7355e-01],
        [2.5592e-03, 9.9744e-01],
        [1.5277e-03, 9.9847e-01],
        [3.2425e-04, 9.9968e-01],
        [1.2076e-03, 9.9879e-01],
        [9.9531e-01, 4.6910e-03],
        [4.2536e-06, 1.0000e+00],
        [9.4846e-04, 9.9905e-01],
        [5.1612e-01, 4.8388e-01],
        [1.9709e-01, 8.0291e-01],
        [9.9069e-01, 9.3122e-03],
        [3.3831e-06, 1.0000e+00],
        [9.9267e-01, 7.3306e-03],
        [7.9498e-04, 9.9921e-01],
        [3.0962e-03, 9.9690e-01],
        [4.6834e-05, 9.9995e-01],
        [2.4841e-05, 9.9998e-01],
        [5.8824e-02, 9.4118e-01],
        [2.7566e-04, 9.9972e-01],
        [5.2291e-04, 9.9948e-01],
        [9.9806e-01, 1.9379e-03],
        [5.7263e-06, 9.9999e-01],
        [1.9496e-04, 9.9981e-01],
        [8.0347e-03, 9.9197e-01],
        [3.8053e-04, 9.9962e-01],
        [2.8561e-06, 1.0000e+00],
        [9.9701e-01, 2.9939e-03],
        [7.2103e-05, 9.9993e-01],
        [1.7480e-03, 9.9825e-01],
        [9.9794e-01, 2.0642e-03],
        [9.4230e-01, 5.7703e-02],
        [2.9208e-05, 9.9997e-01],
        [8.8270e-04, 9.9912e-01],
        [3.4588e-05, 9.9997e-01],
        [8.7320e-02, 9.1268e-01],
        [5.8631e-04, 9.9941e-01],
        [6.9373e-01, 3.0627e-01],
        [5.1980e-02, 9.4802e-01],
        [1.9919e-04, 9.9980e-01],
        [1.1995e-03, 9.9880e-01],
        [2.1619e-03, 9.9784e-01],
        [1.3350e-02, 9.8665e-01],
        [1.4904e-05, 9.9999e-01],
        [5.2648e-05, 9.9995e-01],
        [9.9717e-01, 2.8290e-03],
        [9.4680e-01, 5.3196e-02],
        [9.8638e-01, 1.3623e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9589e-01, 4.1108e-03],
        [9.9993e-01, 7.3167e-05],
        [9.4968e-03, 9.9050e-01],
        [9.9755e-01, 2.4533e-03],
        [8.1098e-06, 9.9999e-01],
        [3.3682e-04, 9.9966e-01],
        [9.9729e-01, 2.7138e-03],
        [1.1368e-02, 9.8863e-01],
        [1.1867e-03, 9.9881e-01],
        [4.2024e-04, 9.9958e-01],
        [6.9961e-04, 9.9930e-01],
        [2.8733e-02, 9.7127e-01],
        [1.5091e-03, 9.9849e-01],
        [9.9888e-01, 1.1152e-03],
        [9.9945e-01, 5.4533e-04],
        [1.3078e-04, 9.9987e-01],
        [3.3474e-04, 9.9967e-01],
        [4.1862e-03, 9.9581e-01],
        [5.5679e-07, 1.0000e+00],
        [5.8103e-05, 9.9994e-01],
        [2.4697e-03, 9.9753e-01],
        [8.2503e-01, 1.7497e-01],
        [9.8023e-01, 1.9766e-02],
        [3.0537e-02, 9.6946e-01],
        [2.7581e-03, 9.9724e-01],
        [1.6678e-05, 9.9998e-01],
        [3.1379e-02, 9.6862e-01],
        [7.4588e-01, 2.5412e-01],
        [2.2147e-04, 9.9978e-01],
        [4.6853e-02, 9.5315e-01],
        [5.4348e-03, 9.9457e-01],
        [4.6637e-04, 9.9953e-01],
        [5.2791e-04, 9.9947e-01],
        [3.9274e-05, 9.9996e-01],
        [9.9313e-01, 6.8702e-03],
        [2.6492e-05, 9.9997e-01],
        [8.4294e-03, 9.9157e-01],
        [9.1131e-06, 9.9999e-01],
        [9.4530e-01, 5.4698e-02],
        [6.4153e-03, 9.9358e-01],
        [1.7083e-06, 1.0000e+00],
        [3.2006e-03, 9.9680e-01],
        [9.9876e-01, 1.2395e-03],
        [2.3254e-03, 9.9767e-01],
        [2.6308e-06, 1.0000e+00],
        [9.9496e-01, 5.0372e-03],
        [8.4975e-03, 9.9150e-01],
        [3.7158e-03, 9.9628e-01],
        [1.0916e-02, 9.8908e-01],
        [7.8073e-03, 9.9219e-01],
        [1.0100e-02, 9.8990e-01],
        [1.7863e-05, 9.9998e-01],
        [2.0298e-06, 1.0000e+00],
        [5.4857e-04, 9.9945e-01],
        [8.4398e-04, 9.9916e-01],
        [9.9525e-01, 4.7465e-03],
        [5.9371e-05, 9.9994e-01],
        [9.9858e-01, 1.4196e-03],
        [6.5710e-06, 9.9999e-01],
        [5.1217e-04, 9.9949e-01],
        [1.0512e-04, 9.9989e-01],
        [2.6287e-04, 9.9974e-01],
        [9.3978e-01, 6.0217e-02],
        [9.8010e-01, 1.9902e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3251e-03, 9.9867e-01],
        [4.2620e-05, 9.9996e-01],
        [9.9849e-01, 1.5067e-03],
        [9.9835e-01, 1.6525e-03],
        [6.3519e-02, 9.3648e-01],
        [9.8973e-01, 1.0274e-02],
        [4.0969e-04, 9.9959e-01],
        [8.2134e-03, 9.9179e-01],
        [6.8247e-01, 3.1753e-01],
        [4.9910e-05, 9.9995e-01],
        [9.5752e-01, 4.2478e-02],
        [8.5617e-06, 9.9999e-01],
        [3.2523e-06, 1.0000e+00],
        [9.9305e-01, 6.9482e-03],
        [4.4826e-05, 9.9996e-01],
        [3.6970e-03, 9.9630e-01],
        [8.3185e-04, 9.9917e-01],
        [9.9422e-01, 5.7768e-03],
        [1.5186e-02, 9.8481e-01],
        [4.8176e-07, 1.0000e+00],
        [4.0124e-04, 9.9960e-01],
        [1.3260e-04, 9.9987e-01],
        [8.5525e-01, 1.4475e-01],
        [3.2423e-06, 1.0000e+00],
        [8.7628e-01, 1.2372e-01],
        [1.4373e-05, 9.9999e-01],
        [1.2800e-05, 9.9999e-01],
        [1.9056e-05, 9.9998e-01],
        [8.3333e-03, 9.9167e-01],
        [6.1291e-05, 9.9994e-01],
        [9.9724e-01, 2.7594e-03],
        [3.0251e-04, 9.9970e-01],
        [9.9488e-01, 5.1232e-03],
        [8.9550e-03, 9.9104e-01],
        [1.1717e-05, 9.9999e-01],
        [9.9677e-01, 3.2341e-03],
        [2.5825e-03, 9.9742e-01],
        [7.5317e-06, 9.9999e-01],
        [4.0201e-04, 9.9960e-01],
        [3.2976e-04, 9.9967e-01],
        [9.9914e-01, 8.6459e-04],
        [9.9669e-01, 3.3072e-03],
        [1.8365e-03, 9.9816e-01],
        [2.2883e-04, 9.9977e-01],
        [9.8113e-01, 1.8873e-02],
        [3.1545e-03, 9.9685e-01],
        [7.5692e-02, 9.2431e-01],
        [3.9327e-06, 1.0000e+00],
        [2.1642e-04, 9.9978e-01],
        [1.2035e-02, 9.8796e-01],
        [1.8542e-02, 9.8146e-01],
        [9.8113e-01, 1.8867e-02],
        [9.9870e-01, 1.3040e-03],
        [2.5215e-05, 9.9997e-01],
        [3.6444e-02, 9.6356e-01],
        [9.9235e-01, 7.6460e-03],
        [1.5307e-03, 9.9847e-01],
        [9.4435e-01, 5.5648e-02],
        [7.5024e-01, 2.4976e-01],
        [3.7412e-04, 9.9963e-01],
        [6.0378e-03, 9.9396e-01],
        [2.6823e-03, 9.9732e-01],
        [1.5379e-02, 9.8462e-01],
        [7.5591e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8839e-01, 1.1612e-02],
        [9.9056e-01, 9.4385e-03],
        [1.8525e-04, 9.9981e-01],
        [9.9926e-01, 7.4318e-04],
        [1.3271e-05, 9.9999e-01],
        [9.9972e-01, 2.7789e-04],
        [8.4621e-06, 9.9999e-01],
        [9.9825e-01, 1.7540e-03],
        [9.9138e-01, 8.6150e-03],
        [2.1065e-02, 9.7894e-01],
        [2.9232e-04, 9.9971e-01],
        [8.0635e-04, 9.9919e-01],
        [2.4001e-02, 9.7600e-01],
        [1.4354e-04, 9.9986e-01],
        [9.9476e-01, 5.2387e-03],
        [5.8928e-06, 9.9999e-01],
        [9.5894e-04, 9.9904e-01],
        [9.9989e-01, 1.0581e-04],
        [9.6209e-01, 3.7913e-02],
        [9.8138e-01, 1.8623e-02],
        [7.2129e-03, 9.9279e-01],
        [3.7523e-03, 9.9625e-01],
        [9.9397e-01, 6.0305e-03],
        [9.6346e-02, 9.0365e-01],
        [1.8913e-03, 9.9811e-01],
        [8.2610e-06, 9.9999e-01],
        [8.8368e-06, 9.9999e-01],
        [9.9712e-01, 2.8787e-03],
        [6.8054e-04, 9.9932e-01],
        [7.8757e-06, 9.9999e-01],
        [3.4308e-01, 6.5692e-01],
        [3.5066e-04, 9.9965e-01],
        [7.9170e-05, 9.9992e-01],
        [4.4457e-03, 9.9555e-01],
        [9.8293e-01, 1.7065e-02],
        [5.7241e-07, 1.0000e+00],
        [9.9056e-01, 9.4401e-03],
        [9.9864e-01, 1.3645e-03],
        [3.6878e-03, 9.9631e-01],
        [1.1120e-02, 9.8888e-01],
        [9.9304e-01, 6.9644e-03],
        [9.2244e-01, 7.7558e-02],
        [8.0139e-01, 1.9861e-01],
        [9.5937e-05, 9.9990e-01],
        [1.6158e-04, 9.9984e-01],
        [5.9110e-06, 9.9999e-01],
        [6.4160e-03, 9.9358e-01],
        [9.9710e-01, 2.8983e-03],
        [5.7154e-02, 9.4285e-01],
        [1.0520e-02, 9.8948e-01],
        [8.2227e-04, 9.9918e-01],
        [2.2316e-02, 9.7768e-01],
        [3.6906e-07, 1.0000e+00],
        [1.2713e-04, 9.9987e-01],
        [2.5629e-04, 9.9974e-01],
        [5.5754e-06, 9.9999e-01],
        [9.9813e-01, 1.8713e-03],
        [8.0848e-07, 1.0000e+00],
        [1.2477e-03, 9.9875e-01],
        [6.3043e-01, 3.6957e-01],
        [9.9619e-05, 9.9990e-01],
        [1.0542e-02, 9.8946e-01],
        [2.3332e-03, 9.9767e-01],
        [9.8927e-01, 1.0726e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[6.8499e-05, 9.9993e-01],
        [9.9662e-01, 3.3843e-03],
        [1.9779e-05, 9.9998e-01],
        [3.7007e-02, 9.6299e-01],
        [1.8558e-04, 9.9981e-01],
        [1.2089e-05, 9.9999e-01],
        [3.4460e-03, 9.9655e-01],
        [9.4980e-01, 5.0203e-02],
        [2.7685e-01, 7.2315e-01],
        [4.6455e-02, 9.5355e-01],
        [1.8832e-02, 9.8117e-01],
        [5.8529e-02, 9.4147e-01],
        [9.9557e-01, 4.4339e-03],
        [9.9787e-01, 2.1293e-03],
        [1.7936e-04, 9.9982e-01],
        [1.2619e-02, 9.8738e-01],
        [8.5339e-03, 9.9147e-01],
        [9.0755e-03, 9.9092e-01],
        [7.2444e-05, 9.9993e-01],
        [2.4372e-03, 9.9756e-01],
        [5.6742e-02, 9.4326e-01],
        [8.2862e-01, 1.7138e-01],
        [9.8902e-01, 1.0976e-02],
        [8.1044e-04, 9.9919e-01],
        [2.9048e-02, 9.7095e-01],
        [9.9923e-03, 9.9001e-01],
        [7.4828e-05, 9.9993e-01],
        [5.7753e-02, 9.4225e-01],
        [7.1379e-03, 9.9286e-01],
        [3.9768e-03, 9.9602e-01],
        [9.9533e-01, 4.6671e-03],
        [9.9963e-01, 3.6972e-04],
        [9.9659e-01, 3.4093e-03],
        [3.7295e-04, 9.9963e-01],
        [6.4360e-04, 9.9936e-01],
        [2.8588e-04, 9.9971e-01],
        [1.3857e-06, 1.0000e+00],
        [2.9289e-03, 9.9707e-01],
        [2.0161e-03, 9.9798e-01],
        [9.9820e-01, 1.8003e-03],
        [9.9440e-01, 5.6050e-03],
        [7.7068e-03, 9.9229e-01],
        [2.0386e-02, 9.7961e-01],
        [6.8091e-01, 3.1909e-01],
        [1.8379e-03, 9.9816e-01],
        [1.1529e-02, 9.8847e-01],
        [8.4628e-04, 9.9915e-01],
        [5.6518e-03, 9.9435e-01],
        [5.2681e-03, 9.9473e-01],
        [5.2628e-02, 9.4737e-01],
        [1.8563e-02, 9.8144e-01],
        [1.7575e-02, 9.8242e-01],
        [9.9621e-01, 3.7900e-03],
        [1.7328e-05, 9.9998e-01],
        [1.2400e-03, 9.9876e-01],
        [5.7411e-05, 9.9994e-01],
        [1.1083e-02, 9.8892e-01],
        [2.6615e-03, 9.9734e-01],
        [5.1343e-04, 9.9949e-01],
        [9.9874e-01, 1.2555e-03],
        [2.0011e-05, 9.9998e-01],
        [5.3750e-04, 9.9946e-01],
        [9.9174e-01, 8.2635e-03],
        [1.5586e-02, 9.8441e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.6800e-03, 9.9632e-01],
        [9.9743e-01, 2.5702e-03],
        [9.9835e-01, 1.6465e-03],
        [5.8226e-04, 9.9942e-01],
        [1.9743e-02, 9.8026e-01],
        [6.3426e-03, 9.9366e-01],
        [2.5421e-03, 9.9746e-01],
        [1.0667e-04, 9.9989e-01],
        [6.7863e-03, 9.9321e-01],
        [5.5166e-06, 9.9999e-01],
        [3.5710e-02, 9.6429e-01],
        [1.3732e-03, 9.9863e-01],
        [7.4320e-05, 9.9993e-01],
        [1.0008e-05, 9.9999e-01],
        [5.8197e-05, 9.9994e-01],
        [2.2436e-03, 9.9776e-01],
        [3.4845e-05, 9.9997e-01],
        [4.6321e-03, 9.9537e-01],
        [7.2096e-05, 9.9993e-01],
        [9.9830e-01, 1.7040e-03],
        [9.8188e-01, 1.8118e-02],
        [2.6948e-03, 9.9731e-01],
        [7.4323e-04, 9.9926e-01],
        [4.6252e-03, 9.9537e-01],
        [9.5129e-03, 9.9049e-01],
        [2.3413e-03, 9.9766e-01],
        [1.1792e-01, 8.8208e-01],
        [1.3942e-03, 9.9861e-01],
        [3.4327e-04, 9.9966e-01],
        [1.1272e-12, 1.0000e+00],
        [1.2876e-03, 9.9871e-01],
        [1.1058e-05, 9.9999e-01],
        [8.7509e-06, 9.9999e-01],
        [3.7329e-05, 9.9996e-01],
        [1.8136e-02, 9.8186e-01],
        [1.4118e-05, 9.9999e-01],
        [7.1494e-04, 9.9929e-01],
        [4.3455e-04, 9.9957e-01],
        [6.0328e-05, 9.9994e-01],
        [2.2579e-01, 7.7421e-01],
        [7.3719e-03, 9.9263e-01],
        [4.7723e-06, 1.0000e+00],
        [3.5876e-06, 1.0000e+00],
        [5.3245e-03, 9.9468e-01],
        [2.6299e-02, 9.7370e-01],
        [1.7260e-02, 9.8274e-01],
        [1.3732e-03, 9.9863e-01],
        [1.9681e-01, 8.0319e-01],
        [2.1370e-02, 9.7863e-01],
        [1.0213e-05, 9.9999e-01],
        [2.1728e-03, 9.9783e-01],
        [3.7848e-05, 9.9996e-01],
        [2.0305e-03, 9.9797e-01],
        [3.5251e-02, 9.6475e-01],
        [9.8744e-01, 1.2560e-02],
        [1.3329e-05, 9.9999e-01],
        [9.9947e-01, 5.2984e-04],
        [3.4858e-04, 9.9965e-01],
        [2.7227e-04, 9.9973e-01],
        [3.5655e-01, 6.4345e-01],
        [1.4776e-03, 9.9852e-01],
        [7.4684e-03, 9.9253e-01],
        [1.2581e-03, 9.9874e-01],
        [9.9382e-01, 6.1782e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9895e-01, 1.0500e-03],
        [3.7651e-03, 9.9623e-01],
        [3.4581e-03, 9.9654e-01],
        [1.3372e-03, 9.9866e-01],
        [2.6953e-03, 9.9730e-01],
        [2.3032e-04, 9.9977e-01],
        [4.1650e-06, 1.0000e+00],
        [5.8302e-03, 9.9417e-01],
        [9.5608e-04, 9.9904e-01],
        [9.5296e-01, 4.7035e-02],
        [3.2877e-03, 9.9671e-01],
        [6.0174e-04, 9.9940e-01],
        [6.2905e-04, 9.9937e-01],
        [2.0053e-06, 1.0000e+00],
        [4.1271e-05, 9.9996e-01],
        [9.9948e-01, 5.2127e-04],
        [4.5919e-04, 9.9954e-01],
        [1.6281e-04, 9.9984e-01],
        [5.8021e-03, 9.9420e-01],
        [1.8778e-04, 9.9981e-01],
        [9.9779e-01, 2.2091e-03],
        [4.9382e-03, 9.9506e-01],
        [1.8059e-04, 9.9982e-01],
        [9.9816e-01, 1.8402e-03],
        [2.6397e-04, 9.9974e-01],
        [7.0775e-03, 9.9292e-01],
        [3.3051e-03, 9.9669e-01],
        [8.3157e-06, 9.9999e-01],
        [4.2757e-02, 9.5724e-01],
        [2.7575e-01, 7.2425e-01],
        [1.3429e-01, 8.6571e-01],
        [3.5293e-03, 9.9647e-01],
        [1.8965e-05, 9.9998e-01],
        [9.8171e-01, 1.8285e-02],
        [1.9454e-03, 9.9805e-01],
        [5.6257e-05, 9.9994e-01],
        [1.4547e-05, 9.9999e-01],
        [2.4064e-06, 1.0000e+00],
        [9.9721e-01, 2.7853e-03],
        [9.9840e-01, 1.6022e-03],
        [2.4833e-04, 9.9975e-01],
        [1.8839e-02, 9.8116e-01],
        [9.6769e-01, 3.2308e-02],
        [5.0612e-03, 9.9494e-01],
        [2.3383e-03, 9.9766e-01],
        [3.1333e-04, 9.9969e-01],
        [3.7059e-06, 1.0000e+00],
        [1.9571e-01, 8.0429e-01],
        [5.1956e-03, 9.9480e-01],
        [9.3448e-01, 6.5522e-02],
        [8.9599e-04, 9.9910e-01],
        [9.9695e-01, 3.0485e-03],
        [7.7343e-06, 9.9999e-01],
        [8.1651e-05, 9.9992e-01],
        [1.1117e-05, 9.9999e-01],
        [2.1070e-07, 1.0000e+00],
        [5.0879e-03, 9.9491e-01],
        [8.0711e-01, 1.9289e-01],
        [6.4496e-02, 9.3550e-01],
        [2.5806e-02, 9.7419e-01],
        [3.9144e-03, 9.9609e-01],
        [2.5357e-04, 9.9975e-01],
        [9.9912e-01, 8.8331e-04],
        [9.1308e-01, 8.6916e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0520e-02, 9.8948e-01],
        [9.3735e-03, 9.9063e-01],
        [8.9711e-01, 1.0289e-01],
        [1.1352e-03, 9.9886e-01],
        [1.2840e-05, 9.9999e-01],
        [5.3438e-06, 9.9999e-01],
        [7.5370e-01, 2.4630e-01],
        [1.1011e-03, 9.9890e-01],
        [1.0065e-04, 9.9990e-01],
        [7.3369e-05, 9.9993e-01],
        [3.0184e-01, 6.9816e-01],
        [9.9774e-01, 2.2642e-03],
        [1.8045e-02, 9.8195e-01],
        [3.2059e-05, 9.9997e-01],
        [1.7195e-01, 8.2805e-01],
        [1.0166e-02, 9.8983e-01],
        [7.4396e-03, 9.9256e-01],
        [1.5633e-04, 9.9984e-01],
        [7.4939e-04, 9.9925e-01],
        [3.3716e-03, 9.9663e-01],
        [9.9802e-01, 1.9802e-03],
        [1.3778e-03, 9.9862e-01],
        [9.9797e-01, 2.0316e-03],
        [1.0782e-03, 9.9892e-01],
        [4.4377e-05, 9.9996e-01],
        [1.6760e-04, 9.9983e-01],
        [9.2782e-01, 7.2177e-02],
        [4.2031e-05, 9.9996e-01],
        [1.6559e-05, 9.9998e-01],
        [1.4529e-03, 9.9855e-01],
        [9.7868e-01, 2.1324e-02],
        [1.4301e-02, 9.8570e-01],
        [9.9497e-01, 5.0301e-03],
        [8.2325e-03, 9.9177e-01],
        [1.0331e-04, 9.9990e-01],
        [9.9868e-01, 1.3169e-03],
        [3.5340e-04, 9.9965e-01],
        [1.5091e-03, 9.9849e-01],
        [8.2892e-02, 9.1711e-01],
        [1.6074e-04, 9.9984e-01],
        [1.9225e-01, 8.0775e-01],
        [5.4437e-04, 9.9946e-01],
        [3.7730e-01, 6.2270e-01],
        [4.8028e-04, 9.9952e-01],
        [2.7609e-05, 9.9997e-01],
        [4.9646e-03, 9.9504e-01],
        [1.0803e-04, 9.9989e-01],
        [7.6505e-05, 9.9992e-01],
        [1.1176e-04, 9.9989e-01],
        [2.5682e-03, 9.9743e-01],
        [9.9494e-01, 5.0582e-03],
        [1.1142e-04, 9.9989e-01],
        [2.9781e-01, 7.0219e-01],
        [8.5315e-05, 9.9991e-01],
        [1.2845e-04, 9.9987e-01],
        [9.8833e-01, 1.1671e-02],
        [1.7504e-03, 9.9825e-01],
        [1.5158e-04, 9.9985e-01],
        [1.6129e-04, 9.9984e-01],
        [1.9992e-03, 9.9800e-01],
        [8.7291e-04, 9.9913e-01],
        [5.7505e-01, 4.2495e-01],
        [9.9210e-01, 7.8967e-03],
        [9.9013e-01, 9.8735e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[3.1202e-03, 9.9688e-01],
        [2.2843e-03, 9.9772e-01],
        [9.9858e-01, 1.4244e-03],
        [9.9879e-01, 1.2103e-03],
        [6.1760e-03, 9.9382e-01],
        [9.9886e-01, 1.1413e-03],
        [1.1907e-05, 9.9999e-01],
        [5.3149e-04, 9.9947e-01],
        [1.8093e-02, 9.8191e-01],
        [6.3090e-07, 1.0000e+00],
        [8.9473e-01, 1.0527e-01],
        [8.8830e-01, 1.1170e-01],
        [8.1888e-03, 9.9181e-01],
        [9.9551e-01, 4.4881e-03],
        [3.6800e-04, 9.9963e-01],
        [7.8741e-01, 2.1259e-01],
        [9.9876e-01, 1.2354e-03],
        [2.7124e-03, 9.9729e-01],
        [6.9275e-06, 9.9999e-01],
        [7.0915e-04, 9.9929e-01],
        [9.9412e-01, 5.8768e-03],
        [8.9747e-06, 9.9999e-01],
        [3.0636e-05, 9.9997e-01],
        [5.6775e-05, 9.9994e-01],
        [1.7195e-02, 9.8280e-01],
        [4.0573e-03, 9.9594e-01],
        [6.3745e-01, 3.6255e-01],
        [9.9831e-01, 1.6915e-03],
        [3.7818e-03, 9.9622e-01],
        [2.9624e-02, 9.7038e-01],
        [1.9570e-04, 9.9980e-01],
        [5.9658e-02, 9.4034e-01],
        [8.7668e-02, 9.1233e-01],
        [3.0072e-04, 9.9970e-01],
        [9.9827e-01, 1.7284e-03],
        [9.9829e-01, 1.7086e-03],
        [9.8925e-01, 1.0746e-02],
        [9.9284e-01, 7.1553e-03],
        [9.9911e-01, 8.8875e-04],
        [1.2546e-02, 9.8745e-01],
        [9.9839e-01, 1.6126e-03],
        [8.0776e-01, 1.9224e-01],
        [8.4541e-04, 9.9915e-01],
        [1.0184e-05, 9.9999e-01],
        [4.8269e-04, 9.9952e-01],
        [1.9709e-06, 1.0000e+00],
        [3.0514e-05, 9.9997e-01],
        [9.9458e-01, 5.4162e-03],
        [3.0880e-05, 9.9997e-01],
        [9.9821e-01, 1.7865e-03],
        [1.0927e-03, 9.9891e-01],
        [9.9926e-01, 7.3697e-04],
        [5.2210e-03, 9.9478e-01],
        [6.9305e-05, 9.9993e-01],
        [6.9645e-03, 9.9304e-01],
        [3.1423e-06, 1.0000e+00],
        [5.5572e-02, 9.4443e-01],
        [3.2597e-04, 9.9967e-01],
        [5.0531e-03, 9.9495e-01],
        [3.7836e-05, 9.9996e-01],
        [7.6500e-03, 9.9235e-01],
        [2.4229e-03, 9.9758e-01],
        [3.3353e-03, 9.9666e-01],
        [1.6363e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1967e-05, 9.9999e-01],
        [9.9814e-01, 1.8595e-03],
        [1.7016e-03, 9.9830e-01],
        [9.4953e-01, 5.0471e-02],
        [5.7542e-05, 9.9994e-01],
        [6.6517e-06, 9.9999e-01],
        [6.2520e-01, 3.7480e-01],
        [1.1751e-05, 9.9999e-01],
        [9.8721e-01, 1.2791e-02],
        [6.4329e-01, 3.5671e-01],
        [2.4604e-03, 9.9754e-01],
        [1.7438e-06, 1.0000e+00],
        [1.4291e-04, 9.9986e-01],
        [9.8102e-01, 1.8977e-02],
        [9.9699e-01, 3.0128e-03],
        [3.3437e-02, 9.6656e-01],
        [5.0505e-03, 9.9495e-01],
        [1.0247e-04, 9.9990e-01],
        [5.0848e-04, 9.9949e-01],
        [9.2432e-04, 9.9908e-01],
        [1.4511e-03, 9.9855e-01],
        [8.8536e-01, 1.1464e-01],
        [1.4176e-03, 9.9858e-01],
        [9.4780e-05, 9.9991e-01],
        [1.0285e-06, 1.0000e+00],
        [9.9805e-01, 1.9539e-03],
        [7.6769e-04, 9.9923e-01],
        [1.9795e-03, 9.9802e-01],
        [6.2591e-01, 3.7409e-01],
        [9.9838e-01, 1.6159e-03],
        [2.1294e-02, 9.7871e-01],
        [9.8874e-01, 1.1256e-02],
        [7.4040e-01, 2.5960e-01],
        [7.2359e-01, 2.7641e-01],
        [1.3622e-03, 9.9864e-01],
        [9.9592e-01, 4.0767e-03],
        [1.7993e-03, 9.9820e-01],
        [3.7997e-05, 9.9996e-01],
        [4.7059e-05, 9.9995e-01],
        [7.2524e-06, 9.9999e-01],
        [1.4600e-04, 9.9985e-01],
        [8.3522e-04, 9.9916e-01],
        [1.0607e-01, 8.9393e-01],
        [9.8402e-01, 1.5984e-02],
        [1.8562e-03, 9.9814e-01],
        [8.8361e-04, 9.9912e-01],
        [1.2438e-05, 9.9999e-01],
        [3.6897e-05, 9.9996e-01],
        [9.9825e-01, 1.7456e-03],
        [1.2464e-01, 8.7536e-01],
        [7.9223e-05, 9.9992e-01],
        [5.0619e-07, 1.0000e+00],
        [2.6070e-05, 9.9997e-01],
        [1.5950e-01, 8.4050e-01],
        [2.4809e-03, 9.9752e-01],
        [9.5662e-01, 4.3377e-02],
        [5.9782e-06, 9.9999e-01],
        [1.6609e-04, 9.9983e-01],
        [1.0617e-04, 9.9989e-01],
        [5.8537e-06, 9.9999e-01],
        [1.3342e-04, 9.9987e-01],
        [1.7768e-04, 9.9982e-01],
        [6.5675e-03, 9.9343e-01],
        [2.7411e-04, 9.9973e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9825e-01, 1.7543e-03],
        [3.4352e-05, 9.9997e-01],
        [7.8225e-05, 9.9992e-01],
        [5.5032e-02, 9.4497e-01],
        [3.2194e-04, 9.9968e-01],
        [9.9850e-01, 1.5045e-03],
        [9.9962e-01, 3.7893e-04],
        [7.4431e-01, 2.5569e-01],
        [2.1263e-02, 9.7874e-01],
        [1.1569e-01, 8.8431e-01],
        [9.9390e-01, 6.0989e-03],
        [1.0279e-04, 9.9990e-01],
        [2.0437e-06, 1.0000e+00],
        [9.9469e-01, 5.3120e-03],
        [1.2064e-05, 9.9999e-01],
        [3.7306e-04, 9.9963e-01],
        [2.0617e-04, 9.9979e-01],
        [9.9801e-01, 1.9920e-03],
        [2.6963e-04, 9.9973e-01],
        [1.1127e-04, 9.9989e-01],
        [9.9628e-01, 3.7188e-03],
        [9.5052e-01, 4.9478e-02],
        [4.0753e-02, 9.5925e-01],
        [2.6083e-02, 9.7392e-01],
        [3.7120e-03, 9.9629e-01],
        [9.9647e-01, 3.5302e-03],
        [1.5763e-03, 9.9842e-01],
        [7.8445e-07, 1.0000e+00],
        [9.8196e-03, 9.9018e-01],
        [8.6660e-01, 1.3340e-01],
        [6.2563e-02, 9.3744e-01],
        [9.9933e-01, 6.6975e-04],
        [4.0987e-02, 9.5901e-01],
        [1.2261e-04, 9.9988e-01],
        [9.9901e-01, 9.8595e-04],
        [9.4308e-03, 9.9057e-01],
        [8.1254e-05, 9.9992e-01],
        [1.3119e-04, 9.9987e-01],
        [8.0972e-05, 9.9992e-01],
        [5.8608e-03, 9.9414e-01],
        [3.5604e-03, 9.9644e-01],
        [9.9461e-01, 5.3928e-03],
        [9.3104e-01, 6.8961e-02],
        [1.3369e-03, 9.9866e-01],
        [8.6292e-02, 9.1371e-01],
        [9.5306e-01, 4.6937e-02],
        [9.9842e-01, 1.5819e-03],
        [9.9478e-01, 5.2245e-03],
        [1.8506e-05, 9.9998e-01],
        [4.8630e-03, 9.9514e-01],
        [1.7830e-04, 9.9982e-01],
        [1.3589e-02, 9.8641e-01],
        [2.0912e-02, 9.7909e-01],
        [9.9499e-01, 5.0106e-03],
        [2.8120e-03, 9.9719e-01],
        [2.7030e-06, 1.0000e+00],
        [9.3517e-01, 6.4834e-02],
        [1.2812e-03, 9.9872e-01],
        [9.9823e-01, 1.7705e-03],
        [9.9025e-04, 9.9901e-01],
        [9.9914e-01, 8.6256e-04],
        [8.0946e-06, 9.9999e-01],
        [9.9832e-01, 1.6841e-03],
        [9.9814e-01, 1.8633e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6833e-04, 9.9983e-01],
        [9.9058e-01, 9.4171e-03],
        [7.2046e-03, 9.9280e-01],
        [1.0678e-06, 1.0000e+00],
        [8.8048e-03, 9.9120e-01],
        [9.9904e-01, 9.5859e-04],
        [3.3707e-03, 9.9663e-01],
        [9.0797e-02, 9.0920e-01],
        [9.0138e-06, 9.9999e-01],
        [6.0596e-08, 1.0000e+00],
        [1.6385e-03, 9.9836e-01],
        [6.7295e-02, 9.3270e-01],
        [4.4333e-05, 9.9996e-01],
        [7.9252e-05, 9.9992e-01],
        [1.6170e-03, 9.9838e-01],
        [3.9079e-04, 9.9961e-01],
        [3.0476e-02, 9.6952e-01],
        [9.6630e-01, 3.3701e-02],
        [3.2470e-03, 9.9675e-01],
        [1.7178e-03, 9.9828e-01],
        [1.1095e-02, 9.8891e-01],
        [4.0498e-04, 9.9959e-01],
        [9.7284e-01, 2.7158e-02],
        [2.6746e-04, 9.9973e-01],
        [3.6110e-04, 9.9964e-01],
        [9.9741e-01, 2.5932e-03],
        [4.4634e-06, 1.0000e+00],
        [3.4168e-03, 9.9658e-01],
        [3.5881e-01, 6.4119e-01],
        [9.8653e-01, 1.3469e-02],
        [3.5419e-04, 9.9965e-01],
        [2.0154e-03, 9.9798e-01],
        [1.3574e-04, 9.9986e-01],
        [5.2549e-04, 9.9947e-01],
        [1.4221e-04, 9.9986e-01],
        [1.0685e-02, 9.8931e-01],
        [4.2043e-01, 5.7957e-01],
        [1.3601e-01, 8.6399e-01],
        [2.1913e-03, 9.9781e-01],
        [6.5570e-03, 9.9344e-01],
        [1.4696e-04, 9.9985e-01],
        [2.5721e-03, 9.9743e-01],
        [5.4141e-05, 9.9995e-01],
        [8.1748e-01, 1.8252e-01],
        [2.4673e-03, 9.9753e-01],
        [1.3861e-03, 9.9861e-01],
        [1.3454e-05, 9.9999e-01],
        [9.9793e-01, 2.0700e-03],
        [1.3212e-03, 9.9868e-01],
        [5.4031e-09, 1.0000e+00],
        [3.9343e-02, 9.6066e-01],
        [9.7154e-01, 2.8465e-02],
        [1.2776e-03, 9.9872e-01],
        [4.9107e-05, 9.9995e-01],
        [5.8940e-01, 4.1060e-01],
        [1.2059e-02, 9.8794e-01],
        [3.6952e-04, 9.9963e-01],
        [1.6412e-05, 9.9998e-01],
        [2.7934e-04, 9.9972e-01],
        [9.9136e-01, 8.6367e-03],
        [1.0896e-04, 9.9989e-01],
        [1.7207e-03, 9.9828e-01],
        [9.8873e-01, 1.1266e-02],
        [1.0488e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0170e-04, 9.9990e-01],
        [4.9686e-02, 9.5031e-01],
        [9.9861e-01, 1.3890e-03],
        [5.8760e-05, 9.9994e-01],
        [9.9739e-01, 2.6056e-03],
        [9.9756e-01, 2.4394e-03],
        [9.9326e-01, 6.7435e-03],
        [1.3656e-04, 9.9986e-01],
        [9.9203e-01, 7.9729e-03],
        [5.8339e-02, 9.4166e-01],
        [3.6898e-03, 9.9631e-01],
        [1.9718e-04, 9.9980e-01],
        [3.0447e-05, 9.9997e-01],
        [4.6374e-06, 1.0000e+00],
        [6.7110e-08, 1.0000e+00],
        [1.9916e-05, 9.9998e-01],
        [1.5841e-02, 9.8416e-01],
        [4.1172e-04, 9.9959e-01],
        [9.9912e-01, 8.8329e-04],
        [4.1045e-04, 9.9959e-01],
        [9.9729e-01, 2.7101e-03],
        [1.0585e-04, 9.9989e-01],
        [2.9965e-04, 9.9970e-01],
        [1.5367e-03, 9.9846e-01],
        [5.7999e-05, 9.9994e-01],
        [1.0689e-02, 9.8931e-01],
        [4.7681e-04, 9.9952e-01],
        [9.8383e-01, 1.6175e-02],
        [8.0123e-01, 1.9877e-01],
        [9.9825e-01, 1.7468e-03],
        [9.9958e-01, 4.1792e-04],
        [5.9210e-04, 9.9941e-01],
        [9.8809e-01, 1.1915e-02],
        [9.9700e-01, 3.0022e-03],
        [1.4824e-02, 9.8518e-01],
        [3.2418e-04, 9.9968e-01],
        [7.9131e-04, 9.9921e-01],
        [2.7547e-06, 1.0000e+00],
        [7.8790e-01, 2.1210e-01],
        [1.0717e-03, 9.9893e-01],
        [1.9713e-05, 9.9998e-01],
        [9.2486e-01, 7.5138e-02],
        [1.2795e-03, 9.9872e-01],
        [7.7489e-04, 9.9923e-01],
        [4.1940e-03, 9.9581e-01],
        [9.9390e-01, 6.1042e-03],
        [7.5835e-06, 9.9999e-01],
        [9.5045e-01, 4.9549e-02],
        [1.7607e-02, 9.8239e-01],
        [2.7305e-06, 1.0000e+00],
        [4.6490e-05, 9.9995e-01],
        [9.9835e-01, 1.6474e-03],
        [5.4285e-04, 9.9946e-01],
        [3.6798e-01, 6.3202e-01],
        [6.6474e-04, 9.9934e-01],
        [3.7685e-03, 9.9623e-01],
        [2.5526e-03, 9.9745e-01],
        [2.3962e-03, 9.9760e-01],
        [1.0856e-03, 9.9891e-01],
        [1.3733e-04, 9.9986e-01],
        [6.5542e-04, 9.9934e-01],
        [1.9718e-03, 9.9803e-01],
        [9.5563e-01, 4.4368e-02],
        [2.8346e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.6382e-03, 9.9436e-01],
        [9.9959e-01, 4.1446e-04],
        [3.8128e-04, 9.9962e-01],
        [3.5766e-02, 9.6423e-01],
        [1.1057e-04, 9.9989e-01],
        [9.4013e-01, 5.9865e-02],
        [1.7755e-05, 9.9998e-01],
        [3.2448e-02, 9.6755e-01],
        [1.5766e-05, 9.9998e-01],
        [1.4608e-06, 1.0000e+00],
        [4.9505e-05, 9.9995e-01],
        [2.5102e-03, 9.9749e-01],
        [9.3629e-04, 9.9906e-01],
        [1.5751e-05, 9.9998e-01],
        [7.4752e-04, 9.9925e-01],
        [2.1855e-04, 9.9978e-01],
        [4.1341e-04, 9.9959e-01],
        [2.0825e-04, 9.9979e-01],
        [9.9837e-01, 1.6259e-03],
        [9.9826e-01, 1.7446e-03],
        [9.9840e-01, 1.6013e-03],
        [5.7417e-05, 9.9994e-01],
        [2.4587e-05, 9.9998e-01],
        [1.2782e-03, 9.9872e-01],
        [6.8308e-05, 9.9993e-01],
        [9.9865e-01, 1.3480e-03],
        [9.9864e-01, 1.3644e-03],
        [6.0268e-03, 9.9397e-01],
        [1.0874e-02, 9.8913e-01],
        [5.9641e-07, 1.0000e+00],
        [3.7385e-06, 1.0000e+00],
        [9.9868e-01, 1.3218e-03],
        [1.3370e-03, 9.9866e-01],
        [9.6549e-06, 9.9999e-01],
        [7.8564e-05, 9.9992e-01],
        [1.7612e-03, 9.9824e-01],
        [1.4447e-03, 9.9856e-01],
        [9.5992e-01, 4.0077e-02],
        [2.0456e-03, 9.9795e-01],
        [2.4903e-04, 9.9975e-01],
        [2.5314e-04, 9.9975e-01],
        [1.7527e-04, 9.9982e-01],
        [1.6220e-03, 9.9838e-01],
        [2.2524e-05, 9.9998e-01],
        [9.9988e-01, 1.2388e-04],
        [4.2243e-06, 1.0000e+00],
        [2.8896e-06, 1.0000e+00],
        [1.6483e-03, 9.9835e-01],
        [2.3972e-03, 9.9760e-01],
        [1.6315e-05, 9.9998e-01],
        [9.9803e-01, 1.9713e-03],
        [1.1020e-04, 9.9989e-01],
        [3.7599e-03, 9.9624e-01],
        [9.9912e-01, 8.8497e-04],
        [7.3692e-04, 9.9926e-01],
        [4.2335e-01, 5.7665e-01],
        [7.4726e-04, 9.9925e-01],
        [4.5395e-07, 1.0000e+00],
        [3.4900e-03, 9.9651e-01],
        [4.6023e-06, 1.0000e+00],
        [7.1975e-06, 9.9999e-01],
        [8.6904e-01, 1.3096e-01],
        [9.9718e-01, 2.8161e-03],
        [9.9816e-01, 1.8423e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2362e-04, 9.9958e-01],
        [8.5696e-01, 1.4304e-01],
        [6.5361e-04, 9.9935e-01],
        [9.9647e-01, 3.5260e-03],
        [5.4729e-04, 9.9945e-01],
        [1.2939e-01, 8.7061e-01],
        [1.8395e-05, 9.9998e-01],
        [1.1714e-05, 9.9999e-01],
        [1.0540e-03, 9.9895e-01],
        [2.9858e-06, 1.0000e+00],
        [4.7228e-05, 9.9995e-01],
        [2.2495e-03, 9.9775e-01],
        [5.8462e-01, 4.1538e-01],
        [7.5560e-04, 9.9924e-01],
        [9.3966e-01, 6.0339e-02],
        [5.7229e-03, 9.9428e-01],
        [6.0960e-06, 9.9999e-01],
        [8.3109e-05, 9.9992e-01],
        [9.1268e-05, 9.9991e-01],
        [3.0203e-05, 9.9997e-01],
        [7.3786e-03, 9.9262e-01],
        [7.1742e-02, 9.2826e-01],
        [5.3429e-07, 1.0000e+00],
        [9.9774e-01, 2.2559e-03],
        [2.9544e-04, 9.9970e-01],
        [6.8350e-04, 9.9932e-01],
        [2.3917e-05, 9.9998e-01],
        [9.9344e-01, 6.5569e-03],
        [3.3337e-03, 9.9667e-01],
        [8.8454e-01, 1.1546e-01],
        [8.7643e-04, 9.9912e-01],
        [2.6081e-02, 9.7392e-01],
        [9.4143e-04, 9.9906e-01],
        [4.8145e-01, 5.1855e-01],
        [7.3902e-03, 9.9261e-01],
        [1.9741e-05, 9.9998e-01],
        [6.6315e-02, 9.3368e-01],
        [7.8871e-06, 9.9999e-01],
        [9.8967e-01, 1.0330e-02],
        [1.6155e-05, 9.9998e-01],
        [3.4208e-06, 1.0000e+00],
        [2.1433e-05, 9.9998e-01],
        [6.0338e-03, 9.9397e-01],
        [6.7114e-03, 9.9329e-01],
        [1.1784e-07, 1.0000e+00],
        [1.6765e-03, 9.9832e-01],
        [1.0060e-03, 9.9899e-01],
        [9.2841e-01, 7.1593e-02],
        [4.5738e-05, 9.9995e-01],
        [3.1699e-04, 9.9968e-01],
        [9.9720e-01, 2.7998e-03],
        [3.4590e-05, 9.9997e-01],
        [1.3572e-02, 9.8643e-01],
        [5.0095e-07, 1.0000e+00],
        [4.0890e-05, 9.9996e-01],
        [5.3476e-05, 9.9995e-01],
        [1.6914e-03, 9.9831e-01],
        [1.0405e-02, 9.8960e-01],
        [1.3304e-05, 9.9999e-01],
        [9.8813e-01, 1.1869e-02],
        [3.5900e-02, 9.6410e-01],
        [9.9782e-01, 2.1776e-03],
        [5.1320e-03, 9.9487e-01],
        [5.8560e-02, 9.4144e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2670e-04, 9.9967e-01],
        [3.1906e-02, 9.6809e-01],
        [6.3093e-03, 9.9369e-01],
        [9.9951e-01, 4.8643e-04],
        [2.5906e-06, 1.0000e+00],
        [6.7584e-01, 3.2416e-01],
        [9.3998e-04, 9.9906e-01],
        [1.4304e-01, 8.5696e-01],
        [6.0435e-03, 9.9396e-01],
        [9.7001e-01, 2.9993e-02],
        [3.3289e-06, 1.0000e+00],
        [1.3336e-04, 9.9987e-01],
        [9.9361e-06, 9.9999e-01],
        [2.0858e-07, 1.0000e+00],
        [5.4949e-05, 9.9995e-01],
        [8.4551e-04, 9.9915e-01],
        [9.9908e-01, 9.2314e-04],
        [9.9912e-01, 8.8390e-04],
        [5.5380e-05, 9.9994e-01],
        [2.6643e-06, 1.0000e+00],
        [3.9317e-02, 9.6068e-01],
        [9.6828e-01, 3.1715e-02],
        [8.2635e-04, 9.9917e-01],
        [5.2690e-06, 9.9999e-01],
        [4.9026e-07, 1.0000e+00],
        [1.4974e-02, 9.8503e-01],
        [8.9390e-05, 9.9991e-01],
        [9.9091e-01, 9.0894e-03],
        [5.4722e-07, 1.0000e+00],
        [5.0272e-03, 9.9497e-01],
        [5.1962e-05, 9.9995e-01],
        [3.2174e-04, 9.9968e-01],
        [3.3690e-03, 9.9663e-01],
        [9.0441e-06, 9.9999e-01],
        [2.8228e-06, 1.0000e+00],
        [6.6943e-03, 9.9331e-01],
        [2.4568e-05, 9.9998e-01],
        [6.2241e-04, 9.9938e-01],
        [4.7711e-05, 9.9995e-01],
        [2.3125e-06, 1.0000e+00],
        [2.9289e-03, 9.9707e-01],
        [9.8138e-01, 1.8623e-02],
        [5.1393e-05, 9.9995e-01],
        [1.4335e-01, 8.5665e-01],
        [4.3663e-06, 1.0000e+00],
        [3.3905e-04, 9.9966e-01],
        [3.5112e-05, 9.9996e-01],
        [1.8171e-01, 8.1829e-01],
        [4.9930e-02, 9.5007e-01],
        [4.8083e-02, 9.5192e-01],
        [3.1008e-03, 9.9690e-01],
        [2.2888e-01, 7.7112e-01],
        [9.3612e-03, 9.9064e-01],
        [2.4237e-05, 9.9998e-01],
        [9.6609e-01, 3.3908e-02],
        [2.5891e-02, 9.7411e-01],
        [9.9646e-01, 3.5433e-03],
        [4.7398e-06, 1.0000e+00],
        [1.1908e-02, 9.8809e-01],
        [2.0300e-04, 9.9980e-01],
        [1.1779e-03, 9.9882e-01],
        [2.1847e-04, 9.9978e-01],
        [2.8289e-03, 9.9717e-01],
        [5.7526e-04, 9.9942e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2836e-04, 9.9987e-01],
        [4.9956e-03, 9.9500e-01],
        [9.8216e-01, 1.7842e-02],
        [9.1448e-05, 9.9991e-01],
        [8.0103e-03, 9.9199e-01],
        [2.1028e-03, 9.9790e-01],
        [2.2295e-02, 9.7770e-01],
        [1.4169e-04, 9.9986e-01],
        [4.6030e-04, 9.9954e-01],
        [9.6320e-01, 3.6801e-02],
        [6.9611e-01, 3.0389e-01],
        [1.8510e-05, 9.9998e-01],
        [1.1889e-03, 9.9881e-01],
        [5.3573e-01, 4.6427e-01],
        [6.9628e-01, 3.0372e-01],
        [4.7295e-05, 9.9995e-01],
        [9.9913e-01, 8.7076e-04],
        [4.8693e-05, 9.9995e-01],
        [1.2255e-05, 9.9999e-01],
        [6.1487e-01, 3.8513e-01],
        [3.0169e-04, 9.9970e-01],
        [1.4391e-03, 9.9856e-01],
        [9.9858e-01, 1.4201e-03],
        [1.2194e-03, 9.9878e-01],
        [1.1946e-06, 1.0000e+00],
        [9.9779e-01, 2.2113e-03],
        [9.2155e-01, 7.8446e-02],
        [1.3972e-02, 9.8603e-01],
        [8.0511e-06, 9.9999e-01],
        [6.5886e-04, 9.9934e-01],
        [6.9144e-04, 9.9931e-01],
        [2.1273e-03, 9.9787e-01],
        [9.9881e-01, 1.1882e-03],
        [3.6300e-07, 1.0000e+00],
        [9.9945e-01, 5.5259e-04],
        [9.5352e-01, 4.6475e-02],
        [1.8775e-01, 8.1225e-01],
        [1.1674e-03, 9.9883e-01],
        [4.0956e-05, 9.9996e-01],
        [4.4395e-01, 5.5605e-01],
        [9.8334e-01, 1.6655e-02],
        [9.8210e-01, 1.7903e-02],
        [3.8127e-06, 1.0000e+00],
        [2.2228e-03, 9.9778e-01],
        [6.8387e-06, 9.9999e-01],
        [4.4700e-04, 9.9955e-01],
        [9.7356e-01, 2.6441e-02],
        [3.1651e-04, 9.9968e-01],
        [1.0599e-03, 9.9894e-01],
        [1.0703e-03, 9.9893e-01],
        [6.2082e-03, 9.9379e-01],
        [2.4970e-04, 9.9975e-01],
        [6.5522e-06, 9.9999e-01],
        [1.5803e-05, 9.9998e-01],
        [6.1871e-06, 9.9999e-01],
        [1.0801e-04, 9.9989e-01],
        [1.4642e-04, 9.9985e-01],
        [1.1975e-04, 9.9988e-01],
        [1.6785e-03, 9.9832e-01],
        [7.4563e-03, 9.9254e-01],
        [9.4618e-05, 9.9991e-01],
        [6.2440e-05, 9.9994e-01],
        [6.8208e-01, 3.1792e-01],
        [1.8745e-03, 9.9813e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.0442e-06, 9.9999e-01],
        [6.3196e-06, 9.9999e-01],
        [1.9504e-03, 9.9805e-01],
        [3.6661e-04, 9.9963e-01],
        [1.4585e-05, 9.9999e-01],
        [2.5304e-05, 9.9997e-01],
        [1.2363e-06, 1.0000e+00],
        [2.0600e-04, 9.9979e-01],
        [1.0535e-04, 9.9989e-01],
        [1.2950e-05, 9.9999e-01],
        [5.2329e-01, 4.7671e-01],
        [9.9850e-01, 1.4994e-03],
        [9.9756e-01, 2.4351e-03],
        [9.9981e-01, 1.9098e-04],
        [5.9479e-04, 9.9941e-01],
        [9.9754e-01, 2.4556e-03],
        [9.8724e-01, 1.2762e-02],
        [8.7978e-05, 9.9991e-01],
        [2.2473e-04, 9.9978e-01],
        [1.0176e-03, 9.9898e-01],
        [9.4256e-03, 9.9057e-01],
        [2.2610e-04, 9.9977e-01],
        [4.1390e-04, 9.9959e-01],
        [7.1222e-05, 9.9993e-01],
        [9.7383e-01, 2.6166e-02],
        [4.4253e-05, 9.9996e-01],
        [1.4615e-03, 9.9854e-01],
        [5.2704e-04, 9.9947e-01],
        [4.1292e-04, 9.9959e-01],
        [1.7477e-06, 1.0000e+00],
        [2.7579e-06, 1.0000e+00],
        [5.3225e-04, 9.9947e-01],
        [9.9503e-01, 4.9699e-03],
        [5.5759e-05, 9.9994e-01],
        [8.5010e-01, 1.4990e-01],
        [1.5373e-05, 9.9998e-01],
        [2.1188e-02, 9.7881e-01],
        [6.7776e-02, 9.3222e-01],
        [1.4288e-05, 9.9999e-01],
        [6.4889e-03, 9.9351e-01],
        [7.1041e-05, 9.9993e-01],
        [2.1369e-03, 9.9786e-01],
        [3.0265e-05, 9.9997e-01],
        [9.5561e-01, 4.4394e-02],
        [8.1017e-04, 9.9919e-01],
        [1.3697e-07, 1.0000e+00],
        [2.4627e-02, 9.7537e-01],
        [1.4144e-02, 9.8586e-01],
        [7.7910e-01, 2.2090e-01],
        [7.6813e-06, 9.9999e-01],
        [9.9889e-01, 1.1104e-03],
        [1.1732e-03, 9.9883e-01],
        [2.1363e-04, 9.9979e-01],
        [9.9823e-01, 1.7696e-03],
        [2.8972e-04, 9.9971e-01],
        [4.0897e-04, 9.9959e-01],
        [2.6410e-03, 9.9736e-01],
        [1.2208e-04, 9.9988e-01],
        [2.5579e-05, 9.9997e-01],
        [2.5427e-02, 9.7457e-01],
        [9.6150e-01, 3.8503e-02],
        [9.8736e-01, 1.2645e-02],
        [9.9590e-01, 4.1021e-03],
        [9.3709e-05, 9.9991e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.4950e-05, 9.9993e-01],
        [9.9908e-01, 9.2062e-04],
        [2.2853e-04, 9.9977e-01],
        [2.6650e-04, 9.9973e-01],
        [7.5160e-05, 9.9992e-01],
        [2.6803e-05, 9.9997e-01],
        [1.7265e-06, 1.0000e+00],
        [1.5064e-01, 8.4936e-01],
        [1.6998e-03, 9.9830e-01],
        [3.4300e-06, 1.0000e+00],
        [9.8137e-07, 1.0000e+00],
        [1.5552e-02, 9.8445e-01],
        [2.1290e-04, 9.9979e-01],
        [2.4115e-01, 7.5885e-01],
        [1.1902e-06, 1.0000e+00],
        [9.9281e-01, 7.1920e-03],
        [3.6899e-03, 9.9631e-01],
        [4.0970e-03, 9.9590e-01],
        [9.9973e-01, 2.7340e-04],
        [8.5684e-01, 1.4316e-01],
        [9.9861e-01, 1.3922e-03],
        [4.9786e-04, 9.9950e-01],
        [7.4904e-07, 1.0000e+00],
        [1.5529e-04, 9.9984e-01],
        [9.9693e-01, 3.0672e-03],
        [1.3451e-02, 9.8655e-01],
        [8.0620e-02, 9.1938e-01],
        [6.4611e-02, 9.3539e-01],
        [2.3717e-06, 1.0000e+00],
        [9.9802e-01, 1.9756e-03],
        [3.5655e-05, 9.9996e-01],
        [8.6005e-01, 1.3995e-01],
        [1.0899e-05, 9.9999e-01],
        [5.9787e-03, 9.9402e-01],
        [2.3197e-02, 9.7680e-01],
        [1.6459e-03, 9.9835e-01],
        [9.7951e-01, 2.0491e-02],
        [1.3744e-04, 9.9986e-01],
        [9.9580e-01, 4.2042e-03],
        [2.6345e-03, 9.9737e-01],
        [5.3263e-02, 9.4674e-01],
        [1.5270e-03, 9.9847e-01],
        [8.1948e-05, 9.9992e-01],
        [1.2668e-04, 9.9987e-01],
        [9.8512e-01, 1.4878e-02],
        [9.6572e-01, 3.4280e-02],
        [8.6615e-01, 1.3385e-01],
        [1.6160e-03, 9.9838e-01],
        [2.7571e-05, 9.9997e-01],
        [3.1734e-05, 9.9997e-01],
        [9.7617e-01, 2.3832e-02],
        [1.0253e-05, 9.9999e-01],
        [7.4304e-03, 9.9257e-01],
        [9.9918e-01, 8.2021e-04],
        [2.8420e-01, 7.1580e-01],
        [8.9779e-02, 9.1022e-01],
        [2.7944e-01, 7.2056e-01],
        [1.0308e-05, 9.9999e-01],
        [9.7907e-01, 2.0931e-02],
        [5.0612e-01, 4.9388e-01],
        [9.9796e-01, 2.0431e-03],
        [8.0191e-05, 9.9992e-01],
        [2.5510e-02, 9.7449e-01],
        [1.7175e-03, 9.9828e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.6856e-04, 9.9963e-01],
        [2.4185e-04, 9.9976e-01],
        [1.5363e-03, 9.9846e-01],
        [4.3986e-07, 1.0000e+00],
        [1.6763e-04, 9.9983e-01],
        [4.9179e-04, 9.9951e-01],
        [2.3856e-03, 9.9761e-01],
        [1.0069e-04, 9.9990e-01],
        [9.9924e-01, 7.6233e-04],
        [1.3078e-06, 1.0000e+00],
        [7.8199e-04, 9.9922e-01],
        [1.8662e-04, 9.9981e-01],
        [6.6521e-04, 9.9933e-01],
        [6.9827e-03, 9.9302e-01],
        [9.9746e-01, 2.5351e-03],
        [9.4584e-06, 9.9999e-01],
        [4.8226e-01, 5.1774e-01],
        [9.9586e-01, 4.1364e-03],
        [7.5091e-01, 2.4909e-01],
        [9.1814e-05, 9.9991e-01],
        [2.9501e-04, 9.9971e-01],
        [6.7125e-04, 9.9933e-01],
        [5.5400e-01, 4.4600e-01],
        [1.9509e-04, 9.9980e-01],
        [9.9779e-01, 2.2070e-03],
        [1.7020e-03, 9.9830e-01],
        [4.1051e-03, 9.9589e-01],
        [3.5063e-04, 9.9965e-01],
        [4.3255e-02, 9.5675e-01],
        [1.2947e-04, 9.9987e-01],
        [1.7668e-03, 9.9823e-01],
        [4.9805e-01, 5.0195e-01],
        [2.2373e-05, 9.9998e-01],
        [9.9824e-01, 1.7612e-03],
        [2.8711e-01, 7.1289e-01],
        [9.9718e-01, 2.8244e-03],
        [9.5281e-03, 9.9047e-01],
        [1.3016e-03, 9.9870e-01],
        [5.6786e-01, 4.3214e-01],
        [9.6478e-01, 3.5217e-02],
        [9.9938e-01, 6.2005e-04],
        [1.3522e-01, 8.6478e-01],
        [1.0523e-03, 9.9895e-01],
        [1.0869e-02, 9.8913e-01],
        [1.7983e-04, 9.9982e-01],
        [3.2520e-03, 9.9675e-01],
        [1.1883e-04, 9.9988e-01],
        [9.9824e-01, 1.7583e-03],
        [1.4046e-03, 9.9860e-01],
        [8.3426e-03, 9.9166e-01],
        [2.0477e-02, 9.7952e-01],
        [8.2046e-06, 9.9999e-01],
        [1.0390e-05, 9.9999e-01],
        [3.7621e-05, 9.9996e-01],
        [2.3364e-02, 9.7664e-01],
        [3.0672e-03, 9.9693e-01],
        [1.2692e-04, 9.9987e-01],
        [9.9955e-01, 4.5289e-04],
        [3.8448e-05, 9.9996e-01],
        [9.9929e-01, 7.1276e-04],
        [6.5938e-01, 3.4062e-01],
        [3.1471e-04, 9.9969e-01],
        [3.9444e-05, 9.9996e-01],
        [9.2859e-01, 7.1408e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9974e-01, 2.6038e-04],
        [9.3603e-01, 6.3974e-02],
        [3.0254e-04, 9.9970e-01],
        [9.7476e-01, 2.5242e-02],
        [6.2799e-05, 9.9994e-01],
        [9.9422e-01, 5.7819e-03],
        [9.8322e-01, 1.6783e-02],
        [7.1218e-05, 9.9993e-01],
        [2.2705e-04, 9.9977e-01],
        [9.9474e-01, 5.2551e-03],
        [1.5064e-03, 9.9849e-01],
        [6.2067e-07, 1.0000e+00],
        [9.7101e-01, 2.8990e-02],
        [9.7039e-01, 2.9613e-02],
        [9.9704e-01, 2.9573e-03],
        [6.2681e-02, 9.3732e-01],
        [4.1664e-01, 5.8336e-01],
        [1.1330e-03, 9.9887e-01],
        [2.7356e-05, 9.9997e-01],
        [1.3788e-04, 9.9986e-01],
        [9.9877e-01, 1.2293e-03],
        [5.5845e-03, 9.9442e-01],
        [6.3005e-02, 9.3700e-01],
        [8.3871e-01, 1.6129e-01],
        [1.7203e-03, 9.9828e-01],
        [9.8934e-01, 1.0658e-02],
        [9.8525e-01, 1.4749e-02],
        [1.2556e-03, 9.9874e-01],
        [2.1546e-02, 9.7845e-01],
        [9.9647e-01, 3.5281e-03],
        [4.5821e-04, 9.9954e-01],
        [2.2901e-03, 9.9771e-01],
        [1.9233e-04, 9.9981e-01],
        [7.0236e-01, 2.9764e-01],
        [9.9868e-01, 1.3174e-03],
        [3.3439e-06, 1.0000e+00],
        [2.5582e-04, 9.9974e-01],
        [2.5177e-06, 1.0000e+00],
        [6.8343e-04, 9.9932e-01],
        [1.0565e-03, 9.9894e-01],
        [8.3899e-06, 9.9999e-01],
        [4.7506e-03, 9.9525e-01],
        [4.6312e-01, 5.3688e-01],
        [9.9674e-01, 3.2552e-03],
        [9.9958e-01, 4.1887e-04],
        [9.0618e-05, 9.9991e-01],
        [1.7411e-03, 9.9826e-01],
        [8.6469e-06, 9.9999e-01],
        [1.4138e-01, 8.5862e-01],
        [3.9739e-03, 9.9603e-01],
        [2.8058e-03, 9.9719e-01],
        [8.2345e-04, 9.9918e-01],
        [9.9966e-01, 3.4193e-04],
        [1.4965e-02, 9.8503e-01],
        [1.1293e-03, 9.9887e-01],
        [2.1632e-06, 1.0000e+00],
        [7.6936e-03, 9.9231e-01],
        [1.7362e-03, 9.9826e-01],
        [4.3644e-03, 9.9564e-01],
        [3.4513e-04, 9.9965e-01],
        [8.8714e-04, 9.9911e-01],
        [6.4767e-06, 9.9999e-01],
        [9.9949e-01, 5.0590e-04],
        [6.7774e-04, 9.9932e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0068e-01, 8.9932e-01],
        [3.7681e-05, 9.9996e-01],
        [1.1990e-03, 9.9880e-01],
        [1.3435e-03, 9.9866e-01],
        [9.5428e-01, 4.5719e-02],
        [1.0583e-03, 9.9894e-01],
        [3.2461e-02, 9.6754e-01],
        [9.9888e-01, 1.1218e-03],
        [2.8630e-05, 9.9997e-01],
        [1.3856e-04, 9.9986e-01],
        [4.6555e-05, 9.9995e-01],
        [3.4496e-04, 9.9966e-01],
        [4.9433e-05, 9.9995e-01],
        [7.1760e-03, 9.9282e-01],
        [2.6531e-04, 9.9973e-01],
        [1.2008e-05, 9.9999e-01],
        [9.6260e-01, 3.7401e-02],
        [5.2358e-06, 9.9999e-01],
        [9.9999e-01, 6.4807e-06],
        [7.3741e-03, 9.9263e-01],
        [1.8418e-03, 9.9816e-01],
        [9.4158e-01, 5.8424e-02],
        [2.9966e-04, 9.9970e-01],
        [1.5078e-02, 9.8492e-01],
        [9.9839e-01, 1.6106e-03],
        [9.9767e-01, 2.3302e-03],
        [2.3600e-03, 9.9764e-01],
        [1.0364e-03, 9.9896e-01],
        [2.8411e-01, 7.1589e-01],
        [6.8434e-04, 9.9932e-01],
        [1.1240e-01, 8.8760e-01],
        [8.3013e-05, 9.9992e-01],
        [3.8540e-05, 9.9996e-01],
        [3.3842e-02, 9.6616e-01],
        [1.2718e-02, 9.8728e-01],
        [1.6179e-05, 9.9998e-01],
        [2.0237e-03, 9.9798e-01],
        [9.8945e-01, 1.0549e-02],
        [9.9313e-04, 9.9901e-01],
        [1.0900e-04, 9.9989e-01],
        [6.9116e-01, 3.0884e-01],
        [1.9032e-04, 9.9981e-01],
        [5.6389e-02, 9.4361e-01],
        [9.9719e-01, 2.8141e-03],
        [4.2849e-06, 1.0000e+00],
        [1.2369e-05, 9.9999e-01],
        [1.0342e-04, 9.9990e-01],
        [1.5411e-03, 9.9846e-01],
        [9.7956e-01, 2.0438e-02],
        [6.3168e-04, 9.9937e-01],
        [5.5022e-04, 9.9945e-01],
        [9.4068e-01, 5.9324e-02],
        [9.9015e-01, 9.8544e-03],
        [2.3973e-03, 9.9760e-01],
        [9.4456e-01, 5.5442e-02],
        [6.8737e-01, 3.1263e-01],
        [9.8476e-01, 1.5242e-02],
        [1.0372e-02, 9.8963e-01],
        [9.9842e-01, 1.5841e-03],
        [9.6616e-01, 3.3842e-02],
        [5.0826e-04, 9.9949e-01],
        [9.9961e-01, 3.8986e-04],
        [6.0521e-05, 9.9994e-01],
        [2.5949e-01, 7.4051e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7599e-03, 9.9624e-01],
        [5.9894e-01, 4.0106e-01],
        [1.2333e-04, 9.9988e-01],
        [8.9584e-03, 9.9104e-01],
        [6.3496e-03, 9.9365e-01],
        [7.2463e-02, 9.2754e-01],
        [9.9766e-01, 2.3388e-03],
        [9.9064e-01, 9.3599e-03],
        [3.0366e-03, 9.9696e-01],
        [1.1310e-05, 9.9999e-01],
        [1.4246e-04, 9.9986e-01],
        [8.7505e-01, 1.2495e-01],
        [1.8844e-02, 9.8116e-01],
        [8.4036e-05, 9.9992e-01],
        [1.5544e-02, 9.8446e-01],
        [2.8337e-03, 9.9717e-01],
        [1.9108e-03, 9.9809e-01],
        [9.9654e-01, 3.4601e-03],
        [1.1180e-03, 9.9888e-01],
        [8.5699e-03, 9.9143e-01],
        [9.9845e-01, 1.5514e-03],
        [6.7280e-03, 9.9327e-01],
        [9.9954e-01, 4.5724e-04],
        [2.2672e-04, 9.9977e-01],
        [1.2819e-04, 9.9987e-01],
        [5.2946e-01, 4.7054e-01],
        [9.9757e-01, 2.4280e-03],
        [3.0327e-06, 1.0000e+00],
        [9.8600e-01, 1.3999e-02],
        [7.3033e-05, 9.9993e-01],
        [9.9645e-01, 3.5525e-03],
        [1.7492e-03, 9.9825e-01],
        [1.0234e-02, 9.8977e-01],
        [9.9896e-01, 1.0397e-03],
        [1.2487e-02, 9.8751e-01],
        [6.3013e-03, 9.9370e-01],
        [2.5113e-04, 9.9975e-01],
        [9.8116e-01, 1.8837e-02],
        [1.9275e-04, 9.9981e-01],
        [1.0078e-03, 9.9899e-01],
        [9.9788e-01, 2.1214e-03],
        [3.6067e-04, 9.9964e-01],
        [3.9773e-04, 9.9960e-01],
        [9.9873e-01, 1.2674e-03],
        [7.3435e-01, 2.6565e-01],
        [3.8239e-03, 9.9618e-01],
        [1.7748e-01, 8.2252e-01],
        [9.9826e-01, 1.7417e-03],
        [9.9549e-01, 4.5147e-03],
        [1.5429e-03, 9.9846e-01],
        [2.8229e-03, 9.9718e-01],
        [3.0424e-04, 9.9970e-01],
        [1.8749e-05, 9.9998e-01],
        [1.6282e-05, 9.9998e-01],
        [5.4858e-04, 9.9945e-01],
        [9.9837e-01, 1.6268e-03],
        [5.3602e-06, 9.9999e-01],
        [9.9010e-01, 9.9026e-03],
        [6.5343e-05, 9.9993e-01],
        [4.0574e-02, 9.5943e-01],
        [1.1169e-02, 9.8883e-01],
        [1.3137e-03, 9.9869e-01],
        [2.2321e-01, 7.7679e-01],
        [5.0828e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.0977e-02, 9.4902e-01],
        [7.2853e-04, 9.9927e-01],
        [1.2140e-02, 9.8786e-01],
        [2.4567e-05, 9.9998e-01],
        [3.5824e-06, 1.0000e+00],
        [4.5626e-06, 1.0000e+00],
        [3.1809e-04, 9.9968e-01],
        [3.6450e-03, 9.9635e-01],
        [2.2766e-06, 1.0000e+00],
        [1.1468e-04, 9.9989e-01],
        [2.6497e-01, 7.3503e-01],
        [9.9800e-01, 1.9951e-03],
        [9.9790e-01, 2.0978e-03],
        [5.6508e-04, 9.9943e-01],
        [8.9347e-06, 9.9999e-01],
        [9.8160e-01, 1.8401e-02],
        [2.2710e-04, 9.9977e-01],
        [4.7939e-02, 9.5206e-01],
        [1.0085e-02, 9.8991e-01],
        [4.0456e-06, 1.0000e+00],
        [3.5117e-05, 9.9996e-01],
        [2.7248e-04, 9.9973e-01],
        [4.6614e-02, 9.5339e-01],
        [9.9607e-01, 3.9337e-03],
        [1.7756e-04, 9.9982e-01],
        [4.1876e-04, 9.9958e-01],
        [3.0858e-03, 9.9691e-01],
        [2.5093e-04, 9.9975e-01],
        [1.6000e-02, 9.8400e-01],
        [9.9537e-01, 4.6282e-03],
        [9.9753e-01, 2.4738e-03],
        [2.6942e-03, 9.9731e-01],
        [9.3439e-02, 9.0656e-01],
        [1.4688e-04, 9.9985e-01],
        [9.9969e-01, 3.1263e-04],
        [9.7758e-03, 9.9022e-01],
        [7.4634e-05, 9.9993e-01],
        [2.0665e-04, 9.9979e-01],
        [8.3935e-05, 9.9992e-01],
        [2.2197e-05, 9.9998e-01],
        [2.3863e-01, 7.6137e-01],
        [9.9542e-01, 4.5832e-03],
        [9.9882e-01, 1.1772e-03],
        [1.7672e-04, 9.9982e-01],
        [1.1331e-05, 9.9999e-01],
        [5.4618e-01, 4.5382e-01],
        [1.9136e-03, 9.9809e-01],
        [3.2837e-03, 9.9672e-01],
        [4.9444e-04, 9.9951e-01],
        [5.3900e-03, 9.9461e-01],
        [1.8422e-03, 9.9816e-01],
        [4.7956e-04, 9.9952e-01],
        [9.9116e-01, 8.8365e-03],
        [2.6909e-06, 1.0000e+00],
        [7.0948e-06, 9.9999e-01],
        [3.2488e-06, 1.0000e+00],
        [4.6431e-04, 9.9954e-01],
        [1.4132e-03, 9.9859e-01],
        [9.7784e-03, 9.9022e-01],
        [3.2846e-04, 9.9967e-01],
        [2.2587e-02, 9.7741e-01],
        [1.0971e-05, 9.9999e-01],
        [5.9196e-05, 9.9994e-01],
        [6.7989e-04, 9.9932e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3340e-02, 9.7666e-01],
        [2.6223e-02, 9.7378e-01],
        [5.6216e-04, 9.9944e-01],
        [1.9856e-05, 9.9998e-01],
        [9.3364e-05, 9.9991e-01],
        [9.9774e-01, 2.2588e-03],
        [4.9063e-03, 9.9509e-01],
        [6.8572e-01, 3.1428e-01],
        [7.9864e-04, 9.9920e-01],
        [3.6443e-02, 9.6356e-01],
        [1.0415e-04, 9.9990e-01],
        [4.1930e-01, 5.8070e-01],
        [1.2029e-03, 9.9880e-01],
        [1.2425e-03, 9.9876e-01],
        [3.9675e-05, 9.9996e-01],
        [1.1359e-04, 9.9989e-01],
        [1.8811e-05, 9.9998e-01],
        [6.1246e-05, 9.9994e-01],
        [7.3163e-06, 9.9999e-01],
        [1.5745e-05, 9.9998e-01],
        [9.0673e-03, 9.9093e-01],
        [1.0784e-02, 9.8922e-01],
        [4.4158e-02, 9.5584e-01],
        [7.8809e-06, 9.9999e-01],
        [1.9033e-03, 9.9810e-01],
        [2.6881e-06, 1.0000e+00],
        [1.7233e-03, 9.9828e-01],
        [5.5799e-05, 9.9994e-01],
        [2.9978e-04, 9.9970e-01],
        [1.1833e-03, 9.9882e-01],
        [3.1583e-02, 9.6842e-01],
        [9.9511e-01, 4.8913e-03],
        [1.4861e-03, 9.9851e-01],
        [5.5196e-04, 9.9945e-01],
        [9.9773e-01, 2.2687e-03],
        [2.7238e-04, 9.9973e-01],
        [3.7328e-04, 9.9963e-01],
        [9.1676e-01, 8.3244e-02],
        [1.5746e-04, 9.9984e-01],
        [1.3341e-05, 9.9999e-01],
        [3.5156e-06, 1.0000e+00],
        [2.1047e-04, 9.9979e-01],
        [1.4318e-03, 9.9857e-01],
        [6.6041e-04, 9.9934e-01],
        [2.3026e-05, 9.9998e-01],
        [2.6800e-03, 9.9732e-01],
        [1.4641e-02, 9.8536e-01],
        [7.0808e-04, 9.9929e-01],
        [1.4987e-04, 9.9985e-01],
        [3.8961e-04, 9.9961e-01],
        [2.1184e-05, 9.9998e-01],
        [7.3526e-07, 1.0000e+00],
        [6.6880e-02, 9.3312e-01],
        [9.9816e-01, 1.8352e-03],
        [2.1954e-03, 9.9780e-01],
        [4.0925e-05, 9.9996e-01],
        [3.5355e-05, 9.9996e-01],
        [9.9822e-01, 1.7830e-03],
        [9.9917e-01, 8.2560e-04],
        [9.9407e-01, 5.9257e-03],
        [6.9878e-01, 3.0122e-01],
        [6.7124e-02, 9.3288e-01],
        [3.5402e-02, 9.6460e-01],
        [9.8966e-01, 1.0336e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[8.6537e-01, 1.3463e-01],
        [8.7134e-06, 9.9999e-01],
        [8.6743e-06, 9.9999e-01],
        [9.9886e-01, 1.1404e-03],
        [1.4460e-04, 9.9986e-01],
        [9.9847e-01, 1.5252e-03],
        [1.3590e-05, 9.9999e-01],
        [4.1199e-05, 9.9996e-01],
        [9.6928e-01, 3.0720e-02],
        [9.7303e-01, 2.6971e-02],
        [4.5168e-07, 1.0000e+00],
        [4.7288e-02, 9.5271e-01],
        [4.6541e-06, 1.0000e+00],
        [3.5256e-03, 9.9647e-01],
        [1.3156e-03, 9.9868e-01],
        [2.0500e-05, 9.9998e-01],
        [3.6147e-04, 9.9964e-01],
        [6.0713e-03, 9.9393e-01],
        [9.9978e-01, 2.1837e-04],
        [9.5583e-01, 4.4171e-02],
        [1.3765e-04, 9.9986e-01],
        [8.9253e-01, 1.0747e-01],
        [4.6182e-06, 1.0000e+00],
        [8.5036e-05, 9.9992e-01],
        [8.3037e-05, 9.9992e-01],
        [6.8310e-06, 9.9999e-01],
        [2.5856e-01, 7.4144e-01],
        [1.0605e-03, 9.9894e-01],
        [1.5387e-05, 9.9998e-01],
        [5.3116e-01, 4.6884e-01],
        [9.9530e-01, 4.6978e-03],
        [2.4738e-05, 9.9998e-01],
        [6.2404e-05, 9.9994e-01],
        [8.4460e-03, 9.9155e-01],
        [2.4899e-06, 1.0000e+00],
        [1.0858e-06, 1.0000e+00],
        [1.3930e-04, 9.9986e-01],
        [1.9656e-04, 9.9980e-01],
        [1.3901e-02, 9.8610e-01],
        [7.2465e-04, 9.9928e-01],
        [9.6313e-01, 3.6871e-02],
        [9.3153e-04, 9.9907e-01],
        [9.3917e-04, 9.9906e-01],
        [5.7840e-04, 9.9942e-01],
        [6.3012e-07, 1.0000e+00],
        [9.7016e-01, 2.9843e-02],
        [1.7394e-05, 9.9998e-01],
        [3.5405e-05, 9.9996e-01],
        [7.9770e-04, 9.9920e-01],
        [4.4331e-06, 1.0000e+00],
        [1.8492e-04, 9.9982e-01],
        [8.1006e-04, 9.9919e-01],
        [1.2953e-05, 9.9999e-01],
        [9.9669e-01, 3.3121e-03],
        [9.7032e-01, 2.9680e-02],
        [7.6775e-04, 9.9923e-01],
        [5.2797e-06, 9.9999e-01],
        [2.3962e-05, 9.9998e-01],
        [4.4955e-03, 9.9550e-01],
        [1.7889e-03, 9.9821e-01],
        [9.3513e-02, 9.0649e-01],
        [6.8156e-05, 9.9993e-01],
        [7.8540e-04, 9.9921e-01],
        [9.9708e-01, 2.9222e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5898e-02, 9.8410e-01],
        [7.6219e-01, 2.3781e-01],
        [3.1945e-03, 9.9681e-01],
        [9.6793e-01, 3.2073e-02],
        [7.1897e-06, 9.9999e-01],
        [9.6683e-03, 9.9033e-01],
        [9.9885e-01, 1.1532e-03],
        [3.6412e-06, 1.0000e+00],
        [6.2653e-05, 9.9994e-01],
        [2.0742e-04, 9.9979e-01],
        [7.9572e-01, 2.0428e-01],
        [2.9928e-01, 7.0072e-01],
        [9.9758e-01, 2.4228e-03],
        [4.2903e-03, 9.9571e-01],
        [9.9373e-01, 6.2657e-03],
        [1.8477e-06, 1.0000e+00],
        [1.3170e-03, 9.9868e-01],
        [9.8867e-01, 1.1328e-02],
        [5.1429e-06, 9.9999e-01],
        [9.9483e-01, 5.1665e-03],
        [1.4993e-03, 9.9850e-01],
        [5.9041e-04, 9.9941e-01],
        [1.6019e-02, 9.8398e-01],
        [8.5736e-04, 9.9914e-01],
        [9.9974e-01, 2.5680e-04],
        [1.1539e-03, 9.9885e-01],
        [1.6463e-04, 9.9984e-01],
        [1.3531e-05, 9.9999e-01],
        [2.3925e-04, 9.9976e-01],
        [2.4070e-05, 9.9998e-01],
        [9.9894e-01, 1.0604e-03],
        [1.8476e-03, 9.9815e-01],
        [3.5506e-05, 9.9996e-01],
        [9.9811e-01, 1.8887e-03],
        [1.9726e-07, 1.0000e+00],
        [8.8548e-03, 9.9115e-01],
        [4.2764e-05, 9.9996e-01],
        [9.1248e-05, 9.9991e-01],
        [7.3141e-04, 9.9927e-01],
        [4.8510e-06, 1.0000e+00],
        [9.9019e-01, 9.8099e-03],
        [1.4750e-03, 9.9853e-01],
        [1.6077e-05, 9.9998e-01],
        [2.2778e-04, 9.9977e-01],
        [1.7668e-04, 9.9982e-01],
        [9.0338e-06, 9.9999e-01],
        [5.6752e-06, 9.9999e-01],
        [1.9561e-05, 9.9998e-01],
        [5.2841e-04, 9.9947e-01],
        [9.3345e-01, 6.6548e-02],
        [2.6506e-02, 9.7349e-01],
        [1.4384e-03, 9.9856e-01],
        [9.9542e-01, 4.5793e-03],
        [8.6001e-07, 1.0000e+00],
        [9.9993e-01, 7.3994e-05],
        [2.2391e-03, 9.9776e-01],
        [9.9980e-01, 1.9520e-04],
        [4.3084e-04, 9.9957e-01],
        [6.0679e-05, 9.9994e-01],
        [3.9631e-06, 1.0000e+00],
        [9.9862e-01, 1.3774e-03],
        [9.9880e-01, 1.1989e-03],
        [6.4138e-03, 9.9359e-01],
        [3.1378e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.9426e-04, 9.9951e-01],
        [3.6935e-04, 9.9963e-01],
        [1.7275e-04, 9.9983e-01],
        [9.9241e-01, 7.5896e-03],
        [4.2098e-04, 9.9958e-01],
        [9.9828e-01, 1.7207e-03],
        [2.2545e-05, 9.9998e-01],
        [9.7082e-01, 2.9185e-02],
        [1.1629e-04, 9.9988e-01],
        [3.8355e-05, 9.9996e-01],
        [7.1432e-01, 2.8568e-01],
        [9.9865e-01, 1.3516e-03],
        [8.3354e-02, 9.1665e-01],
        [1.3355e-02, 9.8664e-01],
        [6.9351e-05, 9.9993e-01],
        [8.0348e-06, 9.9999e-01],
        [4.5717e-03, 9.9543e-01],
        [6.8087e-04, 9.9932e-01],
        [9.9630e-01, 3.7013e-03],
        [6.5733e-01, 3.4267e-01],
        [5.7511e-02, 9.4249e-01],
        [8.8371e-01, 1.1629e-01],
        [5.7406e-03, 9.9426e-01],
        [3.2917e-05, 9.9997e-01],
        [3.4227e-03, 9.9658e-01],
        [1.5392e-05, 9.9998e-01],
        [1.0815e-04, 9.9989e-01],
        [9.6997e-01, 3.0034e-02],
        [6.7456e-07, 1.0000e+00],
        [1.8804e-02, 9.8120e-01],
        [2.8839e-04, 9.9971e-01],
        [9.9991e-01, 9.2395e-05],
        [2.4756e-06, 1.0000e+00],
        [9.9991e-01, 9.1746e-05],
        [2.3380e-04, 9.9977e-01],
        [2.4659e-04, 9.9975e-01],
        [5.1626e-03, 9.9484e-01],
        [9.2879e-01, 7.1212e-02],
        [8.7176e-01, 1.2824e-01],
        [5.8776e-05, 9.9994e-01],
        [1.6613e-04, 9.9983e-01],
        [1.2684e-03, 9.9873e-01],
        [9.9950e-01, 4.9960e-04],
        [2.2063e-01, 7.7937e-01],
        [3.7013e-05, 9.9996e-01],
        [2.0888e-04, 9.9979e-01],
        [2.6811e-04, 9.9973e-01],
        [1.4486e-02, 9.8551e-01],
        [9.9970e-01, 3.0007e-04],
        [5.4747e-05, 9.9995e-01],
        [4.5972e-05, 9.9995e-01],
        [8.9120e-01, 1.0880e-01],
        [5.0769e-06, 9.9999e-01],
        [7.3952e-01, 2.6048e-01],
        [3.3015e-04, 9.9967e-01],
        [3.2411e-04, 9.9968e-01],
        [6.0707e-02, 9.3929e-01],
        [5.3433e-03, 9.9466e-01],
        [1.0561e-06, 1.0000e+00],
        [1.9346e-02, 9.8065e-01],
        [6.5230e-04, 9.9935e-01],
        [4.7626e-06, 1.0000e+00],
        [6.9359e-07, 1.0000e+00],
        [6.1159e-03, 9.9388e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.5312e-04, 9.9945e-01],
        [3.0144e-07, 1.0000e+00],
        [2.2146e-03, 9.9779e-01],
        [9.9856e-01, 1.4431e-03],
        [8.6474e-03, 9.9135e-01],
        [9.5255e-03, 9.9047e-01],
        [5.1208e-06, 9.9999e-01],
        [7.4866e-06, 9.9999e-01],
        [3.3538e-02, 9.6646e-01],
        [6.0696e-06, 9.9999e-01],
        [1.3163e-05, 9.9999e-01],
        [1.0309e-03, 9.9897e-01],
        [3.3421e-04, 9.9967e-01],
        [3.9101e-06, 1.0000e+00],
        [5.9086e-04, 9.9941e-01],
        [3.2290e-03, 9.9677e-01],
        [8.6891e-01, 1.3109e-01],
        [2.7815e-04, 9.9972e-01],
        [9.9494e-01, 5.0594e-03],
        [3.8545e-03, 9.9615e-01],
        [9.7430e-01, 2.5701e-02],
        [4.7290e-04, 9.9953e-01],
        [9.9805e-01, 1.9524e-03],
        [1.3262e-02, 9.8674e-01],
        [1.3386e-04, 9.9987e-01],
        [9.9335e-01, 6.6529e-03],
        [1.7350e-04, 9.9983e-01],
        [9.9779e-01, 2.2132e-03],
        [1.0887e-03, 9.9891e-01],
        [9.9877e-01, 1.2282e-03],
        [9.9631e-01, 3.6864e-03],
        [2.7410e-01, 7.2590e-01],
        [9.9950e-01, 5.0048e-04],
        [9.9417e-01, 5.8283e-03],
        [6.4262e-06, 9.9999e-01],
        [1.7058e-06, 1.0000e+00],
        [1.6684e-02, 9.8332e-01],
        [2.8775e-05, 9.9997e-01],
        [8.8990e-01, 1.1010e-01],
        [2.3549e-04, 9.9976e-01],
        [9.8471e-01, 1.5294e-02],
        [5.2962e-05, 9.9995e-01],
        [6.2931e-03, 9.9371e-01],
        [1.2113e-04, 9.9988e-01],
        [5.0633e-05, 9.9995e-01],
        [1.1080e-03, 9.9889e-01],
        [9.9967e-01, 3.2863e-04],
        [9.0363e-02, 9.0964e-01],
        [4.5362e-03, 9.9546e-01],
        [9.6824e-01, 3.1758e-02],
        [9.9555e-01, 4.4467e-03],
        [9.2809e-03, 9.9072e-01],
        [5.7707e-05, 9.9994e-01],
        [8.6598e-01, 1.3402e-01],
        [3.5888e-04, 9.9964e-01],
        [1.1920e-04, 9.9988e-01],
        [1.1307e-05, 9.9999e-01],
        [4.6341e-04, 9.9954e-01],
        [2.8340e-05, 9.9997e-01],
        [3.2460e-05, 9.9997e-01],
        [1.8895e-06, 1.0000e+00],
        [1.3802e-04, 9.9986e-01],
        [6.2998e-02, 9.3700e-01],
        [1.8060e-03, 9.9819e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9910e-01, 9.0442e-04],
        [1.4982e-04, 9.9985e-01],
        [2.5374e-06, 1.0000e+00],
        [8.4619e-01, 1.5381e-01],
        [9.1014e-01, 8.9856e-02],
        [3.9942e-03, 9.9601e-01],
        [7.2313e-05, 9.9993e-01],
        [9.9512e-06, 9.9999e-01],
        [2.2839e-02, 9.7716e-01],
        [9.9999e-01, 8.4319e-06],
        [9.9965e-01, 3.4915e-04],
        [9.5090e-03, 9.9049e-01],
        [3.5139e-02, 9.6486e-01],
        [5.8018e-05, 9.9994e-01],
        [2.0686e-05, 9.9998e-01],
        [5.0014e-04, 9.9950e-01],
        [9.9898e-01, 1.0171e-03],
        [5.8406e-04, 9.9942e-01],
        [3.5756e-03, 9.9642e-01],
        [1.2655e-03, 9.9873e-01],
        [1.5470e-07, 1.0000e+00],
        [5.9321e-03, 9.9407e-01],
        [4.5535e-03, 9.9545e-01],
        [1.2818e-05, 9.9999e-01],
        [4.1027e-04, 9.9959e-01],
        [3.8451e-02, 9.6155e-01],
        [1.8145e-04, 9.9982e-01],
        [9.9114e-01, 8.8634e-03],
        [1.0843e-01, 8.9157e-01],
        [2.5049e-04, 9.9975e-01],
        [5.8059e-04, 9.9942e-01],
        [4.1826e-03, 9.9582e-01],
        [8.9815e-01, 1.0185e-01],
        [1.5978e-06, 1.0000e+00],
        [2.9592e-05, 9.9997e-01],
        [2.7062e-05, 9.9997e-01],
        [5.5121e-05, 9.9994e-01],
        [1.4263e-03, 9.9857e-01],
        [3.2246e-04, 9.9968e-01],
        [9.9979e-01, 2.1200e-04],
        [5.5373e-05, 9.9994e-01],
        [9.8642e-01, 1.3581e-02],
        [6.3883e-04, 9.9936e-01],
        [1.7354e-06, 1.0000e+00],
        [6.9215e-03, 9.9308e-01],
        [1.5972e-03, 9.9840e-01],
        [9.9892e-01, 1.0767e-03],
        [5.9403e-04, 9.9941e-01],
        [3.2997e-04, 9.9967e-01],
        [9.9979e-01, 2.1169e-04],
        [1.7900e-05, 9.9998e-01],
        [7.8023e-01, 2.1977e-01],
        [9.9910e-01, 9.0196e-04],
        [2.0630e-04, 9.9979e-01],
        [1.4415e-05, 9.9999e-01],
        [1.1028e-01, 8.8972e-01],
        [1.3185e-02, 9.8681e-01],
        [5.8619e-05, 9.9994e-01],
        [1.8045e-02, 9.8196e-01],
        [9.9829e-01, 1.7089e-03],
        [9.9697e-01, 3.0286e-03],
        [6.9929e-05, 9.9993e-01],
        [3.1950e-05, 9.9997e-01],
        [7.7660e-04, 9.9922e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5710e-06, 1.0000e+00],
        [6.7192e-05, 9.9993e-01],
        [8.0521e-05, 9.9992e-01],
        [2.0150e-05, 9.9998e-01],
        [8.7688e-01, 1.2312e-01],
        [9.9985e-01, 1.5371e-04],
        [1.0800e-01, 8.9200e-01],
        [9.3224e-05, 9.9991e-01],
        [9.4267e-05, 9.9991e-01],
        [9.7445e-01, 2.5553e-02],
        [9.5027e-01, 4.9726e-02],
        [2.6100e-06, 1.0000e+00],
        [5.9194e-01, 4.0806e-01],
        [7.5546e-01, 2.4454e-01],
        [4.1046e-04, 9.9959e-01],
        [2.0471e-03, 9.9795e-01],
        [9.9798e-01, 2.0248e-03],
        [4.0886e-04, 9.9959e-01],
        [1.1479e-03, 9.9885e-01],
        [9.9457e-01, 5.4300e-03],
        [1.8745e-03, 9.9813e-01],
        [1.9961e-03, 9.9800e-01],
        [6.7373e-05, 9.9993e-01],
        [9.4653e-01, 5.3469e-02],
        [4.0737e-03, 9.9593e-01],
        [9.0847e-01, 9.1532e-02],
        [2.8689e-05, 9.9997e-01],
        [9.5776e-01, 4.2238e-02],
        [7.7600e-06, 9.9999e-01],
        [1.5140e-02, 9.8486e-01],
        [3.8398e-02, 9.6160e-01],
        [1.0698e-04, 9.9989e-01],
        [9.9968e-01, 3.2248e-04],
        [1.0581e-02, 9.8942e-01],
        [9.9251e-01, 7.4856e-03],
        [2.0919e-02, 9.7908e-01],
        [3.6454e-02, 9.6355e-01],
        [2.2007e-03, 9.9780e-01],
        [9.6899e-05, 9.9990e-01],
        [3.1852e-02, 9.6815e-01],
        [1.1919e-04, 9.9988e-01],
        [6.8010e-05, 9.9993e-01],
        [5.1817e-05, 9.9995e-01],
        [5.3677e-04, 9.9946e-01],
        [1.8265e-04, 9.9982e-01],
        [1.5538e-06, 1.0000e+00],
        [1.5267e-03, 9.9847e-01],
        [3.5873e-02, 9.6413e-01],
        [9.9204e-01, 7.9622e-03],
        [9.9806e-01, 1.9438e-03],
        [9.3119e-01, 6.8808e-02],
        [1.5795e-02, 9.8421e-01],
        [6.0001e-04, 9.9940e-01],
        [7.3462e-03, 9.9265e-01],
        [3.7701e-05, 9.9996e-01],
        [2.1463e-05, 9.9998e-01],
        [3.7978e-04, 9.9962e-01],
        [2.1114e-02, 9.7889e-01],
        [2.1937e-04, 9.9978e-01],
        [2.9433e-04, 9.9971e-01],
        [9.9805e-01, 1.9542e-03],
        [1.5127e-02, 9.8487e-01],
        [9.9792e-01, 2.0773e-03],
        [2.0680e-03, 9.9793e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1477e-01, 7.8523e-01],
        [9.9152e-01, 8.4772e-03],
        [9.9956e-01, 4.4153e-04],
        [5.7395e-06, 9.9999e-01],
        [9.7473e-01, 2.5271e-02],
        [1.8002e-04, 9.9982e-01],
        [1.6507e-02, 9.8349e-01],
        [9.8391e-01, 1.6089e-02],
        [9.9976e-01, 2.3533e-04],
        [9.7900e-01, 2.0997e-02],
        [9.0393e-05, 9.9991e-01],
        [9.9853e-01, 1.4748e-03],
        [4.1313e-05, 9.9996e-01],
        [9.9845e-01, 1.5534e-03],
        [3.7937e-04, 9.9962e-01],
        [4.9651e-04, 9.9950e-01],
        [3.9345e-04, 9.9961e-01],
        [8.3272e-01, 1.6728e-01],
        [9.9747e-01, 2.5277e-03],
        [9.9032e-01, 9.6835e-03],
        [6.6966e-05, 9.9993e-01],
        [2.7778e-03, 9.9722e-01],
        [8.2251e-04, 9.9918e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3875e-03, 9.9761e-01],
        [6.4837e-03, 9.9352e-01],
        [3.9014e-03, 9.9610e-01],
        [9.3779e-01, 6.2214e-02],
        [2.5996e-03, 9.9740e-01],
        [9.1278e-01, 8.7224e-02],
        [7.9255e-04, 9.9921e-01],
        [9.9067e-01, 9.3350e-03],
        [2.7579e-03, 9.9724e-01],
        [1.3307e-03, 9.9867e-01],
        [2.9210e-05, 9.9997e-01],
        [9.8480e-01, 1.5200e-02],
        [3.8560e-05, 9.9996e-01],
        [9.9897e-01, 1.0344e-03],
        [1.0203e-03, 9.9898e-01],
        [6.7831e-02, 9.3217e-01],
        [4.2433e-03, 9.9576e-01],
        [5.9349e-05, 9.9994e-01],
        [2.9922e-04, 9.9970e-01],
        [4.5863e-06, 1.0000e+00],
        [3.8378e-04, 9.9962e-01],
        [1.5488e-06, 1.0000e+00],
        [4.5087e-03, 9.9549e-01],
        [1.3124e-02, 9.8688e-01],
        [6.1856e-03, 9.9381e-01],
        [9.9771e-01, 2.2866e-03],
        [1.4541e-01, 8.5459e-01],
        [1.7153e-03, 9.9828e-01],
        [9.7802e-01, 2.1976e-02],
        [2.1104e-03, 9.9789e-01],
        [1.2174e-03, 9.9878e-01],
        [9.9840e-01, 1.5994e-03],
        [6.8353e-01, 3.1647e-01],
        [3.8505e-02, 9.6150e-01],
        [9.6184e-01, 3.8157e-02],
        [2.5204e-04, 9.9975e-01],
        [1.0874e-01, 8.9126e-01],
        [2.7500e-02, 9.7250e-01],
        [6.1999e-06, 9.9999e-01],
        [9.3796e-06, 9.9999e-01],
        [5.6010e-03, 9.9440e-01],
        [1.0082e-03, 9.9899e-01],
        [1.6214e-04, 9.9984e-01],
        [3.8377e-07, 1.0000e+00],
        [9.9741e-01, 2.5938e-03],
        [7.4790e-04, 9.9925e-01],
        [7.7517e-01, 2.2483e-01],
        [1.3088e-02, 9.8691e-01],
        [8.3643e-04, 9.9916e-01],
        [9.5813e-01, 4.1874e-02],
        [6.8780e-04, 9.9931e-01],
        [3.0543e-02, 9.6946e-01],
        [3.2900e-04, 9.9967e-01],
        [1.4048e-05, 9.9999e-01],
        [9.9661e-01, 3.3912e-03],
        [4.8265e-03, 9.9517e-01],
        [6.0599e-06, 9.9999e-01],
        [6.4898e-05, 9.9994e-01],
        [9.9997e-01, 2.8389e-05],
        [4.3788e-05, 9.9996e-01],
        [1.1218e-03, 9.9888e-01],
        [5.2981e-06, 9.9999e-01],
        [2.3160e-01, 7.6840e-01],
        [8.6250e-04, 9.9914e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7813e-03, 9.9822e-01],
        [4.8624e-02, 9.5138e-01],
        [3.2005e-03, 9.9680e-01],
        [9.9964e-01, 3.6368e-04],
        [3.1550e-06, 1.0000e+00],
        [5.2597e-02, 9.4740e-01],
        [8.6918e-04, 9.9913e-01],
        [1.0449e-04, 9.9990e-01],
        [6.4603e-01, 3.5397e-01],
        [3.2979e-08, 1.0000e+00],
        [1.9942e-03, 9.9801e-01],
        [2.5668e-01, 7.4332e-01],
        [4.1292e-03, 9.9587e-01],
        [1.4449e-03, 9.9856e-01],
        [9.8617e-01, 1.3834e-02],
        [6.4936e-03, 9.9351e-01],
        [1.5824e-04, 9.9984e-01],
        [9.9449e-01, 5.5139e-03],
        [8.0404e-05, 9.9992e-01],
        [9.9994e-01, 6.2514e-05],
        [5.8366e-04, 9.9942e-01],
        [2.0779e-05, 9.9998e-01],
        [3.0291e-04, 9.9970e-01],
        [2.1956e-03, 9.9780e-01],
        [2.6749e-02, 9.7325e-01],
        [9.9417e-01, 5.8314e-03],
        [5.7642e-05, 9.9994e-01],
        [3.1844e-03, 9.9682e-01],
        [1.9910e-03, 9.9801e-01],
        [9.1835e-01, 8.1654e-02],
        [4.5810e-05, 9.9995e-01],
        [9.6206e-01, 3.7944e-02],
        [1.6964e-02, 9.8304e-01],
        [4.1792e-03, 9.9582e-01],
        [2.6428e-04, 9.9974e-01],
        [8.2574e-03, 9.9174e-01],
        [1.2152e-04, 9.9988e-01],
        [7.8099e-01, 2.1901e-01],
        [2.1666e-03, 9.9783e-01],
        [2.5604e-01, 7.4396e-01],
        [5.9960e-03, 9.9400e-01],
        [3.3139e-01, 6.6861e-01],
        [1.9951e-04, 9.9980e-01],
        [5.6808e-04, 9.9943e-01],
        [1.7166e-05, 9.9998e-01],
        [4.6382e-04, 9.9954e-01],
        [6.0415e-02, 9.3958e-01],
        [9.9777e-01, 2.2251e-03],
        [1.1996e-01, 8.8004e-01],
        [1.6836e-03, 9.9832e-01],
        [4.9357e-02, 9.5064e-01],
        [3.0423e-01, 6.9577e-01],
        [9.1340e-06, 9.9999e-01],
        [6.8600e-01, 3.1400e-01],
        [2.9559e-05, 9.9997e-01],
        [2.3559e-04, 9.9976e-01],
        [9.9944e-01, 5.5658e-04],
        [3.9563e-06, 1.0000e+00],
        [1.3394e-02, 9.8661e-01],
        [1.4278e-02, 9.8572e-01],
        [1.3298e-05, 9.9999e-01],
        [6.0965e-05, 9.9994e-01],
        [9.9081e-01, 9.1942e-03],
        [7.2525e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1742e-02, 9.7826e-01],
        [7.3526e-04, 9.9926e-01],
        [1.3838e-01, 8.6162e-01],
        [1.5071e-03, 9.9849e-01],
        [5.5141e-04, 9.9945e-01],
        [3.4078e-04, 9.9966e-01],
        [7.3342e-02, 9.2666e-01],
        [4.4605e-05, 9.9996e-01],
        [6.7600e-05, 9.9993e-01],
        [6.1502e-05, 9.9994e-01],
        [1.7712e-05, 9.9998e-01],
        [9.3784e-03, 9.9062e-01],
        [4.0491e-03, 9.9595e-01],
        [3.0116e-04, 9.9970e-01],
        [4.4223e-05, 9.9996e-01],
        [5.5809e-02, 9.4419e-01],
        [1.0135e-03, 9.9899e-01],
        [9.9963e-01, 3.7284e-04],
        [7.6789e-02, 9.2321e-01],
        [4.4616e-01, 5.5384e-01],
        [1.1351e-04, 9.9989e-01],
        [4.2080e-04, 9.9958e-01],
        [4.0049e-03, 9.9600e-01],
        [3.1095e-03, 9.9689e-01],
        [3.5449e-06, 1.0000e+00],
        [4.3819e-01, 5.6181e-01],
        [9.9764e-01, 2.3599e-03],
        [9.2418e-01, 7.5820e-02],
        [9.9889e-01, 1.1057e-03],
        [9.1998e-06, 9.9999e-01],
        [1.2618e-04, 9.9987e-01],
        [1.2263e-01, 8.7737e-01],
        [1.5463e-04, 9.9985e-01],
        [4.5455e-05, 9.9995e-01],
        [7.0226e-05, 9.9993e-01],
        [1.3886e-01, 8.6114e-01],
        [9.9793e-01, 2.0663e-03],
        [1.0507e-05, 9.9999e-01],
        [7.3138e-01, 2.6862e-01],
        [1.9391e-05, 9.9998e-01],
        [1.0979e-03, 9.9890e-01],
        [6.7367e-04, 9.9933e-01],
        [8.7612e-04, 9.9912e-01],
        [4.5926e-05, 9.9995e-01],
        [1.3028e-03, 9.9870e-01],
        [9.6562e-03, 9.9034e-01],
        [9.9783e-01, 2.1746e-03],
        [1.3889e-04, 9.9986e-01],
        [9.9756e-01, 2.4370e-03],
        [5.4627e-04, 9.9945e-01],
        [4.3065e-03, 9.9569e-01],
        [9.9981e-01, 1.9315e-04],
        [1.2406e-04, 9.9988e-01],
        [9.9915e-01, 8.5247e-04],
        [1.5734e-04, 9.9984e-01],
        [1.2330e-03, 9.9877e-01],
        [3.3947e-03, 9.9661e-01],
        [3.1129e-01, 6.8871e-01],
        [8.4383e-04, 9.9916e-01],
        [2.0406e-05, 9.9998e-01],
        [3.9534e-05, 9.9996e-01],
        [1.7659e-03, 9.9823e-01],
        [1.9763e-02, 9.8024e-01],
        [1.4785e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.9848e-03, 9.9502e-01],
        [1.3195e-03, 9.9868e-01],
        [1.8806e-02, 9.8119e-01],
        [7.2221e-07, 1.0000e+00],
        [9.9949e-01, 5.1055e-04],
        [5.6940e-01, 4.3060e-01],
        [1.4228e-02, 9.8577e-01],
        [7.1031e-05, 9.9993e-01],
        [9.9891e-01, 1.0907e-03],
        [6.5305e-06, 9.9999e-01],
        [8.0408e-02, 9.1959e-01],
        [1.3645e-07, 1.0000e+00],
        [1.0984e-03, 9.9890e-01],
        [4.4938e-04, 9.9955e-01],
        [5.7364e-05, 9.9994e-01],
        [9.9874e-01, 1.2646e-03],
        [1.5949e-03, 9.9841e-01],
        [9.9879e-03, 9.9001e-01],
        [4.4308e-03, 9.9557e-01],
        [2.8849e-05, 9.9997e-01],
        [9.9564e-01, 4.3596e-03],
        [1.5363e-04, 9.9985e-01],
        [6.8914e-04, 9.9931e-01],
        [6.2616e-05, 9.9994e-01],
        [5.5761e-05, 9.9994e-01],
        [2.9156e-04, 9.9971e-01],
        [9.9819e-01, 1.8136e-03],
        [2.0753e-02, 9.7925e-01],
        [9.8583e-01, 1.4173e-02],
        [2.8307e-01, 7.1693e-01],
        [1.1899e-07, 1.0000e+00],
        [7.5069e-04, 9.9925e-01],
        [9.9808e-01, 1.9244e-03],
        [2.4401e-06, 1.0000e+00],
        [2.5780e-04, 9.9974e-01],
        [1.0964e-03, 9.9890e-01],
        [2.5244e-01, 7.4756e-01],
        [4.3505e-01, 5.6495e-01],
        [5.2387e-03, 9.9476e-01],
        [6.9270e-02, 9.3073e-01],
        [1.1039e-03, 9.9890e-01],
        [1.3342e-03, 9.9867e-01],
        [9.9824e-01, 1.7565e-03],
        [7.5362e-04, 9.9925e-01],
        [1.8103e-03, 9.9819e-01],
        [9.9949e-01, 5.1029e-04],
        [9.5154e-01, 4.8458e-02],
        [9.7273e-01, 2.7274e-02],
        [1.0758e-05, 9.9999e-01],
        [3.7070e-04, 9.9963e-01],
        [9.3012e-06, 9.9999e-01],
        [8.0573e-03, 9.9194e-01],
        [8.2732e-03, 9.9173e-01],
        [3.1873e-04, 9.9968e-01],
        [4.7245e-03, 9.9528e-01],
        [4.6487e-05, 9.9995e-01],
        [9.1496e-02, 9.0850e-01],
        [8.9735e-01, 1.0265e-01],
        [2.2700e-04, 9.9977e-01],
        [7.9957e-04, 9.9920e-01],
        [9.9506e-01, 4.9448e-03],
        [1.8205e-05, 9.9998e-01],
        [2.5391e-04, 9.9975e-01],
        [1.4796e-03, 9.9852e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5090e-03, 9.9849e-01],
        [2.5409e-06, 1.0000e+00],
        [9.9892e-01, 1.0823e-03],
        [1.1460e-02, 9.8854e-01],
        [1.9296e-02, 9.8070e-01],
        [2.2132e-04, 9.9978e-01],
        [2.0764e-05, 9.9998e-01],
        [6.8206e-03, 9.9318e-01],
        [6.1844e-04, 9.9938e-01],
        [5.0337e-04, 9.9950e-01],
        [2.1359e-04, 9.9979e-01],
        [9.2969e-03, 9.9070e-01],
        [3.0589e-03, 9.9694e-01],
        [1.5082e-03, 9.9849e-01],
        [8.8278e-04, 9.9912e-01],
        [7.5996e-03, 9.9240e-01],
        [3.8300e-05, 9.9996e-01],
        [2.6372e-03, 9.9736e-01],
        [1.5503e-03, 9.9845e-01],
        [1.7210e-02, 9.8279e-01],
        [1.5841e-01, 8.4159e-01],
        [9.8525e-01, 1.4754e-02],
        [7.0232e-05, 9.9993e-01],
        [2.6695e-01, 7.3305e-01],
        [3.5595e-04, 9.9964e-01],
        [4.5805e-04, 9.9954e-01],
        [8.1814e-03, 9.9182e-01],
        [6.7597e-02, 9.3240e-01],
        [1.0166e-05, 9.9999e-01],
        [1.5928e-04, 9.9984e-01],
        [1.0095e-03, 9.9899e-01],
        [9.9716e-03, 9.9003e-01],
        [4.6023e-06, 1.0000e+00],
        [6.8919e-01, 3.1081e-01],
        [3.3262e-04, 9.9967e-01],
        [8.4674e-06, 9.9999e-01],
        [3.6753e-03, 9.9632e-01],
        [1.1719e-04, 9.9988e-01],
        [1.5871e-03, 9.9841e-01],
        [9.9643e-01, 3.5670e-03],
        [2.2833e-03, 9.9772e-01],
        [4.8205e-04, 9.9952e-01],
        [7.9815e-03, 9.9202e-01],
        [1.2820e-02, 9.8718e-01],
        [9.9403e-01, 5.9725e-03],
        [1.6545e-02, 9.8345e-01],
        [6.4996e-05, 9.9994e-01],
        [1.1277e-03, 9.9887e-01],
        [2.3221e-05, 9.9998e-01],
        [9.8244e-01, 1.7564e-02],
        [3.3455e-06, 1.0000e+00],
        [2.4512e-01, 7.5488e-01],
        [9.9906e-01, 9.4127e-04],
        [6.1029e-04, 9.9939e-01],
        [1.3260e-02, 9.8674e-01],
        [1.3509e-03, 9.9865e-01],
        [1.7671e-05, 9.9998e-01],
        [2.4606e-04, 9.9975e-01],
        [9.9432e-01, 5.6758e-03],
        [9.9612e-01, 3.8818e-03],
        [3.4503e-04, 9.9966e-01],
        [8.2872e-03, 9.9171e-01],
        [8.9580e-04, 9.9910e-01],
        [9.9839e-01, 1.6117e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7050e-04, 9.9983e-01],
        [1.9598e-02, 9.8040e-01],
        [1.6801e-04, 9.9983e-01],
        [7.4547e-03, 9.9255e-01],
        [4.9161e-05, 9.9995e-01],
        [5.3795e-02, 9.4621e-01],
        [7.0773e-03, 9.9292e-01],
        [1.9771e-04, 9.9980e-01],
        [3.6710e-05, 9.9996e-01],
        [1.3276e-03, 9.9867e-01],
        [5.1004e-04, 9.9949e-01],
        [9.9854e-01, 1.4617e-03],
        [1.4034e-03, 9.9860e-01],
        [4.3680e-06, 1.0000e+00],
        [2.7493e-03, 9.9725e-01],
        [1.1702e-03, 9.9883e-01],
        [8.9870e-03, 9.9101e-01],
        [9.1691e-03, 9.9083e-01],
        [2.5641e-03, 9.9744e-01],
        [2.5908e-04, 9.9974e-01],
        [9.9793e-01, 2.0687e-03],
        [9.9186e-01, 8.1416e-03],
        [1.7889e-03, 9.9821e-01],
        [2.3976e-05, 9.9998e-01],
        [9.9244e-01, 7.5587e-03],
        [3.2015e-02, 9.6798e-01],
        [9.5009e-01, 4.9911e-02],
        [9.9565e-01, 4.3489e-03],
        [4.8406e-01, 5.1594e-01],
        [8.7063e-01, 1.2937e-01],
        [6.3075e-03, 9.9369e-01],
        [1.3488e-05, 9.9999e-01],
        [9.9821e-01, 1.7937e-03],
        [2.3777e-05, 9.9998e-01],
        [2.9514e-03, 9.9705e-01],
        [3.5573e-04, 9.9964e-01],
        [9.7239e-06, 9.9999e-01],
        [9.9012e-01, 9.8793e-03],
        [1.4808e-05, 9.9999e-01],
        [1.7581e-01, 8.2419e-01],
        [9.8728e-01, 1.2715e-02],
        [9.2677e-06, 9.9999e-01],
        [2.2945e-04, 9.9977e-01],
        [3.8562e-04, 9.9961e-01],
        [9.6844e-01, 3.1556e-02],
        [1.1822e-03, 9.9882e-01],
        [9.8901e-01, 1.0991e-02],
        [9.1738e-02, 9.0826e-01],
        [9.9453e-01, 5.4704e-03],
        [9.9837e-01, 1.6320e-03],
        [8.6523e-01, 1.3477e-01],
        [9.9886e-01, 1.1389e-03],
        [9.3404e-01, 6.5955e-02],
        [1.4723e-03, 9.9853e-01],
        [1.4225e-04, 9.9986e-01],
        [3.8254e-04, 9.9962e-01],
        [2.3051e-04, 9.9977e-01],
        [3.4368e-05, 9.9997e-01],
        [7.3981e-04, 9.9926e-01],
        [1.6930e-03, 9.9831e-01],
        [2.7075e-02, 9.7293e-01],
        [1.4045e-03, 9.9860e-01],
        [5.7266e-04, 9.9943e-01],
        [4.9078e-03, 9.9509e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9013e-01, 9.8736e-03],
        [2.0794e-03, 9.9792e-01],
        [9.9036e-01, 9.6398e-03],
        [1.5126e-02, 9.8487e-01],
        [9.9839e-01, 1.6132e-03],
        [1.0340e-03, 9.9897e-01],
        [1.8783e-03, 9.9812e-01],
        [9.9950e-01, 5.0203e-04],
        [9.9050e-01, 9.5007e-03],
        [3.2818e-05, 9.9997e-01],
        [2.2509e-03, 9.9775e-01],
        [2.0060e-03, 9.9799e-01],
        [1.0519e-04, 9.9989e-01],
        [6.5064e-04, 9.9935e-01],
        [1.7866e-03, 9.9821e-01],
        [9.9861e-01, 1.3892e-03],
        [1.6183e-06, 1.0000e+00],
        [9.9929e-01, 7.0920e-04],
        [5.0496e-06, 9.9999e-01],
        [3.7782e-01, 6.2218e-01],
        [1.6607e-07, 1.0000e+00],
        [2.2602e-04, 9.9977e-01],
        [6.4442e-02, 9.3556e-01],
        [5.8469e-02, 9.4153e-01],
        [3.7485e-02, 9.6251e-01],
        [4.3462e-06, 1.0000e+00],
        [3.1644e-04, 9.9968e-01],
        [9.1640e-01, 8.3602e-02],
        [1.2652e-05, 9.9999e-01],
        [4.1635e-02, 9.5836e-01],
        [9.4924e-01, 5.0761e-02],
        [2.3853e-05, 9.9998e-01],
        [9.1018e-04, 9.9909e-01],
        [4.0342e-05, 9.9996e-01],
        [2.3082e-05, 9.9998e-01],
        [3.4212e-04, 9.9966e-01],
        [5.0269e-03, 9.9497e-01],
        [7.7386e-05, 9.9992e-01],
        [1.0519e-04, 9.9989e-01],
        [6.3946e-04, 9.9936e-01],
        [2.1869e-04, 9.9978e-01],
        [1.0444e-05, 9.9999e-01],
        [8.9414e-07, 1.0000e+00],
        [1.8608e-03, 9.9814e-01],
        [1.8749e-06, 1.0000e+00],
        [9.9810e-01, 1.8993e-03],
        [1.0357e-05, 9.9999e-01],
        [4.9074e-04, 9.9951e-01],
        [4.7587e-02, 9.5241e-01],
        [6.1194e-04, 9.9939e-01],
        [3.6750e-03, 9.9632e-01],
        [2.3122e-05, 9.9998e-01],
        [1.0124e-03, 9.9899e-01],
        [4.6129e-06, 1.0000e+00],
        [5.1620e-06, 9.9999e-01],
        [1.1386e-02, 9.8861e-01],
        [9.9479e-01, 5.2082e-03],
        [4.6527e-06, 1.0000e+00],
        [3.1011e-06, 1.0000e+00],
        [2.0193e-06, 1.0000e+00],
        [2.1191e-01, 7.8809e-01],
        [2.7486e-01, 7.2514e-01],
        [1.6531e-05, 9.9998e-01],
        [2.2211e-04, 9.9978e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2290e-07, 1.0000e+00],
        [1.4431e-03, 9.9856e-01],
        [4.1172e-06, 1.0000e+00],
        [4.7975e-06, 1.0000e+00],
        [7.3338e-05, 9.9993e-01],
        [7.7875e-05, 9.9992e-01],
        [3.1915e-06, 1.0000e+00],
        [8.8985e-06, 9.9999e-01],
        [9.9614e-01, 3.8585e-03],
        [3.1392e-02, 9.6861e-01],
        [5.5206e-05, 9.9994e-01],
        [2.9524e-06, 1.0000e+00],
        [9.7832e-01, 2.1676e-02],
        [8.6286e-01, 1.3714e-01],
        [4.0267e-06, 1.0000e+00],
        [5.1493e-03, 9.9485e-01],
        [5.3196e-06, 9.9999e-01],
        [9.9991e-01, 8.7225e-05],
        [9.8795e-01, 1.2052e-02],
        [3.6057e-01, 6.3943e-01],
        [2.4115e-04, 9.9976e-01],
        [8.3942e-05, 9.9992e-01],
        [2.1765e-03, 9.9782e-01],
        [1.7604e-04, 9.9982e-01],
        [3.5130e-06, 1.0000e+00],
        [8.4417e-04, 9.9916e-01],
        [4.0316e-05, 9.9996e-01],
        [9.6914e-04, 9.9903e-01],
        [6.4984e-06, 9.9999e-01],
        [9.9933e-01, 6.6782e-04],
        [3.7983e-05, 9.9996e-01],
        [9.9882e-01, 1.1785e-03],
        [1.3890e-03, 9.9861e-01],
        [1.4185e-05, 9.9999e-01],
        [8.0349e-04, 9.9920e-01],
        [9.0731e-04, 9.9909e-01],
        [1.8436e-03, 9.9816e-01],
        [9.1496e-05, 9.9991e-01],
        [1.1756e-02, 9.8824e-01],
        [1.0865e-03, 9.9891e-01],
        [3.1538e-04, 9.9968e-01],
        [2.0880e-03, 9.9791e-01],
        [4.5130e-04, 9.9955e-01],
        [9.9637e-01, 3.6324e-03],
        [9.9734e-01, 2.6584e-03],
        [4.0722e-03, 9.9593e-01],
        [9.5106e-06, 9.9999e-01],
        [9.9996e-01, 3.9291e-05],
        [2.2839e-05, 9.9998e-01],
        [9.9767e-01, 2.3316e-03],
        [5.8837e-02, 9.4116e-01],
        [7.3148e-04, 9.9927e-01],
        [9.9454e-01, 5.4624e-03],
        [2.3819e-05, 9.9998e-01],
        [2.3633e-02, 9.7637e-01],
        [2.1103e-05, 9.9998e-01],
        [9.9847e-01, 1.5315e-03],
        [3.5405e-04, 9.9965e-01],
        [9.9596e-01, 4.0379e-03],
        [9.9950e-01, 5.0349e-04],
        [4.7878e-06, 1.0000e+00],
        [9.0730e-03, 9.9093e-01],
        [4.4601e-05, 9.9996e-01],
        [3.2734e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3723e-03, 9.9863e-01],
        [2.0664e-06, 1.0000e+00],
        [6.5605e-04, 9.9934e-01],
        [9.9941e-01, 5.9180e-04],
        [1.3102e-03, 9.9869e-01],
        [9.2190e-06, 9.9999e-01],
        [2.0218e-05, 9.9998e-01],
        [1.7297e-03, 9.9827e-01],
        [1.9701e-03, 9.9803e-01],
        [2.1009e-02, 9.7899e-01],
        [2.3488e-07, 1.0000e+00],
        [1.9875e-05, 9.9998e-01],
        [5.5359e-07, 1.0000e+00],
        [2.2652e-06, 1.0000e+00],
        [4.9048e-06, 1.0000e+00],
        [9.8584e-01, 1.4162e-02],
        [2.4212e-06, 1.0000e+00],
        [2.5707e-05, 9.9997e-01],
        [2.8092e-04, 9.9972e-01],
        [5.0908e-04, 9.9949e-01],
        [1.3294e-03, 9.9867e-01],
        [9.5546e-01, 4.4539e-02],
        [1.4052e-02, 9.8595e-01],
        [1.7707e-05, 9.9998e-01],
        [1.0053e-05, 9.9999e-01],
        [5.4597e-05, 9.9995e-01],
        [9.9887e-01, 1.1330e-03],
        [8.6693e-04, 9.9913e-01],
        [9.9496e-09, 1.0000e+00],
        [3.1095e-05, 9.9997e-01],
        [1.8651e-04, 9.9981e-01],
        [1.0609e-05, 9.9999e-01],
        [4.9757e-05, 9.9995e-01],
        [7.7602e-01, 2.2398e-01],
        [2.2911e-06, 1.0000e+00],
        [2.1635e-03, 9.9784e-01],
        [1.3377e-06, 1.0000e+00],
        [2.3492e-02, 9.7651e-01],
        [2.3862e-01, 7.6138e-01],
        [8.1386e-04, 9.9919e-01],
        [5.7982e-06, 9.9999e-01],
        [4.8367e-06, 1.0000e+00],
        [9.8859e-01, 1.1405e-02],
        [5.8499e-02, 9.4150e-01],
        [2.6734e-05, 9.9997e-01],
        [1.7375e-07, 1.0000e+00],
        [1.5989e-04, 9.9984e-01],
        [2.4080e-06, 1.0000e+00],
        [5.1456e-06, 9.9999e-01],
        [2.3613e-06, 1.0000e+00],
        [9.9963e-01, 3.6627e-04],
        [9.9879e-01, 1.2093e-03],
        [9.9802e-01, 1.9822e-03],
        [2.9003e-03, 9.9710e-01],
        [3.4448e-06, 1.0000e+00],
        [3.7904e-02, 9.6210e-01],
        [2.7992e-03, 9.9720e-01],
        [1.0368e-04, 9.9990e-01],
        [7.0845e-06, 9.9999e-01],
        [6.9791e-02, 9.3021e-01],
        [9.9651e-01, 3.4871e-03],
        [2.4917e-07, 1.0000e+00],
        [2.6313e-04, 9.9974e-01],
        [9.8234e-01, 1.7664e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5560e-05, 9.9995e-01],
        [2.3542e-04, 9.9976e-01],
        [9.9662e-04, 9.9900e-01],
        [7.4064e-02, 9.2594e-01],
        [9.7879e-01, 2.1208e-02],
        [8.3728e-02, 9.1627e-01],
        [6.4298e-01, 3.5702e-01],
        [7.0957e-07, 1.0000e+00],
        [1.2513e-03, 9.9875e-01],
        [9.5730e-01, 4.2704e-02],
        [1.1558e-03, 9.9884e-01],
        [9.9157e-01, 8.4259e-03],
        [9.8679e-01, 1.3209e-02],
        [3.0180e-05, 9.9997e-01],
        [6.9023e-01, 3.0977e-01],
        [1.2455e-05, 9.9999e-01],
        [2.8334e-05, 9.9997e-01],
        [9.9884e-01, 1.1624e-03],
        [3.3089e-04, 9.9967e-01],
        [5.0132e-03, 9.9499e-01],
        [8.5753e-08, 1.0000e+00],
        [8.2266e-05, 9.9992e-01],
        [1.8096e-04, 9.9982e-01],
        [5.0510e-05, 9.9995e-01],
        [1.3011e-03, 9.9870e-01],
        [2.9299e-04, 9.9971e-01],
        [2.7883e-06, 1.0000e+00],
        [1.0724e-05, 9.9999e-01],
        [9.8229e-01, 1.7709e-02],
        [9.0998e-02, 9.0900e-01],
        [9.9535e-01, 4.6509e-03],
        [9.6452e-04, 9.9904e-01],
        [3.2093e-04, 9.9968e-01],
        [9.8093e-01, 1.9066e-02],
        [3.1580e-07, 1.0000e+00],
        [9.8203e-01, 1.7969e-02],
        [8.2018e-07, 1.0000e+00],
        [2.1169e-05, 9.9998e-01],
        [2.1173e-03, 9.9788e-01],
        [9.9357e-01, 6.4262e-03],
        [3.8219e-06, 1.0000e+00],
        [3.4774e-06, 1.0000e+00],
        [9.8898e-01, 1.1022e-02],
        [3.3075e-06, 1.0000e+00],
        [9.1447e-03, 9.9086e-01],
        [9.6141e-01, 3.8590e-02],
        [1.7950e-03, 9.9821e-01],
        [1.1681e-01, 8.8319e-01],
        [2.5002e-07, 1.0000e+00],
        [3.3633e-03, 9.9664e-01],
        [8.8592e-04, 9.9911e-01],
        [9.9922e-01, 7.7877e-04],
        [5.9417e-05, 9.9994e-01],
        [4.6635e-06, 1.0000e+00],
        [2.5868e-05, 9.9997e-01],
        [1.3312e-04, 9.9987e-01],
        [4.4214e-05, 9.9996e-01],
        [2.6386e-05, 9.9997e-01],
        [1.8849e-04, 9.9981e-01],
        [1.3436e-03, 9.9866e-01],
        [1.4622e-06, 1.0000e+00],
        [1.1425e-04, 9.9989e-01],
        [9.8750e-01, 1.2503e-02],
        [6.2806e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.6657e-01, 3.3435e-02],
        [2.5147e-03, 9.9749e-01],
        [1.6975e-03, 9.9830e-01],
        [3.5575e-03, 9.9644e-01],
        [2.2076e-05, 9.9998e-01],
        [9.9757e-01, 2.4333e-03],
        [9.9853e-01, 1.4720e-03],
        [8.7759e-01, 1.2241e-01],
        [6.5742e-05, 9.9993e-01],
        [4.2198e-04, 9.9958e-01],
        [4.4867e-06, 1.0000e+00],
        [2.6608e-04, 9.9973e-01],
        [9.5695e-01, 4.3046e-02],
        [4.1957e-04, 9.9958e-01],
        [1.7145e-06, 1.0000e+00],
        [1.3285e-05, 9.9999e-01],
        [8.9378e-06, 9.9999e-01],
        [9.8394e-06, 9.9999e-01],
        [9.9927e-01, 7.3049e-04],
        [9.9216e-01, 7.8418e-03],
        [6.5204e-05, 9.9993e-01],
        [1.4824e-04, 9.9985e-01],
        [5.2439e-05, 9.9995e-01],
        [3.9075e-05, 9.9996e-01],
        [6.4640e-03, 9.9354e-01],
        [2.6206e-04, 9.9974e-01],
        [1.1773e-05, 9.9999e-01],
        [2.1514e-06, 1.0000e+00],
        [1.1789e-05, 9.9999e-01],
        [4.9777e-03, 9.9502e-01],
        [1.2377e-06, 1.0000e+00],
        [1.1939e-04, 9.9988e-01],
        [6.6521e-06, 9.9999e-01],
        [3.5528e-05, 9.9996e-01],
        [3.7070e-03, 9.9629e-01],
        [9.9756e-01, 2.4411e-03],
        [5.8905e-04, 9.9941e-01],
        [1.9776e-05, 9.9998e-01],
        [1.2116e-05, 9.9999e-01],
        [7.0380e-07, 1.0000e+00],
        [8.8340e-01, 1.1660e-01],
        [8.8293e-01, 1.1707e-01],
        [5.8386e-01, 4.1614e-01],
        [1.1291e-06, 1.0000e+00],
        [9.9294e-01, 7.0609e-03],
        [3.8609e-07, 1.0000e+00],
        [3.8682e-01, 6.1318e-01],
        [9.9800e-01, 2.0016e-03],
        [3.0714e-03, 9.9693e-01],
        [1.3054e-05, 9.9999e-01],
        [1.7509e-03, 9.9825e-01],
        [5.6922e-05, 9.9994e-01],
        [9.9768e-01, 2.3206e-03],
        [3.4374e-02, 9.6563e-01],
        [9.9633e-01, 3.6670e-03],
        [1.1551e-02, 9.8845e-01],
        [7.2158e-05, 9.9993e-01],
        [5.1880e-03, 9.9481e-01],
        [3.0330e-06, 1.0000e+00],
        [9.3623e-01, 6.3774e-02],
        [1.9401e-04, 9.9981e-01],
        [3.8462e-03, 9.9615e-01],
        [9.8556e-01, 1.4441e-02],
        [1.0060e-01, 8.9940e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2358e-06, 1.0000e+00],
        [7.6696e-02, 9.2330e-01],
        [2.7754e-06, 1.0000e+00],
        [9.3149e-04, 9.9907e-01],
        [5.2023e-03, 9.9480e-01],
        [4.9198e-03, 9.9508e-01],
        [3.2758e-03, 9.9672e-01],
        [6.1828e-01, 3.8172e-01],
        [9.9703e-01, 2.9716e-03],
        [1.9703e-06, 1.0000e+00],
        [2.5197e-03, 9.9748e-01],
        [9.9787e-01, 2.1318e-03],
        [1.3182e-06, 1.0000e+00],
        [9.1903e-01, 8.0970e-02],
        [3.4142e-05, 9.9997e-01],
        [1.7819e-04, 9.9982e-01],
        [9.7242e-01, 2.7577e-02],
        [2.6583e-04, 9.9973e-01],
        [9.5992e-01, 4.0081e-02],
        [9.9715e-01, 2.8491e-03],
        [9.9803e-01, 1.9712e-03],
        [8.8970e-05, 9.9991e-01],
        [7.9175e-05, 9.9992e-01],
        [1.1146e-07, 1.0000e+00],
        [3.6547e-06, 1.0000e+00],
        [1.0018e-03, 9.9900e-01],
        [5.1507e-05, 9.9995e-01],
        [3.8844e-04, 9.9961e-01],
        [3.0308e-03, 9.9697e-01],
        [1.8333e-05, 9.9998e-01],
        [2.5706e-06, 1.0000e+00],
        [4.9753e-06, 9.9999e-01],
        [9.9929e-01, 7.1253e-04],
        [1.1536e-04, 9.9988e-01],
        [6.0349e-05, 9.9994e-01],
        [2.5250e-06, 1.0000e+00],
        [1.1762e-06, 1.0000e+00],
        [7.9243e-01, 2.0757e-01],
        [4.4220e-06, 1.0000e+00],
        [1.9444e-03, 9.9806e-01],
        [1.1774e-07, 1.0000e+00],
        [1.9685e-03, 9.9803e-01],
        [1.5803e-06, 1.0000e+00],
        [4.7721e-05, 9.9995e-01],
        [2.8723e-04, 9.9971e-01],
        [9.9814e-01, 1.8618e-03],
        [9.6746e-08, 1.0000e+00],
        [9.4851e-05, 9.9991e-01],
        [2.9822e-01, 7.0178e-01],
        [6.1980e-04, 9.9938e-01],
        [1.4612e-05, 9.9999e-01],
        [9.9487e-01, 5.1297e-03],
        [4.0220e-05, 9.9996e-01],
        [1.4693e-05, 9.9999e-01],
        [1.3935e-03, 9.9861e-01],
        [7.2633e-02, 9.2737e-01],
        [9.9880e-01, 1.1981e-03],
        [5.6258e-05, 9.9994e-01],
        [9.9752e-01, 2.4818e-03],
        [1.2329e-05, 9.9999e-01],
        [9.9937e-01, 6.3485e-04],
        [5.3605e-04, 9.9946e-01],
        [2.0837e-05, 9.9998e-01],
        [1.2779e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.3910e-04, 9.9956e-01],
        [2.4099e-06, 1.0000e+00],
        [4.1600e-03, 9.9584e-01],
        [9.6355e-01, 3.6448e-02],
        [1.6100e-05, 9.9998e-01],
        [5.1889e-06, 9.9999e-01],
        [8.6784e-01, 1.3216e-01],
        [2.8769e-02, 9.7123e-01],
        [1.3909e-01, 8.6091e-01],
        [1.1594e-01, 8.8406e-01],
        [1.7815e-04, 9.9982e-01],
        [9.3797e-01, 6.2033e-02],
        [9.9635e-01, 3.6453e-03],
        [8.9339e-01, 1.0661e-01],
        [1.0815e-04, 9.9989e-01],
        [9.9854e-01, 1.4608e-03],
        [5.1391e-08, 1.0000e+00],
        [9.7149e-07, 1.0000e+00],
        [9.9448e-01, 5.5233e-03],
        [9.6073e-01, 3.9273e-02],
        [3.7150e-07, 1.0000e+00],
        [9.9807e-01, 1.9253e-03],
        [1.3917e-03, 9.9861e-01],
        [8.7220e-01, 1.2780e-01],
        [2.2567e-06, 1.0000e+00],
        [9.4413e-01, 5.5874e-02],
        [9.9821e-01, 1.7920e-03],
        [7.9134e-05, 9.9992e-01],
        [9.5594e-01, 4.4058e-02],
        [3.7536e-03, 9.9625e-01],
        [1.4606e-06, 1.0000e+00],
        [7.0780e-03, 9.9292e-01],
        [4.8672e-03, 9.9513e-01],
        [1.7394e-06, 1.0000e+00],
        [5.3654e-07, 1.0000e+00],
        [1.3490e-05, 9.9999e-01],
        [5.3521e-04, 9.9946e-01],
        [1.0021e-04, 9.9990e-01],
        [2.3016e-03, 9.9770e-01],
        [7.7507e-05, 9.9992e-01],
        [6.9894e-04, 9.9930e-01],
        [1.1017e-05, 9.9999e-01],
        [3.3625e-03, 9.9664e-01],
        [3.1098e-01, 6.8902e-01],
        [1.2779e-03, 9.9872e-01],
        [2.8776e-04, 9.9971e-01],
        [8.0945e-05, 9.9992e-01],
        [1.1729e-04, 9.9988e-01],
        [2.8762e-05, 9.9997e-01],
        [9.1539e-01, 8.4612e-02],
        [9.1785e-01, 8.2148e-02],
        [1.0707e-06, 1.0000e+00],
        [9.8024e-01, 1.9763e-02],
        [1.2981e-04, 9.9987e-01],
        [2.2414e-05, 9.9998e-01],
        [2.1337e-02, 9.7866e-01],
        [1.1033e-06, 1.0000e+00],
        [7.6991e-05, 9.9992e-01],
        [9.4531e-01, 5.4694e-02],
        [7.7476e-01, 2.2524e-01],
        [1.1252e-04, 9.9989e-01],
        [1.7907e-06, 1.0000e+00],
        [2.2404e-04, 9.9978e-01],
        [4.5845e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.1381e-06, 1.0000e+00],
        [9.9841e-01, 1.5927e-03],
        [1.3395e-04, 9.9987e-01],
        [9.9919e-01, 8.1369e-04],
        [4.2983e-07, 1.0000e+00],
        [9.9978e-01, 2.1701e-04],
        [6.0241e-04, 9.9940e-01],
        [2.2986e-05, 9.9998e-01],
        [7.1062e-04, 9.9929e-01],
        [1.6930e-04, 9.9983e-01],
        [9.9404e-01, 5.9594e-03],
        [1.1639e-06, 1.0000e+00],
        [1.5758e-04, 9.9984e-01],
        [9.7315e-03, 9.9027e-01],
        [6.0705e-04, 9.9939e-01],
        [7.6724e-07, 1.0000e+00],
        [3.1793e-04, 9.9968e-01],
        [2.8627e-03, 9.9714e-01],
        [2.7826e-03, 9.9722e-01],
        [8.4053e-05, 9.9992e-01],
        [8.7669e-05, 9.9991e-01],
        [1.2618e-03, 9.9874e-01],
        [7.2104e-03, 9.9279e-01],
        [8.4570e-07, 1.0000e+00],
        [9.9962e-01, 3.8440e-04],
        [2.7474e-03, 9.9725e-01],
        [1.0945e-03, 9.9891e-01],
        [2.4167e-03, 9.9758e-01],
        [1.3144e-02, 9.8686e-01],
        [5.2832e-04, 9.9947e-01],
        [2.7240e-03, 9.9728e-01],
        [3.8883e-06, 1.0000e+00],
        [8.5188e-04, 9.9915e-01],
        [9.9755e-01, 2.4502e-03],
        [9.9864e-01, 1.3644e-03],
        [7.2430e-01, 2.7570e-01],
        [9.7945e-01, 2.0546e-02],
        [2.6462e-02, 9.7354e-01],
        [5.5637e-06, 9.9999e-01],
        [5.8930e-04, 9.9941e-01],
        [3.1943e-02, 9.6806e-01],
        [7.2657e-01, 2.7343e-01],
        [9.9578e-01, 4.2229e-03],
        [1.9053e-03, 9.9809e-01],
        [3.2930e-05, 9.9997e-01],
        [8.7413e-04, 9.9913e-01],
        [9.9668e-01, 3.3161e-03],
        [6.4269e-04, 9.9936e-01],
        [3.9285e-03, 9.9607e-01],
        [7.0927e-05, 9.9993e-01],
        [2.4127e-06, 1.0000e+00],
        [4.2358e-01, 5.7642e-01],
        [7.1172e-08, 1.0000e+00],
        [7.7955e-03, 9.9220e-01],
        [1.6282e-06, 1.0000e+00],
        [2.5688e-05, 9.9997e-01],
        [5.4385e-07, 1.0000e+00],
        [1.3063e-04, 9.9987e-01],
        [1.7251e-04, 9.9983e-01],
        [2.1107e-04, 9.9979e-01],
        [9.9722e-01, 2.7821e-03],
        [9.9512e-01, 4.8773e-03],
        [2.4208e-04, 9.9976e-01],
        [9.9778e-01, 2.2202e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7439e-06, 1.0000e+00],
        [9.4130e-05, 9.9991e-01],
        [9.3103e-01, 6.8974e-02],
        [4.5277e-04, 9.9955e-01],
        [2.4255e-04, 9.9976e-01],
        [5.8887e-04, 9.9941e-01],
        [9.9752e-01, 2.4841e-03],
        [8.6387e-05, 9.9991e-01],
        [5.3009e-03, 9.9470e-01],
        [3.2853e-03, 9.9671e-01],
        [2.4796e-02, 9.7520e-01],
        [1.9882e-05, 9.9998e-01],
        [3.7576e-06, 1.0000e+00],
        [5.1726e-03, 9.9483e-01],
        [1.9515e-06, 1.0000e+00],
        [1.6020e-04, 9.9984e-01],
        [4.1029e-01, 5.8971e-01],
        [2.3798e-06, 1.0000e+00],
        [3.4581e-07, 1.0000e+00],
        [8.2895e-03, 9.9171e-01],
        [3.8219e-05, 9.9996e-01],
        [5.5002e-05, 9.9995e-01],
        [2.4294e-04, 9.9976e-01],
        [1.1161e-04, 9.9989e-01],
        [7.8638e-01, 2.1362e-01],
        [1.3796e-05, 9.9999e-01],
        [1.6529e-05, 9.9998e-01],
        [1.4531e-02, 9.8547e-01],
        [2.1169e-05, 9.9998e-01],
        [1.2741e-03, 9.9873e-01],
        [9.9655e-01, 3.4546e-03],
        [8.3362e-07, 1.0000e+00],
        [3.3343e-05, 9.9997e-01],
        [1.0504e-05, 9.9999e-01],
        [8.5935e-05, 9.9991e-01],
        [1.1079e-04, 9.9989e-01],
        [1.6532e-07, 1.0000e+00],
        [9.9912e-01, 8.8324e-04],
        [3.9676e-06, 1.0000e+00],
        [9.8573e-01, 1.4267e-02],
        [2.6068e-03, 9.9739e-01],
        [4.3181e-05, 9.9996e-01],
        [8.6796e-01, 1.3204e-01],
        [3.6673e-04, 9.9963e-01],
        [8.4686e-03, 9.9153e-01],
        [1.2689e-04, 9.9987e-01],
        [9.9973e-01, 2.6954e-04],
        [1.5549e-04, 9.9984e-01],
        [8.2326e-04, 9.9918e-01],
        [3.4116e-04, 9.9966e-01],
        [4.2638e-04, 9.9957e-01],
        [7.9867e-07, 1.0000e+00],
        [4.3309e-05, 9.9996e-01],
        [7.6114e-04, 9.9924e-01],
        [9.9982e-01, 1.8392e-04],
        [9.9852e-01, 1.4768e-03],
        [9.8099e-01, 1.9007e-02],
        [5.5924e-03, 9.9441e-01],
        [7.6515e-07, 1.0000e+00],
        [3.8295e-05, 9.9996e-01],
        [9.9761e-01, 2.3946e-03],
        [4.7056e-05, 9.9995e-01],
        [1.5245e-05, 9.9998e-01],
        [2.2501e-02, 9.7750e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3736e-04, 9.9976e-01],
        [9.9829e-01, 1.7063e-03],
        [4.4109e-07, 1.0000e+00],
        [1.8031e-04, 9.9982e-01],
        [9.9703e-01, 2.9665e-03],
        [4.5438e-03, 9.9546e-01],
        [1.2133e-05, 9.9999e-01],
        [2.3599e-07, 1.0000e+00],
        [3.7423e-04, 9.9963e-01],
        [9.4094e-01, 5.9059e-02],
        [1.1443e-02, 9.8856e-01],
        [9.9810e-01, 1.9048e-03],
        [9.9972e-01, 2.8005e-04],
        [2.5492e-06, 1.0000e+00],
        [8.5970e-04, 9.9914e-01],
        [1.1082e-05, 9.9999e-01],
        [6.7791e-01, 3.2209e-01],
        [4.0757e-01, 5.9243e-01],
        [5.7265e-06, 9.9999e-01],
        [1.0717e-03, 9.9893e-01],
        [4.8108e-02, 9.5189e-01],
        [4.1629e-04, 9.9958e-01],
        [7.7092e-07, 1.0000e+00],
        [5.2963e-05, 9.9995e-01],
        [1.9847e-04, 9.9980e-01],
        [1.6390e-03, 9.9836e-01],
        [9.7019e-03, 9.9030e-01],
        [9.8184e-01, 1.8157e-02],
        [1.6852e-03, 9.9831e-01],
        [2.4666e-05, 9.9998e-01],
        [7.0374e-04, 9.9930e-01],
        [6.0997e-03, 9.9390e-01],
        [3.1123e-06, 1.0000e+00],
        [2.4049e-06, 1.0000e+00],
        [1.5146e-03, 9.9849e-01],
        [8.1727e-02, 9.1827e-01],
        [7.9526e-02, 9.2047e-01],
        [6.1010e-04, 9.9939e-01],
        [6.6297e-06, 9.9999e-01],
        [5.7732e-04, 9.9942e-01],
        [3.2142e-03, 9.9679e-01],
        [2.7748e-05, 9.9997e-01],
        [5.4549e-01, 4.5451e-01],
        [4.9790e-01, 5.0210e-01],
        [7.0177e-04, 9.9930e-01],
        [9.9913e-01, 8.6982e-04],
        [3.6426e-04, 9.9964e-01],
        [2.7118e-04, 9.9973e-01],
        [4.8208e-05, 9.9995e-01],
        [2.6534e-05, 9.9997e-01],
        [7.3222e-03, 9.9268e-01],
        [7.5442e-06, 9.9999e-01],
        [1.1703e-04, 9.9988e-01],
        [6.7229e-03, 9.9328e-01],
        [9.9873e-01, 1.2662e-03],
        [4.5726e-05, 9.9995e-01],
        [7.2931e-05, 9.9993e-01],
        [1.0640e-03, 9.9894e-01],
        [3.7820e-01, 6.2180e-01],
        [4.1728e-05, 9.9996e-01],
        [2.8655e-03, 9.9713e-01],
        [2.5331e-02, 9.7467e-01],
        [1.2212e-07, 1.0000e+00],
        [4.7088e-03, 9.9529e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9920e-01, 8.0128e-04],
        [1.7637e-07, 1.0000e+00],
        [9.3084e-01, 6.9159e-02],
        [2.9419e-04, 9.9971e-01],
        [4.1191e-04, 9.9959e-01],
        [2.6098e-01, 7.3902e-01],
        [6.8420e-04, 9.9932e-01],
        [2.6003e-07, 1.0000e+00],
        [9.9958e-01, 4.1642e-04],
        [6.4599e-02, 9.3540e-01],
        [1.5748e-03, 9.9843e-01],
        [4.1201e-03, 9.9588e-01],
        [4.6229e-06, 1.0000e+00],
        [4.9808e-05, 9.9995e-01],
        [2.6638e-02, 9.7336e-01],
        [7.3617e-05, 9.9993e-01],
        [9.9882e-01, 1.1828e-03],
        [2.7210e-03, 9.9728e-01],
        [9.5949e-01, 4.0505e-02],
        [8.2757e-01, 1.7243e-01],
        [2.7093e-03, 9.9729e-01],
        [2.2795e-03, 9.9772e-01],
        [9.9253e-01, 7.4689e-03],
        [6.2649e-06, 9.9999e-01],
        [3.1537e-05, 9.9997e-01],
        [3.7023e-01, 6.2977e-01],
        [9.3513e-04, 9.9906e-01],
        [3.4428e-04, 9.9966e-01],
        [1.3078e-03, 9.9869e-01],
        [2.3423e-02, 9.7658e-01],
        [1.0967e-02, 9.8903e-01],
        [2.9447e-03, 9.9706e-01],
        [9.7977e-01, 2.0232e-02],
        [2.0188e-02, 9.7981e-01],
        [1.5293e-02, 9.8471e-01],
        [9.9808e-01, 1.9199e-03],
        [3.8018e-06, 1.0000e+00],
        [1.5654e-04, 9.9984e-01],
        [4.8474e-01, 5.1526e-01],
        [4.9563e-04, 9.9950e-01],
        [9.9963e-01, 3.6684e-04],
        [4.6666e-05, 9.9995e-01],
        [6.8747e-06, 9.9999e-01],
        [1.6300e-01, 8.3700e-01],
        [1.0286e-05, 9.9999e-01],
        [2.6775e-05, 9.9997e-01],
        [9.9490e-01, 5.1046e-03],
        [9.9964e-01, 3.6208e-04],
        [1.9929e-02, 9.8007e-01],
        [3.4779e-08, 1.0000e+00],
        [1.4177e-04, 9.9986e-01],
        [4.2137e-03, 9.9579e-01],
        [5.7697e-05, 9.9994e-01],
        [1.7003e-03, 9.9830e-01],
        [7.3806e-03, 9.9262e-01],
        [8.8605e-06, 9.9999e-01],
        [9.0075e-06, 9.9999e-01],
        [1.3285e-03, 9.9867e-01],
        [9.9903e-01, 9.6800e-04],
        [9.9935e-01, 6.5040e-04],
        [2.5745e-03, 9.9743e-01],
        [1.7019e-03, 9.9830e-01],
        [9.9169e-04, 9.9901e-01],
        [6.6328e-04, 9.9934e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0105e-06, 1.0000e+00],
        [9.9879e-01, 1.2132e-03],
        [1.6025e-03, 9.9840e-01],
        [7.6334e-01, 2.3666e-01],
        [9.9280e-01, 7.2004e-03],
        [1.0809e-06, 1.0000e+00],
        [9.9862e-01, 1.3760e-03],
        [6.6218e-04, 9.9934e-01],
        [1.1562e-04, 9.9988e-01],
        [9.9912e-01, 8.8291e-04],
        [3.0392e-03, 9.9696e-01],
        [1.4209e-01, 8.5791e-01],
        [4.1487e-06, 1.0000e+00],
        [1.4552e-01, 8.5448e-01],
        [9.9921e-01, 7.8697e-04],
        [3.3625e-04, 9.9966e-01],
        [5.0337e-04, 9.9950e-01],
        [9.9994e-01, 6.0333e-05],
        [9.9924e-01, 7.5887e-04],
        [9.9920e-01, 8.0065e-04],
        [2.9682e-06, 1.0000e+00],
        [9.9928e-01, 7.1546e-04],
        [3.9934e-05, 9.9996e-01],
        [9.0397e-01, 9.6030e-02],
        [2.3875e-05, 9.9998e-01],
        [9.9995e-01, 4.9164e-05],
        [4.3667e-03, 9.9563e-01],
        [2.7237e-05, 9.9997e-01],
        [9.9962e-01, 3.7732e-04],
        [4.8207e-03, 9.9518e-01],
        [4.1411e-01, 5.8589e-01],
        [9.9904e-01, 9.6404e-04],
        [1.9741e-05, 9.9998e-01],
        [9.9910e-01, 8.9629e-04],
        [2.8141e-05, 9.9997e-01],
        [1.0113e-02, 9.8989e-01],
        [1.6707e-02, 9.8329e-01],
        [9.9809e-01, 1.9140e-03],
        [3.5460e-03, 9.9645e-01],
        [1.1545e-03, 9.9885e-01],
        [9.9840e-01, 1.6033e-03],
        [9.8713e-03, 9.9013e-01],
        [3.6430e-04, 9.9964e-01],
        [9.9442e-01, 5.5827e-03],
        [9.5541e-01, 4.4592e-02],
        [9.6630e-03, 9.9034e-01],
        [8.6654e-04, 9.9913e-01],
        [2.9911e-04, 9.9970e-01],
        [9.9944e-01, 5.5565e-04],
        [6.2063e-04, 9.9938e-01],
        [5.1243e-04, 9.9949e-01],
        [1.2900e-05, 9.9999e-01],
        [8.2248e-04, 9.9918e-01],
        [5.4745e-01, 4.5255e-01],
        [4.5426e-03, 9.9546e-01],
        [7.7009e-06, 9.9999e-01],
        [7.4266e-03, 9.9257e-01],
        [3.1969e-03, 9.9680e-01],
        [9.9617e-01, 3.8334e-03],
        [9.7364e-01, 2.6364e-02],
        [9.1538e-04, 9.9908e-01],
        [1.2593e-03, 9.9874e-01],
        [9.0179e-06, 9.9999e-01],
        [3.5767e-03, 9.9642e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5073e-03, 9.9849e-01],
        [9.9203e-01, 7.9728e-03],
        [3.3959e-04, 9.9966e-01],
        [3.1989e-04, 9.9968e-01],
        [7.5661e-05, 9.9992e-01],
        [5.0064e-04, 9.9950e-01],
        [2.0221e-03, 9.9798e-01],
        [3.4134e-04, 9.9966e-01],
        [9.9860e-01, 1.4011e-03],
        [7.8258e-06, 9.9999e-01],
        [4.0422e-01, 5.9578e-01],
        [9.9853e-01, 1.4711e-03],
        [2.1284e-04, 9.9979e-01],
        [2.4029e-04, 9.9976e-01],
        [9.9984e-01, 1.6372e-04],
        [3.6673e-08, 1.0000e+00],
        [9.9998e-01, 1.7688e-05],
        [4.3650e-02, 9.5635e-01],
        [2.6084e-01, 7.3916e-01],
        [4.3901e-06, 1.0000e+00],
        [2.6827e-01, 7.3173e-01],
        [3.5563e-03, 9.9644e-01],
        [2.2677e-01, 7.7323e-01],
        [9.9728e-01, 2.7228e-03],
        [9.9810e-01, 1.8978e-03],
        [8.3751e-02, 9.1625e-01],
        [1.0330e-06, 1.0000e+00],
        [2.3300e-04, 9.9977e-01],
        [4.2990e-04, 9.9957e-01],
        [2.3132e-06, 1.0000e+00],
        [8.2056e-06, 9.9999e-01],
        [9.9159e-01, 8.4091e-03],
        [1.6891e-03, 9.9831e-01],
        [9.9646e-01, 3.5422e-03],
        [3.3049e-04, 9.9967e-01],
        [9.9900e-01, 1.0040e-03],
        [6.6243e-05, 9.9993e-01],
        [1.9306e-01, 8.0694e-01],
        [2.3737e-03, 9.9763e-01],
        [4.8278e-03, 9.9517e-01],
        [3.9046e-04, 9.9961e-01],
        [4.8663e-05, 9.9995e-01],
        [1.9318e-02, 9.8068e-01],
        [9.9743e-01, 2.5653e-03],
        [2.6672e-04, 9.9973e-01],
        [7.1906e-05, 9.9993e-01],
        [9.9799e-01, 2.0106e-03],
        [1.9695e-05, 9.9998e-01],
        [2.9709e-03, 9.9703e-01],
        [3.5191e-04, 9.9965e-01],
        [1.0825e-04, 9.9989e-01],
        [1.1444e-04, 9.9989e-01],
        [1.0000e+00, 4.7998e-06],
        [3.3322e-05, 9.9997e-01],
        [7.7193e-06, 9.9999e-01],
        [6.2368e-05, 9.9994e-01],
        [6.1080e-04, 9.9939e-01],
        [4.7690e-03, 9.9523e-01],
        [9.9905e-01, 9.4912e-04],
        [2.4834e-05, 9.9998e-01],
        [1.6749e-03, 9.9833e-01],
        [7.8705e-06, 9.9999e-01],
        [2.4824e-01, 7.5176e-01],
        [3.1807e-04, 9.9968e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9598e-05, 9.9998e-01],
        [9.1593e-03, 9.9084e-01],
        [2.2362e-02, 9.7764e-01],
        [2.0661e-01, 7.9339e-01],
        [5.6566e-03, 9.9434e-01],
        [1.9862e-03, 9.9801e-01],
        [2.8800e-04, 9.9971e-01],
        [8.0465e-04, 9.9920e-01],
        [7.3941e-01, 2.6059e-01],
        [4.3265e-05, 9.9996e-01],
        [4.3760e-04, 9.9956e-01],
        [4.8413e-05, 9.9995e-01],
        [3.5861e-01, 6.4139e-01],
        [9.2857e-05, 9.9991e-01],
        [4.3868e-07, 1.0000e+00],
        [9.4959e-04, 9.9905e-01],
        [5.7796e-01, 4.2204e-01],
        [4.5069e-03, 9.9549e-01],
        [9.9885e-01, 1.1496e-03],
        [3.3014e-04, 9.9967e-01],
        [6.9620e-03, 9.9304e-01],
        [5.5565e-07, 1.0000e+00],
        [1.0345e-05, 9.9999e-01],
        [9.5572e-05, 9.9990e-01],
        [8.5103e-03, 9.9149e-01],
        [6.2848e-04, 9.9937e-01],
        [1.0329e-04, 9.9990e-01],
        [1.3790e-02, 9.8621e-01],
        [9.9925e-01, 7.4822e-04],
        [9.9832e-01, 1.6809e-03],
        [2.4775e-02, 9.7522e-01],
        [9.7511e-01, 2.4888e-02],
        [8.7167e-04, 9.9913e-01],
        [2.4888e-08, 1.0000e+00],
        [9.9971e-01, 2.9040e-04],
        [2.6306e-03, 9.9737e-01],
        [8.7704e-01, 1.2296e-01],
        [4.2298e-04, 9.9958e-01],
        [1.3852e-03, 9.9861e-01],
        [2.1074e-04, 9.9979e-01],
        [9.7969e-01, 2.0312e-02],
        [4.7405e-04, 9.9953e-01],
        [9.6212e-01, 3.7884e-02],
        [4.1421e-03, 9.9586e-01],
        [1.6159e-04, 9.9984e-01],
        [1.0633e-03, 9.9894e-01],
        [5.1676e-06, 9.9999e-01],
        [2.8248e-03, 9.9718e-01],
        [1.7354e-05, 9.9998e-01],
        [6.1620e-02, 9.3838e-01],
        [9.6434e-03, 9.9036e-01],
        [2.3606e-05, 9.9998e-01],
        [9.5810e-01, 4.1903e-02],
        [1.3364e-04, 9.9987e-01],
        [2.1544e-04, 9.9978e-01],
        [8.0888e-04, 9.9919e-01],
        [2.2237e-04, 9.9978e-01],
        [4.9978e-06, 9.9999e-01],
        [6.8393e-04, 9.9932e-01],
        [9.9512e-01, 4.8789e-03],
        [2.1102e-02, 9.7890e-01],
        [3.4193e-06, 1.0000e+00],
        [2.3755e-05, 9.9998e-01],
        [5.1646e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3969e-03, 9.9860e-01],
        [5.6127e-06, 9.9999e-01],
        [2.5870e-02, 9.7413e-01],
        [9.7802e-01, 2.1980e-02],
        [9.9828e-01, 1.7173e-03],
        [3.2754e-05, 9.9997e-01],
        [7.0207e-05, 9.9993e-01],
        [2.7470e-03, 9.9725e-01],
        [9.7939e-04, 9.9902e-01],
        [4.9257e-05, 9.9995e-01],
        [8.4730e-03, 9.9153e-01],
        [3.1428e-03, 9.9686e-01],
        [9.1862e-02, 9.0814e-01],
        [5.1467e-04, 9.9949e-01],
        [1.0686e-04, 9.9989e-01],
        [7.8897e-03, 9.9211e-01],
        [2.6402e-05, 9.9997e-01],
        [9.9936e-01, 6.3543e-04],
        [9.6818e-04, 9.9903e-01],
        [1.0185e-03, 9.9898e-01],
        [3.4525e-01, 6.5475e-01],
        [1.1447e-01, 8.8553e-01],
        [9.9955e-01, 4.4811e-04],
        [2.0166e-05, 9.9998e-01],
        [9.9796e-01, 2.0405e-03],
        [9.9444e-01, 5.5574e-03],
        [3.9122e-02, 9.6088e-01],
        [2.8549e-04, 9.9971e-01],
        [4.3139e-04, 9.9957e-01],
        [2.8725e-03, 9.9713e-01],
        [1.7565e-04, 9.9982e-01],
        [2.5298e-05, 9.9997e-01],
        [1.3017e-01, 8.6983e-01],
        [4.8768e-06, 1.0000e+00],
        [8.3544e-05, 9.9992e-01],
        [1.0793e-02, 9.8921e-01],
        [6.5100e-03, 9.9349e-01],
        [3.4693e-01, 6.5307e-01],
        [6.8671e-04, 9.9931e-01],
        [2.6402e-05, 9.9997e-01],
        [3.4639e-03, 9.9654e-01],
        [8.9346e-01, 1.0654e-01],
        [8.4555e-03, 9.9154e-01],
        [4.6072e-03, 9.9539e-01],
        [9.9932e-01, 6.8419e-04],
        [1.1765e-03, 9.9882e-01],
        [1.8051e-05, 9.9998e-01],
        [9.9797e-01, 2.0291e-03],
        [4.7197e-03, 9.9528e-01],
        [4.6856e-08, 1.0000e+00],
        [1.8012e-02, 9.8199e-01],
        [1.1960e-05, 9.9999e-01],
        [7.0911e-02, 9.2909e-01],
        [5.5460e-07, 1.0000e+00],
        [3.0192e-01, 6.9808e-01],
        [9.4628e-01, 5.3723e-02],
        [9.9985e-01, 1.4526e-04],
        [2.0588e-02, 9.7941e-01],
        [1.1013e-03, 9.9890e-01],
        [9.9525e-01, 4.7534e-03],
        [3.1058e-05, 9.9997e-01],
        [1.0837e-03, 9.9892e-01],
        [9.9897e-01, 1.0322e-03],
        [8.7792e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.7787e-01, 2.2127e-02],
        [9.1437e-03, 9.9086e-01],
        [9.9590e-01, 4.1046e-03],
        [9.9892e-01, 1.0847e-03],
        [9.9977e-01, 2.3245e-04],
        [1.2200e-03, 9.9878e-01],
        [1.5512e-05, 9.9998e-01],
        [3.9789e-02, 9.6021e-01],
        [1.8061e-04, 9.9982e-01],
        [2.5494e-06, 1.0000e+00],
        [2.1560e-01, 7.8440e-01],
        [1.1372e-02, 9.8863e-01],
        [2.7343e-06, 1.0000e+00],
        [7.1986e-05, 9.9993e-01],
        [7.0038e-01, 2.9962e-01],
        [9.9887e-01, 1.1262e-03],
        [3.1545e-05, 9.9997e-01],
        [9.9673e-01, 3.2736e-03],
        [7.8827e-01, 2.1173e-01],
        [9.9831e-01, 1.6933e-03],
        [2.4664e-03, 9.9753e-01],
        [3.7641e-04, 9.9962e-01],
        [1.1181e-04, 9.9989e-01],
        [9.8968e-01, 1.0324e-02],
        [5.4455e-06, 9.9999e-01],
        [7.4540e-07, 1.0000e+00],
        [9.9929e-01, 7.0579e-04],
        [9.9840e-01, 1.6035e-03],
        [8.6431e-06, 9.9999e-01],
        [2.6741e-06, 1.0000e+00],
        [9.5692e-06, 9.9999e-01],
        [9.4213e-02, 9.0579e-01],
        [7.8838e-01, 2.1162e-01],
        [6.9777e-03, 9.9302e-01],
        [2.5000e-05, 9.9997e-01],
        [9.4593e-01, 5.4066e-02],
        [9.9776e-01, 2.2443e-03],
        [5.3772e-04, 9.9946e-01],
        [1.3449e-05, 9.9999e-01],
        [3.7524e-01, 6.2476e-01],
        [5.9648e-02, 9.4035e-01],
        [9.9785e-01, 2.1522e-03],
        [1.1975e-03, 9.9880e-01],
        [9.9776e-01, 2.2413e-03],
        [2.0307e-04, 9.9980e-01],
        [1.7015e-03, 9.9830e-01],
        [2.7132e-06, 1.0000e+00],
        [9.9982e-01, 1.8035e-04],
        [8.3674e-04, 9.9916e-01],
        [2.0649e-05, 9.9998e-01],
        [9.9817e-01, 1.8318e-03],
        [5.3734e-04, 9.9946e-01],
        [1.9771e-05, 9.9998e-01],
        [9.9856e-01, 1.4397e-03],
        [5.5243e-04, 9.9945e-01],
        [3.0995e-04, 9.9969e-01],
        [9.9951e-01, 4.9240e-04],
        [2.9502e-06, 1.0000e+00],
        [9.7448e-06, 9.9999e-01],
        [2.0924e-02, 9.7908e-01],
        [3.2678e-02, 9.6732e-01],
        [1.9243e-06, 1.0000e+00],
        [1.9763e-03, 9.9802e-01],
        [9.4987e-01, 5.0129e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.1593e-05, 9.9991e-01],
        [1.5116e-06, 1.0000e+00],
        [7.0995e-03, 9.9290e-01],
        [7.1907e-02, 9.2809e-01],
        [9.9925e-01, 7.5392e-04],
        [4.0831e-01, 5.9169e-01],
        [9.9966e-01, 3.3732e-04],
        [1.7987e-04, 9.9982e-01],
        [8.9241e-04, 9.9911e-01],
        [4.5695e-07, 1.0000e+00],
        [8.8179e-04, 9.9912e-01],
        [3.4900e-04, 9.9965e-01],
        [9.9813e-01, 1.8666e-03],
        [1.1378e-02, 9.8862e-01],
        [9.9970e-01, 2.9560e-04],
        [7.0163e-02, 9.2984e-01],
        [5.8507e-03, 9.9415e-01],
        [9.3766e-04, 9.9906e-01],
        [2.4754e-04, 9.9975e-01],
        [6.2073e-05, 9.9994e-01],
        [2.5857e-03, 9.9741e-01],
        [6.8456e-06, 9.9999e-01],
        [1.3570e-04, 9.9986e-01],
        [1.1100e-03, 9.9889e-01],
        [1.5690e-07, 1.0000e+00],
        [3.5060e-01, 6.4940e-01],
        [7.9534e-04, 9.9920e-01],
        [2.1431e-02, 9.7857e-01],
        [5.4819e-01, 4.5181e-01],
        [1.1718e-05, 9.9999e-01],
        [9.9934e-01, 6.5819e-04],
        [9.9160e-01, 8.3990e-03],
        [2.4714e-05, 9.9998e-01],
        [6.5567e-04, 9.9934e-01],
        [9.9488e-01, 5.1250e-03],
        [6.5456e-06, 9.9999e-01],
        [7.0130e-07, 1.0000e+00],
        [6.4965e-06, 9.9999e-01],
        [1.4734e-03, 9.9853e-01],
        [1.0039e-04, 9.9990e-01],
        [2.6007e-04, 9.9974e-01],
        [8.1743e-03, 9.9183e-01],
        [5.3066e-05, 9.9995e-01],
        [3.8193e-04, 9.9962e-01],
        [1.6597e-03, 9.9834e-01],
        [9.9730e-01, 2.6976e-03],
        [2.4046e-07, 1.0000e+00],
        [5.3028e-05, 9.9995e-01],
        [9.9764e-01, 2.3566e-03],
        [9.8679e-01, 1.3213e-02],
        [2.4413e-04, 9.9976e-01],
        [9.9975e-01, 2.4733e-04],
        [9.9999e-01, 1.2503e-05],
        [4.2699e-03, 9.9573e-01],
        [9.6946e-01, 3.0537e-02],
        [9.9891e-01, 1.0899e-03],
        [6.0796e-04, 9.9939e-01],
        [3.7903e-05, 9.9996e-01],
        [2.1710e-02, 9.7829e-01],
        [9.9796e-01, 2.0432e-03],
        [1.0297e-04, 9.9990e-01],
        [5.4635e-04, 9.9945e-01],
        [9.9768e-01, 2.3180e-03],
        [6.3823e-05, 9.9994e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.8364e-03, 9.9516e-01],
        [5.3117e-07, 1.0000e+00],
        [9.9773e-01, 2.2664e-03],
        [1.7097e-05, 9.9998e-01],
        [4.0276e-05, 9.9996e-01],
        [1.0977e-05, 9.9999e-01],
        [2.2715e-04, 9.9977e-01],
        [9.5646e-01, 4.3539e-02],
        [1.0066e-07, 1.0000e+00],
        [9.9913e-01, 8.6930e-04],
        [1.4353e-04, 9.9986e-01],
        [1.6187e-04, 9.9984e-01],
        [9.2247e-05, 9.9991e-01],
        [8.3201e-04, 9.9917e-01],
        [2.8338e-02, 9.7166e-01],
        [8.1034e-07, 1.0000e+00],
        [3.6501e-06, 1.0000e+00],
        [9.9720e-01, 2.8031e-03],
        [1.1822e-05, 9.9999e-01],
        [1.3988e-06, 1.0000e+00],
        [9.9940e-01, 5.9807e-04],
        [8.4309e-06, 9.9999e-01],
        [1.0943e-05, 9.9999e-01],
        [9.9952e-01, 4.8400e-04],
        [6.4426e-05, 9.9994e-01],
        [4.2921e-05, 9.9996e-01],
        [3.7373e-02, 9.6263e-01],
        [7.6451e-03, 9.9235e-01],
        [1.2826e-04, 9.9987e-01],
        [6.6720e-06, 9.9999e-01],
        [7.1277e-02, 9.2872e-01],
        [5.4116e-06, 9.9999e-01],
        [1.8474e-05, 9.9998e-01],
        [1.5896e-06, 1.0000e+00],
        [1.6330e-04, 9.9984e-01],
        [6.2950e-07, 1.0000e+00],
        [4.7798e-03, 9.9522e-01],
        [1.2591e-06, 1.0000e+00],
        [2.7334e-03, 9.9727e-01],
        [9.6097e-01, 3.9026e-02],
        [2.4953e-05, 9.9998e-01],
        [3.4644e-07, 1.0000e+00],
        [9.9536e-05, 9.9990e-01],
        [4.5261e-05, 9.9995e-01],
        [2.5525e-04, 9.9974e-01],
        [9.9918e-01, 8.1996e-04],
        [2.7106e-04, 9.9973e-01],
        [9.9999e-01, 8.3757e-06],
        [7.8845e-01, 2.1155e-01],
        [5.4101e-03, 9.9459e-01],
        [8.1841e-01, 1.8159e-01],
        [9.7565e-05, 9.9990e-01],
        [6.7496e-06, 9.9999e-01],
        [9.9921e-01, 7.8745e-04],
        [9.8965e-01, 1.0347e-02],
        [1.2035e-04, 9.9988e-01],
        [3.7076e-05, 9.9996e-01],
        [7.1197e-05, 9.9993e-01],
        [9.9853e-01, 1.4674e-03],
        [3.8365e-05, 9.9996e-01],
        [1.6379e-05, 9.9998e-01],
        [8.9812e-01, 1.0188e-01],
        [1.4447e-03, 9.9856e-01],
        [6.1493e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.4658e-04, 9.9935e-01],
        [1.2634e-05, 9.9999e-01],
        [9.7929e-06, 9.9999e-01],
        [4.1512e-04, 9.9958e-01],
        [9.9459e-01, 5.4110e-03],
        [1.4496e-04, 9.9986e-01],
        [9.9916e-01, 8.4346e-04],
        [1.6300e-02, 9.8370e-01],
        [3.8408e-07, 1.0000e+00],
        [3.7050e-05, 9.9996e-01],
        [3.0947e-04, 9.9969e-01],
        [1.9858e-03, 9.9801e-01],
        [1.6298e-05, 9.9998e-01],
        [3.8182e-05, 9.9996e-01],
        [2.2828e-03, 9.9772e-01],
        [9.9996e-01, 4.1857e-05],
        [9.8631e-01, 1.3689e-02],
        [9.9760e-01, 2.3977e-03],
        [1.8708e-04, 9.9981e-01],
        [9.8333e-01, 1.6672e-02],
        [9.9622e-01, 3.7759e-03],
        [9.2251e-07, 1.0000e+00],
        [1.1801e-06, 1.0000e+00],
        [9.9877e-01, 1.2278e-03],
        [5.9126e-04, 9.9941e-01],
        [1.0572e-02, 9.8943e-01],
        [9.9842e-01, 1.5756e-03],
        [9.9620e-01, 3.7986e-03],
        [3.3196e-03, 9.9668e-01],
        [4.4606e-06, 1.0000e+00],
        [1.0166e-03, 9.9898e-01],
        [1.3010e-04, 9.9987e-01],
        [3.8693e-03, 9.9613e-01],
        [1.1256e-03, 9.9887e-01],
        [4.8050e-01, 5.1950e-01],
        [4.5220e-06, 1.0000e+00],
        [7.9086e-07, 1.0000e+00],
        [2.2150e-06, 1.0000e+00],
        [9.9945e-01, 5.4960e-04],
        [1.2165e-06, 1.0000e+00],
        [3.8661e-04, 9.9961e-01],
        [1.8905e-03, 9.9811e-01],
        [8.9288e-01, 1.0712e-01],
        [9.5418e-01, 4.5823e-02],
        [1.2264e-04, 9.9988e-01],
        [4.4802e-04, 9.9955e-01],
        [1.1056e-02, 9.8894e-01],
        [6.7368e-02, 9.3263e-01],
        [3.0483e-03, 9.9695e-01],
        [7.3228e-08, 1.0000e+00],
        [6.0000e-08, 1.0000e+00],
        [4.8054e-03, 9.9519e-01],
        [5.9469e-02, 9.4053e-01],
        [4.6808e-06, 1.0000e+00],
        [8.2137e-05, 9.9992e-01],
        [4.5029e-07, 1.0000e+00],
        [6.1433e-05, 9.9994e-01],
        [8.2911e-05, 9.9992e-01],
        [8.0570e-04, 9.9919e-01],
        [9.8522e-04, 9.9901e-01],
        [9.9908e-01, 9.2131e-04],
        [9.7735e-01, 2.2649e-02],
        [2.2231e-05, 9.9998e-01],
        [9.9640e-01, 3.5983e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[6.6762e-05, 9.9993e-01],
        [2.0474e-07, 1.0000e+00],
        [2.0914e-04, 9.9979e-01],
        [2.3888e-04, 9.9976e-01],
        [1.0780e-01, 8.9220e-01],
        [1.4540e-02, 9.8546e-01],
        [5.2617e-07, 1.0000e+00],
        [2.3935e-03, 9.9761e-01],
        [1.7973e-05, 9.9998e-01],
        [9.8186e-01, 1.8143e-02],
        [3.5116e-06, 1.0000e+00],
        [7.7607e-01, 2.2393e-01],
        [2.2759e-03, 9.9772e-01],
        [1.3180e-03, 9.9868e-01],
        [7.0999e-07, 1.0000e+00],
        [1.5557e-04, 9.9984e-01],
        [7.6226e-06, 9.9999e-01],
        [9.9796e-01, 2.0446e-03],
        [1.0122e-04, 9.9990e-01],
        [8.4388e-06, 9.9999e-01],
        [4.4879e-03, 9.9551e-01],
        [3.0064e-04, 9.9970e-01],
        [9.9862e-01, 1.3817e-03],
        [2.4511e-04, 9.9975e-01],
        [1.1037e-07, 1.0000e+00],
        [2.4593e-06, 1.0000e+00],
        [7.4988e-07, 1.0000e+00],
        [1.4474e-02, 9.8553e-01],
        [7.2688e-08, 1.0000e+00],
        [9.4959e-01, 5.0414e-02],
        [6.6125e-06, 9.9999e-01],
        [2.8193e-04, 9.9972e-01],
        [9.9929e-01, 7.0660e-04],
        [9.9900e-01, 9.9792e-04],
        [9.8234e-01, 1.7664e-02],
        [1.6379e-07, 1.0000e+00],
        [9.9772e-01, 2.2796e-03],
        [1.6753e-05, 9.9998e-01],
        [6.4013e-03, 9.9360e-01],
        [3.8298e-05, 9.9996e-01],
        [4.7367e-03, 9.9526e-01],
        [1.9503e-04, 9.9981e-01],
        [2.9374e-04, 9.9971e-01],
        [3.7179e-05, 9.9996e-01],
        [1.1022e-02, 9.8898e-01],
        [2.3081e-04, 9.9977e-01],
        [1.6082e-03, 9.9839e-01],
        [1.3973e-04, 9.9986e-01],
        [1.6552e-05, 9.9998e-01],
        [7.2688e-08, 1.0000e+00],
        [9.9994e-01, 5.6713e-05],
        [8.2415e-01, 1.7585e-01],
        [2.1824e-04, 9.9978e-01],
        [2.6896e-04, 9.9973e-01],
        [3.2755e-03, 9.9672e-01],
        [9.5199e-01, 4.8009e-02],
        [3.8026e-06, 1.0000e+00],
        [3.3753e-06, 1.0000e+00],
        [9.8668e-01, 1.3325e-02],
        [5.1354e-04, 9.9949e-01],
        [6.6094e-05, 9.9993e-01],
        [9.9127e-01, 8.7332e-03],
        [4.5471e-04, 9.9955e-01],
        [8.7154e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.6678e-01, 3.3322e-01],
        [9.9880e-01, 1.2028e-03],
        [2.1319e-04, 9.9979e-01],
        [2.0214e-07, 1.0000e+00],
        [3.5962e-04, 9.9964e-01],
        [9.9799e-01, 2.0126e-03],
        [1.4246e-03, 9.9858e-01],
        [1.4367e-02, 9.8563e-01],
        [1.5924e-04, 9.9984e-01],
        [1.7532e-07, 1.0000e+00],
        [2.2304e-05, 9.9998e-01],
        [9.7161e-01, 2.8390e-02],
        [8.3423e-06, 9.9999e-01],
        [1.3975e-03, 9.9860e-01],
        [1.0580e-05, 9.9999e-01],
        [9.9877e-01, 1.2342e-03],
        [5.9342e-04, 9.9941e-01],
        [5.5710e-04, 9.9944e-01],
        [2.0603e-04, 9.9979e-01],
        [5.5267e-03, 9.9447e-01],
        [4.8199e-01, 5.1801e-01],
        [4.8064e-03, 9.9519e-01],
        [4.9541e-04, 9.9950e-01],
        [9.1300e-05, 9.9991e-01],
        [2.0865e-05, 9.9998e-01],
        [4.9579e-05, 9.9995e-01],
        [3.8260e-04, 9.9962e-01],
        [1.3648e-02, 9.8635e-01],
        [1.5151e-01, 8.4849e-01],
        [2.7910e-05, 9.9997e-01],
        [9.6687e-01, 3.3126e-02],
        [5.5521e-01, 4.4479e-01],
        [8.1808e-04, 9.9918e-01],
        [2.0961e-04, 9.9979e-01],
        [6.2294e-06, 9.9999e-01],
        [8.9473e-06, 9.9999e-01],
        [2.5995e-05, 9.9997e-01],
        [2.4870e-06, 1.0000e+00],
        [1.6950e-05, 9.9998e-01],
        [7.4769e-06, 9.9999e-01],
        [6.1918e-06, 9.9999e-01],
        [1.5115e-05, 9.9998e-01],
        [3.4445e-03, 9.9656e-01],
        [9.9984e-01, 1.5648e-04],
        [9.4917e-01, 5.0834e-02],
        [4.2796e-04, 9.9957e-01],
        [9.8798e-01, 1.2022e-02],
        [6.5422e-06, 9.9999e-01],
        [8.5494e-03, 9.9145e-01],
        [5.4303e-03, 9.9457e-01],
        [1.9852e-05, 9.9998e-01],
        [5.9441e-04, 9.9941e-01],
        [1.2184e-02, 9.8782e-01],
        [1.3935e-03, 9.9861e-01],
        [1.2246e-06, 1.0000e+00],
        [5.7399e-04, 9.9943e-01],
        [9.8563e-01, 1.4366e-02],
        [2.2398e-03, 9.9776e-01],
        [1.1695e-04, 9.9988e-01],
        [1.3458e-06, 1.0000e+00],
        [2.3863e-04, 9.9976e-01],
        [9.9966e-01, 3.4081e-04],
        [2.5520e-03, 9.9745e-01],
        [4.8627e-04, 9.9951e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.7787e-08, 1.0000e+00],
        [6.7408e-01, 3.2592e-01],
        [3.4277e-06, 1.0000e+00],
        [9.8345e-01, 1.6546e-02],
        [7.5685e-01, 2.4315e-01],
        [1.3040e-05, 9.9999e-01],
        [2.0623e-03, 9.9794e-01],
        [1.1350e-05, 9.9999e-01],
        [3.9301e-01, 6.0699e-01],
        [4.2478e-03, 9.9575e-01],
        [8.0246e-05, 9.9992e-01],
        [4.4998e-05, 9.9996e-01],
        [3.6346e-04, 9.9964e-01],
        [1.9009e-03, 9.9810e-01],
        [3.0236e-05, 9.9997e-01],
        [3.0729e-02, 9.6927e-01],
        [2.7456e-05, 9.9997e-01],
        [1.3546e-06, 1.0000e+00],
        [1.1434e-07, 1.0000e+00],
        [7.9982e-06, 9.9999e-01],
        [9.5666e-01, 4.3341e-02],
        [2.0744e-03, 9.9793e-01],
        [9.7224e-01, 2.7760e-02],
        [2.2310e-06, 1.0000e+00],
        [8.3902e-04, 9.9916e-01],
        [9.9059e-01, 9.4129e-03],
        [1.4662e-04, 9.9985e-01],
        [8.7465e-01, 1.2535e-01],
        [7.6317e-08, 1.0000e+00],
        [6.8422e-05, 9.9993e-01],
        [9.2202e-01, 7.7977e-02],
        [9.9595e-01, 4.0452e-03],
        [4.0663e-01, 5.9337e-01],
        [9.9196e-01, 8.0403e-03],
        [4.8043e-01, 5.1957e-01],
        [3.6465e-04, 9.9964e-01],
        [3.9869e-07, 1.0000e+00],
        [5.2685e-06, 9.9999e-01],
        [1.0825e-03, 9.9892e-01],
        [9.9953e-01, 4.6644e-04],
        [5.1088e-04, 9.9949e-01],
        [4.6233e-05, 9.9995e-01],
        [1.3847e-06, 1.0000e+00],
        [9.7765e-01, 2.2346e-02],
        [2.3104e-06, 1.0000e+00],
        [9.8078e-01, 1.9218e-02],
        [9.9998e-01, 1.7175e-05],
        [1.5769e-02, 9.8423e-01],
        [9.9413e-01, 5.8716e-03],
        [2.8288e-07, 1.0000e+00],
        [3.4297e-02, 9.6570e-01],
        [4.1392e-08, 1.0000e+00],
        [1.1064e-02, 9.8894e-01],
        [5.4469e-04, 9.9946e-01],
        [1.9310e-04, 9.9981e-01],
        [1.5822e-02, 9.8418e-01],
        [6.5422e-04, 9.9935e-01],
        [3.2227e-05, 9.9997e-01],
        [8.1858e-04, 9.9918e-01],
        [7.7114e-01, 2.2886e-01],
        [1.8110e-06, 1.0000e+00],
        [4.0398e-05, 9.9996e-01],
        [9.1863e-07, 1.0000e+00],
        [9.9506e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9136e-01, 8.6431e-03],
        [1.2386e-05, 9.9999e-01],
        [9.9798e-01, 2.0209e-03],
        [1.1041e-04, 9.9989e-01],
        [3.7146e-05, 9.9996e-01],
        [1.0848e-02, 9.8915e-01],
        [3.6098e-02, 9.6390e-01],
        [7.3304e-01, 2.6696e-01],
        [1.2293e-04, 9.9988e-01],
        [8.5350e-06, 9.9999e-01],
        [1.0515e-02, 9.8948e-01],
        [9.9919e-01, 8.0702e-04],
        [8.3468e-01, 1.6532e-01],
        [8.3453e-05, 9.9992e-01],
        [3.9457e-04, 9.9961e-01],
        [1.1235e-03, 9.9888e-01],
        [4.4390e-05, 9.9996e-01],
        [8.4256e-01, 1.5744e-01],
        [6.5661e-06, 9.9999e-01],
        [1.9693e-04, 9.9980e-01],
        [4.3417e-06, 1.0000e+00],
        [1.2078e-06, 1.0000e+00],
        [3.7733e-02, 9.6227e-01],
        [5.7401e-01, 4.2599e-01],
        [4.1426e-04, 9.9959e-01],
        [9.3688e-04, 9.9906e-01],
        [1.7259e-13, 1.0000e+00],
        [6.2762e-05, 9.9994e-01],
        [6.8246e-05, 9.9993e-01],
        [9.9904e-01, 9.5791e-04],
        [7.7371e-06, 9.9999e-01],
        [7.7404e-06, 9.9999e-01],
        [9.9773e-01, 2.2739e-03],
        [3.2773e-03, 9.9672e-01],
        [7.7959e-05, 9.9992e-01],
        [1.1605e-04, 9.9988e-01],
        [2.0850e-04, 9.9979e-01],
        [9.2821e-01, 7.1785e-02],
        [9.9647e-01, 3.5332e-03],
        [3.2307e-07, 1.0000e+00],
        [4.6738e-06, 1.0000e+00],
        [9.6227e-04, 9.9904e-01],
        [4.7422e-05, 9.9995e-01],
        [1.0222e-05, 9.9999e-01],
        [1.8667e-04, 9.9981e-01],
        [7.4739e-04, 9.9925e-01],
        [1.9658e-03, 9.9803e-01],
        [8.2929e-01, 1.7071e-01],
        [2.3634e-04, 9.9976e-01],
        [4.6553e-04, 9.9953e-01],
        [9.9841e-01, 1.5888e-03],
        [9.9835e-01, 1.6514e-03],
        [9.8890e-05, 9.9990e-01],
        [5.5353e-06, 9.9999e-01],
        [9.9922e-01, 7.7743e-04],
        [9.7450e-01, 2.5498e-02],
        [9.9817e-01, 1.8263e-03],
        [1.5264e-05, 9.9998e-01],
        [1.2400e-01, 8.7600e-01],
        [1.4424e-04, 9.9986e-01],
        [9.5351e-01, 4.6487e-02],
        [2.7383e-03, 9.9726e-01],
        [3.2040e-06, 1.0000e+00],
        [7.3426e-03, 9.9266e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.3723e-04, 9.9956e-01],
        [9.3537e-04, 9.9906e-01],
        [2.3665e-02, 9.7633e-01],
        [4.5273e-01, 5.4727e-01],
        [9.9966e-01, 3.3679e-04],
        [8.7734e-02, 9.1227e-01],
        [8.3608e-06, 9.9999e-01],
        [2.5743e-03, 9.9743e-01],
        [2.1319e-02, 9.7868e-01],
        [9.9922e-01, 7.7951e-04],
        [4.3647e-06, 1.0000e+00],
        [3.4934e-02, 9.6507e-01],
        [7.3680e-04, 9.9926e-01],
        [1.6499e-03, 9.9835e-01],
        [2.5835e-03, 9.9742e-01],
        [9.9731e-01, 2.6897e-03],
        [8.3166e-01, 1.6834e-01],
        [2.4670e-08, 1.0000e+00],
        [1.1622e-01, 8.8378e-01],
        [8.2691e-04, 9.9917e-01],
        [1.0336e-01, 8.9664e-01],
        [9.8070e-01, 1.9300e-02],
        [1.1102e-05, 9.9999e-01],
        [1.2165e-04, 9.9988e-01],
        [1.2399e-04, 9.9988e-01],
        [2.4055e-04, 9.9976e-01],
        [7.6492e-04, 9.9924e-01],
        [1.5035e-03, 9.9850e-01],
        [8.1721e-06, 9.9999e-01],
        [2.6972e-05, 9.9997e-01],
        [4.7706e-02, 9.5229e-01],
        [4.4634e-02, 9.5537e-01],
        [6.5085e-04, 9.9935e-01],
        [9.9555e-01, 4.4522e-03],
        [1.7326e-05, 9.9998e-01],
        [9.9574e-01, 4.2594e-03],
        [2.0617e-04, 9.9979e-01],
        [1.1680e-01, 8.8320e-01],
        [3.5237e-01, 6.4763e-01],
        [9.9972e-01, 2.8484e-04],
        [9.9805e-01, 1.9543e-03],
        [4.2017e-05, 9.9996e-01],
        [9.9190e-01, 8.0997e-03],
        [8.0183e-05, 9.9992e-01],
        [4.6792e-04, 9.9953e-01],
        [9.9940e-01, 5.9815e-04],
        [9.9970e-01, 2.9868e-04],
        [9.9950e-01, 4.9913e-04],
        [9.9814e-01, 1.8617e-03],
        [5.9083e-06, 9.9999e-01],
        [3.4783e-04, 9.9965e-01],
        [1.1342e-02, 9.8866e-01],
        [6.8867e-02, 9.3113e-01],
        [3.1965e-03, 9.9680e-01],
        [2.9128e-03, 9.9709e-01],
        [1.3775e-06, 1.0000e+00],
        [1.8687e-04, 9.9981e-01],
        [9.4581e-04, 9.9905e-01],
        [1.3054e-01, 8.6946e-01],
        [1.9886e-03, 9.9801e-01],
        [7.1678e-03, 9.9283e-01],
        [9.9959e-01, 4.0702e-04],
        [5.7856e-01, 4.2144e-01],
        [6.4923e-03, 9.9351e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.6140e-03, 9.9039e-01],
        [1.6355e-03, 9.9836e-01],
        [1.0404e-01, 8.9596e-01],
        [9.7204e-01, 2.7959e-02],
        [9.9863e-01, 1.3707e-03],
        [7.6551e-02, 9.2345e-01],
        [2.7229e-05, 9.9997e-01],
        [9.9528e-01, 4.7179e-03],
        [1.1150e-05, 9.9999e-01],
        [2.1839e-04, 9.9978e-01],
        [9.9933e-01, 6.6817e-04],
        [7.4957e-04, 9.9925e-01],
        [2.3058e-06, 1.0000e+00],
        [2.9732e-03, 9.9703e-01],
        [9.9795e-01, 2.0510e-03],
        [9.9150e-01, 8.5001e-03],
        [4.3645e-02, 9.5635e-01],
        [9.9899e-01, 1.0095e-03],
        [7.4010e-03, 9.9260e-01],
        [4.1378e-04, 9.9959e-01],
        [1.7591e-05, 9.9998e-01],
        [1.9776e-04, 9.9980e-01],
        [1.2493e-04, 9.9988e-01],
        [9.9892e-01, 1.0801e-03],
        [8.2244e-03, 9.9178e-01],
        [3.3751e-05, 9.9997e-01],
        [2.6197e-03, 9.9738e-01],
        [1.6747e-03, 9.9833e-01],
        [2.0683e-02, 9.7932e-01],
        [4.7423e-02, 9.5258e-01],
        [2.9727e-02, 9.7027e-01],
        [9.9937e-01, 6.2873e-04],
        [9.9955e-01, 4.5343e-04],
        [1.5502e-02, 9.8450e-01],
        [4.2004e-03, 9.9580e-01],
        [1.7802e-05, 9.9998e-01],
        [2.8520e-01, 7.1480e-01],
        [2.2523e-03, 9.9775e-01],
        [3.4203e-03, 9.9658e-01],
        [7.4551e-03, 9.9254e-01],
        [1.5197e-02, 9.8480e-01],
        [6.9809e-04, 9.9930e-01],
        [9.9899e-01, 1.0124e-03],
        [9.8795e-01, 1.2051e-02],
        [9.9971e-01, 2.9291e-04],
        [1.2431e-02, 9.8757e-01],
        [1.5246e-02, 9.8475e-01],
        [5.6090e-04, 9.9944e-01],
        [3.3153e-05, 9.9997e-01],
        [9.4304e-01, 5.6955e-02],
        [9.9962e-01, 3.8340e-04],
        [9.1985e-05, 9.9991e-01],
        [7.8438e-02, 9.2156e-01],
        [1.0111e-03, 9.9899e-01],
        [3.0368e-05, 9.9997e-01],
        [1.1531e-02, 9.8847e-01],
        [1.9877e-06, 1.0000e+00],
        [1.0209e-04, 9.9990e-01],
        [2.6266e-01, 7.3734e-01],
        [9.9904e-01, 9.6455e-04],
        [9.9990e-01, 1.0331e-04],
        [8.2950e-02, 9.1705e-01],
        [7.1851e-01, 2.8149e-01],
        [6.0347e-03, 9.9397e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.4065e-01, 4.5935e-01],
        [9.9834e-01, 1.6554e-03],
        [1.1048e-02, 9.8895e-01],
        [3.4572e-06, 1.0000e+00],
        [9.9959e-01, 4.1455e-04],
        [1.7523e-02, 9.8248e-01],
        [9.9870e-01, 1.3008e-03],
        [1.4392e-03, 9.9856e-01],
        [6.5204e-04, 9.9935e-01],
        [9.9819e-01, 1.8128e-03],
        [6.1142e-04, 9.9939e-01],
        [9.9936e-01, 6.3517e-04],
        [4.2897e-05, 9.9996e-01],
        [2.1814e-07, 1.0000e+00],
        [3.1352e-02, 9.6865e-01],
        [9.8490e-01, 1.5096e-02],
        [1.2928e-04, 9.9987e-01],
        [9.9944e-01, 5.5653e-04],
        [7.1118e-02, 9.2888e-01],
        [9.6286e-06, 9.9999e-01],
        [6.3556e-04, 9.9936e-01],
        [7.3060e-04, 9.9927e-01],
        [1.7972e-05, 9.9998e-01],
        [9.9979e-01, 2.0533e-04],
        [9.9944e-01, 5.6116e-04],
        [5.6239e-07, 1.0000e+00],
        [2.9117e-02, 9.7088e-01],
        [3.1177e-05, 9.9997e-01],
        [9.0228e-06, 9.9999e-01],
        [6.8053e-03, 9.9319e-01],
        [7.8978e-01, 2.1022e-01],
        [3.1660e-06, 1.0000e+00],
        [1.8310e-06, 1.0000e+00],
        [2.7468e-01, 7.2532e-01],
        [8.7572e-04, 9.9912e-01],
        [1.3847e-02, 9.8615e-01],
        [1.1279e-04, 9.9989e-01],
        [1.8142e-02, 9.8186e-01],
        [9.9877e-01, 1.2295e-03],
        [9.8042e-06, 9.9999e-01],
        [3.3913e-03, 9.9661e-01],
        [3.7051e-01, 6.2949e-01],
        [4.6499e-03, 9.9535e-01],
        [2.5000e-02, 9.7500e-01],
        [2.1615e-05, 9.9998e-01],
        [1.5377e-06, 1.0000e+00],
        [2.7396e-06, 1.0000e+00],
        [9.9608e-01, 3.9223e-03],
        [9.9907e-01, 9.3471e-04],
        [1.5145e-02, 9.8485e-01],
        [9.9944e-01, 5.6289e-04],
        [1.3004e-04, 9.9987e-01],
        [1.5925e-03, 9.9841e-01],
        [9.9860e-01, 1.4015e-03],
        [9.9930e-01, 7.0340e-04],
        [9.9995e-01, 5.1523e-05],
        [8.4705e-03, 9.9153e-01],
        [6.3618e-01, 3.6382e-01],
        [1.3718e-02, 9.8628e-01],
        [9.9946e-01, 5.3659e-04],
        [9.4615e-03, 9.9054e-01],
        [2.0798e-03, 9.9792e-01],
        [1.9837e-04, 9.9980e-01],
        [8.1620e-02, 9.1838e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2002e-04, 9.9968e-01],
        [5.6711e-02, 9.4329e-01],
        [3.6005e-04, 9.9964e-01],
        [9.9972e-01, 2.8413e-04],
        [8.5318e-04, 9.9915e-01],
        [2.3561e-03, 9.9764e-01],
        [9.9783e-01, 2.1735e-03],
        [4.5371e-05, 9.9995e-01],
        [4.7439e-01, 5.2561e-01],
        [7.5440e-02, 9.2456e-01],
        [2.5343e-04, 9.9975e-01],
        [4.6666e-01, 5.3334e-01],
        [3.1709e-02, 9.6829e-01],
        [7.5833e-05, 9.9992e-01],
        [5.2345e-05, 9.9995e-01],
        [9.7888e-01, 2.1125e-02],
        [7.9481e-04, 9.9921e-01],
        [2.0007e-02, 9.7999e-01],
        [2.1771e-01, 7.8229e-01],
        [1.4447e-04, 9.9986e-01],
        [8.0626e-05, 9.9992e-01],
        [7.2682e-04, 9.9927e-01],
        [1.9683e-02, 9.8032e-01],
        [1.0199e-01, 8.9801e-01],
        [6.3490e-01, 3.6510e-01],
        [9.9616e-01, 3.8397e-03],
        [9.9069e-01, 9.3119e-03],
        [9.4813e-04, 9.9905e-01],
        [4.7322e-02, 9.5268e-01],
        [1.9295e-04, 9.9981e-01],
        [4.1198e-05, 9.9996e-01],
        [2.0538e-03, 9.9795e-01],
        [1.3943e-01, 8.6057e-01],
        [7.2203e-06, 9.9999e-01],
        [6.7413e-05, 9.9993e-01],
        [9.6578e-02, 9.0342e-01],
        [9.7284e-02, 9.0272e-01],
        [4.9081e-02, 9.5092e-01],
        [4.1573e-04, 9.9958e-01],
        [1.6787e-04, 9.9983e-01],
        [7.3028e-04, 9.9927e-01],
        [9.9919e-01, 8.1165e-04],
        [1.1943e-02, 9.8806e-01],
        [6.9360e-02, 9.3064e-01],
        [9.9963e-05, 9.9990e-01],
        [3.3653e-02, 9.6635e-01],
        [1.1406e-04, 9.9989e-01],
        [1.2598e-04, 9.9987e-01],
        [2.5978e-03, 9.9740e-01],
        [1.0747e-02, 9.8925e-01],
        [9.6077e-01, 3.9233e-02],
        [9.9473e-01, 5.2684e-03],
        [8.6955e-03, 9.9130e-01],
        [9.7619e-06, 9.9999e-01],
        [2.4170e-03, 9.9758e-01],
        [9.1866e-01, 8.1335e-02],
        [5.4351e-07, 1.0000e+00],
        [1.1391e-03, 9.9886e-01],
        [6.0711e-02, 9.3929e-01],
        [2.8558e-01, 7.1442e-01],
        [9.9974e-01, 2.6247e-04],
        [1.2227e-01, 8.7773e-01],
        [5.5575e-04, 9.9944e-01],
        [4.9114e-04, 9.9951e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9914e-01, 8.5747e-04],
        [1.3529e-05, 9.9999e-01],
        [9.8859e-01, 1.1415e-02],
        [3.8441e-04, 9.9962e-01],
        [9.9921e-01, 7.9329e-04],
        [9.9406e-05, 9.9990e-01],
        [9.9763e-04, 9.9900e-01],
        [6.0385e-02, 9.3962e-01],
        [9.5812e-03, 9.9042e-01],
        [9.9953e-01, 4.6539e-04],
        [7.3202e-03, 9.9268e-01],
        [9.8108e-01, 1.8918e-02],
        [4.1821e-04, 9.9958e-01],
        [2.8407e-02, 9.7159e-01],
        [2.9778e-04, 9.9970e-01],
        [5.5920e-01, 4.4080e-01],
        [6.2297e-07, 1.0000e+00],
        [2.2845e-08, 1.0000e+00],
        [1.9762e-02, 9.8024e-01],
        [9.9954e-01, 4.6449e-04],
        [9.8792e-06, 9.9999e-01],
        [3.5814e-03, 9.9642e-01],
        [2.5796e-03, 9.9742e-01],
        [1.2392e-05, 9.9999e-01],
        [9.9969e-01, 3.1168e-04],
        [6.1051e-05, 9.9994e-01],
        [1.0445e-02, 9.8956e-01],
        [2.8149e-02, 9.7185e-01],
        [9.9710e-01, 2.9036e-03],
        [4.6353e-03, 9.9536e-01],
        [2.3787e-03, 9.9762e-01],
        [4.5927e-04, 9.9954e-01],
        [9.8451e-01, 1.5488e-02],
        [1.2479e-03, 9.9875e-01],
        [1.7519e-03, 9.9825e-01],
        [7.1143e-02, 9.2886e-01],
        [1.6508e-07, 1.0000e+00],
        [1.1523e-03, 9.9885e-01],
        [9.9996e-01, 3.7685e-05],
        [9.9062e-01, 9.3772e-03],
        [1.7546e-05, 9.9998e-01],
        [9.9889e-01, 1.1119e-03],
        [4.0321e-02, 9.5968e-01],
        [1.7107e-03, 9.9829e-01],
        [2.3590e-02, 9.7641e-01],
        [1.4007e-03, 9.9860e-01],
        [4.8330e-01, 5.1670e-01],
        [8.7562e-07, 1.0000e+00],
        [9.2773e-05, 9.9991e-01],
        [1.7209e-02, 9.8279e-01],
        [3.3267e-01, 6.6733e-01],
        [5.8487e-02, 9.4151e-01],
        [2.7097e-05, 9.9997e-01],
        [1.1823e-02, 9.8818e-01],
        [5.2846e-01, 4.7154e-01],
        [1.6117e-02, 9.8388e-01],
        [1.6369e-01, 8.3631e-01],
        [5.1122e-01, 4.8878e-01],
        [7.8253e-04, 9.9922e-01],
        [9.8030e-03, 9.9020e-01],
        [3.4705e-01, 6.5295e-01],
        [2.6445e-04, 9.9974e-01],
        [9.9867e-01, 1.3321e-03],
        [4.0063e-02, 9.5994e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3429e-07, 1.0000e+00],
        [2.3626e-02, 9.7637e-01],
        [7.3807e-02, 9.2619e-01],
        [1.1933e-06, 1.0000e+00],
        [9.8974e-01, 1.0257e-02],
        [8.8657e-07, 1.0000e+00],
        [1.1849e-04, 9.9988e-01],
        [9.9816e-01, 1.8398e-03],
        [2.3424e-04, 9.9977e-01],
        [4.5000e-01, 5.5000e-01],
        [1.3760e-03, 9.9862e-01],
        [8.6918e-03, 9.9131e-01],
        [9.9775e-01, 2.2470e-03],
        [2.5057e-06, 1.0000e+00],
        [2.1517e-03, 9.9785e-01],
        [9.9684e-01, 3.1598e-03],
        [2.3678e-05, 9.9998e-01],
        [7.5444e-04, 9.9925e-01],
        [8.6050e-01, 1.3950e-01],
        [2.6962e-06, 1.0000e+00],
        [2.7692e-06, 1.0000e+00],
        [6.4922e-02, 9.3508e-01],
        [2.6520e-05, 9.9997e-01],
        [9.9640e-01, 3.5980e-03],
        [7.4543e-01, 2.5457e-01],
        [9.9934e-01, 6.6455e-04],
        [1.1318e-04, 9.9989e-01],
        [9.9866e-01, 1.3436e-03],
        [1.5725e-05, 9.9998e-01],
        [2.8330e-05, 9.9997e-01],
        [9.9895e-01, 1.0525e-03],
        [1.7142e-03, 9.9829e-01],
        [2.4049e-02, 9.7595e-01],
        [1.1949e-07, 1.0000e+00],
        [3.9056e-04, 9.9961e-01],
        [8.2339e-05, 9.9992e-01],
        [9.9722e-01, 2.7833e-03],
        [9.7299e-01, 2.7008e-02],
        [2.2509e-04, 9.9977e-01],
        [9.8883e-01, 1.1166e-02],
        [9.9943e-01, 5.6925e-04],
        [9.9932e-01, 6.7532e-04],
        [3.1486e-06, 1.0000e+00],
        [7.0059e-02, 9.2994e-01],
        [6.7512e-04, 9.9932e-01],
        [4.1234e-06, 1.0000e+00],
        [9.9636e-01, 3.6419e-03],
        [2.5409e-03, 9.9746e-01],
        [8.0375e-01, 1.9625e-01],
        [9.9824e-01, 1.7609e-03],
        [3.4444e-05, 9.9997e-01],
        [1.3640e-05, 9.9999e-01],
        [8.2001e-06, 9.9999e-01],
        [9.9721e-01, 2.7887e-03],
        [3.3491e-04, 9.9967e-01],
        [2.9250e-01, 7.0750e-01],
        [2.2173e-02, 9.7783e-01],
        [5.9414e-02, 9.4059e-01],
        [9.9817e-01, 1.8278e-03],
        [3.7633e-05, 9.9996e-01],
        [9.9939e-01, 6.1302e-04],
        [1.3421e-02, 9.8658e-01],
        [9.7420e-03, 9.9026e-01],
        [2.5853e-03, 9.9741e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3824e-04, 9.9986e-01],
        [9.9621e-01, 3.7890e-03],
        [4.7939e-05, 9.9995e-01],
        [3.0359e-04, 9.9970e-01],
        [9.8888e-01, 1.1122e-02],
        [4.3059e-01, 5.6941e-01],
        [4.3330e-04, 9.9957e-01],
        [2.4105e-04, 9.9976e-01],
        [1.1411e-05, 9.9999e-01],
        [2.9155e-05, 9.9997e-01],
        [2.6206e-04, 9.9974e-01],
        [6.5453e-06, 9.9999e-01],
        [9.9828e-01, 1.7194e-03],
        [6.5176e-08, 1.0000e+00],
        [9.9967e-01, 3.2750e-04],
        [7.6166e-03, 9.9238e-01],
        [9.8428e-01, 1.5725e-02],
        [2.3208e-03, 9.9768e-01],
        [7.7245e-05, 9.9992e-01],
        [1.6102e-01, 8.3898e-01],
        [5.5091e-03, 9.9449e-01],
        [9.9749e-01, 2.5112e-03],
        [5.9079e-05, 9.9994e-01],
        [4.8354e-02, 9.5165e-01],
        [4.1342e-04, 9.9959e-01],
        [1.3873e-03, 9.9861e-01],
        [1.6808e-05, 9.9998e-01],
        [9.9661e-01, 3.3906e-03],
        [9.2918e-02, 9.0708e-01],
        [1.0181e-03, 9.9898e-01],
        [1.7328e-03, 9.9827e-01],
        [6.8726e-01, 3.1274e-01],
        [3.5285e-06, 1.0000e+00],
        [1.1317e-04, 9.9989e-01],
        [8.1160e-05, 9.9992e-01],
        [2.0444e-03, 9.9796e-01],
        [5.3723e-05, 9.9995e-01],
        [2.4987e-01, 7.5013e-01],
        [3.6682e-05, 9.9996e-01],
        [9.9917e-01, 8.3026e-04],
        [1.8008e-04, 9.9982e-01],
        [2.1505e-04, 9.9979e-01],
        [1.4521e-06, 1.0000e+00],
        [8.3503e-04, 9.9916e-01],
        [1.4798e-05, 9.9999e-01],
        [9.9971e-01, 2.9251e-04],
        [4.3959e-03, 9.9560e-01],
        [9.3811e-01, 6.1892e-02],
        [4.0396e-04, 9.9960e-01],
        [3.5321e-07, 1.0000e+00],
        [1.6595e-03, 9.9834e-01],
        [9.9634e-01, 3.6615e-03],
        [4.7313e-04, 9.9953e-01],
        [9.9885e-01, 1.1473e-03],
        [9.9070e-01, 9.2995e-03],
        [9.8656e-01, 1.3444e-02],
        [1.7832e-07, 1.0000e+00],
        [1.3762e-05, 9.9999e-01],
        [9.0781e-06, 9.9999e-01],
        [9.4287e-07, 1.0000e+00],
        [3.4334e-07, 1.0000e+00],
        [4.4713e-02, 9.5529e-01],
        [9.9898e-01, 1.0220e-03],
        [9.9696e-01, 3.0444e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.7198e-01, 2.8020e-02],
        [3.9350e-06, 1.0000e+00],
        [9.8011e-01, 1.9889e-02],
        [5.8143e-06, 9.9999e-01],
        [9.9345e-01, 6.5491e-03],
        [1.5013e-02, 9.8499e-01],
        [9.6766e-01, 3.2339e-02],
        [6.8840e-01, 3.1160e-01],
        [6.0264e-05, 9.9994e-01],
        [3.9245e-06, 1.0000e+00],
        [1.3708e-03, 9.9863e-01],
        [2.7047e-06, 1.0000e+00],
        [1.0293e-02, 9.8971e-01],
        [3.1957e-08, 1.0000e+00],
        [9.9942e-01, 5.7848e-04],
        [9.4873e-01, 5.1267e-02],
        [4.8402e-04, 9.9952e-01],
        [9.1648e-01, 8.3519e-02],
        [5.5649e-03, 9.9444e-01],
        [3.9613e-05, 9.9996e-01],
        [1.2096e-04, 9.9988e-01],
        [2.1487e-04, 9.9979e-01],
        [9.9295e-01, 7.0523e-03],
        [9.9876e-01, 1.2359e-03],
        [3.6741e-05, 9.9996e-01],
        [9.8527e-01, 1.4731e-02],
        [5.4210e-05, 9.9995e-01],
        [9.5549e-05, 9.9990e-01],
        [7.5020e-05, 9.9993e-01],
        [2.9766e-04, 9.9970e-01],
        [1.0790e-05, 9.9999e-01],
        [9.9204e-01, 7.9612e-03],
        [9.5194e-03, 9.9048e-01],
        [5.1343e-04, 9.9949e-01],
        [4.6969e-08, 1.0000e+00],
        [2.6132e-06, 1.0000e+00],
        [1.9366e-02, 9.8063e-01],
        [9.9769e-01, 2.3136e-03],
        [5.0632e-04, 9.9949e-01],
        [5.8474e-01, 4.1526e-01],
        [4.9410e-07, 1.0000e+00],
        [9.9857e-01, 1.4295e-03],
        [3.3964e-05, 9.9997e-01],
        [1.0226e-07, 1.0000e+00],
        [1.1064e-02, 9.8894e-01],
        [1.1764e-07, 1.0000e+00],
        [1.1557e-06, 1.0000e+00],
        [9.1828e-06, 9.9999e-01],
        [9.1088e-01, 8.9115e-02],
        [9.0538e-01, 9.4624e-02],
        [2.7414e-06, 1.0000e+00],
        [3.5688e-07, 1.0000e+00],
        [6.7189e-07, 1.0000e+00],
        [8.3541e-01, 1.6459e-01],
        [1.0748e-06, 1.0000e+00],
        [1.3810e-05, 9.9999e-01],
        [1.8360e-08, 1.0000e+00],
        [3.6623e-06, 1.0000e+00],
        [4.0363e-05, 9.9996e-01],
        [3.1359e-03, 9.9686e-01],
        [3.9802e-06, 1.0000e+00],
        [2.3124e-05, 9.9998e-01],
        [3.1195e-05, 9.9997e-01],
        [1.0113e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.0188e-05, 9.9992e-01],
        [1.0380e-05, 9.9999e-01],
        [1.9076e-03, 9.9809e-01],
        [6.7810e-04, 9.9932e-01],
        [8.7517e-06, 9.9999e-01],
        [3.2207e-06, 1.0000e+00],
        [3.3725e-04, 9.9966e-01],
        [1.6749e-06, 1.0000e+00],
        [2.2825e-05, 9.9998e-01],
        [9.8148e-01, 1.8518e-02],
        [9.9748e-01, 2.5197e-03],
        [4.9781e-04, 9.9950e-01],
        [9.9555e-01, 4.4538e-03],
        [4.8710e-03, 9.9513e-01],
        [7.8624e-07, 1.0000e+00],
        [1.0198e-02, 9.8980e-01],
        [8.8831e-08, 1.0000e+00],
        [9.9897e-01, 1.0303e-03],
        [9.7867e-01, 2.1335e-02],
        [9.9932e-01, 6.8133e-04],
        [9.8949e-01, 1.0510e-02],
        [6.3278e-04, 9.9937e-01],
        [6.4101e-04, 9.9936e-01],
        [6.2751e-07, 1.0000e+00],
        [1.0117e-07, 1.0000e+00],
        [1.9083e-04, 9.9981e-01],
        [6.3351e-02, 9.3665e-01],
        [9.9703e-01, 2.9689e-03],
        [9.9994e-01, 5.6180e-05],
        [3.9705e-06, 1.0000e+00],
        [2.3383e-03, 9.9766e-01],
        [1.5538e-04, 9.9984e-01],
        [1.1031e-07, 1.0000e+00],
        [9.8346e-01, 1.6540e-02],
        [4.2283e-09, 1.0000e+00],
        [9.3184e-01, 6.8160e-02],
        [1.5932e-02, 9.8407e-01],
        [7.9585e-04, 9.9920e-01],
        [5.2676e-01, 4.7324e-01],
        [1.7616e-05, 9.9998e-01],
        [5.8578e-06, 9.9999e-01],
        [3.8790e-08, 1.0000e+00],
        [5.4330e-04, 9.9946e-01],
        [2.9723e-05, 9.9997e-01],
        [9.8380e-01, 1.6203e-02],
        [3.5862e-03, 9.9641e-01],
        [9.0409e-04, 9.9910e-01],
        [6.9596e-04, 9.9930e-01],
        [2.2485e-04, 9.9978e-01],
        [8.5831e-04, 9.9914e-01],
        [9.9983e-01, 1.7282e-04],
        [9.3302e-01, 6.6980e-02],
        [2.0451e-03, 9.9795e-01],
        [1.3181e-05, 9.9999e-01],
        [9.1882e-01, 8.1184e-02],
        [9.9668e-01, 3.3198e-03],
        [7.0062e-08, 1.0000e+00],
        [9.9925e-01, 7.4778e-04],
        [5.2938e-07, 1.0000e+00],
        [9.9639e-01, 3.6066e-03],
        [6.7255e-05, 9.9993e-01],
        [1.7706e-03, 9.9823e-01],
        [2.2000e-03, 9.9780e-01],
        [7.0013e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0054e-07, 1.0000e+00],
        [1.4822e-01, 8.5178e-01],
        [9.8459e-01, 1.5414e-02],
        [3.6133e-07, 1.0000e+00],
        [4.3737e-06, 1.0000e+00],
        [1.3442e-06, 1.0000e+00],
        [9.9089e-01, 9.1134e-03],
        [8.0852e-05, 9.9992e-01],
        [1.3819e-05, 9.9999e-01],
        [3.3917e-06, 1.0000e+00],
        [4.4773e-03, 9.9552e-01],
        [2.9795e-05, 9.9997e-01],
        [5.9279e-05, 9.9994e-01],
        [1.4949e-05, 9.9999e-01],
        [1.2376e-04, 9.9988e-01],
        [9.3335e-01, 6.6649e-02],
        [2.2929e-05, 9.9998e-01],
        [9.5485e-01, 4.5148e-02],
        [9.7493e-01, 2.5069e-02],
        [3.4504e-06, 1.0000e+00],
        [4.7954e-06, 1.0000e+00],
        [6.8043e-04, 9.9932e-01],
        [8.1463e-07, 1.0000e+00],
        [9.9907e-01, 9.2761e-04],
        [1.2445e-06, 1.0000e+00],
        [5.8913e-04, 9.9941e-01],
        [2.3604e-05, 9.9998e-01],
        [1.6745e-03, 9.9833e-01],
        [9.9894e-01, 1.0634e-03],
        [9.9456e-01, 5.4417e-03],
        [1.5746e-06, 1.0000e+00],
        [1.8185e-02, 9.8182e-01],
        [5.5463e-06, 9.9999e-01],
        [7.2688e-02, 9.2731e-01],
        [1.0916e-06, 1.0000e+00],
        [7.9907e-07, 1.0000e+00],
        [9.5339e-03, 9.9047e-01],
        [3.2603e-05, 9.9997e-01],
        [1.8187e-06, 1.0000e+00],
        [6.6887e-01, 3.3113e-01],
        [5.7707e-05, 9.9994e-01],
        [9.9850e-01, 1.5027e-03],
        [4.6637e-06, 1.0000e+00],
        [7.9094e-08, 1.0000e+00],
        [9.6641e-01, 3.3586e-02],
        [1.1899e-04, 9.9988e-01],
        [9.8214e-06, 9.9999e-01],
        [1.0465e-04, 9.9990e-01],
        [7.7291e-05, 9.9992e-01],
        [1.8206e-02, 9.8179e-01],
        [4.7146e-05, 9.9995e-01],
        [6.9215e-04, 9.9931e-01],
        [9.9909e-01, 9.1278e-04],
        [9.9872e-01, 1.2765e-03],
        [6.3782e-01, 3.6218e-01],
        [9.9403e-01, 5.9685e-03],
        [1.0353e-04, 9.9990e-01],
        [5.9154e-06, 9.9999e-01],
        [1.2046e-04, 9.9988e-01],
        [2.7563e-07, 1.0000e+00],
        [9.6056e-01, 3.9439e-02],
        [9.9937e-01, 6.2867e-04],
        [1.8501e-06, 1.0000e+00],
        [3.1619e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.4462e-05, 9.9992e-01],
        [1.4863e-06, 1.0000e+00],
        [9.9765e-01, 2.3477e-03],
        [9.9807e-01, 1.9327e-03],
        [1.0464e-03, 9.9895e-01],
        [8.5638e-06, 9.9999e-01],
        [4.8194e-04, 9.9952e-01],
        [4.6391e-05, 9.9995e-01],
        [3.6712e-07, 1.0000e+00],
        [9.4131e-07, 1.0000e+00],
        [1.8721e-07, 1.0000e+00],
        [3.0270e-04, 9.9970e-01],
        [2.5637e-03, 9.9744e-01],
        [4.5500e-01, 5.4500e-01],
        [7.6621e-05, 9.9992e-01],
        [4.5160e-03, 9.9548e-01],
        [3.4203e-05, 9.9997e-01],
        [8.8166e-01, 1.1834e-01],
        [2.0008e-04, 9.9980e-01],
        [2.4657e-02, 9.7534e-01],
        [6.0184e-04, 9.9940e-01],
        [9.9903e-01, 9.6956e-04],
        [5.3875e-04, 9.9946e-01],
        [3.1767e-06, 1.0000e+00],
        [9.9194e-01, 8.0605e-03],
        [9.8691e-01, 1.3091e-02],
        [2.9013e-06, 1.0000e+00],
        [9.2759e-03, 9.9072e-01],
        [9.7683e-01, 2.3171e-02],
        [9.9413e-01, 5.8713e-03],
        [2.1211e-04, 9.9979e-01],
        [9.4502e-07, 1.0000e+00],
        [7.2422e-01, 2.7578e-01],
        [1.6880e-05, 9.9998e-01],
        [1.1863e-06, 1.0000e+00],
        [3.8425e-03, 9.9616e-01],
        [5.5881e-04, 9.9944e-01],
        [9.9652e-01, 3.4844e-03],
        [2.9639e-01, 7.0361e-01],
        [8.4093e-04, 9.9916e-01],
        [9.9451e-01, 5.4861e-03],
        [1.7407e-06, 1.0000e+00],
        [9.0954e-01, 9.0458e-02],
        [1.0645e-03, 9.9894e-01],
        [1.0094e-03, 9.9899e-01],
        [2.8290e-08, 1.0000e+00],
        [2.8912e-06, 1.0000e+00],
        [5.1162e-05, 9.9995e-01],
        [5.5394e-06, 9.9999e-01],
        [3.3615e-06, 1.0000e+00],
        [7.1352e-04, 9.9929e-01],
        [4.1235e-06, 1.0000e+00],
        [2.5502e-08, 1.0000e+00],
        [1.1183e-09, 1.0000e+00],
        [9.9835e-01, 1.6507e-03],
        [6.4341e-04, 9.9936e-01],
        [4.5301e-08, 1.0000e+00],
        [9.4125e-01, 5.8749e-02],
        [1.7643e-03, 9.9824e-01],
        [5.2886e-04, 9.9947e-01],
        [9.9936e-01, 6.4212e-04],
        [1.1660e-05, 9.9999e-01],
        [6.1672e-05, 9.9994e-01],
        [9.4446e-04, 9.9906e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9942e-01, 5.8496e-04],
        [5.5290e-05, 9.9994e-01],
        [2.5656e-02, 9.7434e-01],
        [6.2194e-04, 9.9938e-01],
        [2.0094e-03, 9.9799e-01],
        [5.7524e-04, 9.9942e-01],
        [7.6836e-04, 9.9923e-01],
        [2.3562e-06, 1.0000e+00],
        [9.9956e-01, 4.3530e-04],
        [6.3976e-05, 9.9994e-01],
        [9.5972e-05, 9.9990e-01],
        [1.2597e-06, 1.0000e+00],
        [1.7104e-06, 1.0000e+00],
        [9.9558e-01, 4.4201e-03],
        [2.6301e-07, 1.0000e+00],
        [2.1874e-06, 1.0000e+00],
        [1.3242e-04, 9.9987e-01],
        [2.8837e-06, 1.0000e+00],
        [6.3756e-01, 3.6244e-01],
        [2.0044e-05, 9.9998e-01],
        [4.1494e-05, 9.9996e-01],
        [9.8332e-01, 1.6682e-02],
        [1.4732e-05, 9.9999e-01],
        [9.9245e-01, 7.5540e-03],
        [9.9368e-01, 6.3230e-03],
        [9.9974e-01, 2.6410e-04],
        [9.9115e-01, 8.8490e-03],
        [8.9165e-05, 9.9991e-01],
        [9.8644e-05, 9.9990e-01],
        [1.9675e-06, 1.0000e+00],
        [2.7723e-06, 1.0000e+00],
        [7.7065e-04, 9.9923e-01],
        [2.2754e-06, 1.0000e+00],
        [9.9747e-01, 2.5306e-03],
        [4.0364e-06, 1.0000e+00],
        [2.0871e-04, 9.9979e-01],
        [1.4262e-03, 9.9857e-01],
        [7.7675e-01, 2.2325e-01],
        [9.9245e-01, 7.5540e-03],
        [1.9102e-02, 9.8090e-01],
        [1.1868e-07, 1.0000e+00],
        [1.6245e-06, 1.0000e+00],
        [5.2319e-07, 1.0000e+00],
        [6.9839e-03, 9.9302e-01],
        [5.8350e-01, 4.1650e-01],
        [5.7756e-04, 9.9942e-01],
        [3.1801e-05, 9.9997e-01],
        [6.5907e-03, 9.9341e-01],
        [8.6560e-08, 1.0000e+00],
        [2.5242e-06, 1.0000e+00],
        [1.6703e-08, 1.0000e+00],
        [1.9359e-04, 9.9981e-01],
        [7.0442e-05, 9.9993e-01],
        [8.5224e-05, 9.9991e-01],
        [1.4732e-01, 8.5268e-01],
        [3.1504e-05, 9.9997e-01],
        [7.1447e-01, 2.8553e-01],
        [3.3483e-03, 9.9665e-01],
        [3.6371e-06, 1.0000e+00],
        [9.9188e-01, 8.1195e-03],
        [6.0111e-04, 9.9940e-01],
        [2.7312e-07, 1.0000e+00],
        [5.9710e-05, 9.9994e-01],
        [4.5523e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.9567e-06, 9.9999e-01],
        [3.6073e-01, 6.3927e-01],
        [4.9908e-04, 9.9950e-01],
        [3.7488e-04, 9.9963e-01],
        [5.2124e-04, 9.9948e-01],
        [3.8415e-03, 9.9616e-01],
        [5.1966e-04, 9.9948e-01],
        [5.1616e-04, 9.9948e-01],
        [9.9903e-01, 9.6843e-04],
        [2.3095e-02, 9.7690e-01],
        [6.3261e-06, 9.9999e-01],
        [4.0559e-07, 1.0000e+00],
        [1.2460e-06, 1.0000e+00],
        [4.6170e-03, 9.9538e-01],
        [9.7655e-01, 2.3448e-02],
        [9.8143e-01, 1.8571e-02],
        [3.2867e-05, 9.9997e-01],
        [1.2467e-04, 9.9988e-01],
        [1.1167e-05, 9.9999e-01],
        [3.1656e-07, 1.0000e+00],
        [9.0559e-01, 9.4408e-02],
        [2.8473e-04, 9.9972e-01],
        [9.6323e-01, 3.6775e-02],
        [9.3767e-08, 1.0000e+00],
        [7.1830e-03, 9.9282e-01],
        [1.4500e-07, 1.0000e+00],
        [4.4473e-06, 1.0000e+00],
        [9.8581e-01, 1.4188e-02],
        [4.5307e-07, 1.0000e+00],
        [7.6743e-01, 2.3257e-01],
        [9.1325e-03, 9.9087e-01],
        [9.6663e-01, 3.3374e-02],
        [1.6918e-06, 1.0000e+00],
        [3.1265e-05, 9.9997e-01],
        [1.7803e-05, 9.9998e-01],
        [1.4370e-06, 1.0000e+00],
        [5.7819e-09, 1.0000e+00],
        [6.8499e-01, 3.1501e-01],
        [2.4490e-04, 9.9976e-01],
        [1.8895e-06, 1.0000e+00],
        [9.2038e-07, 1.0000e+00],
        [1.4063e-03, 9.9859e-01],
        [9.6853e-04, 9.9903e-01],
        [9.9978e-01, 2.2408e-04],
        [4.7179e-02, 9.5282e-01],
        [1.8718e-04, 9.9981e-01],
        [8.2303e-05, 9.9992e-01],
        [1.6445e-08, 1.0000e+00],
        [3.5748e-07, 1.0000e+00],
        [9.6778e-06, 9.9999e-01],
        [4.8963e-08, 1.0000e+00],
        [5.2149e-06, 9.9999e-01],
        [1.7359e-03, 9.9826e-01],
        [2.8029e-02, 9.7197e-01],
        [4.2158e-04, 9.9958e-01],
        [1.5858e-05, 9.9998e-01],
        [5.0388e-04, 9.9950e-01],
        [2.7867e-05, 9.9997e-01],
        [8.2611e-03, 9.9174e-01],
        [9.9681e-01, 3.1895e-03],
        [6.6533e-07, 1.0000e+00],
        [1.3895e-06, 1.0000e+00],
        [9.9917e-01, 8.2767e-04],
        [3.9449e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2030e-03, 9.9680e-01],
        [9.9908e-01, 9.2451e-04],
        [8.2269e-05, 9.9992e-01],
        [2.1417e-04, 9.9979e-01],
        [2.2678e-04, 9.9977e-01],
        [9.8163e-01, 1.8372e-02],
        [4.1125e-01, 5.8875e-01],
        [6.5679e-03, 9.9343e-01],
        [3.7232e-05, 9.9996e-01],
        [9.9922e-01, 7.8014e-04],
        [2.7753e-06, 1.0000e+00],
        [2.9620e-07, 1.0000e+00],
        [9.5088e-01, 4.9118e-02],
        [3.1455e-06, 1.0000e+00],
        [9.8697e-01, 1.3028e-02],
        [8.8005e-01, 1.1995e-01],
        [2.3676e-05, 9.9998e-01],
        [2.1476e-05, 9.9998e-01],
        [3.9339e-02, 9.6066e-01],
        [7.1139e-06, 9.9999e-01],
        [3.7956e-04, 9.9962e-01],
        [5.0836e-01, 4.9164e-01],
        [3.8436e-02, 9.6156e-01],
        [4.8424e-04, 9.9952e-01],
        [7.6124e-07, 1.0000e+00],
        [7.8877e-01, 2.1123e-01],
        [7.6838e-06, 9.9999e-01],
        [8.2098e-01, 1.7902e-01],
        [2.1823e-05, 9.9998e-01],
        [9.6834e-01, 3.1658e-02],
        [4.5354e-03, 9.9546e-01],
        [9.8393e-01, 1.6071e-02],
        [1.0354e-04, 9.9990e-01],
        [1.6687e-03, 9.9833e-01],
        [9.9951e-01, 4.9366e-04],
        [1.0354e-04, 9.9990e-01],
        [8.7414e-06, 9.9999e-01],
        [7.0932e-06, 9.9999e-01],
        [9.8510e-01, 1.4898e-02],
        [3.1034e-05, 9.9997e-01],
        [8.0842e-04, 9.9919e-01],
        [1.7501e-06, 1.0000e+00],
        [8.0166e-05, 9.9992e-01],
        [9.9983e-01, 1.6927e-04],
        [5.4592e-04, 9.9945e-01],
        [1.2899e-03, 9.9871e-01],
        [1.0415e-04, 9.9990e-01],
        [3.6754e-06, 1.0000e+00],
        [9.9894e-01, 1.0599e-03],
        [9.8118e-01, 1.8821e-02],
        [4.3268e-02, 9.5673e-01],
        [8.4511e-06, 9.9999e-01],
        [7.4301e-07, 1.0000e+00],
        [9.9971e-01, 2.8879e-04],
        [1.9525e-04, 9.9980e-01],
        [8.4074e-06, 9.9999e-01],
        [1.4495e-06, 1.0000e+00],
        [3.9551e-05, 9.9996e-01],
        [3.5346e-06, 1.0000e+00],
        [9.9941e-01, 5.9202e-04],
        [7.5277e-06, 9.9999e-01],
        [1.9739e-06, 1.0000e+00],
        [1.4103e-06, 1.0000e+00],
        [1.0780e-03, 9.9892e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9948e-01, 5.1559e-04],
        [5.5459e-05, 9.9994e-01],
        [8.2326e-05, 9.9992e-01],
        [9.9873e-01, 1.2667e-03],
        [9.9830e-01, 1.7038e-03],
        [3.6542e-04, 9.9963e-01],
        [6.9694e-06, 9.9999e-01],
        [9.5521e-01, 4.4791e-02],
        [5.5175e-04, 9.9945e-01],
        [7.8827e-04, 9.9921e-01],
        [4.1485e-04, 9.9959e-01],
        [5.8829e-07, 1.0000e+00],
        [2.9227e-05, 9.9997e-01],
        [1.4624e-03, 9.9854e-01],
        [9.9994e-01, 5.6605e-05],
        [6.2827e-05, 9.9994e-01],
        [8.7679e-07, 1.0000e+00],
        [2.2149e-03, 9.9779e-01],
        [1.4725e-03, 9.9853e-01],
        [9.9864e-01, 1.3568e-03],
        [1.3147e-05, 9.9999e-01],
        [9.9868e-01, 1.3249e-03],
        [6.3590e-08, 1.0000e+00],
        [5.4883e-06, 9.9999e-01],
        [9.9422e-01, 5.7831e-03],
        [2.3424e-06, 1.0000e+00],
        [1.8629e-05, 9.9998e-01],
        [9.9948e-01, 5.1776e-04],
        [6.8436e-06, 9.9999e-01],
        [9.9918e-01, 8.2261e-04],
        [9.9811e-01, 1.8911e-03],
        [9.9934e-01, 6.6265e-04],
        [2.7842e-07, 1.0000e+00],
        [4.7634e-01, 5.2366e-01],
        [5.2825e-07, 1.0000e+00],
        [2.0416e-01, 7.9584e-01],
        [8.6482e-07, 1.0000e+00],
        [3.6008e-03, 9.9640e-01],
        [3.6223e-05, 9.9996e-01],
        [6.8656e-06, 9.9999e-01],
        [7.2271e-06, 9.9999e-01],
        [1.6929e-04, 9.9983e-01],
        [1.6307e-02, 9.8369e-01],
        [6.4101e-05, 9.9994e-01],
        [1.8339e-01, 8.1661e-01],
        [3.9609e-02, 9.6039e-01],
        [4.1759e-04, 9.9958e-01],
        [9.7825e-01, 2.1750e-02],
        [7.2422e-03, 9.9276e-01],
        [9.9895e-01, 1.0487e-03],
        [4.5890e-06, 1.0000e+00],
        [3.1566e-05, 9.9997e-01],
        [3.8072e-04, 9.9962e-01],
        [9.9540e-01, 4.6012e-03],
        [4.3052e-03, 9.9569e-01],
        [1.0088e-03, 9.9899e-01],
        [1.5730e-03, 9.9843e-01],
        [9.9676e-01, 3.2434e-03],
        [4.3398e-05, 9.9996e-01],
        [6.8558e-02, 9.3144e-01],
        [1.0532e-06, 1.0000e+00],
        [9.9992e-01, 7.7179e-05],
        [9.6425e-01, 3.5748e-02],
        [9.9720e-04, 9.9900e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.2715e-05, 9.9993e-01],
        [3.1723e-03, 9.9683e-01],
        [3.1501e-05, 9.9997e-01],
        [8.1763e-07, 1.0000e+00],
        [3.2081e-01, 6.7919e-01],
        [4.3866e-02, 9.5613e-01],
        [9.9165e-01, 8.3521e-03],
        [9.9651e-01, 3.4933e-03],
        [2.9069e-04, 9.9971e-01],
        [5.8564e-01, 4.1436e-01],
        [9.9690e-01, 3.1005e-03],
        [3.2860e-05, 9.9997e-01],
        [1.0492e-04, 9.9990e-01],
        [9.9974e-01, 2.5711e-04],
        [9.1983e-01, 8.0168e-02],
        [1.8337e-03, 9.9817e-01],
        [9.9494e-01, 5.0640e-03],
        [5.9180e-03, 9.9408e-01],
        [3.4299e-05, 9.9997e-01],
        [3.1617e-05, 9.9997e-01],
        [1.8965e-03, 9.9810e-01],
        [1.1697e-02, 9.8830e-01],
        [1.0883e-03, 9.9891e-01],
        [9.9965e-01, 3.5448e-04],
        [9.7968e-01, 2.0322e-02],
        [3.1558e-05, 9.9997e-01],
        [3.5116e-04, 9.9965e-01],
        [6.3009e-05, 9.9994e-01],
        [4.6586e-07, 1.0000e+00],
        [1.7563e-07, 1.0000e+00],
        [6.0632e-05, 9.9994e-01],
        [2.1147e-01, 7.8853e-01],
        [1.9818e-05, 9.9998e-01],
        [6.6114e-02, 9.3389e-01],
        [9.9806e-01, 1.9358e-03],
        [2.4373e-06, 1.0000e+00],
        [9.4706e-04, 9.9905e-01],
        [1.0293e-05, 9.9999e-01],
        [3.5991e-05, 9.9996e-01],
        [2.3963e-04, 9.9976e-01],
        [9.5860e-01, 4.1397e-02],
        [1.0000e+00, 1.8146e-06],
        [7.9903e-07, 1.0000e+00],
        [7.4474e-05, 9.9993e-01],
        [8.3833e-01, 1.6167e-01],
        [2.8814e-03, 9.9712e-01],
        [4.3210e-04, 9.9957e-01],
        [5.9908e-01, 4.0092e-01],
        [3.2888e-05, 9.9997e-01],
        [1.9438e-07, 1.0000e+00],
        [2.2462e-05, 9.9998e-01],
        [5.2886e-02, 9.4711e-01],
        [5.1658e-06, 9.9999e-01],
        [8.7948e-06, 9.9999e-01],
        [1.1159e-03, 9.9888e-01],
        [2.1658e-03, 9.9783e-01],
        [3.9646e-06, 1.0000e+00],
        [1.0780e-01, 8.9220e-01],
        [6.0364e-05, 9.9994e-01],
        [2.9049e-05, 9.9997e-01],
        [1.3190e-03, 9.9868e-01],
        [9.9836e-01, 1.6431e-03],
        [9.7250e-01, 2.7495e-02],
        [2.5695e-04, 9.9974e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9907e-01, 9.2720e-04],
        [2.7260e-03, 9.9727e-01],
        [9.9601e-01, 3.9880e-03],
        [1.4614e-04, 9.9985e-01],
        [1.5366e-04, 9.9985e-01],
        [5.8546e-08, 1.0000e+00],
        [9.7994e-06, 9.9999e-01],
        [8.6364e-04, 9.9914e-01],
        [3.0165e-06, 1.0000e+00],
        [3.9607e-04, 9.9960e-01],
        [1.6115e-04, 9.9984e-01],
        [2.1146e-02, 9.7885e-01],
        [1.2156e-03, 9.9878e-01],
        [9.9873e-01, 1.2702e-03],
        [9.9939e-01, 6.0555e-04],
        [2.6027e-04, 9.9974e-01],
        [9.9837e-01, 1.6341e-03],
        [3.2007e-05, 9.9997e-01],
        [3.5499e-05, 9.9996e-01],
        [9.8667e-01, 1.3335e-02],
        [9.9299e-01, 7.0145e-03],
        [9.9966e-01, 3.4319e-04],
        [7.9211e-05, 9.9992e-01],
        [1.4870e-02, 9.8513e-01],
        [2.4211e-03, 9.9758e-01],
        [8.7865e-03, 9.9121e-01],
        [9.9967e-01, 3.3302e-04],
        [7.0741e-05, 9.9993e-01],
        [9.7656e-02, 9.0234e-01],
        [3.7398e-01, 6.2602e-01],
        [5.8402e-06, 9.9999e-01],
        [1.5859e-05, 9.9998e-01],
        [9.9851e-01, 1.4866e-03],
        [1.4154e-04, 9.9986e-01],
        [1.3801e-03, 9.9862e-01],
        [5.1034e-05, 9.9995e-01],
        [4.6983e-01, 5.3017e-01],
        [1.7873e-04, 9.9982e-01],
        [8.6799e-04, 9.9913e-01],
        [6.2109e-01, 3.7891e-01],
        [9.9876e-01, 1.2426e-03],
        [7.7725e-04, 9.9922e-01],
        [6.8765e-03, 9.9312e-01],
        [4.6189e-06, 1.0000e+00],
        [9.6578e-01, 3.4216e-02],
        [2.8736e-02, 9.7126e-01],
        [1.9624e-06, 1.0000e+00],
        [6.8483e-04, 9.9932e-01],
        [9.2116e-04, 9.9908e-01],
        [5.1217e-04, 9.9949e-01],
        [1.8842e-05, 9.9998e-01],
        [2.8035e-03, 9.9720e-01],
        [4.2746e-04, 9.9957e-01],
        [6.8687e-04, 9.9931e-01],
        [9.9955e-01, 4.4792e-04],
        [8.0613e-07, 1.0000e+00],
        [1.1798e-05, 9.9999e-01],
        [2.6900e-07, 1.0000e+00],
        [9.9962e-01, 3.8059e-04],
        [1.6150e-04, 9.9984e-01],
        [9.9923e-01, 7.7101e-04],
        [3.4640e-02, 9.6536e-01],
        [4.0305e-05, 9.9996e-01],
        [9.9920e-01, 7.9921e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5036e-04, 9.9965e-01],
        [8.9587e-03, 9.9104e-01],
        [9.9023e-01, 9.7733e-03],
        [6.6386e-04, 9.9934e-01],
        [7.7157e-05, 9.9992e-01],
        [9.9904e-01, 9.6212e-04],
        [4.8102e-04, 9.9952e-01],
        [9.9893e-01, 1.0676e-03],
        [3.9445e-03, 9.9606e-01],
        [6.0545e-01, 3.9455e-01],
        [5.1010e-05, 9.9995e-01],
        [9.9961e-01, 3.8647e-04],
        [7.4046e-04, 9.9926e-01],
        [3.5287e-07, 1.0000e+00],
        [1.6522e-02, 9.8348e-01],
        [3.7032e-05, 9.9996e-01],
        [1.1120e-04, 9.9989e-01],
        [1.4213e-06, 1.0000e+00],
        [9.8998e-01, 1.0023e-02],
        [9.9917e-01, 8.2588e-04],
        [3.7332e-03, 9.9627e-01],
        [8.9344e-06, 9.9999e-01],
        [2.2214e-02, 9.7779e-01],
        [9.9926e-01, 7.3863e-04],
        [9.8843e-01, 1.1571e-02],
        [1.3738e-04, 9.9986e-01],
        [9.9819e-01, 1.8057e-03],
        [3.1576e-02, 9.6842e-01],
        [2.8333e-06, 1.0000e+00],
        [9.6484e-01, 3.5156e-02],
        [1.1411e-06, 1.0000e+00],
        [8.9511e-01, 1.0489e-01],
        [5.2290e-02, 9.4771e-01],
        [6.2664e-03, 9.9373e-01],
        [1.0167e-04, 9.9990e-01],
        [1.9530e-05, 9.9998e-01],
        [1.6562e-05, 9.9998e-01],
        [8.8125e-06, 9.9999e-01],
        [3.7628e-03, 9.9624e-01],
        [1.4184e-03, 9.9858e-01],
        [3.3763e-06, 1.0000e+00],
        [8.7986e-01, 1.2014e-01],
        [1.3291e-03, 9.9867e-01],
        [7.9869e-04, 9.9920e-01],
        [1.2950e-04, 9.9987e-01],
        [6.4361e-05, 9.9994e-01],
        [3.2059e-05, 9.9997e-01],
        [9.9960e-01, 3.9540e-04],
        [4.4952e-05, 9.9996e-01],
        [5.4933e-03, 9.9451e-01],
        [8.9245e-03, 9.9108e-01],
        [3.0387e-03, 9.9696e-01],
        [5.0803e-01, 4.9197e-01],
        [5.1102e-07, 1.0000e+00],
        [7.8108e-01, 2.1892e-01],
        [5.9942e-05, 9.9994e-01],
        [7.3607e-05, 9.9993e-01],
        [2.0225e-04, 9.9980e-01],
        [2.3747e-05, 9.9998e-01],
        [8.6028e-01, 1.3972e-01],
        [3.4617e-05, 9.9997e-01],
        [1.2228e-03, 9.9878e-01],
        [9.9615e-01, 3.8488e-03],
        [9.9638e-01, 3.6185e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[4.4080e-06, 1.0000e+00],
        [1.8521e-06, 1.0000e+00],
        [1.8466e-03, 9.9815e-01],
        [3.9029e-03, 9.9610e-01],
        [1.8622e-04, 9.9981e-01],
        [1.5478e-04, 9.9985e-01],
        [3.4055e-05, 9.9997e-01],
        [3.1313e-05, 9.9997e-01],
        [1.0493e-01, 8.9507e-01],
        [3.2079e-06, 1.0000e+00],
        [8.7987e-01, 1.2013e-01],
        [9.3960e-05, 9.9991e-01],
        [2.1824e-02, 9.7818e-01],
        [2.3815e-04, 9.9976e-01],
        [9.3893e-05, 9.9991e-01],
        [3.8139e-06, 1.0000e+00],
        [2.5198e-03, 9.9748e-01],
        [1.9945e-04, 9.9980e-01],
        [2.9328e-01, 7.0672e-01],
        [1.8632e-02, 9.8137e-01],
        [6.2853e-03, 9.9371e-01],
        [1.5513e-04, 9.9984e-01],
        [5.7587e-03, 9.9424e-01],
        [2.4697e-04, 9.9975e-01],
        [3.9409e-07, 1.0000e+00],
        [6.3933e-05, 9.9994e-01],
        [9.6277e-05, 9.9990e-01],
        [9.8828e-01, 1.1725e-02],
        [9.6963e-03, 9.9030e-01],
        [1.8396e-06, 1.0000e+00],
        [9.9957e-01, 4.3162e-04],
        [6.9749e-03, 9.9303e-01],
        [2.3248e-02, 9.7675e-01],
        [9.9882e-01, 1.1755e-03],
        [9.9960e-01, 4.0155e-04],
        [9.9926e-01, 7.3876e-04],
        [9.7070e-01, 2.9303e-02],
        [5.4655e-05, 9.9995e-01],
        [2.7386e-03, 9.9726e-01],
        [8.1878e-05, 9.9992e-01],
        [9.9806e-01, 1.9431e-03],
        [1.7820e-04, 9.9982e-01],
        [3.7818e-06, 1.0000e+00],
        [1.0745e-02, 9.8926e-01],
        [7.6222e-05, 9.9992e-01],
        [9.9398e-01, 6.0214e-03],
        [3.7660e-02, 9.6234e-01],
        [1.3984e-05, 9.9999e-01],
        [2.8924e-06, 1.0000e+00],
        [6.7287e-04, 9.9933e-01],
        [2.1888e-06, 1.0000e+00],
        [2.3062e-07, 1.0000e+00],
        [1.1518e-06, 1.0000e+00],
        [9.9950e-01, 4.9586e-04],
        [8.9131e-06, 9.9999e-01],
        [1.9057e-01, 8.0943e-01],
        [9.9904e-01, 9.5922e-04],
        [1.0214e-02, 9.8979e-01],
        [9.9447e-01, 5.5260e-03],
        [6.6787e-03, 9.9332e-01],
        [7.6593e-05, 9.9992e-01],
        [8.9056e-05, 9.9991e-01],
        [4.7284e-06, 1.0000e+00],
        [7.2378e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8270e-03, 9.9817e-01],
        [7.9164e-01, 2.0836e-01],
        [3.6366e-05, 9.9996e-01],
        [9.9963e-01, 3.6703e-04],
        [3.5820e-04, 9.9964e-01],
        [1.4550e-02, 9.8545e-01],
        [8.9784e-03, 9.9102e-01],
        [7.3469e-05, 9.9993e-01],
        [9.9889e-01, 1.1112e-03],
        [3.5195e-04, 9.9965e-01],
        [2.4260e-04, 9.9976e-01],
        [9.9952e-01, 4.8118e-04],
        [8.7818e-03, 9.9122e-01],
        [9.2917e-05, 9.9991e-01],
        [3.2309e-06, 1.0000e+00],
        [9.5388e-04, 9.9905e-01],
        [2.1946e-03, 9.9781e-01],
        [3.0285e-05, 9.9997e-01],
        [9.9927e-01, 7.3161e-04],
        [3.2625e-07, 1.0000e+00],
        [9.9981e-01, 1.9046e-04],
        [9.4940e-05, 9.9991e-01],
        [2.3173e-07, 1.0000e+00],
        [3.9301e-05, 9.9996e-01],
        [9.9934e-01, 6.5774e-04],
        [1.1099e-03, 9.9889e-01],
        [1.0796e-04, 9.9989e-01],
        [1.0560e-03, 9.9894e-01],
        [3.3680e-02, 9.6632e-01],
        [7.2909e-03, 9.9271e-01],
        [2.7783e-04, 9.9972e-01],
        [4.8345e-04, 9.9952e-01],
        [2.9135e-02, 9.7087e-01],
        [3.8607e-07, 1.0000e+00],
        [3.4641e-01, 6.5359e-01],
        [4.7169e-06, 1.0000e+00],
        [1.0533e-04, 9.9989e-01],
        [8.2969e-02, 9.1703e-01],
        [2.9003e-05, 9.9997e-01],
        [2.2439e-02, 9.7756e-01],
        [3.6114e-04, 9.9964e-01],
        [4.9690e-06, 9.9999e-01],
        [7.3543e-06, 9.9999e-01],
        [9.9871e-01, 1.2934e-03],
        [1.0170e-04, 9.9990e-01],
        [3.4907e-03, 9.9651e-01],
        [9.9877e-01, 1.2290e-03],
        [1.2812e-03, 9.9872e-01],
        [9.9852e-01, 1.4805e-03],
        [1.6751e-04, 9.9983e-01],
        [9.9890e-01, 1.1016e-03],
        [4.1361e-05, 9.9996e-01],
        [6.9988e-06, 9.9999e-01],
        [8.3440e-03, 9.9166e-01],
        [8.6374e-06, 9.9999e-01],
        [3.8161e-04, 9.9962e-01],
        [9.9888e-01, 1.1157e-03],
        [9.9296e-01, 7.0449e-03],
        [9.1956e-01, 8.0438e-02],
        [1.6737e-06, 1.0000e+00],
        [3.8230e-03, 9.9618e-01],
        [9.3455e-03, 9.9065e-01],
        [1.3751e-06, 1.0000e+00],
        [2.1885e-02, 9.7812e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.6458e-03, 9.9135e-01],
        [9.9903e-01, 9.7240e-04],
        [1.4439e-04, 9.9986e-01],
        [9.9616e-01, 3.8396e-03],
        [1.4680e-06, 1.0000e+00],
        [1.8731e-04, 9.9981e-01],
        [7.2933e-07, 1.0000e+00],
        [8.4115e-05, 9.9992e-01],
        [7.2218e-04, 9.9928e-01],
        [9.9889e-01, 1.1087e-03],
        [8.9983e-02, 9.1002e-01],
        [9.4156e-07, 1.0000e+00],
        [4.4420e-03, 9.9556e-01],
        [4.1882e-03, 9.9581e-01],
        [1.9809e-07, 1.0000e+00],
        [1.3263e-03, 9.9867e-01],
        [3.1623e-05, 9.9997e-01],
        [3.6390e-05, 9.9996e-01],
        [3.2599e-05, 9.9997e-01],
        [6.6268e-07, 1.0000e+00],
        [2.8999e-06, 1.0000e+00],
        [9.9911e-01, 8.9179e-04],
        [9.4381e-05, 9.9991e-01],
        [5.0034e-03, 9.9500e-01],
        [3.5145e-03, 9.9649e-01],
        [1.3162e-05, 9.9999e-01],
        [2.3274e-05, 9.9998e-01],
        [9.9999e-01, 5.3951e-06],
        [9.9982e-01, 1.7941e-04],
        [1.8419e-06, 1.0000e+00],
        [9.9925e-01, 7.4886e-04],
        [9.0032e-05, 9.9991e-01],
        [9.9611e-01, 3.8911e-03],
        [7.8379e-06, 9.9999e-01],
        [9.9980e-01, 1.9763e-04],
        [2.3080e-04, 9.9977e-01],
        [9.9616e-01, 3.8351e-03],
        [9.9942e-01, 5.8017e-04],
        [9.4515e-07, 1.0000e+00],
        [1.9132e-03, 9.9809e-01],
        [3.4194e-06, 1.0000e+00],
        [3.7867e-04, 9.9962e-01],
        [9.9776e-01, 2.2441e-03],
        [3.7091e-05, 9.9996e-01],
        [9.5535e-01, 4.4649e-02],
        [1.6027e-04, 9.9984e-01],
        [1.2525e-03, 9.9875e-01],
        [2.2188e-01, 7.7812e-01],
        [3.2599e-05, 9.9997e-01],
        [2.0561e-06, 1.0000e+00],
        [4.3059e-05, 9.9996e-01],
        [2.0203e-02, 9.7980e-01],
        [1.1670e-02, 9.8833e-01],
        [9.9936e-01, 6.3926e-04],
        [3.2225e-04, 9.9968e-01],
        [4.0019e-03, 9.9600e-01],
        [4.5751e-04, 9.9954e-01],
        [1.3874e-08, 1.0000e+00],
        [9.7227e-01, 2.7733e-02],
        [8.1926e-01, 1.8074e-01],
        [4.0706e-05, 9.9996e-01],
        [2.5109e-04, 9.9975e-01],
        [2.2782e-05, 9.9998e-01],
        [9.4506e-01, 5.4936e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9952e-01, 4.8282e-04],
        [9.8149e-05, 9.9990e-01],
        [2.5021e-04, 9.9975e-01],
        [9.9956e-01, 4.4256e-04],
        [6.3864e-05, 9.9994e-01],
        [9.6047e-01, 3.9526e-02],
        [9.9999e-01, 8.3635e-06],
        [6.9031e-04, 9.9931e-01],
        [1.2430e-03, 9.9876e-01],
        [9.9929e-01, 7.1466e-04],
        [1.2803e-05, 9.9999e-01],
        [4.1430e-04, 9.9959e-01],
        [5.1493e-03, 9.9485e-01],
        [1.5097e-02, 9.8490e-01],
        [3.4835e-03, 9.9652e-01],
        [5.9377e-01, 4.0623e-01],
        [9.9986e-01, 1.4195e-04],
        [1.3263e-04, 9.9987e-01],
        [1.8551e-05, 9.9998e-01],
        [9.9653e-01, 3.4722e-03],
        [9.9890e-01, 1.0974e-03],
        [9.8458e-04, 9.9902e-01],
        [5.3212e-04, 9.9947e-01],
        [9.9878e-01, 1.2170e-03],
        [9.9360e-01, 6.3957e-03],
        [1.2198e-05, 9.9999e-01],
        [9.9422e-01, 5.7796e-03],
        [9.9801e-01, 1.9908e-03],
        [2.9663e-03, 9.9703e-01],
        [2.5237e-06, 1.0000e+00],
        [1.0602e-01, 8.9398e-01],
        [4.2831e-05, 9.9996e-01],
        [1.9977e-04, 9.9980e-01],
        [4.3699e-05, 9.9996e-01],
        [3.7793e-04, 9.9962e-01],
        [1.2655e-05, 9.9999e-01],
        [5.2276e-03, 9.9477e-01],
        [3.9045e-05, 9.9996e-01],
        [6.3240e-03, 9.9368e-01],
        [8.5254e-06, 9.9999e-01],
        [1.9472e-06, 1.0000e+00],
        [6.9182e-04, 9.9931e-01],
        [2.8972e-05, 9.9997e-01],
        [2.9532e-05, 9.9997e-01],
        [2.0288e-02, 9.7971e-01],
        [1.5192e-03, 9.9848e-01],
        [5.2476e-07, 1.0000e+00],
        [4.2112e-03, 9.9579e-01],
        [9.9757e-01, 2.4286e-03],
        [9.5763e-01, 4.2370e-02],
        [9.9987e-01, 1.3135e-04],
        [2.6893e-05, 9.9997e-01],
        [2.1421e-04, 9.9979e-01],
        [2.7941e-05, 9.9997e-01],
        [8.0127e-04, 9.9920e-01],
        [1.8019e-06, 1.0000e+00],
        [9.9952e-01, 4.7509e-04],
        [7.8128e-04, 9.9922e-01],
        [4.5316e-07, 1.0000e+00],
        [1.7755e-05, 9.9998e-01],
        [9.9888e-01, 1.1235e-03],
        [9.5986e-01, 4.0135e-02],
        [4.9324e-04, 9.9951e-01],
        [7.0264e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1334e-06, 1.0000e+00],
        [1.9008e-06, 1.0000e+00],
        [4.0067e-06, 1.0000e+00],
        [4.8132e-06, 1.0000e+00],
        [3.2994e-06, 1.0000e+00],
        [1.9297e-02, 9.8070e-01],
        [4.1147e-02, 9.5885e-01],
        [8.9789e-04, 9.9910e-01],
        [3.0328e-05, 9.9997e-01],
        [5.7395e-04, 9.9943e-01],
        [9.9718e-01, 2.8244e-03],
        [1.8433e-06, 1.0000e+00],
        [5.3324e-04, 9.9947e-01],
        [1.8766e-04, 9.9981e-01],
        [1.6071e-04, 9.9984e-01],
        [9.3303e-05, 9.9991e-01],
        [1.7937e-02, 9.8206e-01],
        [1.0305e-03, 9.9897e-01],
        [9.9544e-01, 4.5628e-03],
        [9.9818e-01, 1.8213e-03],
        [2.0742e-03, 9.9793e-01],
        [7.6500e-01, 2.3500e-01],
        [3.9013e-04, 9.9961e-01],
        [8.2284e-02, 9.1772e-01],
        [2.5062e-05, 9.9997e-01],
        [2.2726e-03, 9.9773e-01],
        [2.8224e-02, 9.7178e-01],
        [6.7818e-06, 9.9999e-01],
        [1.4020e-02, 9.8598e-01],
        [9.6329e-01, 3.6706e-02],
        [4.6429e-05, 9.9995e-01],
        [2.4263e-05, 9.9998e-01],
        [9.4324e-03, 9.9057e-01],
        [1.2575e-04, 9.9987e-01],
        [2.0081e-05, 9.9998e-01],
        [5.7461e-04, 9.9943e-01],
        [9.8889e-07, 1.0000e+00],
        [3.4250e-06, 1.0000e+00],
        [1.4817e-03, 9.9852e-01],
        [1.6825e-05, 9.9998e-01],
        [5.2587e-01, 4.7413e-01],
        [9.6165e-01, 3.8354e-02],
        [4.2840e-04, 9.9957e-01],
        [9.8042e-03, 9.9020e-01],
        [1.1943e-04, 9.9988e-01],
        [1.9340e-02, 9.8066e-01],
        [9.6974e-01, 3.0264e-02],
        [9.9933e-01, 6.6958e-04],
        [9.0944e-06, 9.9999e-01],
        [9.3864e-01, 6.1364e-02],
        [9.9968e-01, 3.2417e-04],
        [9.8937e-01, 1.0632e-02],
        [2.6377e-03, 9.9736e-01],
        [1.4207e-01, 8.5793e-01],
        [1.8984e-04, 9.9981e-01],
        [6.4717e-02, 9.3528e-01],
        [6.5211e-06, 9.9999e-01],
        [9.9383e-01, 6.1742e-03],
        [2.5571e-04, 9.9974e-01],
        [9.5249e-04, 9.9905e-01],
        [1.1401e-01, 8.8599e-01],
        [5.2724e-01, 4.7276e-01],
        [5.4717e-03, 9.9453e-01],
        [9.0019e-01, 9.9811e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2061e-03, 9.9579e-01],
        [9.9906e-01, 9.3574e-04],
        [3.2337e-03, 9.9677e-01],
        [3.1689e-04, 9.9968e-01],
        [5.5649e-07, 1.0000e+00],
        [3.2633e-06, 1.0000e+00],
        [3.7234e-05, 9.9996e-01],
        [4.0630e-05, 9.9996e-01],
        [7.8867e-01, 2.1133e-01],
        [9.3068e-04, 9.9907e-01],
        [1.8741e-04, 9.9981e-01],
        [7.9918e-01, 2.0082e-01],
        [5.9372e-04, 9.9941e-01],
        [4.5978e-05, 9.9995e-01],
        [9.5564e-01, 4.4363e-02],
        [9.7448e-01, 2.5519e-02],
        [3.3367e-02, 9.6663e-01],
        [1.1417e-01, 8.8583e-01],
        [6.4092e-02, 9.3591e-01],
        [9.8971e-01, 1.0288e-02],
        [1.2122e-05, 9.9999e-01],
        [9.8422e-01, 1.5783e-02],
        [3.1742e-05, 9.9997e-01],
        [7.1220e-04, 9.9929e-01],
        [1.9486e-04, 9.9981e-01],
        [5.4990e-02, 9.4501e-01],
        [2.8432e-03, 9.9716e-01],
        [1.4729e-04, 9.9985e-01],
        [9.9958e-01, 4.2208e-04],
        [1.2850e-03, 9.9871e-01],
        [1.2935e-03, 9.9871e-01],
        [1.7165e-08, 1.0000e+00],
        [5.9873e-06, 9.9999e-01],
        [6.3858e-04, 9.9936e-01],
        [9.9966e-01, 3.3867e-04],
        [9.9969e-01, 3.1272e-04],
        [9.9985e-01, 1.4920e-04],
        [9.4791e-01, 5.2088e-02],
        [3.0352e-05, 9.9997e-01],
        [1.8372e-06, 1.0000e+00],
        [8.9841e-01, 1.0159e-01],
        [2.6892e-03, 9.9731e-01],
        [1.5037e-05, 9.9998e-01],
        [1.5407e-03, 9.9846e-01],
        [9.9799e-01, 2.0148e-03],
        [1.5513e-06, 1.0000e+00],
        [4.0775e-06, 1.0000e+00],
        [2.7286e-03, 9.9727e-01],
        [1.8830e-02, 9.8117e-01],
        [7.5547e-05, 9.9992e-01],
        [1.1758e-02, 9.8824e-01],
        [8.7756e-01, 1.2244e-01],
        [1.9823e-03, 9.9802e-01],
        [2.0206e-04, 9.9980e-01],
        [9.9836e-01, 1.6369e-03],
        [9.9992e-01, 8.4035e-05],
        [1.0590e-06, 1.0000e+00],
        [7.2710e-06, 9.9999e-01],
        [2.7450e-02, 9.7255e-01],
        [9.6457e-01, 3.5435e-02],
        [9.9594e-01, 4.0610e-03],
        [4.1390e-04, 9.9959e-01],
        [5.4577e-04, 9.9945e-01],
        [7.8724e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7519e-07, 1.0000e+00],
        [1.0696e-02, 9.8930e-01],
        [2.9952e-01, 7.0048e-01],
        [4.3168e-05, 9.9996e-01],
        [9.9977e-01, 2.3102e-04],
        [9.9611e-01, 3.8910e-03],
        [9.9876e-01, 1.2420e-03],
        [6.1442e-01, 3.8558e-01],
        [7.8979e-07, 1.0000e+00],
        [2.7944e-05, 9.9997e-01],
        [3.5626e-07, 1.0000e+00],
        [6.6255e-04, 9.9934e-01],
        [7.8002e-06, 9.9999e-01],
        [2.8703e-03, 9.9713e-01],
        [9.9929e-01, 7.1096e-04],
        [1.4323e-02, 9.8568e-01],
        [7.4671e-07, 1.0000e+00],
        [2.5128e-02, 9.7487e-01],
        [2.2755e-05, 9.9998e-01],
        [9.9129e-01, 8.7075e-03],
        [3.0596e-06, 1.0000e+00],
        [1.7086e-03, 9.9829e-01],
        [6.1547e-05, 9.9994e-01],
        [1.4411e-04, 9.9986e-01],
        [1.9684e-05, 9.9998e-01],
        [3.6932e-06, 1.0000e+00],
        [1.7598e-01, 8.2402e-01],
        [1.0433e-04, 9.9990e-01],
        [9.8591e-01, 1.4087e-02],
        [1.3945e-02, 9.8605e-01],
        [1.1661e-05, 9.9999e-01],
        [7.4983e-04, 9.9925e-01],
        [7.7997e-01, 2.2003e-01],
        [9.0975e-05, 9.9991e-01],
        [1.9367e-04, 9.9981e-01],
        [9.9898e-01, 1.0160e-03],
        [6.9818e-05, 9.9993e-01],
        [3.6719e-04, 9.9963e-01],
        [9.9904e-01, 9.6290e-04],
        [4.9322e-04, 9.9951e-01],
        [9.8359e-01, 1.6407e-02],
        [4.2404e-04, 9.9958e-01],
        [1.7123e-03, 9.9829e-01],
        [7.0972e-05, 9.9993e-01],
        [9.8993e-01, 1.0071e-02],
        [1.4055e-05, 9.9999e-01],
        [9.9973e-01, 2.7381e-04],
        [1.9836e-05, 9.9998e-01],
        [3.7872e-02, 9.6213e-01],
        [8.6975e-08, 1.0000e+00],
        [8.9706e-03, 9.9103e-01],
        [5.3743e-07, 1.0000e+00],
        [1.1994e-02, 9.8801e-01],
        [6.8766e-05, 9.9993e-01],
        [3.0745e-01, 6.9255e-01],
        [6.5076e-01, 3.4924e-01],
        [6.8994e-09, 1.0000e+00],
        [9.9945e-01, 5.4589e-04],
        [2.7749e-05, 9.9997e-01],
        [1.3910e-05, 9.9999e-01],
        [9.7946e-01, 2.0543e-02],
        [1.4856e-05, 9.9999e-01],
        [3.6285e-05, 9.9996e-01],
        [1.1679e-03, 9.9883e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.7115e-02, 9.4288e-01],
        [1.8434e-04, 9.9982e-01],
        [7.5968e-07, 1.0000e+00],
        [1.9388e-05, 9.9998e-01],
        [7.9867e-05, 9.9992e-01],
        [4.3899e-05, 9.9996e-01],
        [1.9227e-04, 9.9981e-01],
        [1.7122e-03, 9.9829e-01],
        [2.2546e-04, 9.9977e-01],
        [1.2931e-02, 9.8707e-01],
        [9.6201e-03, 9.9038e-01],
        [1.3242e-06, 1.0000e+00],
        [1.1953e-02, 9.8805e-01],
        [9.9776e-01, 2.2410e-03],
        [1.6207e-03, 9.9838e-01],
        [1.7098e-03, 9.9829e-01],
        [3.0209e-06, 1.0000e+00],
        [9.9797e-06, 9.9999e-01],
        [7.7722e-03, 9.9223e-01],
        [9.7097e-01, 2.9026e-02],
        [9.9471e-01, 5.2919e-03],
        [4.5650e-04, 9.9954e-01],
        [9.1723e-01, 8.2768e-02],
        [1.7536e-04, 9.9982e-01],
        [8.8528e-07, 1.0000e+00],
        [1.6512e-05, 9.9998e-01],
        [2.6257e-05, 9.9997e-01],
        [3.6382e-04, 9.9964e-01],
        [8.3651e-05, 9.9992e-01],
        [9.9945e-01, 5.5334e-04],
        [7.9386e-06, 9.9999e-01],
        [9.9948e-01, 5.1756e-04],
        [9.9991e-01, 9.4552e-05],
        [9.8468e-01, 1.5318e-02],
        [4.9259e-03, 9.9507e-01],
        [9.1221e-01, 8.7791e-02],
        [5.6444e-07, 1.0000e+00],
        [7.5622e-05, 9.9992e-01],
        [1.0478e-06, 1.0000e+00],
        [3.5928e-08, 1.0000e+00],
        [1.6512e-06, 1.0000e+00],
        [9.2856e-05, 9.9991e-01],
        [8.1635e-04, 9.9918e-01],
        [6.7656e-04, 9.9932e-01],
        [9.8714e-01, 1.2863e-02],
        [3.6957e-04, 9.9963e-01],
        [1.0334e-04, 9.9990e-01],
        [9.7891e-05, 9.9990e-01],
        [2.9657e-02, 9.7034e-01],
        [7.8940e-07, 1.0000e+00],
        [9.9862e-01, 1.3760e-03],
        [4.0646e-03, 9.9594e-01],
        [9.9948e-01, 5.2273e-04],
        [9.9954e-01, 4.5728e-04],
        [2.8347e-04, 9.9972e-01],
        [9.9975e-01, 2.5159e-04],
        [1.3653e-07, 1.0000e+00],
        [1.3084e-06, 1.0000e+00],
        [2.6347e-03, 9.9737e-01],
        [8.5947e-06, 9.9999e-01],
        [9.4558e-04, 9.9905e-01],
        [9.9956e-01, 4.3625e-04],
        [7.4382e-07, 1.0000e+00],
        [9.7452e-01, 2.5481e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9932e-01, 6.8232e-04],
        [3.9311e-05, 9.9996e-01],
        [5.5334e-03, 9.9447e-01],
        [3.3404e-07, 1.0000e+00],
        [9.9575e-01, 4.2472e-03],
        [9.9868e-01, 1.3210e-03],
        [1.5571e-02, 9.8443e-01],
        [1.7815e-06, 1.0000e+00],
        [1.5614e-02, 9.8439e-01],
        [1.7457e-02, 9.8254e-01],
        [9.9737e-01, 2.6298e-03],
        [5.4641e-05, 9.9995e-01],
        [7.3797e-09, 1.0000e+00],
        [2.0037e-06, 1.0000e+00],
        [3.6222e-04, 9.9964e-01],
        [1.6778e-03, 9.9832e-01],
        [5.3974e-05, 9.9995e-01],
        [9.9851e-01, 1.4918e-03],
        [9.7916e-01, 2.0840e-02],
        [3.3903e-04, 9.9966e-01],
        [9.9933e-01, 6.6967e-04],
        [3.9702e-05, 9.9996e-01],
        [3.1129e-04, 9.9969e-01],
        [5.7745e-04, 9.9942e-01],
        [9.4531e-03, 9.9055e-01],
        [9.2342e-04, 9.9908e-01],
        [5.4387e-06, 9.9999e-01],
        [9.9998e-01, 1.5016e-05],
        [9.9761e-01, 2.3932e-03],
        [4.4870e-04, 9.9955e-01],
        [5.1383e-04, 9.9949e-01],
        [9.9262e-01, 7.3834e-03],
        [2.3600e-04, 9.9976e-01],
        [9.9991e-01, 9.3369e-05],
        [6.7481e-03, 9.9325e-01],
        [1.1828e-07, 1.0000e+00],
        [1.2023e-04, 9.9988e-01],
        [3.8751e-04, 9.9961e-01],
        [9.9900e-01, 1.0018e-03],
        [9.9413e-01, 5.8736e-03],
        [1.1255e-06, 1.0000e+00],
        [3.6452e-06, 1.0000e+00],
        [7.3809e-09, 1.0000e+00],
        [5.8485e-04, 9.9942e-01],
        [9.9857e-01, 1.4329e-03],
        [1.2476e-05, 9.9999e-01],
        [1.0815e-04, 9.9989e-01],
        [2.6168e-02, 9.7383e-01],
        [1.2908e-07, 1.0000e+00],
        [9.9925e-01, 7.5366e-04],
        [1.8959e-04, 9.9981e-01],
        [9.9124e-01, 8.7565e-03],
        [9.9929e-01, 7.1298e-04],
        [1.3449e-05, 9.9999e-01],
        [6.0010e-05, 9.9994e-01],
        [7.4493e-06, 9.9999e-01],
        [2.8467e-06, 1.0000e+00],
        [1.9745e-04, 9.9980e-01],
        [9.9911e-01, 8.9048e-04],
        [9.9919e-01, 8.0650e-04],
        [2.8668e-04, 9.9971e-01],
        [2.7794e-03, 9.9722e-01],
        [9.8040e-04, 9.9902e-01],
        [9.9952e-01, 4.7873e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9877e-01, 1.2321e-03],
        [2.3272e-04, 9.9977e-01],
        [1.0236e-04, 9.9990e-01],
        [3.1095e-06, 1.0000e+00],
        [9.9766e-01, 2.3437e-03],
        [7.8440e-01, 2.1560e-01],
        [3.9615e-02, 9.6039e-01],
        [1.3632e-05, 9.9999e-01],
        [1.1480e-03, 9.9885e-01],
        [6.0472e-01, 3.9528e-01],
        [4.1630e-03, 9.9584e-01],
        [3.5351e-03, 9.9646e-01],
        [5.6796e-06, 9.9999e-01],
        [8.4145e-09, 1.0000e+00],
        [1.2232e-04, 9.9988e-01],
        [9.9928e-01, 7.1666e-04],
        [2.5326e-05, 9.9997e-01],
        [5.1568e-04, 9.9948e-01],
        [3.6302e-03, 9.9637e-01],
        [1.4491e-05, 9.9999e-01],
        [9.2974e-05, 9.9991e-01],
        [3.2039e-04, 9.9968e-01],
        [3.9652e-03, 9.9603e-01],
        [7.5680e-05, 9.9992e-01],
        [5.5302e-01, 4.4698e-01],
        [8.1934e-06, 9.9999e-01],
        [2.2024e-05, 9.9998e-01],
        [1.1615e-07, 1.0000e+00],
        [1.5067e-07, 1.0000e+00],
        [4.9181e-07, 1.0000e+00],
        [9.1213e-04, 9.9909e-01],
        [3.7835e-05, 9.9996e-01],
        [1.1057e-03, 9.9889e-01],
        [9.9986e-01, 1.4127e-04],
        [2.2301e-02, 9.7770e-01],
        [4.8903e-07, 1.0000e+00],
        [1.3246e-07, 1.0000e+00],
        [2.3884e-07, 1.0000e+00],
        [9.9751e-01, 2.4926e-03],
        [2.9033e-03, 9.9710e-01],
        [9.9974e-01, 2.6207e-04],
        [2.6565e-03, 9.9734e-01],
        [9.9768e-01, 2.3239e-03],
        [9.5120e-05, 9.9990e-01],
        [2.5346e-04, 9.9975e-01],
        [9.9971e-01, 2.8704e-04],
        [9.5297e-01, 4.7026e-02],
        [1.3876e-05, 9.9999e-01],
        [9.9850e-01, 1.5017e-03],
        [5.6571e-06, 9.9999e-01],
        [9.6348e-01, 3.6515e-02],
        [2.6341e-05, 9.9997e-01],
        [1.4307e-03, 9.9857e-01],
        [5.7169e-03, 9.9428e-01],
        [1.1374e-04, 9.9989e-01],
        [9.9922e-01, 7.7650e-04],
        [9.3466e-01, 6.5341e-02],
        [2.1827e-04, 9.9978e-01],
        [1.5037e-05, 9.9998e-01],
        [2.8075e-03, 9.9719e-01],
        [3.4840e-03, 9.9652e-01],
        [8.4606e-06, 9.9999e-01],
        [9.9354e-01, 6.4647e-03],
        [2.9859e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0695e-04, 9.9989e-01],
        [8.0472e-04, 9.9920e-01],
        [9.5801e-06, 9.9999e-01],
        [2.0499e-08, 1.0000e+00],
        [6.6196e-05, 9.9993e-01],
        [9.7269e-05, 9.9990e-01],
        [1.0470e-04, 9.9990e-01],
        [1.3332e-06, 1.0000e+00],
        [9.9599e-01, 4.0110e-03],
        [9.9924e-01, 7.6212e-04],
        [9.9152e-01, 8.4767e-03],
        [9.7282e-01, 2.7176e-02],
        [2.0772e-05, 9.9998e-01],
        [9.7914e-01, 2.0858e-02],
        [8.0433e-04, 9.9920e-01],
        [9.9312e-01, 6.8759e-03],
        [9.4069e-04, 9.9906e-01],
        [9.6687e-01, 3.3127e-02],
        [2.7175e-04, 9.9973e-01],
        [1.4308e-04, 9.9986e-01],
        [5.1663e-05, 9.9995e-01],
        [9.0429e-08, 1.0000e+00],
        [8.1155e-02, 9.1884e-01],
        [9.8992e-01, 1.0079e-02],
        [5.9136e-06, 9.9999e-01],
        [3.8356e-02, 9.6164e-01],
        [2.4977e-07, 1.0000e+00],
        [2.4109e-06, 1.0000e+00],
        [4.7991e-05, 9.9995e-01],
        [2.6466e-03, 9.9735e-01],
        [9.9600e-01, 4.0003e-03],
        [1.1344e-05, 9.9999e-01],
        [1.6035e-05, 9.9998e-01],
        [9.9478e-01, 5.2216e-03],
        [1.8704e-06, 1.0000e+00],
        [4.1322e-04, 9.9959e-01],
        [9.9010e-01, 9.9023e-03],
        [6.5486e-01, 3.4514e-01],
        [2.0925e-05, 9.9998e-01],
        [9.9823e-01, 1.7713e-03],
        [9.5671e-01, 4.3293e-02],
        [1.5602e-04, 9.9984e-01],
        [1.8240e-04, 9.9982e-01],
        [1.5190e-05, 9.9998e-01],
        [9.9803e-01, 1.9675e-03],
        [4.5926e-04, 9.9954e-01],
        [2.4700e-03, 9.9753e-01],
        [9.6107e-01, 3.8934e-02],
        [1.7163e-06, 1.0000e+00],
        [7.4714e-06, 9.9999e-01],
        [1.5140e-02, 9.8486e-01],
        [1.1420e-06, 1.0000e+00],
        [2.3235e-03, 9.9768e-01],
        [5.3834e-05, 9.9995e-01],
        [5.8207e-08, 1.0000e+00],
        [4.7932e-06, 1.0000e+00],
        [4.4133e-06, 1.0000e+00],
        [6.7160e-04, 9.9933e-01],
        [9.9664e-01, 3.3566e-03],
        [7.6140e-03, 9.9239e-01],
        [2.5562e-07, 1.0000e+00],
        [2.0329e-03, 9.9797e-01],
        [1.2032e-02, 9.8797e-01],
        [1.5999e-01, 8.4001e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.4759e-05, 9.9996e-01],
        [5.3037e-04, 9.9947e-01],
        [5.9638e-03, 9.9404e-01],
        [2.6444e-04, 9.9974e-01],
        [3.9196e-06, 1.0000e+00],
        [2.9619e-03, 9.9704e-01],
        [3.5207e-04, 9.9965e-01],
        [8.3187e-03, 9.9168e-01],
        [2.9537e-07, 1.0000e+00],
        [4.2041e-03, 9.9580e-01],
        [2.0457e-01, 7.9543e-01],
        [6.5688e-07, 1.0000e+00],
        [9.9591e-01, 4.0913e-03],
        [2.1599e-05, 9.9998e-01],
        [1.8744e-09, 1.0000e+00],
        [1.3234e-03, 9.9868e-01],
        [1.7444e-05, 9.9998e-01],
        [9.9980e-01, 1.9798e-04],
        [4.5205e-06, 1.0000e+00],
        [5.2537e-05, 9.9995e-01],
        [6.7226e-03, 9.9328e-01],
        [5.8564e-08, 1.0000e+00],
        [9.5052e-01, 4.9475e-02],
        [7.7066e-04, 9.9923e-01],
        [2.9354e-04, 9.9971e-01],
        [4.6097e-06, 1.0000e+00],
        [6.0862e-06, 9.9999e-01],
        [6.4574e-03, 9.9354e-01],
        [1.0818e-03, 9.9892e-01],
        [9.9812e-01, 1.8836e-03],
        [6.7743e-07, 1.0000e+00],
        [1.2158e-06, 1.0000e+00],
        [9.9844e-01, 1.5614e-03],
        [4.8216e-02, 9.5178e-01],
        [8.5705e-01, 1.4295e-01],
        [9.9910e-01, 9.0487e-04],
        [8.2298e-04, 9.9918e-01],
        [5.0829e-01, 4.9171e-01],
        [9.7143e-06, 9.9999e-01],
        [9.9141e-01, 8.5902e-03],
        [2.8462e-03, 9.9715e-01],
        [3.4258e-07, 1.0000e+00],
        [6.8964e-04, 9.9931e-01],
        [9.9982e-01, 1.8278e-04],
        [1.2508e-04, 9.9987e-01],
        [1.0845e-08, 1.0000e+00],
        [9.8836e-01, 1.1640e-02],
        [9.6117e-01, 3.8833e-02],
        [2.3842e-05, 9.9998e-01],
        [8.8662e-06, 9.9999e-01],
        [9.9840e-01, 1.5976e-03],
        [9.9963e-01, 3.7252e-04],
        [9.6701e-07, 1.0000e+00],
        [9.9428e-01, 5.7200e-03],
        [9.9573e-01, 4.2658e-03],
        [1.2298e-06, 1.0000e+00],
        [8.9769e-04, 9.9910e-01],
        [4.5986e-06, 1.0000e+00],
        [7.5557e-07, 1.0000e+00],
        [2.3061e-06, 1.0000e+00],
        [2.4777e-06, 1.0000e+00],
        [4.1563e-03, 9.9584e-01],
        [4.7809e-05, 9.9995e-01],
        [7.8281e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1797e-04, 9.9988e-01],
        [9.9985e-01, 1.5253e-04],
        [5.2548e-04, 9.9947e-01],
        [7.9147e-05, 9.9992e-01],
        [1.3837e-07, 1.0000e+00],
        [9.8779e-05, 9.9990e-01],
        [1.2953e-02, 9.8705e-01],
        [1.3835e-05, 9.9999e-01],
        [8.8868e-07, 1.0000e+00],
        [8.0571e-01, 1.9429e-01],
        [4.2870e-03, 9.9571e-01],
        [8.7182e-01, 1.2818e-01],
        [6.2390e-04, 9.9938e-01],
        [9.9785e-01, 2.1546e-03],
        [6.9379e-01, 3.0621e-01],
        [1.4775e-02, 9.8523e-01],
        [3.7155e-07, 1.0000e+00],
        [9.2794e-01, 7.2056e-02],
        [9.7858e-07, 1.0000e+00],
        [9.9992e-01, 8.2764e-05],
        [1.7491e-02, 9.8251e-01],
        [9.9941e-01, 5.9094e-04],
        [1.2405e-03, 9.9876e-01],
        [7.7275e-05, 9.9992e-01],
        [5.2336e-01, 4.7664e-01],
        [1.0278e-06, 1.0000e+00],
        [9.7269e-01, 2.7313e-02],
        [4.5014e-06, 1.0000e+00],
        [1.0033e-03, 9.9900e-01],
        [6.2034e-07, 1.0000e+00],
        [4.9532e-01, 5.0468e-01],
        [1.4260e-04, 9.9986e-01],
        [1.5455e-06, 1.0000e+00],
        [8.5225e-06, 9.9999e-01],
        [9.9878e-01, 1.2190e-03],
        [9.0176e-04, 9.9910e-01],
        [1.4007e-04, 9.9986e-01],
        [9.9843e-01, 1.5661e-03],
        [9.9303e-01, 6.9733e-03],
        [3.4797e-05, 9.9997e-01],
        [1.8604e-04, 9.9981e-01],
        [1.3625e-05, 9.9999e-01],
        [3.8384e-07, 1.0000e+00],
        [9.2046e-06, 9.9999e-01],
        [1.3471e-02, 9.8653e-01],
        [9.7696e-01, 2.3044e-02],
        [9.5102e-07, 1.0000e+00],
        [5.4138e-06, 9.9999e-01],
        [5.8233e-05, 9.9994e-01],
        [6.9397e-04, 9.9931e-01],
        [7.1172e-07, 1.0000e+00],
        [7.4161e-03, 9.9258e-01],
        [9.9947e-01, 5.3294e-04],
        [9.9693e-01, 3.0680e-03],
        [9.8323e-05, 9.9990e-01],
        [6.4929e-04, 9.9935e-01],
        [7.2947e-07, 1.0000e+00],
        [9.9946e-01, 5.3879e-04],
        [1.1196e-05, 9.9999e-01],
        [4.3608e-05, 9.9996e-01],
        [5.9437e-07, 1.0000e+00],
        [2.8098e-01, 7.1902e-01],
        [2.5168e-03, 9.9748e-01],
        [8.0983e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9914e-01, 8.6222e-04],
        [6.9973e-07, 1.0000e+00],
        [9.9793e-01, 2.0684e-03],
        [5.5980e-05, 9.9994e-01],
        [5.4952e-05, 9.9995e-01],
        [3.1772e-01, 6.8228e-01],
        [7.3447e-06, 9.9999e-01],
        [3.2229e-04, 9.9968e-01],
        [9.9639e-01, 3.6112e-03],
        [1.0279e-03, 9.9897e-01],
        [3.7665e-07, 1.0000e+00],
        [3.5646e-03, 9.9644e-01],
        [9.9910e-01, 8.9919e-04],
        [5.5465e-03, 9.9445e-01],
        [1.6297e-03, 9.9837e-01],
        [8.2638e-08, 1.0000e+00],
        [5.7336e-07, 1.0000e+00],
        [9.9928e-01, 7.1978e-04],
        [9.9874e-01, 1.2641e-03],
        [8.4127e-05, 9.9992e-01],
        [1.8210e-04, 9.9982e-01],
        [3.7227e-07, 1.0000e+00],
        [2.4310e-05, 9.9998e-01],
        [5.0986e-06, 9.9999e-01],
        [5.1102e-06, 9.9999e-01],
        [1.9898e-04, 9.9980e-01],
        [4.4064e-04, 9.9956e-01],
        [4.7618e-05, 9.9995e-01],
        [7.5143e-01, 2.4857e-01],
        [2.4221e-03, 9.9758e-01],
        [4.1724e-07, 1.0000e+00],
        [6.6888e-07, 1.0000e+00],
        [6.3858e-01, 3.6142e-01],
        [2.1580e-04, 9.9978e-01],
        [9.9702e-01, 2.9839e-03],
        [9.9913e-01, 8.6896e-04],
        [5.7865e-05, 9.9994e-01],
        [3.6580e-05, 9.9996e-01],
        [4.8642e-06, 1.0000e+00],
        [9.9793e-01, 2.0677e-03],
        [9.5964e-01, 4.0359e-02],
        [9.6156e-01, 3.8436e-02],
        [9.6001e-01, 3.9992e-02],
        [9.9947e-01, 5.3105e-04],
        [2.8177e-05, 9.9997e-01],
        [4.8954e-03, 9.9510e-01],
        [1.5845e-05, 9.9998e-01],
        [3.9263e-07, 1.0000e+00],
        [9.9811e-01, 1.8880e-03],
        [9.9975e-01, 2.5219e-04],
        [1.1796e-05, 9.9999e-01],
        [2.6917e-05, 9.9997e-01],
        [9.9974e-01, 2.6394e-04],
        [9.8838e-01, 1.1621e-02],
        [1.0508e-04, 9.9989e-01],
        [1.5620e-05, 9.9998e-01],
        [7.9556e-05, 9.9992e-01],
        [2.3662e-05, 9.9998e-01],
        [7.8023e-06, 9.9999e-01],
        [1.5982e-06, 1.0000e+00],
        [9.9777e-01, 2.2334e-03],
        [9.1310e-01, 8.6895e-02],
        [2.7632e-02, 9.7237e-01],
        [8.4760e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7820e-04, 9.9982e-01],
        [9.8645e-01, 1.3554e-02],
        [1.7557e-02, 9.8244e-01],
        [1.3327e-06, 1.0000e+00],
        [9.9726e-01, 2.7404e-03],
        [1.8840e-05, 9.9998e-01],
        [6.3147e-05, 9.9994e-01],
        [9.7191e-01, 2.8090e-02],
        [9.1496e-06, 9.9999e-01],
        [1.0300e-03, 9.9897e-01],
        [7.1304e-06, 9.9999e-01],
        [4.1806e-06, 1.0000e+00],
        [2.6236e-03, 9.9738e-01],
        [6.2420e-01, 3.7580e-01],
        [9.9659e-01, 3.4093e-03],
        [5.9971e-04, 9.9940e-01],
        [8.8668e-08, 1.0000e+00],
        [1.1115e-02, 9.8888e-01],
        [9.9983e-01, 1.6549e-04],
        [9.9315e-01, 6.8488e-03],
        [9.9957e-01, 4.2762e-04],
        [9.2482e-05, 9.9991e-01],
        [1.1667e-03, 9.9883e-01],
        [1.6467e-05, 9.9998e-01],
        [4.2408e-08, 1.0000e+00],
        [4.9697e-04, 9.9950e-01],
        [3.9987e-04, 9.9960e-01],
        [1.9210e-05, 9.9998e-01],
        [7.1014e-01, 2.8986e-01],
        [7.0764e-05, 9.9993e-01],
        [2.5695e-02, 9.7430e-01],
        [1.5880e-06, 1.0000e+00],
        [9.9974e-01, 2.5917e-04],
        [1.0896e-05, 9.9999e-01],
        [5.2929e-01, 4.7071e-01],
        [4.8675e-05, 9.9995e-01],
        [8.5267e-04, 9.9915e-01],
        [4.0158e-05, 9.9996e-01],
        [7.4270e-07, 1.0000e+00],
        [8.6609e-01, 1.3391e-01],
        [9.8652e-01, 1.3478e-02],
        [9.9784e-01, 2.1619e-03],
        [9.6072e-01, 3.9278e-02],
        [2.9806e-05, 9.9997e-01],
        [1.4868e-04, 9.9985e-01],
        [5.5507e-04, 9.9944e-01],
        [3.6099e-07, 1.0000e+00],
        [8.3719e-05, 9.9992e-01],
        [5.1801e-01, 4.8199e-01],
        [3.9171e-04, 9.9961e-01],
        [3.1786e-07, 1.0000e+00],
        [2.3393e-04, 9.9977e-01],
        [7.4683e-05, 9.9993e-01],
        [6.1657e-01, 3.8343e-01],
        [9.9918e-01, 8.2138e-04],
        [1.7235e-03, 9.9828e-01],
        [1.5864e-02, 9.8414e-01],
        [3.3706e-05, 9.9997e-01],
        [9.9909e-01, 9.0868e-04],
        [1.5519e-04, 9.9984e-01],
        [9.0350e-07, 1.0000e+00],
        [1.1212e-03, 9.9888e-01],
        [8.1691e-06, 9.9999e-01],
        [3.8695e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.0830e-04, 9.9959e-01],
        [4.1583e-07, 1.0000e+00],
        [8.3642e-01, 1.6358e-01],
        [6.6653e-03, 9.9333e-01],
        [9.9962e-01, 3.8013e-04],
        [9.9962e-01, 3.8167e-04],
        [9.7221e-01, 2.7790e-02],
        [2.5733e-04, 9.9974e-01],
        [8.4481e-04, 9.9916e-01],
        [3.9568e-04, 9.9960e-01],
        [2.8175e-03, 9.9718e-01],
        [9.9420e-01, 5.7972e-03],
        [1.2244e-04, 9.9988e-01],
        [4.4330e-05, 9.9996e-01],
        [9.9084e-05, 9.9990e-01],
        [2.9553e-07, 1.0000e+00],
        [2.2502e-04, 9.9977e-01],
        [2.4498e-03, 9.9755e-01],
        [7.6855e-01, 2.3145e-01],
        [1.7731e-02, 9.8227e-01],
        [9.9891e-01, 1.0857e-03],
        [9.9106e-01, 8.9402e-03],
        [2.6438e-04, 9.9974e-01],
        [9.9726e-01, 2.7448e-03],
        [1.8031e-07, 1.0000e+00],
        [1.7507e-05, 9.9998e-01],
        [3.2371e-04, 9.9968e-01],
        [2.4531e-06, 1.0000e+00],
        [2.5137e-05, 9.9997e-01],
        [1.5564e-05, 9.9998e-01],
        [6.4914e-03, 9.9351e-01],
        [1.4390e-05, 9.9999e-01],
        [1.2128e-02, 9.8787e-01],
        [9.9666e-01, 3.3423e-03],
        [3.1893e-07, 1.0000e+00],
        [1.5122e-04, 9.9985e-01],
        [7.9076e-05, 9.9992e-01],
        [9.8548e-06, 9.9999e-01],
        [6.7709e-07, 1.0000e+00],
        [2.2428e-06, 1.0000e+00],
        [2.7204e-03, 9.9728e-01],
        [4.3895e-02, 9.5610e-01],
        [6.6387e-06, 9.9999e-01],
        [3.5296e-05, 9.9996e-01],
        [2.9994e-03, 9.9700e-01],
        [9.7098e-01, 2.9020e-02],
        [5.9749e-05, 9.9994e-01],
        [5.1947e-07, 1.0000e+00],
        [7.5896e-03, 9.9241e-01],
        [3.0180e-05, 9.9997e-01],
        [1.1553e-05, 9.9999e-01],
        [9.8298e-01, 1.7025e-02],
        [1.4220e-04, 9.9986e-01],
        [1.1084e-05, 9.9999e-01],
        [1.1997e-01, 8.8003e-01],
        [1.0427e-05, 9.9999e-01],
        [2.8729e-07, 1.0000e+00],
        [7.1231e-04, 9.9929e-01],
        [9.1431e-04, 9.9909e-01],
        [8.5845e-06, 9.9999e-01],
        [1.3888e-06, 1.0000e+00],
        [4.5552e-05, 9.9995e-01],
        [8.4194e-07, 1.0000e+00],
        [3.5821e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.8346e-05, 9.9996e-01],
        [3.3042e-02, 9.6696e-01],
        [1.4319e-06, 1.0000e+00],
        [4.8299e-06, 1.0000e+00],
        [9.9962e-01, 3.7781e-04],
        [8.4575e-03, 9.9154e-01],
        [1.0694e-03, 9.9893e-01],
        [1.4880e-04, 9.9985e-01],
        [5.1173e-07, 1.0000e+00],
        [3.3732e-04, 9.9966e-01],
        [1.0676e-07, 1.0000e+00],
        [9.9858e-01, 1.4175e-03],
        [2.5592e-03, 9.9744e-01],
        [9.9795e-01, 2.0453e-03],
        [5.7122e-06, 9.9999e-01],
        [2.6523e-06, 1.0000e+00],
        [1.2006e-03, 9.9880e-01],
        [3.8238e-06, 1.0000e+00],
        [9.9841e-01, 1.5914e-03],
        [3.4788e-05, 9.9997e-01],
        [1.5946e-04, 9.9984e-01],
        [1.9310e-04, 9.9981e-01],
        [4.4406e-05, 9.9996e-01],
        [6.3335e-05, 9.9994e-01],
        [9.6701e-01, 3.2992e-02],
        [7.9227e-07, 1.0000e+00],
        [1.0013e-05, 9.9999e-01],
        [2.2211e-04, 9.9978e-01],
        [9.9900e-01, 9.9645e-04],
        [3.7130e-03, 9.9629e-01],
        [9.9987e-01, 1.2774e-04],
        [9.9784e-01, 2.1572e-03],
        [1.0305e-02, 9.8970e-01],
        [3.1606e-03, 9.9684e-01],
        [1.2758e-04, 9.9987e-01],
        [9.9833e-01, 1.6737e-03],
        [1.3778e-04, 9.9986e-01],
        [7.8178e-08, 1.0000e+00],
        [1.0455e-02, 9.8955e-01],
        [2.7237e-04, 9.9973e-01],
        [1.6940e-03, 9.9831e-01],
        [9.9874e-01, 1.2582e-03],
        [7.4247e-06, 9.9999e-01],
        [2.4123e-05, 9.9998e-01],
        [3.7974e-07, 1.0000e+00],
        [5.8792e-03, 9.9412e-01],
        [7.3752e-01, 2.6248e-01],
        [6.0980e-05, 9.9994e-01],
        [9.9604e-01, 3.9564e-03],
        [2.3189e-03, 9.9768e-01],
        [9.9748e-01, 2.5247e-03],
        [9.9643e-04, 9.9900e-01],
        [6.5408e-01, 3.4592e-01],
        [1.2022e-04, 9.9988e-01],
        [1.0040e-03, 9.9900e-01],
        [4.4792e-06, 1.0000e+00],
        [5.8873e-06, 9.9999e-01],
        [9.9882e-01, 1.1815e-03],
        [1.1164e-03, 9.9888e-01],
        [2.3695e-01, 7.6305e-01],
        [6.9404e-03, 9.9306e-01],
        [1.7665e-07, 1.0000e+00],
        [3.5206e-04, 9.9965e-01],
        [9.0133e-01, 9.8669e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[2.4177e-07, 1.0000e+00],
        [9.9833e-01, 1.6658e-03],
        [9.9974e-01, 2.6220e-04],
        [1.3262e-04, 9.9987e-01],
        [2.6671e-04, 9.9973e-01],
        [7.6939e-03, 9.9231e-01],
        [5.0175e-04, 9.9950e-01],
        [3.2582e-03, 9.9674e-01],
        [1.4998e-07, 1.0000e+00],
        [7.8475e-02, 9.2152e-01],
        [4.3366e-04, 9.9957e-01],
        [9.6031e-01, 3.9687e-02],
        [5.7621e-06, 9.9999e-01],
        [3.5277e-04, 9.9965e-01],
        [7.6854e-01, 2.3146e-01],
        [8.5661e-01, 1.4339e-01],
        [9.9941e-01, 5.9015e-04],
        [6.1437e-07, 1.0000e+00],
        [7.1270e-03, 9.9287e-01],
        [9.9960e-01, 3.9966e-04],
        [6.9397e-06, 9.9999e-01],
        [6.8996e-03, 9.9310e-01],
        [6.5190e-04, 9.9935e-01],
        [2.6896e-04, 9.9973e-01],
        [6.1442e-01, 3.8558e-01],
        [2.3729e-03, 9.9763e-01],
        [8.5730e-03, 9.9143e-01],
        [1.4162e-01, 8.5838e-01],
        [9.7807e-01, 2.1930e-02],
        [9.9922e-01, 7.8287e-04],
        [8.4924e-06, 9.9999e-01],
        [1.6819e-03, 9.9832e-01],
        [2.6803e-01, 7.3197e-01],
        [1.3209e-06, 1.0000e+00],
        [1.8508e-05, 9.9998e-01],
        [9.4139e-01, 5.8614e-02],
        [1.1134e-06, 1.0000e+00],
        [9.9829e-01, 1.7124e-03],
        [9.9631e-01, 3.6940e-03],
        [5.9879e-03, 9.9401e-01],
        [3.4800e-04, 9.9965e-01],
        [9.7910e-01, 2.0903e-02],
        [6.9141e-01, 3.0859e-01],
        [9.8146e-01, 1.8537e-02],
        [6.1687e-04, 9.9938e-01],
        [3.5522e-02, 9.6448e-01],
        [4.5835e-05, 9.9995e-01],
        [8.8908e-06, 9.9999e-01],
        [7.8250e-01, 2.1750e-01],
        [3.6915e-02, 9.6309e-01],
        [6.8374e-09, 1.0000e+00],
        [3.2988e-04, 9.9967e-01],
        [1.8241e-05, 9.9998e-01],
        [6.4744e-04, 9.9935e-01],
        [8.1616e-05, 9.9992e-01],
        [2.5293e-06, 1.0000e+00],
        [9.9941e-01, 5.9039e-04],
        [6.2949e-05, 9.9994e-01],
        [4.8272e-05, 9.9995e-01],
        [1.8103e-05, 9.9998e-01],
        [1.7876e-07, 1.0000e+00],
        [1.4257e-02, 9.8574e-01],
        [2.0495e-06, 1.0000e+00],
        [2.3227e-03, 9.9768e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2363e-03, 9.9776e-01],
        [3.0270e-04, 9.9970e-01],
        [2.6458e-05, 9.9997e-01],
        [3.1990e-04, 9.9968e-01],
        [6.3915e-05, 9.9994e-01],
        [7.0019e-03, 9.9300e-01],
        [9.9969e-01, 3.1254e-04],
        [9.9914e-01, 8.5921e-04],
        [9.1017e-03, 9.9090e-01],
        [1.7740e-06, 1.0000e+00],
        [8.6100e-05, 9.9991e-01],
        [2.2838e-03, 9.9772e-01],
        [1.7494e-03, 9.9825e-01],
        [2.5314e-01, 7.4686e-01],
        [1.0904e-06, 1.0000e+00],
        [2.2731e-03, 9.9773e-01],
        [1.7738e-06, 1.0000e+00],
        [2.3877e-02, 9.7612e-01],
        [2.2791e-05, 9.9998e-01],
        [4.6124e-04, 9.9954e-01],
        [4.1239e-06, 1.0000e+00],
        [8.4867e-06, 9.9999e-01],
        [7.5440e-01, 2.4560e-01],
        [2.1209e-04, 9.9979e-01],
        [1.9256e-04, 9.9981e-01],
        [3.2511e-04, 9.9967e-01],
        [6.9917e-03, 9.9301e-01],
        [4.4056e-09, 1.0000e+00],
        [9.9959e-01, 4.1415e-04],
        [4.1360e-03, 9.9586e-01],
        [3.1854e-04, 9.9968e-01],
        [1.8467e-02, 9.8153e-01],
        [8.7648e-05, 9.9991e-01],
        [6.3446e-06, 9.9999e-01],
        [1.2956e-04, 9.9987e-01],
        [2.2876e-07, 1.0000e+00],
        [4.1889e-03, 9.9581e-01],
        [5.5368e-04, 9.9945e-01],
        [7.9714e-05, 9.9992e-01],
        [8.3326e-05, 9.9992e-01],
        [3.7007e-07, 1.0000e+00],
        [4.7512e-04, 9.9952e-01],
        [5.4674e-01, 4.5326e-01],
        [3.9970e-05, 9.9996e-01],
        [3.0384e-02, 9.6962e-01],
        [9.7003e-05, 9.9990e-01],
        [6.7729e-05, 9.9993e-01],
        [3.8039e-03, 9.9620e-01],
        [9.7962e-01, 2.0379e-02],
        [3.6991e-05, 9.9996e-01],
        [9.9963e-01, 3.7038e-04],
        [5.2276e-06, 9.9999e-01],
        [9.9961e-01, 3.8588e-04],
        [9.9367e-01, 6.3337e-03],
        [9.6118e-01, 3.8819e-02],
        [3.3575e-05, 9.9997e-01],
        [7.3297e-05, 9.9993e-01],
        [9.6810e-01, 3.1900e-02],
        [4.1646e-03, 9.9584e-01],
        [6.8154e-01, 3.1846e-01],
        [9.2163e-01, 7.8366e-02],
        [3.9086e-04, 9.9961e-01],
        [9.2591e-03, 9.9074e-01],
        [6.6225e-04, 9.9934e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7456e-04, 9.9983e-01],
        [9.9894e-01, 1.0626e-03],
        [9.9922e-01, 7.8084e-04],
        [4.5078e-06, 1.0000e+00],
        [9.7666e-05, 9.9990e-01],
        [9.9490e-01, 5.0972e-03],
        [1.7314e-03, 9.9827e-01],
        [2.1486e-03, 9.9785e-01],
        [9.9963e-01, 3.6919e-04],
        [4.8031e-03, 9.9520e-01],
        [9.9911e-01, 8.9111e-04],
        [2.0699e-03, 9.9793e-01],
        [1.0862e-05, 9.9999e-01],
        [9.9912e-01, 8.7596e-04],
        [9.9962e-01, 3.7941e-04],
        [9.9950e-01, 4.9760e-04],
        [1.9170e-04, 9.9981e-01],
        [9.3221e-08, 1.0000e+00],
        [9.9947e-01, 5.2838e-04],
        [4.0087e-03, 9.9599e-01],
        [9.9954e-01, 4.5735e-04],
        [2.3194e-07, 1.0000e+00],
        [5.1669e-04, 9.9948e-01],
        [1.0409e-03, 9.9896e-01],
        [9.9943e-01, 5.6879e-04],
        [6.4109e-06, 9.9999e-01],
        [3.0836e-06, 1.0000e+00],
        [8.9712e-01, 1.0288e-01],
        [2.3950e-05, 9.9998e-01],
        [1.4240e-06, 1.0000e+00],
        [1.9674e-01, 8.0326e-01],
        [5.2081e-03, 9.9479e-01],
        [8.3408e-02, 9.1659e-01],
        [4.5811e-04, 9.9954e-01],
        [5.8513e-04, 9.9941e-01],
        [1.1045e-03, 9.9890e-01],
        [4.5324e-04, 9.9955e-01],
        [7.3138e-01, 2.6862e-01],
        [1.3559e-02, 9.8644e-01],
        [9.9920e-01, 8.0315e-04],
        [2.4186e-04, 9.9976e-01],
        [1.7000e-02, 9.8300e-01],
        [6.6685e-02, 9.3332e-01],
        [4.3195e-05, 9.9996e-01],
        [4.1293e-03, 9.9587e-01],
        [5.7941e-06, 9.9999e-01],
        [2.0506e-03, 9.9795e-01],
        [2.1313e-05, 9.9998e-01],
        [3.4608e-05, 9.9997e-01],
        [8.9424e-04, 9.9911e-01],
        [3.6550e-03, 9.9634e-01],
        [9.9628e-01, 3.7181e-03],
        [1.3920e-04, 9.9986e-01],
        [9.8104e-01, 1.8958e-02],
        [9.2868e-03, 9.9071e-01],
        [9.9999e-01, 6.1252e-06],
        [2.0127e-01, 7.9873e-01],
        [4.6908e-04, 9.9953e-01],
        [3.9778e-03, 9.9602e-01],
        [1.5458e-05, 9.9998e-01],
        [1.9438e-04, 9.9981e-01],
        [1.8318e-05, 9.9998e-01],
        [9.9996e-01, 4.3466e-05],
        [9.5579e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1561e-05, 9.9998e-01],
        [4.7091e-04, 9.9953e-01],
        [7.1972e-05, 9.9993e-01],
        [9.9833e-01, 1.6666e-03],
        [3.2822e-05, 9.9997e-01],
        [2.0901e-04, 9.9979e-01],
        [1.4137e-03, 9.9859e-01],
        [1.5922e-03, 9.9841e-01],
        [1.2402e-05, 9.9999e-01],
        [6.7668e-05, 9.9993e-01],
        [9.9261e-01, 7.3893e-03],
        [9.9887e-01, 1.1262e-03],
        [2.3284e-06, 1.0000e+00],
        [3.0905e-03, 9.9691e-01],
        [9.9919e-01, 8.1264e-04],
        [1.8660e-06, 1.0000e+00],
        [9.9929e-01, 7.0821e-04],
        [1.8231e-03, 9.9818e-01],
        [9.0567e-01, 9.4331e-02],
        [4.9658e-07, 1.0000e+00],
        [2.1279e-03, 9.9787e-01],
        [2.8241e-05, 9.9997e-01],
        [7.7384e-05, 9.9992e-01],
        [1.6778e-06, 1.0000e+00],
        [3.5395e-06, 1.0000e+00],
        [9.9664e-01, 3.3619e-03],
        [2.8308e-04, 9.9972e-01],
        [3.3226e-07, 1.0000e+00],
        [1.6379e-01, 8.3621e-01],
        [9.9802e-01, 1.9777e-03],
        [7.9222e-07, 1.0000e+00],
        [2.0215e-06, 1.0000e+00],
        [9.9914e-01, 8.5587e-04],
        [1.3176e-07, 1.0000e+00],
        [5.4651e-04, 9.9945e-01],
        [3.9450e-05, 9.9996e-01],
        [1.6376e-05, 9.9998e-01],
        [9.9978e-01, 2.1965e-04],
        [5.5828e-05, 9.9994e-01],
        [1.0036e-01, 8.9964e-01],
        [1.9206e-05, 9.9998e-01],
        [2.1365e-04, 9.9979e-01],
        [5.8971e-06, 9.9999e-01],
        [9.9022e-01, 9.7779e-03],
        [2.7417e-03, 9.9726e-01],
        [9.9849e-01, 1.5062e-03],
        [9.7938e-02, 9.0206e-01],
        [2.7008e-03, 9.9730e-01],
        [9.9951e-01, 4.8791e-04],
        [5.4591e-05, 9.9995e-01],
        [9.9179e-01, 8.2053e-03],
        [2.1163e-06, 1.0000e+00],
        [7.9402e-04, 9.9921e-01],
        [7.3376e-02, 9.2662e-01],
        [9.9925e-01, 7.5031e-04],
        [1.6501e-02, 9.8350e-01],
        [7.9391e-05, 9.9992e-01],
        [7.7364e-04, 9.9923e-01],
        [8.1166e-02, 9.1883e-01],
        [9.9943e-01, 5.7088e-04],
        [9.9654e-01, 3.4614e-03],
        [3.7861e-05, 9.9996e-01],
        [2.8659e-05, 9.9997e-01],
        [3.0905e-03, 9.9691e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9953e-01, 4.7379e-04],
        [4.9563e-04, 9.9950e-01],
        [9.9967e-01, 3.2950e-04],
        [3.8033e-04, 9.9962e-01],
        [1.1900e-03, 9.9881e-01],
        [3.1331e-05, 9.9997e-01],
        [9.9779e-01, 2.2114e-03],
        [4.6472e-05, 9.9995e-01],
        [7.0063e-05, 9.9993e-01],
        [2.9346e-03, 9.9707e-01],
        [4.8729e-04, 9.9951e-01],
        [8.9472e-03, 9.9105e-01],
        [1.4314e-03, 9.9857e-01],
        [5.3755e-03, 9.9462e-01],
        [9.9936e-01, 6.3505e-04],
        [4.1948e-01, 5.8052e-01],
        [1.4554e-01, 8.5446e-01],
        [1.1239e-02, 9.8876e-01],
        [3.9673e-05, 9.9996e-01],
        [9.5869e-06, 9.9999e-01],
        [9.9716e-01, 2.8353e-03],
        [9.9930e-01, 6.9927e-04],
        [1.3377e-01, 8.6623e-01],
        [4.9601e-07, 1.0000e+00],
        [4.4534e-05, 9.9996e-01],
        [1.3387e-04, 9.9987e-01],
        [2.5385e-03, 9.9746e-01],
        [6.3658e-05, 9.9994e-01],
        [9.9938e-01, 6.1728e-04],
        [9.9984e-01, 1.5753e-04],
        [9.9952e-01, 4.7786e-04],
        [1.8958e-04, 9.9981e-01],
        [5.1140e-02, 9.4886e-01],
        [3.8618e-01, 6.1382e-01],
        [9.5192e-03, 9.9048e-01],
        [5.1160e-05, 9.9995e-01],
        [7.6838e-03, 9.9232e-01],
        [5.4721e-04, 9.9945e-01],
        [8.5361e-03, 9.9146e-01],
        [1.8457e-02, 9.8154e-01],
        [4.2539e-03, 9.9575e-01],
        [9.9988e-01, 1.1724e-04],
        [7.6283e-06, 9.9999e-01],
        [9.2829e-02, 9.0717e-01],
        [9.9939e-01, 6.1333e-04],
        [9.9842e-01, 1.5762e-03],
        [9.9930e-01, 6.9998e-04],
        [1.5547e-06, 1.0000e+00],
        [9.9968e-01, 3.1884e-04],
        [2.3341e-04, 9.9977e-01],
        [3.3398e-03, 9.9666e-01],
        [2.6930e-03, 9.9731e-01],
        [1.1610e-02, 9.8839e-01],
        [7.6952e-03, 9.9230e-01],
        [3.0366e-05, 9.9997e-01],
        [9.7149e-01, 2.8507e-02],
        [1.3253e-03, 9.9867e-01],
        [6.4963e-04, 9.9935e-01],
        [2.1170e-03, 9.9788e-01],
        [3.7121e-03, 9.9629e-01],
        [6.0918e-07, 1.0000e+00],
        [4.1927e-06, 1.0000e+00],
        [7.4104e-03, 9.9259e-01],
        [9.9505e-01, 4.9519e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[4.6867e-04, 9.9953e-01],
        [8.3084e-01, 1.6916e-01],
        [8.9080e-01, 1.0920e-01],
        [6.7932e-07, 1.0000e+00],
        [9.3174e-07, 1.0000e+00],
        [2.9342e-04, 9.9971e-01],
        [2.8190e-06, 1.0000e+00],
        [9.9732e-01, 2.6768e-03],
        [7.3557e-02, 9.2644e-01],
        [9.9980e-01, 1.9598e-04],
        [2.2461e-03, 9.9775e-01],
        [9.9946e-01, 5.3750e-04],
        [2.5129e-04, 9.9975e-01],
        [1.1349e-03, 9.9887e-01],
        [9.7027e-01, 2.9726e-02],
        [5.1517e-03, 9.9485e-01],
        [1.1300e-04, 9.9989e-01],
        [2.1016e-05, 9.9998e-01],
        [4.0752e-02, 9.5925e-01],
        [1.4064e-05, 9.9999e-01],
        [9.5763e-04, 9.9904e-01],
        [5.2483e-03, 9.9475e-01],
        [8.8973e-01, 1.1027e-01],
        [4.5328e-04, 9.9955e-01],
        [3.1142e-04, 9.9969e-01],
        [6.2385e-04, 9.9938e-01],
        [8.0092e-09, 1.0000e+00],
        [9.9996e-01, 4.1930e-05],
        [2.5110e-06, 1.0000e+00],
        [1.9875e-03, 9.9801e-01],
        [8.3960e-01, 1.6040e-01],
        [9.9914e-01, 8.5677e-04],
        [4.6397e-04, 9.9954e-01],
        [8.7709e-07, 1.0000e+00],
        [9.9956e-01, 4.4413e-04],
        [9.9899e-01, 1.0107e-03],
        [1.1845e-06, 1.0000e+00],
        [5.2713e-03, 9.9473e-01],
        [9.8933e-01, 1.0667e-02],
        [2.0165e-04, 9.9980e-01],
        [1.4663e-04, 9.9985e-01],
        [5.3262e-06, 9.9999e-01],
        [9.9917e-01, 8.3028e-04],
        [1.8047e-03, 9.9820e-01],
        [1.6119e-03, 9.9839e-01],
        [2.7900e-03, 9.9721e-01],
        [9.1215e-04, 9.9909e-01],
        [1.4029e-03, 9.9860e-01],
        [2.0570e-03, 9.9794e-01],
        [9.9840e-01, 1.5992e-03],
        [4.2013e-02, 9.5799e-01],
        [9.9907e-01, 9.2956e-04],
        [2.8663e-05, 9.9997e-01],
        [9.9141e-01, 8.5886e-03],
        [2.0221e-03, 9.9798e-01],
        [1.1073e-05, 9.9999e-01],
        [3.3608e-02, 9.6639e-01],
        [4.8314e-04, 9.9952e-01],
        [4.6926e-01, 5.3074e-01],
        [1.4858e-03, 9.9851e-01],
        [7.5795e-02, 9.2420e-01],
        [7.4313e-01, 2.5687e-01],
        [4.5492e-04, 9.9955e-01],
        [7.7839e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9960e-01, 4.0496e-04],
        [5.2762e-04, 9.9947e-01],
        [3.2058e-03, 9.9679e-01],
        [1.1608e-02, 9.8839e-01],
        [1.8505e-04, 9.9981e-01],
        [1.0163e-03, 9.9898e-01],
        [3.2779e-06, 1.0000e+00],
        [5.1458e-08, 1.0000e+00],
        [2.1457e-04, 9.9979e-01],
        [3.4552e-02, 9.6545e-01],
        [8.7122e-05, 9.9991e-01],
        [2.2554e-06, 1.0000e+00],
        [7.6153e-05, 9.9992e-01],
        [1.0615e-03, 9.9894e-01],
        [1.3813e-01, 8.6187e-01],
        [1.2609e-02, 9.8739e-01],
        [6.8876e-04, 9.9931e-01],
        [2.1652e-01, 7.8348e-01],
        [9.9574e-01, 4.2612e-03],
        [4.8785e-05, 9.9995e-01],
        [2.2974e-04, 9.9977e-01],
        [1.2807e-04, 9.9987e-01],
        [5.3513e-05, 9.9995e-01],
        [2.7880e-04, 9.9972e-01],
        [9.9972e-01, 2.7878e-04],
        [9.5467e-01, 4.5328e-02],
        [9.9964e-01, 3.5631e-04],
        [3.0414e-03, 9.9696e-01],
        [4.3507e-07, 1.0000e+00],
        [2.4593e-05, 9.9998e-01],
        [6.7991e-03, 9.9320e-01],
        [4.4053e-02, 9.5595e-01],
        [7.3411e-02, 9.2659e-01],
        [9.7190e-03, 9.9028e-01],
        [1.0129e-06, 1.0000e+00],
        [1.6699e-05, 9.9998e-01],
        [5.3700e-03, 9.9463e-01],
        [1.6673e-06, 1.0000e+00],
        [1.4716e-02, 9.8528e-01],
        [5.1309e-02, 9.4869e-01],
        [5.0754e-05, 9.9995e-01],
        [1.2434e-04, 9.9988e-01],
        [2.6856e-03, 9.9731e-01],
        [4.9733e-04, 9.9950e-01],
        [9.8956e-01, 1.0440e-02],
        [9.5596e-01, 4.4040e-02],
        [3.9429e-05, 9.9996e-01],
        [9.9962e-01, 3.7599e-04],
        [2.0198e-04, 9.9980e-01],
        [1.6506e-05, 9.9998e-01],
        [1.7464e-06, 1.0000e+00],
        [9.9905e-01, 9.5339e-04],
        [2.1999e-02, 9.7800e-01],
        [1.0983e-04, 9.9989e-01],
        [1.1357e-03, 9.9886e-01],
        [7.3449e-05, 9.9993e-01],
        [3.1740e-05, 9.9997e-01],
        [2.5583e-06, 1.0000e+00],
        [9.8081e-01, 1.9191e-02],
        [8.8174e-05, 9.9991e-01],
        [9.9314e-01, 6.8606e-03],
        [9.9945e-01, 5.5135e-04],
        [6.6760e-03, 9.9332e-01],
        [2.1152e-02, 9.7885e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0982e-06, 1.0000e+00],
        [3.2536e-02, 9.6746e-01],
        [8.4212e-08, 1.0000e+00],
        [7.6208e-05, 9.9992e-01],
        [1.8836e-01, 8.1164e-01],
        [7.6367e-01, 2.3633e-01],
        [8.4055e-07, 1.0000e+00],
        [3.0006e-04, 9.9970e-01],
        [9.9933e-01, 6.6728e-04],
        [4.5960e-06, 1.0000e+00],
        [9.5184e-02, 9.0482e-01],
        [1.3622e-04, 9.9986e-01],
        [2.2728e-04, 9.9977e-01],
        [2.6784e-05, 9.9997e-01],
        [1.9285e-05, 9.9998e-01],
        [1.1272e-04, 9.9989e-01],
        [8.2967e-04, 9.9917e-01],
        [9.9852e-01, 1.4844e-03],
        [1.2380e-03, 9.9876e-01],
        [1.4751e-03, 9.9852e-01],
        [1.7255e-04, 9.9983e-01],
        [2.8587e-05, 9.9997e-01],
        [9.9699e-01, 3.0119e-03],
        [3.2043e-08, 1.0000e+00],
        [9.7457e-07, 1.0000e+00],
        [4.5415e-02, 9.5458e-01],
        [2.0011e-04, 9.9980e-01],
        [9.2384e-07, 1.0000e+00],
        [7.0918e-06, 9.9999e-01],
        [7.9611e-07, 1.0000e+00],
        [2.7873e-03, 9.9721e-01],
        [2.0347e-02, 9.7965e-01],
        [3.1183e-03, 9.9688e-01],
        [2.7285e-03, 9.9727e-01],
        [4.5865e-03, 9.9541e-01],
        [4.7040e-04, 9.9953e-01],
        [1.1831e-05, 9.9999e-01],
        [1.9971e-05, 9.9998e-01],
        [5.7606e-05, 9.9994e-01],
        [1.4259e-04, 9.9986e-01],
        [9.9929e-01, 7.1015e-04],
        [1.6846e-05, 9.9998e-01],
        [2.1076e-02, 9.7892e-01],
        [9.5135e-03, 9.9049e-01],
        [9.9916e-01, 8.3782e-04],
        [7.3212e-06, 9.9999e-01],
        [1.2229e-06, 1.0000e+00],
        [9.9986e-01, 1.4232e-04],
        [1.7551e-04, 9.9982e-01],
        [9.9101e-01, 8.9869e-03],
        [9.9944e-01, 5.6067e-04],
        [3.0745e-04, 9.9969e-01],
        [9.7324e-01, 2.6755e-02],
        [9.9978e-01, 2.2424e-04],
        [1.6686e-08, 1.0000e+00],
        [7.6684e-01, 2.3316e-01],
        [2.5566e-04, 9.9974e-01],
        [4.5404e-04, 9.9955e-01],
        [2.3093e-05, 9.9998e-01],
        [1.7837e-01, 8.2163e-01],
        [9.9864e-01, 1.3623e-03],
        [3.8215e-04, 9.9962e-01],
        [5.3195e-07, 1.0000e+00],
        [9.9798e-01, 2.0198e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5770e-07, 1.0000e+00],
        [2.3122e-02, 9.7688e-01],
        [6.6796e-05, 9.9993e-01],
        [5.3978e-07, 1.0000e+00],
        [3.2902e-05, 9.9997e-01],
        [9.9953e-01, 4.6624e-04],
        [2.0030e-02, 9.7997e-01],
        [1.6606e-07, 1.0000e+00],
        [2.4097e-08, 1.0000e+00],
        [1.1531e-03, 9.9885e-01],
        [2.4209e-04, 9.9976e-01],
        [3.7996e-07, 1.0000e+00],
        [3.7828e-07, 1.0000e+00],
        [1.3383e-06, 1.0000e+00],
        [7.0460e-05, 9.9993e-01],
        [1.1320e-01, 8.8680e-01],
        [3.7925e-04, 9.9962e-01],
        [4.0144e-04, 9.9960e-01],
        [5.5200e-04, 9.9945e-01],
        [5.8648e-04, 9.9941e-01],
        [5.9479e-07, 1.0000e+00],
        [1.7545e-07, 1.0000e+00],
        [5.6552e-04, 9.9943e-01],
        [9.9971e-01, 2.9414e-04],
        [5.7214e-05, 9.9994e-01],
        [5.8922e-03, 9.9411e-01],
        [1.1457e-02, 9.8854e-01],
        [1.2188e-05, 9.9999e-01],
        [1.1543e-04, 9.9988e-01],
        [9.9969e-01, 3.1137e-04],
        [9.9517e-01, 4.8299e-03],
        [5.9463e-05, 9.9994e-01],
        [6.7240e-04, 9.9933e-01],
        [6.1580e-04, 9.9938e-01],
        [2.4472e-05, 9.9998e-01],
        [4.0222e-05, 9.9996e-01],
        [6.0274e-11, 1.0000e+00],
        [9.9815e-01, 1.8502e-03],
        [3.8301e-03, 9.9617e-01],
        [2.1312e-05, 9.9998e-01],
        [4.4357e-05, 9.9996e-01],
        [9.9945e-01, 5.5234e-04],
        [8.1724e-06, 9.9999e-01],
        [9.9081e-01, 9.1944e-03],
        [1.7199e-06, 1.0000e+00],
        [1.1708e-03, 9.9883e-01],
        [4.5757e-04, 9.9954e-01],
        [4.4068e-04, 9.9956e-01],
        [9.8325e-04, 9.9902e-01],
        [1.1917e-06, 1.0000e+00],
        [4.7601e-06, 1.0000e+00],
        [2.5377e-03, 9.9746e-01],
        [2.0793e-05, 9.9998e-01],
        [2.3492e-02, 9.7651e-01],
        [9.6544e-01, 3.4560e-02],
        [1.4287e-05, 9.9999e-01],
        [9.1359e-05, 9.9991e-01],
        [2.1771e-05, 9.9998e-01],
        [4.2540e-06, 1.0000e+00],
        [9.9967e-01, 3.2979e-04],
        [9.5099e-06, 9.9999e-01],
        [9.9950e-01, 5.0327e-04],
        [9.9968e-01, 3.2127e-04],
        [1.7974e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4593e-02, 9.8541e-01],
        [9.8168e-06, 9.9999e-01],
        [9.7409e-08, 1.0000e+00],
        [9.9992e-01, 8.3542e-05],
        [9.9953e-01, 4.6969e-04],
        [9.3855e-01, 6.1446e-02],
        [5.8222e-03, 9.9418e-01],
        [9.9762e-01, 2.3769e-03],
        [1.8015e-06, 1.0000e+00],
        [1.4484e-04, 9.9986e-01],
        [2.5445e-04, 9.9975e-01],
        [9.9970e-01, 3.0231e-04],
        [1.2424e-04, 9.9988e-01],
        [1.1309e-02, 9.8869e-01],
        [3.1019e-04, 9.9969e-01],
        [1.4704e-06, 1.0000e+00],
        [9.9912e-01, 8.8040e-04],
        [2.8095e-02, 9.7190e-01],
        [2.0091e-04, 9.9980e-01],
        [1.4739e-05, 9.9999e-01],
        [9.9914e-01, 8.6262e-04],
        [3.9496e-05, 9.9996e-01],
        [2.6620e-01, 7.3380e-01],
        [1.0865e-02, 9.8913e-01],
        [4.6444e-05, 9.9995e-01],
        [1.1830e-03, 9.9882e-01],
        [9.9833e-01, 1.6717e-03],
        [9.9899e-01, 1.0134e-03],
        [1.5908e-07, 1.0000e+00],
        [3.4911e-05, 9.9997e-01],
        [9.9869e-01, 1.3123e-03],
        [2.7426e-06, 1.0000e+00],
        [9.9993e-01, 6.5496e-05],
        [3.3620e-03, 9.9664e-01],
        [9.5845e-01, 4.1550e-02],
        [2.2878e-04, 9.9977e-01],
        [6.2341e-07, 1.0000e+00],
        [8.8870e-02, 9.1113e-01],
        [9.9872e-01, 1.2768e-03],
        [9.9965e-01, 3.5391e-04],
        [3.2475e-07, 1.0000e+00],
        [2.3920e-05, 9.9998e-01],
        [1.5036e-06, 1.0000e+00],
        [5.5384e-05, 9.9994e-01],
        [1.3393e-07, 1.0000e+00],
        [3.8946e-03, 9.9611e-01],
        [9.9997e-01, 3.3220e-05],
        [1.4491e-04, 9.9986e-01],
        [7.0789e-07, 1.0000e+00],
        [1.6766e-02, 9.8323e-01],
        [7.8921e-03, 9.9211e-01],
        [6.0201e-06, 9.9999e-01],
        [2.2272e-04, 9.9978e-01],
        [1.3044e-02, 9.8696e-01],
        [2.5312e-03, 9.9747e-01],
        [9.9954e-01, 4.5865e-04],
        [9.9926e-01, 7.3507e-04],
        [9.8243e-01, 1.7566e-02],
        [3.8903e-02, 9.6110e-01],
        [1.0636e-04, 9.9989e-01],
        [1.8540e-07, 1.0000e+00],
        [8.5605e-07, 1.0000e+00],
        [1.9928e-05, 9.9998e-01],
        [1.3413e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9950e-01, 4.9787e-04],
        [5.7209e-05, 9.9994e-01],
        [2.4687e-03, 9.9753e-01],
        [2.0546e-07, 1.0000e+00],
        [6.8582e-05, 9.9993e-01],
        [2.5184e-04, 9.9975e-01],
        [6.8790e-01, 3.1210e-01],
        [9.9703e-01, 2.9743e-03],
        [5.6695e-06, 9.9999e-01],
        [1.2833e-03, 9.9872e-01],
        [7.7790e-04, 9.9922e-01],
        [3.0044e-06, 1.0000e+00],
        [1.6862e-04, 9.9983e-01],
        [1.6281e-04, 9.9984e-01],
        [1.9996e-05, 9.9998e-01],
        [1.1012e-02, 9.8899e-01],
        [1.6504e-02, 9.8350e-01],
        [5.8859e-01, 4.1141e-01],
        [8.7495e-04, 9.9913e-01],
        [9.3099e-01, 6.9010e-02],
        [2.4765e-05, 9.9998e-01],
        [2.3753e-07, 1.0000e+00],
        [5.4794e-04, 9.9945e-01],
        [5.8766e-06, 9.9999e-01],
        [2.0904e-05, 9.9998e-01],
        [2.5626e-06, 1.0000e+00],
        [9.9938e-01, 6.2041e-04],
        [2.1209e-05, 9.9998e-01],
        [1.0460e-08, 1.0000e+00],
        [1.0027e-05, 9.9999e-01],
        [3.1907e-07, 1.0000e+00],
        [5.6854e-04, 9.9943e-01],
        [9.9881e-01, 1.1885e-03],
        [9.4858e-04, 9.9905e-01],
        [9.9933e-01, 6.7158e-04],
        [1.2098e-06, 1.0000e+00],
        [9.2183e-01, 7.8172e-02],
        [4.1181e-05, 9.9996e-01],
        [1.6210e-07, 1.0000e+00],
        [1.8284e-07, 1.0000e+00],
        [3.8775e-03, 9.9612e-01],
        [1.3818e-05, 9.9999e-01],
        [2.6815e-05, 9.9997e-01],
        [1.4458e-02, 9.8554e-01],
        [8.9012e-01, 1.0988e-01],
        [9.9945e-01, 5.4805e-04],
        [8.0483e-05, 9.9992e-01],
        [2.4384e-04, 9.9976e-01],
        [2.6132e-05, 9.9997e-01],
        [4.0530e-07, 1.0000e+00],
        [4.9715e-03, 9.9503e-01],
        [9.9809e-01, 1.9087e-03],
        [3.1247e-07, 1.0000e+00],
        [3.1662e-05, 9.9997e-01],
        [1.4483e-05, 9.9999e-01],
        [2.9635e-04, 9.9970e-01],
        [6.5371e-07, 1.0000e+00],
        [2.8517e-04, 9.9971e-01],
        [1.3214e-05, 9.9999e-01],
        [1.5061e-07, 1.0000e+00],
        [9.9894e-01, 1.0611e-03],
        [5.9179e-05, 9.9994e-01],
        [8.6488e-06, 9.9999e-01],
        [4.2710e-03, 9.9573e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9692e-01, 3.0795e-03],
        [3.1205e-03, 9.9688e-01],
        [9.9928e-01, 7.1657e-04],
        [1.0142e-02, 9.8986e-01],
        [2.9486e-01, 7.0514e-01],
        [8.1146e-03, 9.9189e-01],
        [2.2034e-05, 9.9998e-01],
        [3.4361e-04, 9.9966e-01],
        [3.8443e-03, 9.9616e-01],
        [3.5936e-04, 9.9964e-01],
        [5.0949e-06, 9.9999e-01],
        [2.6139e-05, 9.9997e-01],
        [1.3110e-04, 9.9987e-01],
        [1.3808e-03, 9.9862e-01],
        [5.7598e-06, 9.9999e-01],
        [3.7770e-05, 9.9996e-01],
        [1.2348e-06, 1.0000e+00],
        [2.8586e-04, 9.9971e-01],
        [2.1841e-08, 1.0000e+00],
        [9.9906e-01, 9.3614e-04],
        [5.5434e-06, 9.9999e-01],
        [1.1525e-04, 9.9988e-01],
        [6.3111e-09, 1.0000e+00],
        [2.4847e-04, 9.9975e-01],
        [8.1200e-01, 1.8800e-01],
        [2.1953e-06, 1.0000e+00],
        [2.9852e-02, 9.7015e-01],
        [6.1734e-05, 9.9994e-01],
        [9.9797e-01, 2.0263e-03],
        [6.8313e-01, 3.1687e-01],
        [1.0911e-04, 9.9989e-01],
        [3.9387e-06, 1.0000e+00],
        [7.7962e-04, 9.9922e-01],
        [8.0767e-06, 9.9999e-01],
        [2.3368e-06, 1.0000e+00],
        [7.3607e-07, 1.0000e+00],
        [9.6601e-07, 1.0000e+00],
        [9.9424e-06, 9.9999e-01],
        [4.1971e-08, 1.0000e+00],
        [1.2850e-04, 9.9987e-01],
        [1.3655e-03, 9.9863e-01],
        [4.4305e-03, 9.9557e-01],
        [6.9824e-03, 9.9302e-01],
        [4.4408e-01, 5.5592e-01],
        [9.9736e-01, 2.6387e-03],
        [1.9901e-06, 1.0000e+00],
        [3.3875e-03, 9.9661e-01],
        [2.6621e-05, 9.9997e-01],
        [2.2757e-07, 1.0000e+00],
        [1.2734e-05, 9.9999e-01],
        [8.5930e-01, 1.4070e-01],
        [4.3718e-04, 9.9956e-01],
        [5.7250e-04, 9.9943e-01],
        [9.5901e-01, 4.0990e-02],
        [1.1005e-04, 9.9989e-01],
        [1.0631e-07, 1.0000e+00],
        [9.9943e-01, 5.6822e-04],
        [9.9987e-01, 1.2894e-04],
        [9.9570e-01, 4.3016e-03],
        [1.9563e-04, 9.9980e-01],
        [9.7600e-01, 2.3999e-02],
        [3.2835e-03, 9.9672e-01],
        [6.9835e-05, 9.9993e-01],
        [2.2980e-03, 9.9770e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5988e-04, 9.9974e-01],
        [7.6666e-06, 9.9999e-01],
        [2.1199e-03, 9.9788e-01],
        [7.6666e-06, 9.9999e-01],
        [5.4438e-01, 4.5562e-01],
        [9.1625e-01, 8.3748e-02],
        [9.9963e-01, 3.7446e-04],
        [9.9887e-01, 1.1283e-03],
        [9.5076e-05, 9.9990e-01],
        [2.1809e-05, 9.9998e-01],
        [3.7927e-05, 9.9996e-01],
        [7.8155e-06, 9.9999e-01],
        [1.3214e-08, 1.0000e+00],
        [4.5594e-06, 1.0000e+00],
        [2.3093e-05, 9.9998e-01],
        [3.0894e-06, 1.0000e+00],
        [3.6946e-02, 9.6305e-01],
        [9.2356e-04, 9.9908e-01],
        [9.8710e-01, 1.2905e-02],
        [1.3882e-04, 9.9986e-01],
        [5.1360e-05, 9.9995e-01],
        [4.0090e-05, 9.9996e-01],
        [7.8572e-07, 1.0000e+00],
        [2.9431e-04, 9.9971e-01],
        [5.1225e-06, 9.9999e-01],
        [1.3214e-08, 1.0000e+00],
        [7.8171e-08, 1.0000e+00],
        [3.9366e-05, 9.9996e-01],
        [4.8748e-04, 9.9951e-01],
        [1.1274e-05, 9.9999e-01],
        [7.4549e-01, 2.5451e-01],
        [9.9241e-01, 7.5887e-03],
        [4.1991e-07, 1.0000e+00],
        [9.9869e-01, 1.3147e-03],
        [1.0046e-06, 1.0000e+00],
        [6.8607e-06, 9.9999e-01],
        [3.3395e-06, 1.0000e+00],
        [1.4567e-06, 1.0000e+00],
        [1.8218e-05, 9.9998e-01],
        [2.6808e-04, 9.9973e-01],
        [6.6890e-05, 9.9993e-01],
        [4.9472e-04, 9.9951e-01],
        [6.2739e-07, 1.0000e+00],
        [2.3697e-05, 9.9998e-01],
        [9.5766e-01, 4.2345e-02],
        [9.9807e-01, 1.9252e-03],
        [8.8734e-01, 1.1266e-01],
        [7.5807e-08, 1.0000e+00],
        [9.9348e-01, 6.5235e-03],
        [1.5298e-04, 9.9985e-01],
        [3.2937e-06, 1.0000e+00],
        [1.1236e-03, 9.9888e-01],
        [3.6810e-06, 1.0000e+00],
        [8.7659e-01, 1.2341e-01],
        [1.2130e-06, 1.0000e+00],
        [9.9641e-01, 3.5883e-03],
        [6.8166e-01, 3.1834e-01],
        [5.2596e-08, 1.0000e+00],
        [3.7432e-07, 1.0000e+00],
        [6.5057e-08, 1.0000e+00],
        [9.9817e-01, 1.8328e-03],
        [4.5556e-04, 9.9954e-01],
        [9.9546e-07, 1.0000e+00],
        [6.3911e-01, 3.6089e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9868e-01, 1.3221e-03],
        [9.9411e-01, 5.8888e-03],
        [6.4473e-08, 1.0000e+00],
        [7.9601e-07, 1.0000e+00],
        [2.5862e-05, 9.9997e-01],
        [9.8707e-01, 1.2928e-02],
        [9.9986e-01, 1.3697e-04],
        [3.2223e-07, 1.0000e+00],
        [2.3158e-06, 1.0000e+00],
        [1.2445e-04, 9.9988e-01],
        [9.9914e-01, 8.6013e-04],
        [4.5710e-06, 1.0000e+00],
        [8.7923e-05, 9.9991e-01],
        [1.1244e-02, 9.8876e-01],
        [6.1446e-01, 3.8554e-01],
        [3.1075e-05, 9.9997e-01],
        [1.1561e-06, 1.0000e+00],
        [4.5638e-06, 1.0000e+00],
        [4.5464e-09, 1.0000e+00],
        [2.5506e-05, 9.9997e-01],
        [1.8433e-01, 8.1567e-01],
        [8.8471e-08, 1.0000e+00],
        [9.1147e-01, 8.8530e-02],
        [9.9993e-01, 7.1795e-05],
        [9.9911e-01, 8.8877e-04],
        [1.5216e-02, 9.8478e-01],
        [4.7734e-07, 1.0000e+00],
        [9.9908e-01, 9.2120e-04],
        [2.1293e-04, 9.9979e-01],
        [2.6613e-08, 1.0000e+00],
        [4.7796e-05, 9.9995e-01],
        [2.0915e-03, 9.9791e-01],
        [6.5113e-03, 9.9349e-01],
        [9.9699e-01, 3.0055e-03],
        [7.8298e-04, 9.9922e-01],
        [9.1714e-06, 9.9999e-01],
        [9.9947e-01, 5.3502e-04],
        [6.9322e-01, 3.0678e-01],
        [4.7930e-04, 9.9952e-01],
        [9.4968e-06, 9.9999e-01],
        [2.8947e-04, 9.9971e-01],
        [1.2673e-01, 8.7327e-01],
        [2.9910e-03, 9.9701e-01],
        [9.9872e-01, 1.2830e-03],
        [2.5654e-01, 7.4346e-01],
        [1.8663e-06, 1.0000e+00],
        [2.9589e-03, 9.9704e-01],
        [9.9951e-01, 4.8513e-04],
        [1.6673e-03, 9.9833e-01],
        [3.1128e-07, 1.0000e+00],
        [3.1527e-03, 9.9685e-01],
        [9.9783e-01, 2.1732e-03],
        [9.9955e-01, 4.5035e-04],
        [6.5719e-04, 9.9934e-01],
        [8.5011e-07, 1.0000e+00],
        [1.9339e-09, 1.0000e+00],
        [3.3785e-03, 9.9662e-01],
        [9.9950e-01, 4.9757e-04],
        [2.6863e-06, 1.0000e+00],
        [4.5905e-06, 1.0000e+00],
        [9.9734e-01, 2.6565e-03],
        [1.6996e-07, 1.0000e+00],
        [9.9913e-01, 8.6887e-04],
        [5.4853e-02, 9.4515e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0315e-02, 9.7968e-01],
        [6.1226e-01, 3.8774e-01],
        [1.6161e-02, 9.8384e-01],
        [9.9944e-01, 5.6022e-04],
        [9.7349e-01, 2.6506e-02],
        [9.9844e-01, 1.5567e-03],
        [9.9940e-01, 6.0056e-04],
        [7.3023e-06, 9.9999e-01],
        [2.7502e-02, 9.7250e-01],
        [8.5386e-02, 9.1461e-01],
        [9.2738e-03, 9.9073e-01],
        [9.8400e-01, 1.6003e-02],
        [9.9850e-01, 1.5008e-03],
        [1.7616e-08, 1.0000e+00],
        [1.7110e-05, 9.9998e-01],
        [5.7628e-08, 1.0000e+00],
        [1.3159e-07, 1.0000e+00],
        [6.9871e-05, 9.9993e-01],
        [3.0566e-05, 9.9997e-01],
        [1.6417e-02, 9.8358e-01],
        [6.1305e-06, 9.9999e-01],
        [9.7937e-01, 2.0634e-02],
        [4.0066e-05, 9.9996e-01],
        [1.1408e-01, 8.8592e-01],
        [5.4034e-06, 9.9999e-01],
        [3.6055e-07, 1.0000e+00],
        [8.0792e-04, 9.9919e-01],
        [5.0534e-06, 9.9999e-01],
        [2.0162e-03, 9.9798e-01],
        [1.0215e-04, 9.9990e-01],
        [4.8154e-05, 9.9995e-01],
        [3.9167e-07, 1.0000e+00],
        [7.4417e-03, 9.9256e-01],
        [3.6712e-04, 9.9963e-01],
        [9.7052e-01, 2.9480e-02],
        [9.9650e-01, 3.4970e-03],
        [5.4505e-05, 9.9995e-01],
        [1.4097e-02, 9.8590e-01],
        [2.2197e-05, 9.9998e-01],
        [9.9842e-01, 1.5783e-03],
        [1.0581e-07, 1.0000e+00],
        [1.6388e-04, 9.9984e-01],
        [6.0413e-05, 9.9994e-01],
        [1.1776e-03, 9.9882e-01],
        [6.1061e-03, 9.9389e-01],
        [1.5440e-05, 9.9998e-01],
        [1.6259e-05, 9.9998e-01],
        [9.9915e-01, 8.4833e-04],
        [1.7715e-06, 1.0000e+00],
        [1.1622e-06, 1.0000e+00],
        [3.1450e-06, 1.0000e+00],
        [3.5699e-07, 1.0000e+00],
        [2.3384e-03, 9.9766e-01],
        [9.7786e-01, 2.2143e-02],
        [2.0568e-06, 1.0000e+00],
        [4.5752e-01, 5.4248e-01],
        [1.1663e-03, 9.9883e-01],
        [1.3378e-04, 9.9987e-01],
        [7.4719e-06, 9.9999e-01],
        [9.9999e-01, 7.4932e-06],
        [1.3307e-04, 9.9987e-01],
        [5.8619e-08, 1.0000e+00],
        [1.3594e-03, 9.9864e-01],
        [2.6298e-04, 9.9974e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.1437e-04, 9.9909e-01],
        [1.9577e-03, 9.9804e-01],
        [1.3927e-05, 9.9999e-01],
        [8.0443e-06, 9.9999e-01],
        [1.0400e-04, 9.9990e-01],
        [8.6815e-08, 1.0000e+00],
        [1.1548e-01, 8.8452e-01],
        [3.7709e-08, 1.0000e+00],
        [9.9929e-01, 7.0980e-04],
        [8.0933e-05, 9.9992e-01],
        [7.1562e-03, 9.9284e-01],
        [1.1933e-05, 9.9999e-01],
        [9.9955e-01, 4.5460e-04],
        [9.9988e-01, 1.2365e-04],
        [1.8064e-03, 9.9819e-01],
        [1.8266e-04, 9.9982e-01],
        [9.9865e-01, 1.3481e-03],
        [9.9821e-01, 1.7878e-03],
        [5.2122e-06, 9.9999e-01],
        [7.0291e-05, 9.9993e-01],
        [1.7334e-03, 9.9827e-01],
        [1.7490e-07, 1.0000e+00],
        [3.6263e-04, 9.9964e-01],
        [1.7814e-04, 9.9982e-01],
        [9.1716e-01, 8.2844e-02],
        [2.7241e-07, 1.0000e+00],
        [2.5069e-03, 9.9749e-01],
        [2.6233e-05, 9.9997e-01],
        [9.8613e-05, 9.9990e-01],
        [3.6678e-01, 6.3322e-01],
        [1.1335e-05, 9.9999e-01],
        [2.1082e-07, 1.0000e+00],
        [3.6039e-04, 9.9964e-01],
        [9.2881e-01, 7.1192e-02],
        [4.2514e-03, 9.9575e-01],
        [4.0811e-04, 9.9959e-01],
        [7.3218e-05, 9.9993e-01],
        [7.2053e-05, 9.9993e-01],
        [3.6650e-05, 9.9996e-01],
        [4.4770e-05, 9.9996e-01],
        [1.0290e-03, 9.9897e-01],
        [2.0782e-04, 9.9979e-01],
        [1.2813e-07, 1.0000e+00],
        [9.8954e-01, 1.0465e-02],
        [9.9866e-01, 1.3407e-03],
        [4.2968e-04, 9.9957e-01],
        [5.6528e-06, 9.9999e-01],
        [9.9944e-01, 5.6174e-04],
        [2.7661e-05, 9.9997e-01],
        [1.3203e-02, 9.8680e-01],
        [9.9920e-01, 7.9905e-04],
        [9.9891e-01, 1.0922e-03],
        [4.2453e-04, 9.9958e-01],
        [5.3288e-07, 1.0000e+00],
        [4.6374e-04, 9.9954e-01],
        [1.7628e-04, 9.9982e-01],
        [9.4061e-06, 9.9999e-01],
        [1.8082e-06, 1.0000e+00],
        [3.3218e-05, 9.9997e-01],
        [9.7998e-08, 1.0000e+00],
        [2.0194e-02, 9.7981e-01],
        [9.8280e-01, 1.7198e-02],
        [3.2474e-01, 6.7526e-01],
        [4.6028e-02, 9.5397e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2010e-01, 5.7990e-01],
        [9.3193e-01, 6.8069e-02],
        [1.4315e-05, 9.9999e-01],
        [9.1976e-05, 9.9991e-01],
        [8.6990e-04, 9.9913e-01],
        [4.9910e-03, 9.9501e-01],
        [2.4341e-03, 9.9757e-01],
        [9.9029e-01, 9.7070e-03],
        [8.2856e-05, 9.9992e-01],
        [9.9854e-01, 1.4565e-03],
        [1.8845e-06, 1.0000e+00],
        [1.9466e-07, 1.0000e+00],
        [8.3571e-01, 1.6429e-01],
        [1.2720e-07, 1.0000e+00],
        [2.0772e-05, 9.9998e-01],
        [1.4647e-05, 9.9999e-01],
        [1.1229e-04, 9.9989e-01],
        [9.8724e-01, 1.2764e-02],
        [3.6152e-07, 1.0000e+00],
        [9.6598e-01, 3.4019e-02],
        [1.5433e-04, 9.9985e-01],
        [3.0643e-05, 9.9997e-01],
        [9.9933e-01, 6.7049e-04],
        [9.9982e-01, 1.7771e-04],
        [3.0349e-07, 1.0000e+00],
        [8.7404e-02, 9.1260e-01],
        [5.3740e-05, 9.9995e-01],
        [5.6542e-05, 9.9994e-01],
        [6.0313e-05, 9.9994e-01],
        [9.9907e-01, 9.2943e-04],
        [9.3740e-08, 1.0000e+00],
        [5.8300e-01, 4.1700e-01],
        [9.9916e-01, 8.4167e-04],
        [9.9935e-01, 6.5151e-04],
        [9.9959e-01, 4.1093e-04],
        [9.9953e-01, 4.7397e-04],
        [1.3591e-06, 1.0000e+00],
        [2.0521e-04, 9.9979e-01],
        [1.9720e-08, 1.0000e+00],
        [2.9237e-04, 9.9971e-01],
        [9.9970e-01, 3.0378e-04],
        [2.3218e-06, 1.0000e+00],
        [1.6013e-06, 1.0000e+00],
        [1.1265e-04, 9.9989e-01],
        [2.0827e-07, 1.0000e+00],
        [9.9993e-01, 7.3431e-05],
        [1.5040e-03, 9.9850e-01],
        [9.9612e-01, 3.8789e-03],
        [1.2626e-06, 1.0000e+00],
        [1.8079e-05, 9.9998e-01],
        [2.7067e-03, 9.9729e-01],
        [4.0077e-06, 1.0000e+00],
        [3.7541e-05, 9.9996e-01],
        [3.8167e-05, 9.9996e-01],
        [3.4486e-01, 6.5514e-01],
        [5.4943e-01, 4.5057e-01],
        [2.8569e-04, 9.9971e-01],
        [4.7136e-04, 9.9953e-01],
        [3.9691e-07, 1.0000e+00],
        [9.9975e-01, 2.4703e-04],
        [9.1065e-01, 8.9353e-02],
        [3.7722e-04, 9.9962e-01],
        [7.8976e-03, 9.9210e-01],
        [3.1128e-04, 9.9969e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5905e-07, 1.0000e+00],
        [9.9941e-01, 5.9495e-04],
        [9.9953e-01, 4.6911e-04],
        [9.7491e-01, 2.5088e-02],
        [2.4748e-06, 1.0000e+00],
        [3.0298e-03, 9.9697e-01],
        [4.2340e-06, 1.0000e+00],
        [7.0118e-01, 2.9882e-01],
        [9.1625e-05, 9.9991e-01],
        [9.9954e-01, 4.6003e-04],
        [3.3361e-02, 9.6664e-01],
        [8.9635e-01, 1.0365e-01],
        [7.2125e-07, 1.0000e+00],
        [7.0864e-04, 9.9929e-01],
        [2.5381e-04, 9.9975e-01],
        [6.0215e-01, 3.9785e-01],
        [6.1472e-05, 9.9994e-01],
        [3.2857e-04, 9.9967e-01],
        [9.7633e-01, 2.3672e-02],
        [9.3944e-04, 9.9906e-01],
        [9.9944e-01, 5.6363e-04],
        [1.1659e-02, 9.8834e-01],
        [5.9377e-05, 9.9994e-01],
        [3.6208e-06, 1.0000e+00],
        [1.2338e-05, 9.9999e-01],
        [9.9948e-01, 5.2053e-04],
        [9.9961e-05, 9.9990e-01],
        [9.4791e-08, 1.0000e+00],
        [8.2466e-01, 1.7534e-01],
        [5.7502e-04, 9.9943e-01],
        [5.4927e-08, 1.0000e+00],
        [4.3814e-03, 9.9562e-01],
        [7.5142e-05, 9.9992e-01],
        [3.9046e-07, 1.0000e+00],
        [1.0880e-08, 1.0000e+00],
        [9.2769e-01, 7.2305e-02],
        [4.8359e-04, 9.9952e-01],
        [2.4669e-05, 9.9998e-01],
        [1.3136e-02, 9.8686e-01],
        [9.8442e-05, 9.9990e-01],
        [9.5877e-01, 4.1227e-02],
        [1.3390e-05, 9.9999e-01],
        [1.7454e-04, 9.9983e-01],
        [5.1453e-01, 4.8547e-01],
        [9.9913e-01, 8.7498e-04],
        [5.5665e-05, 9.9994e-01],
        [2.7021e-05, 9.9997e-01],
        [2.5660e-01, 7.4340e-01],
        [8.8213e-02, 9.1179e-01],
        [6.0732e-07, 1.0000e+00],
        [1.0000e+00, 2.1854e-06],
        [3.4599e-03, 9.9654e-01],
        [8.5247e-02, 9.1475e-01],
        [1.7674e-06, 1.0000e+00],
        [7.3293e-05, 9.9993e-01],
        [1.4437e-04, 9.9986e-01],
        [9.8221e-01, 1.7788e-02],
        [5.9082e-04, 9.9941e-01],
        [3.0725e-04, 9.9969e-01],
        [8.1957e-01, 1.8043e-01],
        [9.8816e-01, 1.1840e-02],
        [1.7796e-03, 9.9822e-01],
        [7.4913e-05, 9.9993e-01],
        [9.9486e-01, 5.1402e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2766e-03, 9.9572e-01],
        [3.0072e-05, 9.9997e-01],
        [1.0816e-07, 1.0000e+00],
        [1.1566e-01, 8.8434e-01],
        [4.1390e-06, 1.0000e+00],
        [2.6871e-06, 1.0000e+00],
        [9.9828e-01, 1.7169e-03],
        [9.9922e-01, 7.7856e-04],
        [5.7402e-07, 1.0000e+00],
        [4.5807e-04, 9.9954e-01],
        [1.1104e-04, 9.9989e-01],
        [2.4196e-04, 9.9976e-01],
        [6.2636e-03, 9.9374e-01],
        [3.2898e-04, 9.9967e-01],
        [2.3226e-03, 9.9768e-01],
        [9.9962e-01, 3.7612e-04],
        [1.5939e-04, 9.9984e-01],
        [3.1962e-08, 1.0000e+00],
        [2.4208e-05, 9.9998e-01],
        [9.9955e-01, 4.4944e-04],
        [3.3310e-05, 9.9997e-01],
        [1.9831e-03, 9.9802e-01],
        [9.7538e-03, 9.9025e-01],
        [3.1116e-05, 9.9997e-01],
        [9.9940e-01, 6.0012e-04],
        [8.7159e-01, 1.2841e-01],
        [1.2409e-07, 1.0000e+00],
        [1.8408e-03, 9.9816e-01],
        [1.3937e-04, 9.9986e-01],
        [6.0846e-07, 1.0000e+00],
        [9.9978e-01, 2.2062e-04],
        [1.6202e-05, 9.9998e-01],
        [9.9561e-01, 4.3899e-03],
        [9.9993e-01, 6.7455e-05],
        [1.8621e-06, 1.0000e+00],
        [2.4720e-04, 9.9975e-01],
        [3.4579e-04, 9.9965e-01],
        [3.1971e-04, 9.9968e-01],
        [1.3527e-03, 9.9865e-01],
        [1.9867e-03, 9.9801e-01],
        [1.8526e-02, 9.8147e-01],
        [1.1915e-01, 8.8085e-01],
        [7.8279e-01, 2.1721e-01],
        [1.3158e-04, 9.9987e-01],
        [9.8478e-01, 1.5222e-02],
        [4.0488e-03, 9.9595e-01],
        [1.1622e-06, 1.0000e+00],
        [3.2827e-04, 9.9967e-01],
        [3.9571e-07, 1.0000e+00],
        [1.5779e-06, 1.0000e+00],
        [2.4368e-03, 9.9756e-01],
        [9.5669e-07, 1.0000e+00],
        [6.7489e-07, 1.0000e+00],
        [2.9475e-08, 1.0000e+00],
        [1.0841e-06, 1.0000e+00],
        [9.9985e-01, 1.4653e-04],
        [9.9954e-01, 4.6083e-04],
        [2.3226e-03, 9.9768e-01],
        [9.7950e-01, 2.0502e-02],
        [9.9929e-01, 7.0585e-04],
        [4.5989e-05, 9.9995e-01],
        [1.2756e-04, 9.9987e-01],
        [1.3525e-06, 1.0000e+00],
        [9.9918e-01, 8.2420e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5203e-03, 9.9848e-01],
        [8.0661e-08, 1.0000e+00],
        [9.9950e-01, 5.0354e-04],
        [1.2437e-06, 1.0000e+00],
        [9.2161e-01, 7.8394e-02],
        [9.9825e-01, 1.7490e-03],
        [1.2315e-01, 8.7685e-01],
        [7.6363e-05, 9.9992e-01],
        [1.2014e-03, 9.9880e-01],
        [2.1631e-04, 9.9978e-01],
        [3.5383e-04, 9.9965e-01],
        [9.9971e-01, 2.9370e-04],
        [1.4990e-02, 9.8501e-01],
        [7.7734e-03, 9.9223e-01],
        [1.7832e-01, 8.2168e-01],
        [7.5015e-04, 9.9925e-01],
        [2.6882e-04, 9.9973e-01],
        [1.6812e-04, 9.9983e-01],
        [2.2433e-04, 9.9978e-01],
        [4.3018e-03, 9.9570e-01],
        [6.1108e-04, 9.9939e-01],
        [8.1379e-06, 9.9999e-01],
        [3.5115e-04, 9.9965e-01],
        [2.7469e-04, 9.9973e-01],
        [5.3843e-07, 1.0000e+00],
        [4.5650e-06, 1.0000e+00],
        [1.2815e-07, 1.0000e+00],
        [2.1916e-08, 1.0000e+00],
        [1.8643e-04, 9.9981e-01],
        [2.8204e-05, 9.9997e-01],
        [2.8195e-04, 9.9972e-01],
        [1.4078e-08, 1.0000e+00],
        [7.1912e-01, 2.8088e-01],
        [6.5991e-06, 9.9999e-01],
        [7.8972e-07, 1.0000e+00],
        [8.5565e-04, 9.9914e-01],
        [1.1439e-04, 9.9989e-01],
        [9.6893e-01, 3.1066e-02],
        [6.9689e-02, 9.3031e-01],
        [2.3779e-08, 1.0000e+00],
        [8.0418e-06, 9.9999e-01],
        [1.8593e-08, 1.0000e+00],
        [6.2411e-05, 9.9994e-01],
        [6.1937e-04, 9.9938e-01],
        [5.2792e-05, 9.9995e-01],
        [1.2660e-05, 9.9999e-01],
        [4.6239e-07, 1.0000e+00],
        [3.4802e-03, 9.9652e-01],
        [3.9675e-07, 1.0000e+00],
        [9.9352e-01, 6.4778e-03],
        [4.9105e-05, 9.9995e-01],
        [1.2188e-04, 9.9988e-01],
        [2.2678e-05, 9.9998e-01],
        [9.9876e-01, 1.2352e-03],
        [2.0986e-05, 9.9998e-01],
        [1.3092e-03, 9.9869e-01],
        [7.8299e-02, 9.2170e-01],
        [9.9605e-01, 3.9462e-03],
        [8.4880e-02, 9.1512e-01],
        [1.3235e-05, 9.9999e-01],
        [6.9880e-04, 9.9930e-01],
        [2.5442e-06, 1.0000e+00],
        [9.8889e-01, 1.1105e-02],
        [2.0063e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5125e-05, 9.9998e-01],
        [7.0700e-06, 9.9999e-01],
        [1.3450e-04, 9.9987e-01],
        [1.4946e-03, 9.9851e-01],
        [2.2139e-04, 9.9978e-01],
        [5.8381e-01, 4.1619e-01],
        [9.9935e-01, 6.5328e-04],
        [1.3521e-01, 8.6479e-01],
        [2.3285e-01, 7.6715e-01],
        [1.2892e-07, 1.0000e+00],
        [9.9971e-01, 2.9234e-04],
        [9.7825e-01, 2.1749e-02],
        [9.9968e-01, 3.1839e-04],
        [9.9979e-01, 2.0685e-04],
        [7.3802e-01, 2.6198e-01],
        [2.2885e-04, 9.9977e-01],
        [1.8551e-01, 8.1449e-01],
        [1.5626e-04, 9.9984e-01],
        [1.4094e-03, 9.9859e-01],
        [9.0856e-04, 9.9909e-01],
        [9.5734e-04, 9.9904e-01],
        [1.5610e-04, 9.9984e-01],
        [6.1145e-04, 9.9939e-01],
        [4.7848e-06, 1.0000e+00],
        [2.4150e-04, 9.9976e-01],
        [4.6360e-06, 1.0000e+00],
        [6.0512e-03, 9.9395e-01],
        [6.4471e-03, 9.9355e-01],
        [4.0220e-05, 9.9996e-01],
        [1.3088e-05, 9.9999e-01],
        [9.9184e-01, 8.1590e-03],
        [3.7858e-07, 1.0000e+00],
        [3.7658e-06, 1.0000e+00],
        [9.9968e-01, 3.2482e-04],
        [1.9151e-04, 9.9981e-01],
        [9.9453e-01, 5.4701e-03],
        [3.4903e-05, 9.9997e-01],
        [1.8913e-06, 1.0000e+00],
        [8.4701e-06, 9.9999e-01],
        [4.8866e-07, 1.0000e+00],
        [9.9915e-01, 8.4922e-04],
        [9.9750e-01, 2.4959e-03],
        [1.6724e-07, 1.0000e+00],
        [4.9078e-07, 1.0000e+00],
        [1.4808e-02, 9.8519e-01],
        [9.9326e-01, 6.7405e-03],
        [1.4224e-02, 9.8578e-01],
        [9.5197e-04, 9.9905e-01],
        [4.5514e-08, 1.0000e+00],
        [9.9966e-01, 3.3824e-04],
        [1.1820e-06, 1.0000e+00],
        [9.9999e-01, 5.5388e-06],
        [1.8820e-05, 9.9998e-01],
        [1.2647e-02, 9.8735e-01],
        [9.9713e-01, 2.8665e-03],
        [2.5498e-08, 1.0000e+00],
        [5.4058e-06, 9.9999e-01],
        [2.3895e-03, 9.9761e-01],
        [5.2177e-03, 9.9478e-01],
        [9.8219e-01, 1.7813e-02],
        [3.8824e-02, 9.6118e-01],
        [9.9927e-01, 7.3024e-04],
        [2.2789e-02, 9.7721e-01],
        [4.2058e-02, 9.5794e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.9427e-04, 9.9951e-01],
        [8.0399e-01, 1.9601e-01],
        [8.0989e-01, 1.9011e-01],
        [9.9840e-01, 1.5964e-03],
        [7.1253e-07, 1.0000e+00],
        [1.0705e-03, 9.9893e-01],
        [9.9792e-01, 2.0807e-03],
        [1.1483e-03, 9.9885e-01],
        [3.2598e-03, 9.9674e-01],
        [9.9988e-01, 1.1960e-04],
        [2.1011e-08, 1.0000e+00],
        [7.2916e-08, 1.0000e+00],
        [7.9463e-06, 9.9999e-01],
        [4.1931e-02, 9.5807e-01],
        [1.3299e-07, 1.0000e+00],
        [1.0070e-04, 9.9990e-01],
        [1.2614e-04, 9.9987e-01],
        [9.9217e-01, 7.8254e-03],
        [9.9935e-01, 6.5139e-04],
        [9.9654e-01, 3.4554e-03],
        [5.7749e-05, 9.9994e-01],
        [1.5035e-04, 9.9985e-01],
        [9.9752e-05, 9.9990e-01],
        [1.2627e-07, 1.0000e+00],
        [2.1455e-04, 9.9979e-01],
        [8.9776e-06, 9.9999e-01],
        [3.6889e-07, 1.0000e+00],
        [4.7401e-05, 9.9995e-01],
        [7.7606e-06, 9.9999e-01],
        [9.1807e-05, 9.9991e-01],
        [6.4229e-05, 9.9994e-01],
        [4.3263e-03, 9.9567e-01],
        [2.9695e-08, 1.0000e+00],
        [3.6622e-05, 9.9996e-01],
        [1.1321e-02, 9.8868e-01],
        [8.6649e-05, 9.9991e-01],
        [1.0435e-09, 1.0000e+00],
        [1.2250e-04, 9.9988e-01],
        [1.0310e-05, 9.9999e-01],
        [8.1523e-05, 9.9992e-01],
        [1.2796e-05, 9.9999e-01],
        [9.9892e-01, 1.0779e-03],
        [6.2497e-04, 9.9938e-01],
        [9.9861e-01, 1.3880e-03],
        [4.3558e-05, 9.9996e-01],
        [3.3418e-06, 1.0000e+00],
        [2.9927e-05, 9.9997e-01],
        [9.9733e-01, 2.6693e-03],
        [4.9652e-08, 1.0000e+00],
        [6.4634e-05, 9.9994e-01],
        [2.5366e-05, 9.9997e-01],
        [1.3478e-03, 9.9865e-01],
        [9.0737e-05, 9.9991e-01],
        [2.1681e-06, 1.0000e+00],
        [9.9953e-01, 4.7340e-04],
        [3.7191e-01, 6.2809e-01],
        [2.1082e-06, 1.0000e+00],
        [2.8745e-06, 1.0000e+00],
        [1.6150e-07, 1.0000e+00],
        [8.5879e-03, 9.9141e-01],
        [1.6883e-06, 1.0000e+00],
        [3.1252e-04, 9.9969e-01],
        [9.6464e-01, 3.5363e-02],
        [3.6081e-01, 6.3919e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.7023e-01, 2.9772e-02],
        [9.9969e-01, 3.1022e-04],
        [9.9951e-01, 4.9334e-04],
        [1.4445e-03, 9.9856e-01],
        [6.7778e-05, 9.9993e-01],
        [4.9973e-04, 9.9950e-01],
        [3.5393e-04, 9.9965e-01],
        [3.2867e-06, 1.0000e+00],
        [8.2445e-01, 1.7555e-01],
        [9.9784e-01, 2.1615e-03],
        [9.9844e-01, 1.5631e-03],
        [2.3885e-08, 1.0000e+00],
        [1.7621e-06, 1.0000e+00],
        [4.2866e-04, 9.9957e-01],
        [1.0777e-05, 9.9999e-01],
        [1.3710e-07, 1.0000e+00],
        [2.4807e-01, 7.5193e-01],
        [3.6440e-03, 9.9636e-01],
        [7.1674e-07, 1.0000e+00],
        [3.3004e-06, 1.0000e+00],
        [6.7116e-01, 3.2884e-01],
        [9.4235e-05, 9.9991e-01],
        [9.2801e-05, 9.9991e-01],
        [9.5206e-07, 1.0000e+00],
        [5.7658e-05, 9.9994e-01],
        [1.2731e-03, 9.9873e-01],
        [1.8150e-07, 1.0000e+00],
        [2.9244e-07, 1.0000e+00],
        [9.8168e-01, 1.8319e-02],
        [2.2957e-04, 9.9977e-01],
        [2.1811e-04, 9.9978e-01],
        [9.8077e-01, 1.9225e-02],
        [1.8253e-05, 9.9998e-01],
        [6.0738e-09, 1.0000e+00],
        [1.2648e-05, 9.9999e-01],
        [1.9914e-01, 8.0086e-01],
        [8.6706e-05, 9.9991e-01],
        [2.0939e-06, 1.0000e+00],
        [3.1790e-01, 6.8210e-01],
        [1.7115e-03, 9.9829e-01],
        [2.9229e-04, 9.9971e-01],
        [1.1532e-04, 9.9988e-01],
        [1.0584e-04, 9.9989e-01],
        [1.9381e-03, 9.9806e-01],
        [9.6338e-01, 3.6619e-02],
        [9.7601e-01, 2.3993e-02],
        [1.2656e-04, 9.9987e-01],
        [6.0983e-06, 9.9999e-01],
        [1.1435e-03, 9.9886e-01],
        [1.8193e-03, 9.9818e-01],
        [1.7786e-05, 9.9998e-01],
        [1.6727e-05, 9.9998e-01],
        [4.6797e-07, 1.0000e+00],
        [7.5554e-09, 1.0000e+00],
        [8.5026e-01, 1.4974e-01],
        [6.0040e-04, 9.9940e-01],
        [1.3129e-02, 9.8687e-01],
        [2.0518e-01, 7.9482e-01],
        [1.4758e-04, 9.9985e-01],
        [1.0111e-04, 9.9990e-01],
        [8.2834e-07, 1.0000e+00],
        [5.8437e-07, 1.0000e+00],
        [9.9990e-01, 1.0166e-04],
        [4.6435e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9975e-01, 2.5354e-04],
        [1.3430e-03, 9.9866e-01],
        [3.6655e-07, 1.0000e+00],
        [9.8653e-07, 1.0000e+00],
        [4.7640e-03, 9.9524e-01],
        [1.8799e-06, 1.0000e+00],
        [5.3632e-06, 9.9999e-01],
        [1.0000e+00, 2.2862e-07],
        [9.7914e-01, 2.0862e-02],
        [5.8329e-07, 1.0000e+00],
        [8.9915e-01, 1.0085e-01],
        [2.0196e-07, 1.0000e+00],
        [3.6144e-04, 9.9964e-01],
        [9.9739e-01, 2.6077e-03],
        [2.2226e-03, 9.9778e-01],
        [7.9073e-01, 2.0927e-01],
        [6.1670e-04, 9.9938e-01],
        [2.0838e-07, 1.0000e+00],
        [5.9013e-08, 1.0000e+00],
        [4.1304e-04, 9.9959e-01],
        [2.1280e-07, 1.0000e+00],
        [5.1511e-06, 9.9999e-01],
        [3.7918e-05, 9.9996e-01],
        [9.9906e-07, 1.0000e+00],
        [1.8399e-07, 1.0000e+00],
        [9.9513e-01, 4.8665e-03],
        [8.1937e-01, 1.8063e-01],
        [1.5534e-01, 8.4466e-01],
        [2.3875e-06, 1.0000e+00],
        [5.0189e-07, 1.0000e+00],
        [9.7870e-01, 2.1300e-02],
        [3.8674e-06, 1.0000e+00],
        [9.9598e-01, 4.0184e-03],
        [3.0583e-06, 1.0000e+00],
        [2.2737e-03, 9.9773e-01],
        [5.2119e-03, 9.9479e-01],
        [7.5997e-06, 9.9999e-01],
        [9.7053e-01, 2.9467e-02],
        [9.7010e-01, 2.9899e-02],
        [8.0312e-01, 1.9688e-01],
        [1.8896e-02, 9.8110e-01],
        [1.3686e-04, 9.9986e-01],
        [5.1969e-04, 9.9948e-01],
        [1.5443e-03, 9.9846e-01],
        [1.2586e-03, 9.9874e-01],
        [2.4423e-06, 1.0000e+00],
        [8.3862e-05, 9.9992e-01],
        [9.9982e-01, 1.7591e-04],
        [2.7518e-07, 1.0000e+00],
        [1.9407e-07, 1.0000e+00],
        [5.0610e-04, 9.9949e-01],
        [7.9174e-06, 9.9999e-01],
        [7.1309e-04, 9.9929e-01],
        [9.8523e-08, 1.0000e+00],
        [1.5485e-03, 9.9845e-01],
        [9.9969e-01, 3.1423e-04],
        [3.0063e-06, 1.0000e+00],
        [9.9927e-01, 7.3290e-04],
        [3.0755e-04, 9.9969e-01],
        [6.0814e-06, 9.9999e-01],
        [8.3718e-03, 9.9163e-01],
        [2.4003e-04, 9.9976e-01],
        [9.9624e-01, 3.7618e-03],
        [1.5176e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.8959e-06, 1.0000e+00],
        [9.9927e-01, 7.2982e-04],
        [1.3414e-07, 1.0000e+00],
        [1.6235e-03, 9.9838e-01],
        [7.2017e-08, 1.0000e+00],
        [5.6002e-05, 9.9994e-01],
        [7.2462e-02, 9.2754e-01],
        [8.8326e-08, 1.0000e+00],
        [3.4392e-05, 9.9997e-01],
        [2.5435e-04, 9.9975e-01],
        [5.6519e-04, 9.9943e-01],
        [2.9416e-05, 9.9997e-01],
        [4.2911e-05, 9.9996e-01],
        [2.5435e-04, 9.9975e-01],
        [4.5656e-07, 1.0000e+00],
        [1.3105e-04, 9.9987e-01],
        [4.4800e-04, 9.9955e-01],
        [7.0587e-03, 9.9294e-01],
        [7.5897e-01, 2.4103e-01],
        [2.3318e-04, 9.9977e-01],
        [1.4816e-07, 1.0000e+00],
        [4.6200e-06, 1.0000e+00],
        [5.9007e-03, 9.9410e-01],
        [5.3098e-04, 9.9947e-01],
        [1.1186e-03, 9.9888e-01],
        [2.7262e-04, 9.9973e-01],
        [5.7126e-03, 9.9429e-01],
        [7.3239e-06, 9.9999e-01],
        [9.9420e-01, 5.7998e-03],
        [7.1944e-05, 9.9993e-01],
        [2.3583e-07, 1.0000e+00],
        [7.8333e-04, 9.9922e-01],
        [9.6848e-01, 3.1519e-02],
        [9.5929e-10, 1.0000e+00],
        [9.1423e-05, 9.9991e-01],
        [9.9964e-01, 3.5877e-04],
        [2.1020e-08, 1.0000e+00],
        [9.9964e-01, 3.5629e-04],
        [9.9984e-01, 1.6413e-04],
        [2.9724e-03, 9.9703e-01],
        [2.0466e-04, 9.9980e-01],
        [1.7947e-04, 9.9982e-01],
        [1.4797e-04, 9.9985e-01],
        [2.7695e-04, 9.9972e-01],
        [9.8664e-08, 1.0000e+00],
        [9.9532e-01, 4.6833e-03],
        [5.5786e-07, 1.0000e+00],
        [5.1891e-04, 9.9948e-01],
        [3.2312e-06, 1.0000e+00],
        [8.0315e-01, 1.9685e-01],
        [1.0511e-06, 1.0000e+00],
        [1.2594e-02, 9.8741e-01],
        [9.9892e-01, 1.0834e-03],
        [3.0071e-05, 9.9997e-01],
        [1.2059e-02, 9.8794e-01],
        [1.0994e-06, 1.0000e+00],
        [9.9454e-01, 5.4566e-03],
        [2.6093e-04, 9.9974e-01],
        [1.2901e-04, 9.9987e-01],
        [9.9948e-01, 5.2202e-04],
        [1.5272e-01, 8.4728e-01],
        [1.4166e-03, 9.9858e-01],
        [3.1132e-03, 9.9689e-01],
        [1.5893e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9974e-01, 2.5733e-04],
        [1.2159e-07, 1.0000e+00],
        [7.7472e-01, 2.2528e-01],
        [2.7438e-03, 9.9726e-01],
        [2.4730e-04, 9.9975e-01],
        [1.2320e-04, 9.9988e-01],
        [4.3482e-06, 1.0000e+00],
        [9.5752e-06, 9.9999e-01],
        [5.0768e-04, 9.9949e-01],
        [2.0013e-03, 9.9800e-01],
        [1.1573e-03, 9.9884e-01],
        [1.1737e-03, 9.9883e-01],
        [9.8646e-01, 1.3542e-02],
        [1.5182e-06, 1.0000e+00],
        [2.0108e-07, 1.0000e+00],
        [6.4658e-08, 1.0000e+00],
        [5.0778e-08, 1.0000e+00],
        [4.0630e-06, 1.0000e+00],
        [2.2607e-05, 9.9998e-01],
        [4.6895e-06, 1.0000e+00],
        [9.0905e-05, 9.9991e-01],
        [9.9858e-01, 1.4237e-03],
        [2.2507e-04, 9.9977e-01],
        [7.0878e-05, 9.9993e-01],
        [6.2687e-06, 9.9999e-01],
        [2.1652e-07, 1.0000e+00],
        [8.4332e-05, 9.9992e-01],
        [7.3503e-08, 1.0000e+00],
        [7.3068e-03, 9.9269e-01],
        [7.3569e-06, 9.9999e-01],
        [2.1063e-01, 7.8937e-01],
        [3.6439e-04, 9.9964e-01],
        [4.8402e-03, 9.9516e-01],
        [9.9982e-01, 1.8470e-04],
        [1.9045e-05, 9.9998e-01],
        [1.2745e-04, 9.9987e-01],
        [2.3449e-04, 9.9977e-01],
        [9.9971e-01, 2.9334e-04],
        [4.4944e-08, 1.0000e+00],
        [9.9883e-01, 1.1730e-03],
        [3.5761e-04, 9.9964e-01],
        [8.8065e-01, 1.1935e-01],
        [8.7274e-01, 1.2726e-01],
        [2.3230e-05, 9.9998e-01],
        [2.9133e-03, 9.9709e-01],
        [2.5935e-06, 1.0000e+00],
        [7.3359e-07, 1.0000e+00],
        [2.4859e-05, 9.9998e-01],
        [7.2867e-09, 1.0000e+00],
        [9.3092e-03, 9.9069e-01],
        [5.5550e-06, 9.9999e-01],
        [3.8299e-05, 9.9996e-01],
        [3.6673e-04, 9.9963e-01],
        [5.0977e-06, 9.9999e-01],
        [4.2313e-04, 9.9958e-01],
        [6.5073e-06, 9.9999e-01],
        [9.9972e-01, 2.8360e-04],
        [8.6972e-07, 1.0000e+00],
        [5.1816e-07, 1.0000e+00],
        [2.9859e-08, 1.0000e+00],
        [7.3249e-05, 9.9993e-01],
        [1.0546e-06, 1.0000e+00],
        [4.5368e-07, 1.0000e+00],
        [2.9999e-03, 9.9700e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4806e-03, 9.9852e-01],
        [1.4782e-07, 1.0000e+00],
        [3.5549e-06, 1.0000e+00],
        [9.9541e-01, 4.5932e-03],
        [7.0885e-06, 9.9999e-01],
        [4.7730e-05, 9.9995e-01],
        [6.9445e-06, 9.9999e-01],
        [4.3990e-05, 9.9996e-01],
        [9.9964e-01, 3.6092e-04],
        [9.8611e-01, 1.3885e-02],
        [9.3648e-06, 9.9999e-01],
        [4.4791e-05, 9.9996e-01],
        [7.4829e-05, 9.9993e-01],
        [9.8752e-01, 1.2476e-02],
        [8.0274e-03, 9.9197e-01],
        [1.1989e-06, 1.0000e+00],
        [1.6872e-02, 9.8313e-01],
        [4.4861e-06, 1.0000e+00],
        [2.9642e-03, 9.9704e-01],
        [1.7021e-03, 9.9830e-01],
        [9.9987e-01, 1.2853e-04],
        [2.1739e-06, 1.0000e+00],
        [9.9141e-01, 8.5859e-03],
        [1.4353e-03, 9.9856e-01],
        [9.9932e-01, 6.7997e-04],
        [9.6751e-01, 3.2489e-02],
        [2.0706e-07, 1.0000e+00],
        [9.9972e-01, 2.8386e-04],
        [4.9284e-05, 9.9995e-01],
        [1.1125e-04, 9.9989e-01],
        [8.7524e-01, 1.2476e-01],
        [2.3605e-06, 1.0000e+00],
        [1.7469e-05, 9.9998e-01],
        [5.3683e-07, 1.0000e+00],
        [6.3404e-05, 9.9994e-01],
        [5.9795e-05, 9.9994e-01],
        [1.8945e-03, 9.9811e-01],
        [7.2940e-05, 9.9993e-01],
        [9.9864e-01, 1.3561e-03],
        [1.4280e-07, 1.0000e+00],
        [9.9982e-01, 1.7767e-04],
        [2.1723e-07, 1.0000e+00],
        [2.2471e-04, 9.9978e-01],
        [1.0337e-05, 9.9999e-01],
        [2.6894e-01, 7.3106e-01],
        [9.9958e-01, 4.1979e-04],
        [9.9985e-01, 1.4612e-04],
        [2.2486e-03, 9.9775e-01],
        [7.6202e-04, 9.9924e-01],
        [3.0597e-05, 9.9997e-01],
        [1.6476e-08, 1.0000e+00],
        [1.3890e-05, 9.9999e-01],
        [1.0684e-06, 1.0000e+00],
        [5.9421e-01, 4.0579e-01],
        [2.5799e-05, 9.9997e-01],
        [9.9896e-01, 1.0362e-03],
        [9.9975e-01, 2.4700e-04],
        [3.6111e-04, 9.9964e-01],
        [6.8631e-06, 9.9999e-01],
        [1.8107e-07, 1.0000e+00],
        [1.5775e-05, 9.9998e-01],
        [9.9947e-01, 5.3184e-04],
        [9.9980e-01, 2.0431e-04],
        [6.9091e-03, 9.9309e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.7338e-07, 1.0000e+00],
        [9.9967e-01, 3.2796e-04],
        [9.9910e-01, 8.9917e-04],
        [9.9770e-01, 2.3021e-03],
        [2.1831e-05, 9.9998e-01],
        [1.5547e-06, 1.0000e+00],
        [1.0778e-05, 9.9999e-01],
        [9.9978e-01, 2.2129e-04],
        [1.0134e-07, 1.0000e+00],
        [2.1430e-06, 1.0000e+00],
        [1.0042e-06, 1.0000e+00],
        [6.5265e-08, 1.0000e+00],
        [9.9908e-01, 9.2072e-04],
        [2.2150e-02, 9.7785e-01],
        [1.5262e-04, 9.9985e-01],
        [9.2285e-01, 7.7147e-02],
        [1.0979e-07, 1.0000e+00],
        [1.1871e-03, 9.9881e-01],
        [9.9990e-01, 9.5403e-05],
        [9.9718e-01, 2.8195e-03],
        [9.9894e-01, 1.0600e-03],
        [5.7489e-04, 9.9943e-01],
        [3.3233e-04, 9.9967e-01]], grad_fn=<SoftmaxBackward0>)
Epoch 3, loss: 0.3380
tensor([[9.9559e-05, 9.9990e-01],
        [9.9737e-01, 2.6267e-03],
        [4.5604e-07, 1.0000e+00],
        [9.9989e-01, 1.0725e-04],
        [1.6607e-05, 9.9998e-01],
        [9.9927e-01, 7.2644e-04],
        [4.2903e-01, 5.7097e-01],
        [2.3453e-02, 9.7655e-01],
        [1.7043e-05, 9.9998e-01],
        [9.1903e-02, 9.0810e-01],
        [8.0566e-03, 9.9194e-01],
        [7.8818e-05, 9.9992e-01],
        [8.7032e-04, 9.9913e-01],
        [8.9182e-05, 9.9991e-01],
        [2.2262e-05, 9.9998e-01],
        [4.3364e-05, 9.9996e-01],
        [7.9841e-04, 9.9920e-01],
        [7.4719e-02, 9.2528e-01],
        [1.1525e-04, 9.9988e-01],
        [9.6941e-06, 9.9999e-01],
        [9.9110e-01, 8.9024e-03],
        [2.0164e-07, 1.0000e+00],
        [4.9862e-06, 9.9999e-01],
        [9.2134e-01, 7.8663e-02],
        [9.9384e-01, 6.1570e-03],
        [9.9532e-01, 4.6754e-03],
        [4.8265e-04, 9.9952e-01],
        [4.7098e-03, 9.9529e-01],
        [9.1349e-05, 9.9991e-01],
        [2.3469e-02, 9.7653e-01],
        [9.9951e-01, 4.8989e-04],
        [1.1160e-03, 9.9888e-01],
        [9.9996e-01, 4.2770e-05],
        [3.8475e-02, 9.6153e-01],
        [1.7156e-05, 9.9998e-01],
        [9.4452e-06, 9.9999e-01],
        [3.1712e-06, 1.0000e+00],
        [9.9123e-01, 8.7688e-03],
        [1.2393e-03, 9.9876e-01],
        [9.9946e-01, 5.4067e-04],
        [9.9970e-01, 3.0202e-04],
        [2.0224e-04, 9.9980e-01],
        [7.8089e-05, 9.9992e-01],
        [2.1089e-06, 1.0000e+00],
        [1.0716e-05, 9.9999e-01],
        [9.9993e-01, 7.1611e-05],
        [1.8159e-04, 9.9982e-01],
        [4.2101e-06, 1.0000e+00],
        [4.9543e-07, 1.0000e+00],
        [4.5780e-03, 9.9542e-01],
        [9.9992e-01, 8.2428e-05],
        [1.0399e-02, 9.8960e-01],
        [7.0267e-05, 9.9993e-01],
        [4.5180e-05, 9.9995e-01],
        [8.5195e-01, 1.4805e-01],
        [2.6602e-06, 1.0000e+00],
        [9.9427e-01, 5.7316e-03],
        [9.9945e-01, 5.5363e-04],
        [8.9687e-01, 1.0313e-01],
        [1.8325e-04, 9.9982e-01],
        [9.7852e-01, 2.1480e-02],
        [9.9966e-01, 3.3617e-04],
        [9.9470e-01, 5.3009e-03],
        [9.9951e-01, 4.9475e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[4.3132e-02, 9.5687e-01],
        [9.9966e-01, 3.3511e-04],
        [1.6240e-03, 9.9838e-01],
        [1.5422e-03, 9.9846e-01],
        [3.8254e-08, 1.0000e+00],
        [3.3360e-05, 9.9997e-01],
        [9.9977e-01, 2.2693e-04],
        [1.7215e-05, 9.9998e-01],
        [2.0377e-03, 9.9796e-01],
        [9.9969e-01, 3.1055e-04],
        [2.2804e-07, 1.0000e+00],
        [9.9833e-01, 1.6677e-03],
        [1.4986e-04, 9.9985e-01],
        [7.0364e-04, 9.9930e-01],
        [1.9589e-03, 9.9804e-01],
        [4.6504e-02, 9.5350e-01],
        [3.6372e-07, 1.0000e+00],
        [4.0163e-04, 9.9960e-01],
        [1.7437e-06, 1.0000e+00],
        [1.2972e-04, 9.9987e-01],
        [9.7347e-01, 2.6534e-02],
        [3.3093e-03, 9.9669e-01],
        [9.9564e-03, 9.9004e-01],
        [1.8772e-03, 9.9812e-01],
        [4.3207e-04, 9.9957e-01],
        [3.2736e-04, 9.9967e-01],
        [7.9573e-02, 9.2043e-01],
        [4.2430e-03, 9.9576e-01],
        [6.3457e-03, 9.9365e-01],
        [2.8002e-05, 9.9997e-01],
        [1.3374e-03, 9.9866e-01],
        [1.2308e-04, 9.9988e-01],
        [2.5760e-05, 9.9997e-01],
        [2.2765e-05, 9.9998e-01],
        [1.0926e-06, 1.0000e+00],
        [4.5526e-05, 9.9995e-01],
        [1.4693e-03, 9.9853e-01],
        [5.5415e-03, 9.9446e-01],
        [4.2277e-07, 1.0000e+00],
        [9.9966e-01, 3.4464e-04],
        [1.0509e-05, 9.9999e-01],
        [3.1727e-04, 9.9968e-01],
        [1.0395e-01, 8.9605e-01],
        [9.9561e-01, 4.3890e-03],
        [9.9963e-01, 3.6926e-04],
        [1.4498e-05, 9.9999e-01],
        [9.9923e-01, 7.7178e-04],
        [7.0293e-05, 9.9993e-01],
        [9.9913e-01, 8.7112e-04],
        [1.7018e-04, 9.9983e-01],
        [9.9925e-01, 7.4868e-04],
        [2.3725e-05, 9.9998e-01],
        [2.3775e-05, 9.9998e-01],
        [3.6429e-01, 6.3571e-01],
        [4.4277e-03, 9.9557e-01],
        [1.3767e-05, 9.9999e-01],
        [1.2099e-07, 1.0000e+00],
        [2.3339e-01, 7.6661e-01],
        [1.2087e-05, 9.9999e-01],
        [1.5874e-01, 8.4126e-01],
        [7.1130e-01, 2.8870e-01],
        [6.1023e-01, 3.8977e-01],
        [1.0608e-02, 9.8939e-01],
        [1.6579e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.7847e-05, 9.9992e-01],
        [2.5912e-02, 9.7409e-01],
        [9.9489e-01, 5.1096e-03],
        [3.1292e-02, 9.6871e-01],
        [2.5198e-05, 9.9997e-01],
        [9.8885e-01, 1.1150e-02],
        [1.6734e-04, 9.9983e-01],
        [9.9963e-01, 3.7444e-04],
        [3.1781e-05, 9.9997e-01],
        [9.9923e-01, 7.6851e-04],
        [5.8883e-03, 9.9411e-01],
        [2.5882e-01, 7.4118e-01],
        [6.4271e-06, 9.9999e-01],
        [5.4518e-03, 9.9455e-01],
        [1.6221e-03, 9.9838e-01],
        [3.5754e-06, 1.0000e+00],
        [2.9053e-06, 1.0000e+00],
        [8.7586e-04, 9.9912e-01],
        [5.4778e-03, 9.9452e-01],
        [1.0823e-02, 9.8918e-01],
        [9.9933e-01, 6.6821e-04],
        [6.7538e-01, 3.2462e-01],
        [9.4316e-03, 9.9057e-01],
        [2.3681e-04, 9.9976e-01],
        [3.1537e-01, 6.8463e-01],
        [1.9824e-05, 9.9998e-01],
        [9.9947e-01, 5.2520e-04],
        [7.4094e-05, 9.9993e-01],
        [2.0492e-05, 9.9998e-01],
        [8.1782e-01, 1.8218e-01],
        [9.9979e-01, 2.0808e-04],
        [4.3829e-02, 9.5617e-01],
        [2.6467e-05, 9.9997e-01],
        [4.6192e-07, 1.0000e+00],
        [1.8498e-04, 9.9982e-01],
        [9.9959e-01, 4.1083e-04],
        [1.2627e-03, 9.9874e-01],
        [1.3925e-06, 1.0000e+00],
        [7.7098e-03, 9.9229e-01],
        [2.6216e-03, 9.9738e-01],
        [1.8154e-03, 9.9818e-01],
        [6.3202e-03, 9.9368e-01],
        [1.0761e-04, 9.9989e-01],
        [9.9938e-01, 6.1986e-04],
        [2.7848e-03, 9.9722e-01],
        [1.6392e-03, 9.9836e-01],
        [5.2493e-06, 9.9999e-01],
        [9.9966e-01, 3.4475e-04],
        [9.9936e-01, 6.4166e-04],
        [9.9737e-01, 2.6294e-03],
        [2.1350e-03, 9.9787e-01],
        [1.8585e-05, 9.9998e-01],
        [2.1803e-01, 7.8197e-01],
        [9.9942e-01, 5.8197e-04],
        [6.6309e-05, 9.9993e-01],
        [8.9411e-01, 1.0589e-01],
        [2.1472e-05, 9.9998e-01],
        [1.5024e-04, 9.9985e-01],
        [9.8611e-01, 1.3893e-02],
        [5.2447e-04, 9.9948e-01],
        [1.7385e-02, 9.8261e-01],
        [1.7066e-07, 1.0000e+00],
        [9.9987e-01, 1.3303e-04],
        [1.2347e-01, 8.7653e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1842e-06, 1.0000e+00],
        [9.9952e-01, 4.7709e-04],
        [6.8548e-05, 9.9993e-01],
        [9.0305e-03, 9.9097e-01],
        [8.9602e-06, 9.9999e-01],
        [4.7597e-03, 9.9524e-01],
        [4.3954e-06, 1.0000e+00],
        [4.2170e-04, 9.9958e-01],
        [9.9962e-01, 3.7995e-04],
        [4.4112e-04, 9.9956e-01],
        [9.9887e-01, 1.1267e-03],
        [3.7998e-05, 9.9996e-01],
        [5.0802e-05, 9.9995e-01],
        [4.1103e-06, 1.0000e+00],
        [6.6512e-05, 9.9993e-01],
        [2.9524e-04, 9.9970e-01],
        [7.7123e-05, 9.9992e-01],
        [5.2656e-05, 9.9995e-01],
        [3.4635e-03, 9.9654e-01],
        [1.1462e-04, 9.9989e-01],
        [1.0300e-05, 9.9999e-01],
        [1.4209e-03, 9.9858e-01],
        [3.5781e-07, 1.0000e+00],
        [9.9947e-01, 5.3401e-04],
        [1.0226e-07, 1.0000e+00],
        [1.4245e-03, 9.9858e-01],
        [1.0570e-03, 9.9894e-01],
        [1.2647e-03, 9.9874e-01],
        [9.9069e-01, 9.3141e-03],
        [3.4873e-02, 9.6513e-01],
        [9.4052e-05, 9.9991e-01],
        [4.4521e-02, 9.5548e-01],
        [9.9955e-01, 4.5326e-04],
        [3.4300e-03, 9.9657e-01],
        [4.5229e-05, 9.9995e-01],
        [7.0434e-04, 9.9930e-01],
        [1.4888e-02, 9.8511e-01],
        [9.4704e-01, 5.2964e-02],
        [2.8640e-02, 9.7136e-01],
        [3.8738e-01, 6.1262e-01],
        [9.4938e-03, 9.9051e-01],
        [9.4118e-02, 9.0588e-01],
        [7.7248e-05, 9.9992e-01],
        [1.2821e-03, 9.9872e-01],
        [2.7631e-03, 9.9724e-01],
        [9.9912e-01, 8.7797e-04],
        [4.4370e-04, 9.9956e-01],
        [8.3189e-01, 1.6811e-01],
        [9.5216e-03, 9.9048e-01],
        [5.1951e-07, 1.0000e+00],
        [1.0202e-03, 9.9898e-01],
        [1.5583e-04, 9.9984e-01],
        [1.6656e-03, 9.9833e-01],
        [3.1208e-06, 1.0000e+00],
        [3.2570e-02, 9.6743e-01],
        [7.3118e-05, 9.9993e-01],
        [1.4573e-03, 9.9854e-01],
        [9.9959e-01, 4.1167e-04],
        [8.3532e-04, 9.9916e-01],
        [1.5869e-02, 9.8413e-01],
        [2.9392e-04, 9.9971e-01],
        [2.8795e-02, 9.7121e-01],
        [4.0271e-07, 1.0000e+00],
        [9.9959e-01, 4.1492e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9972e-01, 2.7987e-04],
        [4.1864e-06, 1.0000e+00],
        [9.9970e-01, 2.9613e-04],
        [3.7425e-05, 9.9996e-01],
        [1.5599e-04, 9.9984e-01],
        [8.0083e-02, 9.1992e-01],
        [4.5712e-05, 9.9995e-01],
        [3.6898e-01, 6.3102e-01],
        [8.0217e-02, 9.1978e-01],
        [1.1233e-03, 9.9888e-01],
        [9.7649e-01, 2.3509e-02],
        [4.6275e-05, 9.9995e-01],
        [6.8784e-07, 1.0000e+00],
        [9.9865e-01, 1.3476e-03],
        [9.9951e-01, 4.9149e-04],
        [8.8016e-05, 9.9991e-01],
        [1.0721e-07, 1.0000e+00],
        [9.9926e-01, 7.4491e-04],
        [9.9973e-01, 2.7482e-04],
        [5.8387e-05, 9.9994e-01],
        [2.2224e-04, 9.9978e-01],
        [1.1164e-04, 9.9989e-01],
        [5.4999e-05, 9.9995e-01],
        [1.1231e-05, 9.9999e-01],
        [9.9966e-01, 3.3863e-04],
        [9.9931e-01, 6.9268e-04],
        [1.1164e-04, 9.9989e-01],
        [8.7766e-03, 9.9122e-01],
        [3.1974e-05, 9.9997e-01],
        [9.9973e-01, 2.6594e-04],
        [1.1657e-03, 9.9883e-01],
        [3.8244e-04, 9.9962e-01],
        [7.4208e-05, 9.9993e-01],
        [2.0123e-05, 9.9998e-01],
        [6.0199e-05, 9.9994e-01],
        [3.3569e-05, 9.9997e-01],
        [9.9964e-01, 3.5848e-04],
        [4.3620e-04, 9.9956e-01],
        [9.9972e-01, 2.7966e-04],
        [1.9711e-06, 1.0000e+00],
        [1.6782e-03, 9.9832e-01],
        [6.0200e-04, 9.9940e-01],
        [5.7296e-03, 9.9427e-01],
        [6.9352e-01, 3.0648e-01],
        [9.9952e-01, 4.7604e-04],
        [7.4425e-06, 9.9999e-01],
        [1.0477e-01, 8.9523e-01],
        [1.3206e-03, 9.9868e-01],
        [2.1691e-01, 7.8309e-01],
        [2.4024e-02, 9.7598e-01],
        [3.3796e-05, 9.9997e-01],
        [3.1977e-03, 9.9680e-01],
        [9.9316e-01, 6.8432e-03],
        [9.1323e-07, 1.0000e+00],
        [7.0248e-05, 9.9993e-01],
        [8.4885e-01, 1.5115e-01],
        [9.5932e-01, 4.0684e-02],
        [8.3982e-01, 1.6018e-01],
        [1.4203e-03, 9.9858e-01],
        [9.1257e-02, 9.0874e-01],
        [3.6872e-01, 6.3128e-01],
        [1.0792e-04, 9.9989e-01],
        [9.0461e-01, 9.5389e-02],
        [9.9963e-01, 3.7313e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9063e-04, 9.9901e-01],
        [2.1190e-08, 1.0000e+00],
        [4.1279e-04, 9.9959e-01],
        [6.0231e-07, 1.0000e+00],
        [1.5593e-03, 9.9844e-01],
        [4.1710e-01, 5.8290e-01],
        [1.3326e-03, 9.9867e-01],
        [2.4843e-06, 1.0000e+00],
        [1.4671e-05, 9.9999e-01],
        [5.2197e-03, 9.9478e-01],
        [3.6639e-06, 1.0000e+00],
        [2.1746e-01, 7.8254e-01],
        [9.9936e-01, 6.3569e-04],
        [9.9948e-01, 5.1623e-04],
        [4.1900e-05, 9.9996e-01],
        [5.4013e-06, 9.9999e-01],
        [9.9912e-01, 8.8443e-04],
        [9.7999e-06, 9.9999e-01],
        [2.9348e-09, 1.0000e+00],
        [7.4074e-04, 9.9926e-01],
        [9.9971e-01, 2.9344e-04],
        [9.5478e-01, 4.5219e-02],
        [9.9927e-01, 7.3503e-04],
        [3.3816e-02, 9.6618e-01],
        [9.9944e-01, 5.6217e-04],
        [1.4468e-05, 9.9999e-01],
        [1.8136e-05, 9.9998e-01],
        [9.3780e-07, 1.0000e+00],
        [3.6013e-05, 9.9996e-01],
        [9.9887e-01, 1.1273e-03],
        [4.2108e-07, 1.0000e+00],
        [2.7788e-02, 9.7221e-01],
        [1.5511e-03, 9.9845e-01],
        [5.3741e-06, 9.9999e-01],
        [1.1077e-04, 9.9989e-01],
        [9.9958e-01, 4.2181e-04],
        [6.0393e-03, 9.9396e-01],
        [3.6927e-03, 9.9631e-01],
        [2.2735e-07, 1.0000e+00],
        [1.2195e-05, 9.9999e-01],
        [9.9988e-01, 1.2353e-04],
        [9.9921e-01, 7.9093e-04],
        [5.6571e-06, 9.9999e-01],
        [9.9911e-01, 8.9417e-04],
        [4.4333e-02, 9.5567e-01],
        [7.9745e-04, 9.9920e-01],
        [3.6551e-03, 9.9634e-01],
        [1.2892e-05, 9.9999e-01],
        [8.4102e-07, 1.0000e+00],
        [7.6885e-01, 2.3115e-01],
        [3.7882e-03, 9.9621e-01],
        [8.1051e-06, 9.9999e-01],
        [9.9550e-01, 4.5036e-03],
        [9.9999e-01, 1.3920e-05],
        [6.1788e-04, 9.9938e-01],
        [2.4521e-06, 1.0000e+00],
        [8.5708e-03, 9.9143e-01],
        [1.1988e-03, 9.9880e-01],
        [2.1015e-07, 1.0000e+00],
        [4.1346e-04, 9.9959e-01],
        [4.2098e-07, 1.0000e+00],
        [9.9964e-01, 3.5963e-04],
        [2.1567e-03, 9.9784e-01],
        [6.9233e-03, 9.9308e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9945e-01, 5.5078e-04],
        [2.6730e-06, 1.0000e+00],
        [1.8794e-05, 9.9998e-01],
        [3.4402e-05, 9.9997e-01],
        [9.9970e-01, 2.9592e-04],
        [1.0877e-04, 9.9989e-01],
        [3.8770e-04, 9.9961e-01],
        [2.8626e-05, 9.9997e-01],
        [9.9867e-01, 1.3302e-03],
        [4.3674e-03, 9.9563e-01],
        [2.1061e-06, 1.0000e+00],
        [6.2289e-01, 3.7711e-01],
        [9.9252e-01, 7.4783e-03],
        [2.3224e-06, 1.0000e+00],
        [1.4043e-07, 1.0000e+00],
        [3.4994e-06, 1.0000e+00],
        [9.9508e-05, 9.9990e-01],
        [9.9146e-01, 8.5402e-03],
        [1.2617e-04, 9.9987e-01],
        [1.1449e-01, 8.8551e-01],
        [1.0669e-02, 9.8933e-01],
        [2.8410e-06, 1.0000e+00],
        [6.8832e-01, 3.1168e-01],
        [8.1297e-02, 9.1870e-01],
        [4.4397e-07, 1.0000e+00],
        [9.9765e-01, 2.3536e-03],
        [1.0612e-06, 1.0000e+00],
        [1.0663e-03, 9.9893e-01],
        [9.9968e-01, 3.1554e-04],
        [6.0287e-03, 9.9397e-01],
        [1.0394e-03, 9.9896e-01],
        [7.8129e-01, 2.1871e-01],
        [9.4524e-05, 9.9991e-01],
        [6.1297e-08, 1.0000e+00],
        [9.9935e-01, 6.5235e-04],
        [3.9299e-03, 9.9607e-01],
        [1.2387e-08, 1.0000e+00],
        [9.9334e-01, 6.6632e-03],
        [2.8679e-05, 9.9997e-01],
        [2.6548e-05, 9.9997e-01],
        [1.3982e-05, 9.9999e-01],
        [9.2766e-08, 1.0000e+00],
        [2.5380e-04, 9.9975e-01],
        [1.5538e-02, 9.8446e-01],
        [2.7084e-06, 1.0000e+00],
        [1.9988e-02, 9.8001e-01],
        [3.7086e-05, 9.9996e-01],
        [5.1381e-03, 9.9486e-01],
        [3.2102e-07, 1.0000e+00],
        [9.9929e-01, 7.0732e-04],
        [9.8909e-01, 1.0913e-02],
        [3.3277e-02, 9.6672e-01],
        [1.0363e-03, 9.9896e-01],
        [6.0219e-05, 9.9994e-01],
        [3.4061e-01, 6.5939e-01],
        [1.9068e-05, 9.9998e-01],
        [9.1915e-06, 9.9999e-01],
        [8.7011e-01, 1.2989e-01],
        [5.0646e-05, 9.9995e-01],
        [4.4397e-07, 1.0000e+00],
        [9.9968e-01, 3.1643e-04],
        [2.7119e-02, 9.7288e-01],
        [9.9911e-01, 8.8621e-04],
        [4.3997e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9960e-01, 3.9972e-04],
        [5.3801e-03, 9.9462e-01],
        [1.8231e-03, 9.9818e-01],
        [5.7726e-05, 9.9994e-01],
        [3.1831e-07, 1.0000e+00],
        [3.8651e-05, 9.9996e-01],
        [3.1654e-06, 1.0000e+00],
        [9.8858e-05, 9.9990e-01],
        [9.9965e-01, 3.4558e-04],
        [7.7519e-06, 9.9999e-01],
        [7.4007e-06, 9.9999e-01],
        [9.9523e-01, 4.7708e-03],
        [3.8478e-06, 1.0000e+00],
        [2.9056e-07, 1.0000e+00],
        [9.9942e-01, 5.7561e-04],
        [5.4276e-06, 9.9999e-01],
        [9.4558e-01, 5.4421e-02],
        [3.7224e-03, 9.9628e-01],
        [8.8573e-01, 1.1427e-01],
        [1.8637e-01, 8.1363e-01],
        [9.9352e-04, 9.9901e-01],
        [1.7622e-04, 9.9982e-01],
        [3.0290e-04, 9.9970e-01],
        [9.9620e-01, 3.7991e-03],
        [1.1238e-07, 1.0000e+00],
        [1.9690e-03, 9.9803e-01],
        [2.7759e-04, 9.9972e-01],
        [2.1281e-02, 9.7872e-01],
        [1.4387e-04, 9.9986e-01],
        [9.9641e-01, 3.5853e-03],
        [9.9863e-01, 1.3658e-03],
        [3.6203e-05, 9.9996e-01],
        [2.1330e-04, 9.9979e-01],
        [8.6307e-06, 9.9999e-01],
        [9.9941e-01, 5.9035e-04],
        [9.9975e-01, 2.4623e-04],
        [1.6853e-03, 9.9831e-01],
        [9.9268e-01, 7.3237e-03],
        [1.0061e-03, 9.9899e-01],
        [4.4459e-05, 9.9996e-01],
        [8.5584e-05, 9.9991e-01],
        [3.9051e-06, 1.0000e+00],
        [8.6889e-05, 9.9991e-01],
        [3.2479e-04, 9.9968e-01],
        [4.9579e-04, 9.9950e-01],
        [9.9865e-01, 1.3477e-03],
        [9.9753e-01, 2.4725e-03],
        [9.9830e-01, 1.6980e-03],
        [9.9807e-01, 1.9251e-03],
        [1.6433e-05, 9.9998e-01],
        [3.6910e-07, 1.0000e+00],
        [3.9080e-06, 1.0000e+00],
        [5.1012e-03, 9.9490e-01],
        [9.7610e-01, 2.3895e-02],
        [9.9958e-01, 4.2152e-04],
        [1.2079e-02, 9.8792e-01],
        [8.1474e-04, 9.9919e-01],
        [9.9863e-01, 1.3658e-03],
        [3.1425e-06, 1.0000e+00],
        [7.0644e-05, 9.9993e-01],
        [1.5763e-04, 9.9984e-01],
        [2.2681e-05, 9.9998e-01],
        [9.9954e-01, 4.5868e-04],
        [7.6545e-03, 9.9235e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9150e-01, 8.5006e-03],
        [3.9067e-06, 1.0000e+00],
        [7.2418e-01, 2.7582e-01],
        [2.9582e-07, 1.0000e+00],
        [9.9993e-01, 6.5140e-05],
        [9.3591e-08, 1.0000e+00],
        [9.9978e-01, 2.2280e-04],
        [9.1619e-09, 1.0000e+00],
        [9.8885e-06, 9.9999e-01],
        [4.2891e-05, 9.9996e-01],
        [9.9804e-01, 1.9648e-03],
        [1.0614e-05, 9.9999e-01],
        [2.1290e-05, 9.9998e-01],
        [3.0395e-01, 6.9605e-01],
        [7.9549e-06, 9.9999e-01],
        [1.0873e-06, 1.0000e+00],
        [1.8388e-04, 9.9982e-01],
        [4.4028e-04, 9.9956e-01],
        [6.0429e-01, 3.9571e-01],
        [2.8612e-02, 9.7139e-01],
        [3.2313e-07, 1.0000e+00],
        [1.0127e-04, 9.9990e-01],
        [6.1610e-08, 1.0000e+00],
        [9.9825e-01, 1.7505e-03],
        [1.0425e-03, 9.9896e-01],
        [9.9961e-01, 3.8959e-04],
        [9.7300e-03, 9.9027e-01],
        [1.7294e-02, 9.8271e-01],
        [7.3884e-05, 9.9993e-01],
        [4.5480e-06, 1.0000e+00],
        [9.9818e-01, 1.8197e-03],
        [2.7142e-06, 1.0000e+00],
        [6.6259e-07, 1.0000e+00],
        [9.7237e-08, 1.0000e+00],
        [1.6370e-04, 9.9984e-01],
        [1.6957e-06, 1.0000e+00],
        [3.7444e-09, 1.0000e+00],
        [5.1325e-05, 9.9995e-01],
        [3.4219e-02, 9.6578e-01],
        [8.9292e-06, 9.9999e-01],
        [1.4330e-08, 1.0000e+00],
        [1.1279e-04, 9.9989e-01],
        [1.1145e-07, 1.0000e+00],
        [8.9586e-01, 1.0414e-01],
        [2.0747e-04, 9.9979e-01],
        [4.3846e-06, 1.0000e+00],
        [2.1691e-05, 9.9998e-01],
        [8.0550e-08, 1.0000e+00],
        [5.0249e-06, 9.9999e-01],
        [4.0949e-05, 9.9996e-01],
        [9.5626e-01, 4.3744e-02],
        [2.3216e-07, 1.0000e+00],
        [2.1900e-05, 9.9998e-01],
        [6.1793e-04, 9.9938e-01],
        [9.7832e-01, 2.1681e-02],
        [9.9977e-01, 2.3370e-04],
        [8.6901e-04, 9.9913e-01],
        [9.8882e-01, 1.1179e-02],
        [4.9637e-04, 9.9950e-01],
        [2.4629e-03, 9.9754e-01],
        [9.9945e-01, 5.5431e-04],
        [9.6716e-01, 3.2839e-02],
        [7.9565e-05, 9.9992e-01],
        [8.1241e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.0140e-08, 1.0000e+00],
        [7.5129e-07, 1.0000e+00],
        [1.4874e-04, 9.9985e-01],
        [5.8642e-07, 1.0000e+00],
        [9.9911e-01, 8.9143e-04],
        [1.5973e-06, 1.0000e+00],
        [3.2160e-09, 1.0000e+00],
        [3.0722e-05, 9.9997e-01],
        [1.5930e-04, 9.9984e-01],
        [2.5660e-06, 1.0000e+00],
        [8.8420e-01, 1.1580e-01],
        [9.3814e-01, 6.1862e-02],
        [4.6508e-03, 9.9535e-01],
        [2.6571e-03, 9.9734e-01],
        [1.6475e-04, 9.9984e-01],
        [9.8704e-01, 1.2964e-02],
        [4.9620e-03, 9.9504e-01],
        [9.9933e-01, 6.6874e-04],
        [9.9126e-01, 8.7404e-03],
        [9.9499e-01, 5.0090e-03],
        [1.9989e-05, 9.9998e-01],
        [4.1711e-03, 9.9583e-01],
        [1.3625e-02, 9.8638e-01],
        [6.9055e-04, 9.9931e-01],
        [1.1011e-07, 1.0000e+00],
        [9.9386e-01, 6.1368e-03],
        [5.5175e-06, 9.9999e-01],
        [9.9663e-01, 3.3692e-03],
        [3.8154e-06, 1.0000e+00],
        [1.7960e-04, 9.9982e-01],
        [7.0224e-07, 1.0000e+00],
        [4.5002e-07, 1.0000e+00],
        [3.1742e-07, 1.0000e+00],
        [3.5431e-01, 6.4569e-01],
        [5.1962e-05, 9.9995e-01],
        [1.1524e-03, 9.9885e-01],
        [7.3484e-09, 1.0000e+00],
        [1.0608e-04, 9.9989e-01],
        [6.8130e-09, 1.0000e+00],
        [9.9791e-01, 2.0898e-03],
        [5.7924e-05, 9.9994e-01],
        [2.7587e-07, 1.0000e+00],
        [1.4328e-04, 9.9986e-01],
        [3.8560e-06, 1.0000e+00],
        [9.9856e-01, 1.4365e-03],
        [6.8742e-01, 3.1258e-01],
        [5.6139e-06, 9.9999e-01],
        [9.9968e-01, 3.1981e-04],
        [9.9972e-01, 2.7812e-04],
        [2.2158e-04, 9.9978e-01],
        [9.9954e-01, 4.6219e-04],
        [2.9134e-06, 1.0000e+00],
        [3.2402e-07, 1.0000e+00],
        [2.0625e-05, 9.9998e-01],
        [6.1998e-08, 1.0000e+00],
        [9.9933e-01, 6.6738e-04],
        [8.3697e-07, 1.0000e+00],
        [4.3212e-06, 1.0000e+00],
        [9.9970e-01, 3.0113e-04],
        [3.4339e-04, 9.9966e-01],
        [9.5281e-01, 4.7189e-02],
        [9.0165e-08, 1.0000e+00],
        [3.4952e-06, 1.0000e+00],
        [3.9578e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5225e-04, 9.9975e-01],
        [5.3995e-05, 9.9995e-01],
        [2.3058e-06, 1.0000e+00],
        [8.2924e-06, 9.9999e-01],
        [2.2384e-09, 1.0000e+00],
        [9.1816e-04, 9.9908e-01],
        [1.5238e-03, 9.9848e-01],
        [8.0437e-05, 9.9992e-01],
        [5.6178e-01, 4.3822e-01],
        [2.5109e-03, 9.9749e-01],
        [5.4172e-06, 9.9999e-01],
        [1.2151e-06, 1.0000e+00],
        [9.9548e-01, 4.5242e-03],
        [6.9012e-05, 9.9993e-01],
        [3.1068e-03, 9.9689e-01],
        [9.9714e-01, 2.8586e-03],
        [1.1527e-06, 1.0000e+00],
        [3.1821e-06, 1.0000e+00],
        [9.9043e-01, 9.5673e-03],
        [6.4424e-04, 9.9936e-01],
        [2.5469e-02, 9.7453e-01],
        [5.7008e-01, 4.2992e-01],
        [1.8997e-08, 1.0000e+00],
        [9.8488e-01, 1.5120e-02],
        [9.9927e-01, 7.3423e-04],
        [7.9453e-05, 9.9992e-01],
        [2.5086e-07, 1.0000e+00],
        [1.0658e-05, 9.9999e-01],
        [3.3031e-07, 1.0000e+00],
        [1.4693e-03, 9.9853e-01],
        [3.2661e-05, 9.9997e-01],
        [9.8758e-01, 1.2421e-02],
        [2.3040e-05, 9.9998e-01],
        [2.0711e-04, 9.9979e-01],
        [7.1632e-08, 1.0000e+00],
        [6.8150e-04, 9.9932e-01],
        [6.3924e-05, 9.9994e-01],
        [1.5429e-03, 9.9846e-01],
        [7.8350e-08, 1.0000e+00],
        [9.8888e-01, 1.1122e-02],
        [2.0086e-05, 9.9998e-01],
        [9.8237e-01, 1.7633e-02],
        [3.7554e-04, 9.9962e-01],
        [2.1432e-07, 1.0000e+00],
        [3.2065e-09, 1.0000e+00],
        [4.7917e-07, 1.0000e+00],
        [5.8882e-04, 9.9941e-01],
        [1.7862e-06, 1.0000e+00],
        [1.0888e-05, 9.9999e-01],
        [5.8168e-02, 9.4183e-01],
        [2.7327e-05, 9.9997e-01],
        [1.0640e-05, 9.9999e-01],
        [2.7927e-04, 9.9972e-01],
        [1.1081e-05, 9.9999e-01],
        [9.9645e-01, 3.5501e-03],
        [9.9764e-01, 2.3623e-03],
        [9.8837e-01, 1.1626e-02],
        [6.2270e-07, 1.0000e+00],
        [1.1358e-06, 1.0000e+00],
        [1.6190e-06, 1.0000e+00],
        [2.6929e-06, 1.0000e+00],
        [1.3208e-07, 1.0000e+00],
        [9.9800e-01, 2.0033e-03],
        [1.1799e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9969e-04, 9.9900e-01],
        [1.3863e-03, 9.9861e-01],
        [9.6670e-03, 9.9033e-01],
        [3.0632e-07, 1.0000e+00],
        [9.9425e-01, 5.7505e-03],
        [1.0343e-04, 9.9990e-01],
        [5.9310e-03, 9.9407e-01],
        [5.0957e-06, 9.9999e-01],
        [6.6230e-06, 9.9999e-01],
        [4.3567e-07, 1.0000e+00],
        [5.2800e-04, 9.9947e-01],
        [9.9978e-01, 2.1713e-04],
        [2.3167e-09, 1.0000e+00],
        [3.1913e-03, 9.9681e-01],
        [6.6328e-07, 1.0000e+00],
        [5.9880e-05, 9.9994e-01],
        [8.9675e-01, 1.0325e-01],
        [9.8818e-01, 1.1817e-02],
        [9.7305e-01, 2.6949e-02],
        [3.0962e-06, 1.0000e+00],
        [1.3812e-04, 9.9986e-01],
        [9.9895e-01, 1.0499e-03],
        [1.0451e-05, 9.9999e-01],
        [9.9436e-01, 5.6443e-03],
        [9.9339e-01, 6.6149e-03],
        [4.0370e-05, 9.9996e-01],
        [2.3205e-01, 7.6795e-01],
        [1.8842e-03, 9.9812e-01],
        [6.2420e-01, 3.7580e-01],
        [1.3812e-03, 9.9862e-01],
        [1.1803e-01, 8.8197e-01],
        [3.0283e-08, 1.0000e+00],
        [1.3188e-06, 1.0000e+00],
        [7.5611e-05, 9.9992e-01],
        [2.6011e-05, 9.9997e-01],
        [1.7949e-04, 9.9982e-01],
        [9.9891e-01, 1.0939e-03],
        [1.9887e-06, 1.0000e+00],
        [3.7731e-05, 9.9996e-01],
        [1.3555e-04, 9.9986e-01],
        [5.4722e-05, 9.9995e-01],
        [1.5644e-03, 9.9844e-01],
        [7.6048e-02, 9.2395e-01],
        [9.9990e-01, 1.0130e-04],
        [9.9990e-01, 9.7136e-05],
        [9.9959e-01, 4.1051e-04],
        [4.8924e-03, 9.9511e-01],
        [3.2317e-05, 9.9997e-01],
        [1.9795e-04, 9.9980e-01],
        [9.9846e-01, 1.5379e-03],
        [5.0106e-03, 9.9499e-01],
        [2.4013e-06, 1.0000e+00],
        [3.8386e-08, 1.0000e+00],
        [3.0339e-06, 1.0000e+00],
        [1.4685e-04, 9.9985e-01],
        [1.5080e-02, 9.8492e-01],
        [3.3320e-06, 1.0000e+00],
        [6.6461e-01, 3.3539e-01],
        [3.6850e-05, 9.9996e-01],
        [1.6863e-07, 1.0000e+00],
        [3.3192e-01, 6.6808e-01],
        [9.9598e-01, 4.0192e-03],
        [9.9886e-01, 1.1396e-03],
        [9.9842e-01, 1.5837e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3470e-05, 9.9999e-01],
        [6.7652e-06, 9.9999e-01],
        [7.1218e-05, 9.9993e-01],
        [3.4660e-03, 9.9653e-01],
        [3.7151e-03, 9.9628e-01],
        [1.8055e-04, 9.9982e-01],
        [9.8890e-07, 1.0000e+00],
        [3.8919e-04, 9.9961e-01],
        [1.1442e-08, 1.0000e+00],
        [2.6341e-03, 9.9737e-01],
        [6.1600e-02, 9.3840e-01],
        [3.2131e-04, 9.9968e-01],
        [1.1745e-03, 9.9883e-01],
        [2.5295e-04, 9.9975e-01],
        [4.9837e-02, 9.5016e-01],
        [6.1737e-05, 9.9994e-01],
        [9.9972e-01, 2.7567e-04],
        [3.3484e-07, 1.0000e+00],
        [1.6679e-07, 1.0000e+00],
        [9.9948e-01, 5.2013e-04],
        [9.9959e-01, 4.0929e-04],
        [1.7040e-04, 9.9983e-01],
        [4.6158e-05, 9.9995e-01],
        [9.9197e-01, 8.0280e-03],
        [2.5894e-03, 9.9741e-01],
        [6.0814e-03, 9.9392e-01],
        [4.5862e-01, 5.4138e-01],
        [5.4477e-04, 9.9946e-01],
        [3.7099e-02, 9.6290e-01],
        [3.1955e-05, 9.9997e-01],
        [9.9938e-01, 6.2245e-04],
        [9.9916e-01, 8.3940e-04],
        [9.9703e-01, 2.9736e-03],
        [2.7740e-03, 9.9723e-01],
        [1.0570e-03, 9.9894e-01],
        [7.9492e-06, 9.9999e-01],
        [3.1115e-03, 9.9689e-01],
        [3.4240e-04, 9.9966e-01],
        [4.1488e-06, 1.0000e+00],
        [2.1845e-06, 1.0000e+00],
        [6.4462e-06, 9.9999e-01],
        [1.3912e-06, 1.0000e+00],
        [1.6705e-02, 9.8329e-01],
        [2.1886e-06, 1.0000e+00],
        [9.8542e-01, 1.4578e-02],
        [1.0777e-07, 1.0000e+00],
        [9.1663e-05, 9.9991e-01],
        [8.3824e-01, 1.6176e-01],
        [1.9791e-03, 9.9802e-01],
        [5.1419e-07, 1.0000e+00],
        [1.6216e-04, 9.9984e-01],
        [9.9994e-01, 6.4171e-05],
        [3.3524e-06, 1.0000e+00],
        [8.9285e-03, 9.9107e-01],
        [5.2941e-03, 9.9471e-01],
        [6.0189e-06, 9.9999e-01],
        [1.5684e-03, 9.9843e-01],
        [9.7924e-01, 2.0762e-02],
        [1.5028e-04, 9.9985e-01],
        [5.3397e-08, 1.0000e+00],
        [2.3561e-06, 1.0000e+00],
        [1.1772e-04, 9.9988e-01],
        [4.2267e-07, 1.0000e+00],
        [9.9974e-01, 2.5583e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7366e-06, 1.0000e+00],
        [5.5196e-03, 9.9448e-01],
        [4.0972e-06, 1.0000e+00],
        [7.3660e-06, 9.9999e-01],
        [4.3747e-06, 1.0000e+00],
        [3.0778e-07, 1.0000e+00],
        [1.4721e-06, 1.0000e+00],
        [1.4999e-04, 9.9985e-01],
        [9.9974e-01, 2.6162e-04],
        [3.5556e-06, 1.0000e+00],
        [5.5558e-10, 1.0000e+00],
        [1.6495e-04, 9.9984e-01],
        [4.3256e-04, 9.9957e-01],
        [8.9237e-04, 9.9911e-01],
        [4.5500e-03, 9.9545e-01],
        [9.9980e-01, 2.0092e-04],
        [6.3385e-07, 1.0000e+00],
        [6.1750e-04, 9.9938e-01],
        [1.4818e-06, 1.0000e+00],
        [5.4017e-09, 1.0000e+00],
        [8.6173e-04, 9.9914e-01],
        [1.5241e-06, 1.0000e+00],
        [8.4244e-01, 1.5756e-01],
        [9.9968e-01, 3.1877e-04],
        [3.5020e-07, 1.0000e+00],
        [3.9731e-06, 1.0000e+00],
        [9.9714e-01, 2.8595e-03],
        [9.9871e-01, 1.2931e-03],
        [1.5719e-05, 9.9998e-01],
        [2.0048e-04, 9.9980e-01],
        [1.2198e-06, 1.0000e+00],
        [4.2257e-03, 9.9577e-01],
        [5.5258e-07, 1.0000e+00],
        [5.8299e-05, 9.9994e-01],
        [5.5120e-06, 9.9999e-01],
        [5.2997e-04, 9.9947e-01],
        [4.1163e-07, 1.0000e+00],
        [6.5149e-05, 9.9993e-01],
        [2.8030e-03, 9.9720e-01],
        [9.0320e-04, 9.9910e-01],
        [3.6358e-07, 1.0000e+00],
        [6.0418e-04, 9.9940e-01],
        [1.5241e-06, 1.0000e+00],
        [9.0960e-05, 9.9991e-01],
        [2.2478e-08, 1.0000e+00],
        [1.8900e-01, 8.1100e-01],
        [2.9427e-05, 9.9997e-01],
        [3.5217e-05, 9.9996e-01],
        [2.7174e-05, 9.9997e-01],
        [1.4850e-02, 9.8515e-01],
        [1.8074e-04, 9.9982e-01],
        [6.2540e-05, 9.9994e-01],
        [2.8078e-07, 1.0000e+00],
        [1.3653e-03, 9.9863e-01],
        [4.0787e-05, 9.9996e-01],
        [1.5473e-05, 9.9998e-01],
        [1.5782e-07, 1.0000e+00],
        [5.5196e-03, 9.9448e-01],
        [9.9999e-01, 8.4604e-06],
        [7.0845e-01, 2.9155e-01],
        [1.2418e-02, 9.8758e-01],
        [6.6733e-06, 9.9999e-01],
        [4.5192e-05, 9.9995e-01],
        [2.3119e-04, 9.9977e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.1790e-03, 9.9082e-01],
        [9.9958e-01, 4.1754e-04],
        [6.5415e-04, 9.9935e-01],
        [2.9002e-06, 1.0000e+00],
        [2.1145e-05, 9.9998e-01],
        [2.8562e-08, 1.0000e+00],
        [5.4320e-03, 9.9457e-01],
        [9.9901e-01, 9.8679e-04],
        [2.6366e-05, 9.9997e-01],
        [9.9872e-01, 1.2815e-03],
        [9.9971e-01, 2.8964e-04],
        [1.9098e-04, 9.9981e-01],
        [8.3942e-05, 9.9992e-01],
        [9.9605e-01, 3.9545e-03],
        [5.1832e-04, 9.9948e-01],
        [2.2309e-05, 9.9998e-01],
        [3.0733e-05, 9.9997e-01],
        [8.5929e-05, 9.9991e-01],
        [6.1921e-06, 9.9999e-01],
        [1.3030e-03, 9.9870e-01],
        [4.9877e-04, 9.9950e-01],
        [9.7889e-01, 2.1114e-02],
        [1.0808e-05, 9.9999e-01],
        [1.2550e-03, 9.9874e-01],
        [3.7203e-03, 9.9628e-01],
        [2.0670e-04, 9.9979e-01],
        [4.5891e-03, 9.9541e-01],
        [1.0738e-05, 9.9999e-01],
        [5.8367e-02, 9.4163e-01],
        [9.9897e-01, 1.0316e-03],
        [9.9971e-01, 2.9128e-04],
        [1.5197e-02, 9.8480e-01],
        [9.9421e-01, 5.7918e-03],
        [5.0285e-03, 9.9497e-01],
        [7.2801e-03, 9.9272e-01],
        [9.9988e-01, 1.2122e-04],
        [2.6244e-06, 1.0000e+00],
        [3.2722e-04, 9.9967e-01],
        [5.0493e-05, 9.9995e-01],
        [1.5191e-07, 1.0000e+00],
        [9.9897e-01, 1.0254e-03],
        [2.7969e-03, 9.9720e-01],
        [1.8902e-07, 1.0000e+00],
        [1.3722e-04, 9.9986e-01],
        [9.9962e-01, 3.8120e-04],
        [2.5558e-06, 1.0000e+00],
        [1.6673e-06, 1.0000e+00],
        [9.9976e-01, 2.3902e-04],
        [2.4297e-07, 1.0000e+00],
        [2.2947e-03, 9.9771e-01],
        [5.5146e-05, 9.9994e-01],
        [2.7585e-01, 7.2415e-01],
        [6.5420e-05, 9.9993e-01],
        [3.2924e-03, 9.9671e-01],
        [1.0851e-06, 1.0000e+00],
        [7.9672e-04, 9.9920e-01],
        [9.9981e-01, 1.9431e-04],
        [4.0591e-04, 9.9959e-01],
        [4.3680e-06, 1.0000e+00],
        [4.0788e-04, 9.9959e-01],
        [2.9810e-03, 9.9702e-01],
        [2.3380e-03, 9.9766e-01],
        [2.1126e-06, 1.0000e+00],
        [4.9285e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3945e-03, 9.9761e-01],
        [1.3615e-03, 9.9864e-01],
        [1.6974e-04, 9.9983e-01],
        [9.9943e-01, 5.6602e-04],
        [2.4206e-03, 9.9758e-01],
        [5.8567e-06, 9.9999e-01],
        [9.9666e-01, 3.3398e-03],
        [4.3216e-03, 9.9568e-01],
        [9.9987e-01, 1.3295e-04],
        [1.1560e-01, 8.8440e-01],
        [3.8934e-04, 9.9961e-01],
        [1.0979e-04, 9.9989e-01],
        [4.7840e-06, 1.0000e+00],
        [4.0150e-01, 5.9850e-01],
        [3.2122e-03, 9.9679e-01],
        [7.1922e-05, 9.9993e-01],
        [9.4447e-06, 9.9999e-01],
        [4.2345e-05, 9.9996e-01],
        [1.2822e-05, 9.9999e-01],
        [3.6464e-05, 9.9996e-01],
        [1.9356e-07, 1.0000e+00],
        [2.6442e-04, 9.9974e-01],
        [1.5844e-02, 9.8416e-01],
        [1.3558e-05, 9.9999e-01],
        [9.9876e-01, 1.2366e-03],
        [9.6234e-01, 3.7662e-02],
        [6.9843e-04, 9.9930e-01],
        [6.1705e-05, 9.9994e-01],
        [3.9860e-03, 9.9601e-01],
        [9.9982e-01, 1.7727e-04],
        [8.7577e-06, 9.9999e-01],
        [2.2317e-06, 1.0000e+00],
        [2.3251e-05, 9.9998e-01],
        [2.9844e-06, 1.0000e+00],
        [5.0468e-04, 9.9950e-01],
        [9.5737e-01, 4.2629e-02],
        [1.1963e-06, 1.0000e+00],
        [1.8731e-02, 9.8127e-01],
        [9.2266e-02, 9.0773e-01],
        [3.4010e-04, 9.9966e-01],
        [1.3665e-05, 9.9999e-01],
        [4.4900e-05, 9.9996e-01],
        [1.0387e-06, 1.0000e+00],
        [8.1215e-08, 1.0000e+00],
        [4.4704e-04, 9.9955e-01],
        [9.3818e-01, 6.1818e-02],
        [1.8335e-06, 1.0000e+00],
        [7.3845e-04, 9.9926e-01],
        [7.2274e-01, 2.7726e-01],
        [1.2008e-05, 9.9999e-01],
        [8.3737e-01, 1.6263e-01],
        [1.2931e-02, 9.8707e-01],
        [9.9975e-01, 2.4875e-04],
        [9.9962e-01, 3.7714e-04],
        [5.6336e-09, 1.0000e+00],
        [1.0002e-03, 9.9900e-01],
        [8.8326e-01, 1.1674e-01],
        [2.0936e-04, 9.9979e-01],
        [9.8322e-01, 1.6779e-02],
        [6.7095e-08, 1.0000e+00],
        [2.4490e-04, 9.9976e-01],
        [4.1276e-03, 9.9587e-01],
        [1.4875e-06, 1.0000e+00],
        [2.5380e-03, 9.9746e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9986e-01, 1.3636e-04],
        [1.7537e-05, 9.9998e-01],
        [5.0023e-09, 1.0000e+00],
        [6.0157e-06, 9.9999e-01],
        [9.9958e-01, 4.1780e-04],
        [2.6852e-07, 1.0000e+00],
        [3.7407e-02, 9.6259e-01],
        [1.7633e-04, 9.9982e-01],
        [9.9949e-01, 5.1290e-04],
        [7.6609e-01, 2.3391e-01],
        [9.9912e-01, 8.8239e-04],
        [2.6355e-02, 9.7365e-01],
        [5.5503e-03, 9.9445e-01],
        [2.3950e-07, 1.0000e+00],
        [2.3859e-03, 9.9761e-01],
        [1.1609e-05, 9.9999e-01],
        [6.2149e-02, 9.3785e-01],
        [9.1991e-06, 9.9999e-01],
        [9.9926e-01, 7.3614e-04],
        [1.0019e-06, 1.0000e+00],
        [8.7663e-01, 1.2337e-01],
        [6.7299e-06, 9.9999e-01],
        [3.1524e-01, 6.8476e-01],
        [3.2198e-04, 9.9968e-01],
        [2.7696e-04, 9.9972e-01],
        [1.4100e-02, 9.8590e-01],
        [9.9507e-01, 4.9279e-03],
        [2.0012e-01, 7.9988e-01],
        [3.0356e-03, 9.9696e-01],
        [9.9885e-01, 1.1495e-03],
        [7.2732e-02, 9.2727e-01],
        [1.0862e-04, 9.9989e-01],
        [9.9463e-01, 5.3711e-03],
        [1.8003e-06, 1.0000e+00],
        [1.0175e-06, 1.0000e+00],
        [1.4278e-03, 9.9857e-01],
        [2.9344e-06, 1.0000e+00],
        [1.4199e-04, 9.9986e-01],
        [3.4777e-04, 9.9965e-01],
        [3.8266e-01, 6.1734e-01],
        [1.1330e-01, 8.8670e-01],
        [1.7458e-05, 9.9998e-01],
        [9.9741e-01, 2.5914e-03],
        [7.9590e-05, 9.9992e-01],
        [9.9451e-01, 5.4860e-03],
        [9.9537e-01, 4.6291e-03],
        [2.2764e-03, 9.9772e-01],
        [2.2871e-05, 9.9998e-01],
        [5.4059e-01, 4.5941e-01],
        [3.2552e-03, 9.9674e-01],
        [1.2769e-06, 1.0000e+00],
        [9.9841e-01, 1.5937e-03],
        [1.1347e-04, 9.9989e-01],
        [3.9830e-03, 9.9602e-01],
        [9.9985e-01, 1.4996e-04],
        [3.2767e-05, 9.9997e-01],
        [9.9828e-01, 1.7176e-03],
        [7.1345e-02, 9.2866e-01],
        [9.9745e-01, 2.5497e-03],
        [9.6961e-03, 9.9030e-01],
        [2.2156e-04, 9.9978e-01],
        [9.9108e-01, 8.9150e-03],
        [3.1241e-04, 9.9969e-01],
        [9.9983e-01, 1.7215e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9473e-06, 1.0000e+00],
        [1.1546e-08, 1.0000e+00],
        [2.2918e-03, 9.9771e-01],
        [9.9895e-01, 1.0475e-03],
        [2.5381e-04, 9.9975e-01],
        [3.0943e-08, 1.0000e+00],
        [2.7542e-02, 9.7246e-01],
        [1.3448e-04, 9.9987e-01],
        [1.0292e-06, 1.0000e+00],
        [9.0743e-05, 9.9991e-01],
        [9.9984e-01, 1.6434e-04],
        [3.9822e-07, 1.0000e+00],
        [1.3640e-05, 9.9999e-01],
        [1.8379e-01, 8.1621e-01],
        [1.2404e-04, 9.9988e-01],
        [9.8586e-01, 1.4144e-02],
        [3.7251e-02, 9.6275e-01],
        [9.9021e-01, 9.7919e-03],
        [9.9941e-01, 5.8530e-04],
        [1.3999e-03, 9.9860e-01],
        [4.9153e-04, 9.9951e-01],
        [1.3269e-03, 9.9867e-01],
        [9.9960e-01, 3.9834e-04],
        [6.3309e-04, 9.9937e-01],
        [9.9992e-01, 7.9260e-05],
        [6.9591e-08, 1.0000e+00],
        [9.9435e-05, 9.9990e-01],
        [9.9979e-01, 2.1464e-04],
        [9.0465e-06, 9.9999e-01],
        [2.4976e-01, 7.5024e-01],
        [9.5545e-05, 9.9990e-01],
        [4.3670e-05, 9.9996e-01],
        [2.0605e-04, 9.9979e-01],
        [4.1886e-05, 9.9996e-01],
        [3.0434e-06, 1.0000e+00],
        [4.4230e-06, 1.0000e+00],
        [1.7851e-05, 9.9998e-01],
        [2.4107e-05, 9.9998e-01],
        [3.4098e-07, 1.0000e+00],
        [9.8635e-04, 9.9901e-01],
        [1.9678e-02, 9.8032e-01],
        [9.9284e-01, 7.1593e-03],
        [1.0233e-06, 1.0000e+00],
        [3.9562e-04, 9.9960e-01],
        [1.4860e-02, 9.8514e-01],
        [3.8294e-04, 9.9962e-01],
        [5.3662e-04, 9.9946e-01],
        [8.0849e-02, 9.1915e-01],
        [9.8940e-01, 1.0605e-02],
        [9.9877e-05, 9.9990e-01],
        [9.9935e-01, 6.5502e-04],
        [2.1146e-04, 9.9979e-01],
        [4.7001e-07, 1.0000e+00],
        [6.3745e-03, 9.9363e-01],
        [3.4668e-06, 1.0000e+00],
        [3.6359e-05, 9.9996e-01],
        [7.4363e-05, 9.9993e-01],
        [7.8399e-03, 9.9216e-01],
        [2.6471e-06, 1.0000e+00],
        [6.1701e-05, 9.9994e-01],
        [9.9972e-01, 2.7763e-04],
        [1.7525e-03, 9.9825e-01],
        [1.2454e-03, 9.9875e-01],
        [1.5032e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5815e-02, 9.6419e-01],
        [9.9747e-01, 2.5256e-03],
        [4.6915e-07, 1.0000e+00],
        [6.5423e-04, 9.9935e-01],
        [1.1517e-05, 9.9999e-01],
        [2.9538e-01, 7.0462e-01],
        [1.1243e-05, 9.9999e-01],
        [9.9943e-01, 5.7084e-04],
        [9.9899e-01, 1.0058e-03],
        [1.9931e-05, 9.9998e-01],
        [3.8664e-04, 9.9961e-01],
        [5.6468e-07, 1.0000e+00],
        [1.2485e-04, 9.9988e-01],
        [1.9675e-06, 1.0000e+00],
        [6.6796e-05, 9.9993e-01],
        [8.6480e-05, 9.9991e-01],
        [5.0000e-04, 9.9950e-01],
        [9.1681e-01, 8.3194e-02],
        [2.0069e-05, 9.9998e-01],
        [2.5780e-01, 7.4220e-01],
        [3.2556e-03, 9.9674e-01],
        [1.1562e-05, 9.9999e-01],
        [1.3003e-03, 9.9870e-01],
        [2.5371e-01, 7.4629e-01],
        [2.9266e-06, 1.0000e+00],
        [1.0538e-01, 8.9462e-01],
        [9.9998e-01, 1.5152e-05],
        [3.9024e-04, 9.9961e-01],
        [2.1373e-03, 9.9786e-01],
        [1.1493e-06, 1.0000e+00],
        [4.2473e-01, 5.7527e-01],
        [2.1162e-03, 9.9788e-01],
        [2.0239e-02, 9.7976e-01],
        [7.5720e-03, 9.9243e-01],
        [4.7828e-04, 9.9952e-01],
        [3.7325e-05, 9.9996e-01],
        [3.9921e-04, 9.9960e-01],
        [9.6438e-05, 9.9990e-01],
        [9.9912e-01, 8.7921e-04],
        [1.1170e-03, 9.9888e-01],
        [2.5525e-05, 9.9997e-01],
        [9.8848e-01, 1.1520e-02],
        [9.9990e-01, 1.0416e-04],
        [1.4113e-04, 9.9986e-01],
        [1.6548e-05, 9.9998e-01],
        [3.0856e-04, 9.9969e-01],
        [9.6995e-01, 3.0049e-02],
        [7.6177e-06, 9.9999e-01],
        [8.7556e-06, 9.9999e-01],
        [9.9962e-01, 3.7957e-04],
        [1.4890e-05, 9.9999e-01],
        [5.6117e-05, 9.9994e-01],
        [1.6592e-05, 9.9998e-01],
        [1.1249e-04, 9.9989e-01],
        [9.9802e-01, 1.9785e-03],
        [9.9995e-01, 4.6017e-05],
        [3.1061e-02, 9.6894e-01],
        [9.9972e-01, 2.8100e-04],
        [5.4406e-05, 9.9995e-01],
        [5.5980e-07, 1.0000e+00],
        [4.6642e-05, 9.9995e-01],
        [1.0714e-01, 8.9286e-01],
        [2.4249e-05, 9.9998e-01],
        [9.9964e-01, 3.6365e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3984e-05, 9.9999e-01],
        [4.7753e-06, 1.0000e+00],
        [6.7011e-03, 9.9330e-01],
        [9.7860e-01, 2.1403e-02],
        [3.7346e-05, 9.9996e-01],
        [2.7170e-06, 1.0000e+00],
        [2.6006e-04, 9.9974e-01],
        [1.7462e-05, 9.9998e-01],
        [5.6507e-03, 9.9435e-01],
        [2.4716e-04, 9.9975e-01],
        [2.5504e-03, 9.9745e-01],
        [7.5267e-06, 9.9999e-01],
        [9.9953e-01, 4.7296e-04],
        [9.6492e-04, 9.9904e-01],
        [9.9986e-01, 1.3732e-04],
        [2.4228e-03, 9.9758e-01],
        [9.5712e-06, 9.9999e-01],
        [4.1240e-03, 9.9588e-01],
        [5.1630e-04, 9.9948e-01],
        [2.9064e-04, 9.9971e-01],
        [1.9310e-03, 9.9807e-01],
        [8.5575e-06, 9.9999e-01],
        [1.5452e-06, 1.0000e+00],
        [2.6890e-04, 9.9973e-01],
        [1.7380e-02, 9.8262e-01],
        [1.2946e-03, 9.9871e-01],
        [9.2296e-01, 7.7044e-02],
        [9.8392e-01, 1.6076e-02],
        [2.6442e-08, 1.0000e+00],
        [2.0270e-05, 9.9998e-01],
        [2.7150e-05, 9.9997e-01],
        [9.7656e-06, 9.9999e-01],
        [2.5056e-04, 9.9975e-01],
        [2.9162e-05, 9.9997e-01],
        [1.6256e-01, 8.3744e-01],
        [9.9966e-01, 3.3692e-04],
        [1.2375e-01, 8.7625e-01],
        [9.3429e-01, 6.5712e-02],
        [1.4450e-03, 9.9856e-01],
        [5.4524e-03, 9.9455e-01],
        [1.0220e-03, 9.9898e-01],
        [5.9813e-04, 9.9940e-01],
        [6.2298e-07, 1.0000e+00],
        [5.8419e-02, 9.4158e-01],
        [2.0525e-04, 9.9979e-01],
        [9.5471e-01, 4.5293e-02],
        [2.9515e-05, 9.9997e-01],
        [9.9902e-01, 9.7939e-04],
        [4.2893e-03, 9.9571e-01],
        [9.7078e-01, 2.9225e-02],
        [1.5644e-02, 9.8436e-01],
        [1.3678e-04, 9.9986e-01],
        [9.6853e-01, 3.1468e-02],
        [1.4798e-04, 9.9985e-01],
        [4.2136e-04, 9.9958e-01],
        [3.3749e-09, 1.0000e+00],
        [1.0032e-07, 1.0000e+00],
        [9.9992e-01, 8.3735e-05],
        [5.1668e-05, 9.9995e-01],
        [3.9638e-01, 6.0362e-01],
        [1.6006e-04, 9.9984e-01],
        [6.8906e-05, 9.9993e-01],
        [3.7586e-05, 9.9996e-01],
        [7.8409e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9893e-01, 1.0653e-03],
        [9.9944e-01, 5.6427e-04],
        [6.1085e-01, 3.8915e-01],
        [1.4368e-07, 1.0000e+00],
        [7.9013e-06, 9.9999e-01],
        [1.2123e-08, 1.0000e+00],
        [1.1979e-05, 9.9999e-01],
        [9.9994e-01, 6.4140e-05],
        [9.9916e-01, 8.3812e-04],
        [1.6083e-04, 9.9984e-01],
        [9.5077e-01, 4.9232e-02],
        [9.9999e-01, 5.1521e-06],
        [9.2015e-04, 9.9908e-01],
        [7.7925e-03, 9.9221e-01],
        [5.0645e-05, 9.9995e-01],
        [1.9462e-03, 9.9805e-01],
        [1.6692e-06, 1.0000e+00],
        [1.7925e-02, 9.8207e-01],
        [4.4255e-04, 9.9956e-01],
        [8.7870e-01, 1.2130e-01],
        [9.4901e-02, 9.0510e-01],
        [9.6453e-01, 3.5466e-02],
        [9.9990e-01, 1.0146e-04],
        [5.3727e-02, 9.4627e-01],
        [5.8918e-02, 9.4108e-01],
        [3.3020e-05, 9.9997e-01],
        [1.6800e-08, 1.0000e+00],
        [9.6422e-01, 3.5775e-02],
        [9.9965e-01, 3.4582e-04],
        [9.9621e-01, 3.7856e-03],
        [9.9966e-01, 3.3858e-04],
        [6.9423e-07, 1.0000e+00],
        [3.2562e-05, 9.9997e-01],
        [9.9678e-01, 3.2215e-03],
        [2.8705e-04, 9.9971e-01],
        [9.9767e-01, 2.3270e-03],
        [1.9471e-05, 9.9998e-01],
        [6.8513e-07, 1.0000e+00],
        [9.9948e-01, 5.1622e-04],
        [2.1133e-06, 1.0000e+00],
        [4.2276e-07, 1.0000e+00],
        [1.0442e-02, 9.8956e-01],
        [1.9164e-04, 9.9981e-01],
        [1.6360e-02, 9.8364e-01],
        [4.0577e-04, 9.9959e-01],
        [5.9794e-03, 9.9402e-01],
        [6.6788e-04, 9.9933e-01],
        [9.9935e-01, 6.5368e-04],
        [1.7462e-01, 8.2538e-01],
        [2.0772e-07, 1.0000e+00],
        [3.9054e-03, 9.9609e-01],
        [2.0191e-02, 9.7981e-01],
        [1.6788e-03, 9.9832e-01],
        [1.7190e-05, 9.9998e-01],
        [1.8710e-04, 9.9981e-01],
        [1.7148e-05, 9.9998e-01],
        [3.0493e-04, 9.9970e-01],
        [9.8894e-01, 1.1063e-02],
        [7.2912e-07, 1.0000e+00],
        [7.5450e-01, 2.4550e-01],
        [9.9440e-01, 5.5962e-03],
        [2.2932e-03, 9.9771e-01],
        [9.9456e-01, 5.4431e-03],
        [1.9772e-03, 9.9802e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.4015e-07, 1.0000e+00],
        [1.3206e-06, 1.0000e+00],
        [9.9772e-01, 2.2755e-03],
        [9.9968e-01, 3.2404e-04],
        [9.8217e-01, 1.7830e-02],
        [5.2466e-08, 1.0000e+00],
        [1.7097e-03, 9.9829e-01],
        [6.6790e-04, 9.9933e-01],
        [1.1189e-07, 1.0000e+00],
        [9.7656e-03, 9.9023e-01],
        [1.0000e+00, 2.0381e-06],
        [5.9450e-03, 9.9405e-01],
        [9.2085e-05, 9.9991e-01],
        [3.3712e-03, 9.9663e-01],
        [1.0856e-04, 9.9989e-01],
        [2.7305e-06, 1.0000e+00],
        [4.8659e-06, 1.0000e+00],
        [9.1898e-01, 8.1019e-02],
        [1.4511e-08, 1.0000e+00],
        [1.7106e-04, 9.9983e-01],
        [1.0204e-07, 1.0000e+00],
        [8.2280e-03, 9.9177e-01],
        [1.3206e-06, 1.0000e+00],
        [9.9961e-01, 3.8579e-04],
        [4.4998e-03, 9.9550e-01],
        [4.0342e-06, 1.0000e+00],
        [1.4169e-06, 1.0000e+00],
        [1.9317e-05, 9.9998e-01],
        [7.5482e-03, 9.9245e-01],
        [1.3431e-07, 1.0000e+00],
        [2.0624e-04, 9.9979e-01],
        [9.9953e-01, 4.6705e-04],
        [1.0494e-03, 9.9895e-01],
        [9.9986e-01, 1.4474e-04],
        [9.9930e-01, 6.9770e-04],
        [8.0508e-01, 1.9492e-01],
        [1.3446e-04, 9.9987e-01],
        [7.5995e-03, 9.9240e-01],
        [1.0144e-02, 9.8986e-01],
        [9.4398e-07, 1.0000e+00],
        [6.4998e-05, 9.9994e-01],
        [9.9953e-01, 4.6996e-04],
        [9.9991e-01, 8.8512e-05],
        [3.2293e-07, 1.0000e+00],
        [7.0001e-05, 9.9993e-01],
        [3.1594e-04, 9.9968e-01],
        [3.8188e-08, 1.0000e+00],
        [9.9939e-01, 6.0790e-04],
        [2.1120e-05, 9.9998e-01],
        [3.2455e-05, 9.9997e-01],
        [2.3798e-05, 9.9998e-01],
        [7.8086e-06, 9.9999e-01],
        [3.4951e-04, 9.9965e-01],
        [3.3087e-07, 1.0000e+00],
        [2.8284e-03, 9.9717e-01],
        [9.9933e-01, 6.7169e-04],
        [1.7933e-04, 9.9982e-01],
        [9.9925e-01, 7.5143e-04],
        [9.2917e-05, 9.9991e-01],
        [1.0599e-02, 9.8940e-01],
        [9.9943e-01, 5.7240e-04],
        [3.0477e-01, 6.9523e-01],
        [7.4301e-02, 9.2570e-01],
        [5.4968e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2598e-02, 9.8740e-01],
        [6.6319e-06, 9.9999e-01],
        [1.7149e-04, 9.9983e-01],
        [9.5174e-05, 9.9990e-01],
        [7.9468e-02, 9.2053e-01],
        [2.0484e-03, 9.9795e-01],
        [9.4324e-05, 9.9991e-01],
        [2.4777e-04, 9.9975e-01],
        [1.7415e-07, 1.0000e+00],
        [2.0117e-04, 9.9980e-01],
        [1.1705e-04, 9.9988e-01],
        [5.5311e-04, 9.9945e-01],
        [4.0451e-04, 9.9960e-01],
        [2.1773e-05, 9.9998e-01],
        [4.7809e-07, 1.0000e+00],
        [1.0658e-05, 9.9999e-01],
        [1.9287e-04, 9.9981e-01],
        [3.4813e-06, 1.0000e+00],
        [6.0253e-04, 9.9940e-01],
        [1.7375e-05, 9.9998e-01],
        [9.9963e-01, 3.7278e-04],
        [7.7892e-05, 9.9992e-01],
        [7.6738e-03, 9.9233e-01],
        [9.7755e-01, 2.2447e-02],
        [9.9872e-01, 1.2802e-03],
        [2.5244e-03, 9.9748e-01],
        [6.0658e-05, 9.9994e-01],
        [1.0147e-03, 9.9899e-01],
        [1.4696e-02, 9.8530e-01],
        [3.8850e-03, 9.9611e-01],
        [1.5300e-02, 9.8470e-01],
        [9.9940e-01, 6.0109e-04],
        [9.9809e-01, 1.9117e-03],
        [1.0578e-05, 9.9999e-01],
        [7.4037e-05, 9.9993e-01],
        [3.2130e-06, 1.0000e+00],
        [6.0311e-04, 9.9940e-01],
        [5.8556e-05, 9.9994e-01],
        [5.3033e-05, 9.9995e-01],
        [2.7304e-04, 9.9973e-01],
        [3.3542e-01, 6.6458e-01],
        [2.1669e-06, 1.0000e+00],
        [3.4507e-03, 9.9655e-01],
        [1.0593e-06, 1.0000e+00],
        [3.9330e-03, 9.9607e-01],
        [9.8881e-01, 1.1187e-02],
        [9.7548e-01, 2.4523e-02],
        [9.2545e-09, 1.0000e+00],
        [2.8881e-03, 9.9711e-01],
        [9.9725e-01, 2.7483e-03],
        [1.7373e-08, 1.0000e+00],
        [1.2267e-01, 8.7733e-01],
        [8.8291e-07, 1.0000e+00],
        [1.2493e-03, 9.9875e-01],
        [9.9979e-01, 2.1430e-04],
        [9.9962e-01, 3.8222e-04],
        [9.9815e-01, 1.8509e-03],
        [9.9960e-01, 4.0238e-04],
        [9.9903e-01, 9.7104e-04],
        [9.4944e-05, 9.9990e-01],
        [2.9176e-05, 9.9997e-01],
        [3.4212e-04, 9.9966e-01],
        [6.1579e-04, 9.9938e-01],
        [8.2654e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.4485e-05, 9.9993e-01],
        [9.9972e-01, 2.7871e-04],
        [9.9766e-08, 1.0000e+00],
        [7.1161e-03, 9.9288e-01],
        [3.8457e-06, 1.0000e+00],
        [4.7729e-07, 1.0000e+00],
        [7.4613e-07, 1.0000e+00],
        [9.9981e-01, 1.9113e-04],
        [1.6857e-05, 9.9998e-01],
        [2.4193e-04, 9.9976e-01],
        [9.9208e-01, 7.9227e-03],
        [9.3454e-02, 9.0655e-01],
        [9.2529e-01, 7.4710e-02],
        [9.9996e-01, 3.9397e-05],
        [5.0068e-08, 1.0000e+00],
        [9.5695e-01, 4.3048e-02],
        [2.4787e-01, 7.5213e-01],
        [9.4384e-05, 9.9991e-01],
        [2.5246e-07, 1.0000e+00],
        [1.0705e-04, 9.9989e-01],
        [8.2386e-04, 9.9918e-01],
        [1.4000e-03, 9.9860e-01],
        [9.9975e-01, 2.5392e-04],
        [4.0592e-04, 9.9959e-01],
        [9.6604e-01, 3.3961e-02],
        [7.4824e-07, 1.0000e+00],
        [1.0343e-07, 1.0000e+00],
        [9.3609e-07, 1.0000e+00],
        [1.9472e-05, 9.9998e-01],
        [2.0064e-07, 1.0000e+00],
        [3.6642e-04, 9.9963e-01],
        [1.1479e-04, 9.9989e-01],
        [1.3655e-05, 9.9999e-01],
        [9.9565e-01, 4.3531e-03],
        [3.0333e-03, 9.9697e-01],
        [1.0000e+00, 3.0155e-06],
        [5.8199e-06, 9.9999e-01],
        [1.7623e-02, 9.8238e-01],
        [9.9823e-01, 1.7745e-03],
        [6.4084e-06, 9.9999e-01],
        [9.9333e-01, 6.6708e-03],
        [9.9962e-01, 3.8312e-04],
        [1.8795e-06, 1.0000e+00],
        [1.1574e-09, 1.0000e+00],
        [1.8227e-06, 1.0000e+00],
        [5.3904e-04, 9.9946e-01],
        [7.7184e-09, 1.0000e+00],
        [4.4830e-06, 1.0000e+00],
        [9.5537e-01, 4.4630e-02],
        [2.7330e-05, 9.9997e-01],
        [9.9898e-01, 1.0191e-03],
        [1.5316e-07, 1.0000e+00],
        [8.3406e-03, 9.9166e-01],
        [9.9327e-01, 6.7326e-03],
        [9.8216e-01, 1.7838e-02],
        [1.3937e-09, 1.0000e+00],
        [1.8169e-04, 9.9982e-01],
        [1.2037e-07, 1.0000e+00],
        [6.6470e-05, 9.9993e-01],
        [2.5288e-03, 9.9747e-01],
        [2.4788e-03, 9.9752e-01],
        [8.1659e-05, 9.9992e-01],
        [2.4408e-06, 1.0000e+00],
        [9.9782e-01, 2.1841e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[5.0350e-06, 9.9999e-01],
        [1.2880e-02, 9.8712e-01],
        [1.0052e-07, 1.0000e+00],
        [1.1748e-03, 9.9883e-01],
        [9.9897e-01, 1.0275e-03],
        [9.8984e-04, 9.9901e-01],
        [1.6633e-07, 1.0000e+00],
        [5.0853e-06, 9.9999e-01],
        [9.9812e-01, 1.8753e-03],
        [7.4205e-07, 1.0000e+00],
        [1.6027e-04, 9.9984e-01],
        [9.9942e-01, 5.7802e-04],
        [5.9383e-09, 1.0000e+00],
        [1.4699e-05, 9.9999e-01],
        [4.2598e-03, 9.9574e-01],
        [9.9898e-01, 1.0167e-03],
        [9.9129e-06, 9.9999e-01],
        [2.1239e-05, 9.9998e-01],
        [5.3577e-05, 9.9995e-01],
        [3.7465e-07, 1.0000e+00],
        [5.3852e-05, 9.9995e-01],
        [9.9527e-01, 4.7269e-03],
        [4.5967e-05, 9.9995e-01],
        [6.9368e-05, 9.9993e-01],
        [5.0230e-07, 1.0000e+00],
        [9.9462e-01, 5.3779e-03],
        [6.5360e-05, 9.9993e-01],
        [3.3224e-07, 1.0000e+00],
        [4.3385e-05, 9.9996e-01],
        [8.4118e-03, 9.9159e-01],
        [3.2615e-06, 1.0000e+00],
        [9.9607e-05, 9.9990e-01],
        [9.9958e-01, 4.2126e-04],
        [2.0976e-06, 1.0000e+00],
        [2.9809e-04, 9.9970e-01],
        [7.1293e-05, 9.9993e-01],
        [7.0779e-04, 9.9929e-01],
        [2.0556e-02, 9.7944e-01],
        [9.8525e-01, 1.4754e-02],
        [7.1709e-04, 9.9928e-01],
        [1.1871e-07, 1.0000e+00],
        [3.8517e-05, 9.9996e-01],
        [9.2101e-01, 7.8990e-02],
        [9.9929e-01, 7.1152e-04],
        [2.1048e-04, 9.9979e-01],
        [1.1989e-04, 9.9988e-01],
        [9.9945e-01, 5.4679e-04],
        [2.2512e-06, 1.0000e+00],
        [2.5347e-06, 1.0000e+00],
        [1.5229e-03, 9.9848e-01],
        [1.7869e-08, 1.0000e+00],
        [1.6853e-08, 1.0000e+00],
        [5.1245e-09, 1.0000e+00],
        [5.6930e-08, 1.0000e+00],
        [9.6319e-01, 3.6815e-02],
        [1.9152e-06, 1.0000e+00],
        [9.8953e-01, 1.0469e-02],
        [1.9688e-07, 1.0000e+00],
        [1.6451e-05, 9.9998e-01],
        [7.9142e-01, 2.0858e-01],
        [1.8748e-05, 9.9998e-01],
        [9.9966e-01, 3.4233e-04],
        [9.2493e-05, 9.9991e-01],
        [8.3100e-04, 9.9917e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9552e-04, 9.9980e-01],
        [3.2043e-04, 9.9968e-01],
        [5.9526e-01, 4.0474e-01],
        [1.2647e-07, 1.0000e+00],
        [1.4473e-01, 8.5527e-01],
        [2.1010e-03, 9.9790e-01],
        [8.9704e-01, 1.0296e-01],
        [5.9776e-10, 1.0000e+00],
        [1.8543e-01, 8.1457e-01],
        [6.3042e-04, 9.9937e-01],
        [9.8449e-01, 1.5513e-02],
        [4.9051e-06, 1.0000e+00],
        [4.8225e-05, 9.9995e-01],
        [1.1029e-07, 1.0000e+00],
        [1.0541e-02, 9.8946e-01],
        [9.6240e-01, 3.7598e-02],
        [2.0267e-06, 1.0000e+00],
        [2.0953e-05, 9.9998e-01],
        [2.8240e-05, 9.9997e-01],
        [4.8344e-06, 1.0000e+00],
        [7.6386e-07, 1.0000e+00],
        [1.3209e-08, 1.0000e+00],
        [1.8110e-05, 9.9998e-01],
        [9.5348e-01, 4.6517e-02],
        [4.0743e-04, 9.9959e-01],
        [5.7235e-06, 9.9999e-01],
        [9.9938e-01, 6.1738e-04],
        [5.7556e-04, 9.9942e-01],
        [1.2418e-08, 1.0000e+00],
        [6.0133e-06, 9.9999e-01],
        [1.3544e-02, 9.8646e-01],
        [2.9178e-06, 1.0000e+00],
        [9.9958e-01, 4.1713e-04],
        [7.7299e-05, 9.9992e-01],
        [1.1732e-05, 9.9999e-01],
        [1.2210e-04, 9.9988e-01],
        [3.4514e-05, 9.9997e-01],
        [2.0315e-04, 9.9980e-01],
        [8.6038e-01, 1.3962e-01],
        [9.9811e-01, 1.8948e-03],
        [1.1322e-03, 9.9887e-01],
        [4.1476e-04, 9.9959e-01],
        [2.4982e-08, 1.0000e+00],
        [9.9982e-01, 1.8213e-04],
        [9.9999e-01, 5.4098e-06],
        [2.2279e-03, 9.9777e-01],
        [5.1626e-04, 9.9948e-01],
        [9.4959e-02, 9.0504e-01],
        [2.8956e-05, 9.9997e-01],
        [1.0073e-08, 1.0000e+00],
        [5.3560e-02, 9.4644e-01],
        [9.9868e-01, 1.3222e-03],
        [2.6387e-02, 9.7361e-01],
        [3.0300e-03, 9.9697e-01],
        [3.6692e-06, 1.0000e+00],
        [3.8585e-08, 1.0000e+00],
        [1.5010e-02, 9.8499e-01],
        [3.7801e-06, 1.0000e+00],
        [2.1050e-02, 9.7895e-01],
        [3.3783e-06, 1.0000e+00],
        [3.3393e-05, 9.9997e-01],
        [2.8532e-07, 1.0000e+00],
        [1.5440e-08, 1.0000e+00],
        [8.8691e-03, 9.9113e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5279e-01, 6.4721e-01],
        [2.9666e-07, 1.0000e+00],
        [1.3530e-05, 9.9999e-01],
        [8.4845e-01, 1.5155e-01],
        [1.5762e-06, 1.0000e+00],
        [4.5291e-05, 9.9995e-01],
        [3.6963e-01, 6.3037e-01],
        [1.0503e-06, 1.0000e+00],
        [1.6767e-05, 9.9998e-01],
        [2.7446e-05, 9.9997e-01],
        [3.1310e-03, 9.9687e-01],
        [1.6999e-02, 9.8300e-01],
        [2.3686e-04, 9.9976e-01],
        [5.5547e-04, 9.9944e-01],
        [9.9869e-01, 1.3092e-03],
        [1.8433e-06, 1.0000e+00],
        [7.4278e-01, 2.5722e-01],
        [2.8502e-05, 9.9997e-01],
        [8.1684e-09, 1.0000e+00],
        [2.2295e-05, 9.9998e-01],
        [3.5239e-05, 9.9996e-01],
        [2.7410e-02, 9.7259e-01],
        [1.3034e-09, 1.0000e+00],
        [7.7721e-05, 9.9992e-01],
        [1.4498e-05, 9.9999e-01],
        [5.5009e-01, 4.4991e-01],
        [1.5288e-08, 1.0000e+00],
        [1.5997e-05, 9.9998e-01],
        [1.2065e-06, 1.0000e+00],
        [6.7340e-05, 9.9993e-01],
        [6.1654e-06, 9.9999e-01],
        [4.9578e-03, 9.9504e-01],
        [1.8044e-06, 1.0000e+00],
        [3.5627e-04, 9.9964e-01],
        [8.7770e-02, 9.1223e-01],
        [1.0012e-05, 9.9999e-01],
        [2.2452e-07, 1.0000e+00],
        [9.9610e-01, 3.9037e-03],
        [4.3730e-03, 9.9563e-01],
        [1.0386e-06, 1.0000e+00],
        [6.1328e-10, 1.0000e+00],
        [9.9889e-01, 1.1072e-03],
        [2.0390e-08, 1.0000e+00],
        [1.8630e-02, 9.8137e-01],
        [6.6228e-01, 3.3772e-01],
        [4.4020e-06, 1.0000e+00],
        [1.0752e-05, 9.9999e-01],
        [1.5575e-04, 9.9984e-01],
        [1.8016e-03, 9.9820e-01],
        [2.5523e-04, 9.9974e-01],
        [5.9432e-05, 9.9994e-01],
        [3.3793e-08, 1.0000e+00],
        [1.1952e-03, 9.9880e-01],
        [5.0204e-05, 9.9995e-01],
        [1.6045e-03, 9.9840e-01],
        [1.0942e-06, 1.0000e+00],
        [3.1636e-03, 9.9684e-01],
        [3.7136e-04, 9.9963e-01],
        [9.7902e-01, 2.0980e-02],
        [2.6743e-08, 1.0000e+00],
        [1.2596e-07, 1.0000e+00],
        [9.9315e-01, 6.8506e-03],
        [2.5202e-06, 1.0000e+00],
        [6.8248e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9967e-01, 3.3079e-04],
        [9.6217e-01, 3.7833e-02],
        [9.5522e-05, 9.9990e-01],
        [5.6579e-06, 9.9999e-01],
        [1.0661e-06, 1.0000e+00],
        [4.6257e-04, 9.9954e-01],
        [9.9853e-01, 1.4731e-03],
        [1.5736e-05, 9.9998e-01],
        [9.9945e-01, 5.5486e-04],
        [8.8816e-08, 1.0000e+00],
        [9.9927e-01, 7.2900e-04],
        [6.7764e-01, 3.2236e-01],
        [2.5045e-04, 9.9975e-01],
        [9.3846e-08, 1.0000e+00],
        [1.9660e-08, 1.0000e+00],
        [4.8936e-09, 1.0000e+00],
        [7.5790e-05, 9.9992e-01],
        [9.9365e-01, 6.3538e-03],
        [3.2063e-07, 1.0000e+00],
        [2.1144e-06, 1.0000e+00],
        [9.9328e-01, 6.7151e-03],
        [2.7689e-07, 1.0000e+00],
        [2.8127e-04, 9.9972e-01],
        [7.5960e-05, 9.9992e-01],
        [3.7354e-03, 9.9626e-01],
        [2.0668e-09, 1.0000e+00],
        [9.4608e-01, 5.3918e-02],
        [1.7194e-03, 9.9828e-01],
        [1.4990e-08, 1.0000e+00],
        [9.9114e-01, 8.8635e-03],
        [9.9036e-04, 9.9901e-01],
        [2.7870e-04, 9.9972e-01],
        [1.9908e-07, 1.0000e+00],
        [9.9446e-01, 5.5433e-03],
        [5.3696e-03, 9.9463e-01],
        [5.2949e-05, 9.9995e-01],
        [3.0279e-05, 9.9997e-01],
        [2.3991e-08, 1.0000e+00],
        [5.7220e-05, 9.9994e-01],
        [7.2931e-03, 9.9271e-01],
        [9.5986e-06, 9.9999e-01],
        [1.0072e-04, 9.9990e-01],
        [2.1832e-08, 1.0000e+00],
        [4.8055e-04, 9.9952e-01],
        [1.4845e-06, 1.0000e+00],
        [4.6496e-06, 1.0000e+00],
        [1.3511e-06, 1.0000e+00],
        [9.9953e-01, 4.7183e-04],
        [2.9314e-04, 9.9971e-01],
        [1.4375e-07, 1.0000e+00],
        [1.8023e-04, 9.9982e-01],
        [5.7979e-05, 9.9994e-01],
        [1.1442e-07, 1.0000e+00],
        [3.2995e-10, 1.0000e+00],
        [8.7172e-04, 9.9913e-01],
        [3.1592e-07, 1.0000e+00],
        [9.8508e-01, 1.4921e-02],
        [3.6898e-06, 1.0000e+00],
        [7.3864e-08, 1.0000e+00],
        [1.9443e-04, 9.9981e-01],
        [4.5548e-07, 1.0000e+00],
        [7.2843e-01, 2.7157e-01],
        [4.5324e-05, 9.9995e-01],
        [8.2271e-04, 9.9918e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8446e-01, 1.5543e-02],
        [4.2101e-01, 5.7899e-01],
        [3.1799e-03, 9.9682e-01],
        [9.6856e-01, 3.1437e-02],
        [9.9047e-01, 9.5319e-03],
        [2.2227e-05, 9.9998e-01],
        [8.0614e-05, 9.9992e-01],
        [2.8302e-03, 9.9717e-01],
        [6.5260e-03, 9.9347e-01],
        [2.7250e-02, 9.7275e-01],
        [2.3353e-01, 7.6647e-01],
        [7.8289e-05, 9.9992e-01],
        [3.2709e-05, 9.9997e-01],
        [1.3936e-08, 1.0000e+00],
        [9.9895e-01, 1.0515e-03],
        [5.3927e-04, 9.9946e-01],
        [9.9694e-01, 3.0600e-03],
        [1.2764e-06, 1.0000e+00],
        [9.9882e-01, 1.1829e-03],
        [8.1695e-08, 1.0000e+00],
        [6.3359e-06, 9.9999e-01],
        [9.8984e-01, 1.0161e-02],
        [5.4946e-06, 9.9999e-01],
        [6.0749e-04, 9.9939e-01],
        [9.9434e-01, 5.6587e-03],
        [7.3448e-07, 1.0000e+00],
        [2.3719e-06, 1.0000e+00],
        [5.1363e-06, 9.9999e-01],
        [2.0128e-06, 1.0000e+00],
        [6.4250e-01, 3.5750e-01],
        [8.8611e-01, 1.1389e-01],
        [1.2141e-08, 1.0000e+00],
        [4.8093e-07, 1.0000e+00],
        [7.2803e-01, 2.7197e-01],
        [2.1348e-05, 9.9998e-01],
        [9.8677e-01, 1.3230e-02],
        [3.2674e-05, 9.9997e-01],
        [6.3335e-01, 3.6665e-01],
        [9.8288e-01, 1.7122e-02],
        [5.0279e-05, 9.9995e-01],
        [7.3417e-01, 2.6583e-01],
        [9.9008e-01, 9.9237e-03],
        [1.3057e-08, 1.0000e+00],
        [1.2726e-07, 1.0000e+00],
        [1.1085e-07, 1.0000e+00],
        [2.0918e-06, 1.0000e+00],
        [1.3377e-06, 1.0000e+00],
        [1.1353e-07, 1.0000e+00],
        [4.2845e-03, 9.9572e-01],
        [2.9339e-05, 9.9997e-01],
        [9.9998e-01, 1.8480e-05],
        [7.3501e-01, 2.6499e-01],
        [6.0752e-07, 1.0000e+00],
        [3.7395e-08, 1.0000e+00],
        [6.3593e-08, 1.0000e+00],
        [5.1664e-06, 9.9999e-01],
        [4.3166e-07, 1.0000e+00],
        [9.8386e-01, 1.6141e-02],
        [4.9329e-06, 1.0000e+00],
        [5.4791e-06, 9.9999e-01],
        [1.3693e-07, 1.0000e+00],
        [3.9929e-07, 1.0000e+00],
        [7.5739e-09, 1.0000e+00],
        [1.0058e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2540e-03, 9.9575e-01],
        [5.0983e-05, 9.9995e-01],
        [2.6136e-03, 9.9739e-01],
        [5.5467e-05, 9.9994e-01],
        [2.4106e-07, 1.0000e+00],
        [9.9958e-01, 4.1617e-04],
        [1.9724e-05, 9.9998e-01],
        [1.3617e-07, 1.0000e+00],
        [7.8213e-03, 9.9218e-01],
        [1.5426e-04, 9.9985e-01],
        [2.1138e-05, 9.9998e-01],
        [2.9770e-06, 1.0000e+00],
        [3.7440e-08, 1.0000e+00],
        [2.4020e-06, 1.0000e+00],
        [8.4288e-06, 9.9999e-01],
        [9.8602e-01, 1.3984e-02],
        [9.6036e-04, 9.9904e-01],
        [7.0151e-05, 9.9993e-01],
        [7.6480e-06, 9.9999e-01],
        [4.7114e-05, 9.9995e-01],
        [9.9993e-01, 7.4581e-05],
        [8.2245e-04, 9.9918e-01],
        [9.9932e-01, 6.8297e-04],
        [1.0968e-05, 9.9999e-01],
        [2.1181e-01, 7.8819e-01],
        [1.6554e-07, 1.0000e+00],
        [9.9743e-01, 2.5726e-03],
        [9.9916e-01, 8.3984e-04],
        [2.9324e-07, 1.0000e+00],
        [9.6871e-04, 9.9903e-01],
        [1.1779e-06, 1.0000e+00],
        [9.7932e-08, 1.0000e+00],
        [9.9046e-04, 9.9901e-01],
        [2.1742e-07, 1.0000e+00],
        [9.9703e-01, 2.9669e-03],
        [1.0336e-02, 9.8966e-01],
        [4.5257e-07, 1.0000e+00],
        [9.9987e-01, 1.2679e-04],
        [2.8850e-03, 9.9712e-01],
        [9.9974e-01, 2.5969e-04],
        [1.9019e-04, 9.9981e-01],
        [6.9628e-02, 9.3037e-01],
        [4.9961e-04, 9.9950e-01],
        [9.9929e-01, 7.0515e-04],
        [1.3568e-05, 9.9999e-01],
        [2.1212e-05, 9.9998e-01],
        [6.9714e-07, 1.0000e+00],
        [9.9049e-01, 9.5147e-03],
        [4.7602e-06, 1.0000e+00],
        [2.2618e-04, 9.9977e-01],
        [5.6722e-05, 9.9994e-01],
        [5.4739e-08, 1.0000e+00],
        [8.6292e-05, 9.9991e-01],
        [4.7720e-08, 1.0000e+00],
        [1.0000e+00, 7.5730e-07],
        [8.7250e-01, 1.2750e-01],
        [3.7738e-05, 9.9996e-01],
        [2.2806e-06, 1.0000e+00],
        [2.9704e-05, 9.9997e-01],
        [9.9903e-01, 9.7148e-04],
        [1.4681e-02, 9.8532e-01],
        [1.8110e-05, 9.9998e-01],
        [3.2843e-04, 9.9967e-01],
        [7.3724e-01, 2.6276e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9940e-01, 5.9649e-04],
        [9.9992e-01, 7.5092e-05],
        [2.8238e-07, 1.0000e+00],
        [8.4516e-05, 9.9992e-01],
        [7.3510e-03, 9.9265e-01],
        [2.0530e-05, 9.9998e-01],
        [1.7983e-04, 9.9982e-01],
        [4.6747e-02, 9.5325e-01],
        [7.4286e-05, 9.9993e-01],
        [1.3559e-02, 9.8644e-01],
        [3.8297e-07, 1.0000e+00],
        [1.8550e-05, 9.9998e-01],
        [8.8243e-06, 9.9999e-01],
        [9.0756e-01, 9.2441e-02],
        [4.9399e-02, 9.5060e-01],
        [5.7854e-06, 9.9999e-01],
        [3.5143e-07, 1.0000e+00],
        [6.1899e-05, 9.9994e-01],
        [1.2788e-04, 9.9987e-01],
        [1.1587e-02, 9.8841e-01],
        [1.1937e-04, 9.9988e-01],
        [5.2062e-05, 9.9995e-01],
        [9.9941e-01, 5.9386e-04],
        [7.4601e-07, 1.0000e+00],
        [9.9723e-01, 2.7720e-03],
        [1.7794e-03, 9.9822e-01],
        [7.7344e-04, 9.9923e-01],
        [2.7351e-02, 9.7265e-01],
        [1.2135e-03, 9.9879e-01],
        [2.3855e-03, 9.9761e-01],
        [1.9693e-03, 9.9803e-01],
        [1.9521e-04, 9.9980e-01],
        [2.4246e-04, 9.9976e-01],
        [3.1117e-04, 9.9969e-01],
        [2.6136e-06, 1.0000e+00],
        [4.8169e-03, 9.9518e-01],
        [3.7850e-07, 1.0000e+00],
        [2.4270e-04, 9.9976e-01],
        [5.5637e-04, 9.9944e-01],
        [6.9282e-05, 9.9993e-01],
        [1.2697e-07, 1.0000e+00],
        [8.0191e-07, 1.0000e+00],
        [9.9790e-01, 2.1043e-03],
        [3.2353e-03, 9.9676e-01],
        [3.4679e-04, 9.9965e-01],
        [9.9953e-01, 4.7162e-04],
        [2.6882e-04, 9.9973e-01],
        [6.2115e-05, 9.9994e-01],
        [9.6188e-01, 3.8124e-02],
        [1.3746e-03, 9.9863e-01],
        [9.9967e-01, 3.3257e-04],
        [8.4828e-03, 9.9152e-01],
        [5.7558e-08, 1.0000e+00],
        [5.0179e-04, 9.9950e-01],
        [9.1324e-05, 9.9991e-01],
        [9.9432e-01, 5.6800e-03],
        [2.2411e-02, 9.7759e-01],
        [2.7486e-07, 1.0000e+00],
        [7.4210e-04, 9.9926e-01],
        [4.3235e-06, 1.0000e+00],
        [1.8069e-05, 9.9998e-01],
        [5.4352e-03, 9.9456e-01],
        [3.7431e-02, 9.6257e-01],
        [1.8550e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.9605e-05, 9.9994e-01],
        [8.8753e-02, 9.1125e-01],
        [2.6952e-04, 9.9973e-01],
        [3.4359e-02, 9.6564e-01],
        [1.4417e-04, 9.9986e-01],
        [7.3675e-04, 9.9926e-01],
        [4.4355e-05, 9.9996e-01],
        [9.9998e-01, 2.1138e-05],
        [1.1317e-01, 8.8683e-01],
        [7.5688e-01, 2.4312e-01],
        [2.9503e-03, 9.9705e-01],
        [6.2639e-02, 9.3736e-01],
        [9.9976e-01, 2.3705e-04],
        [9.9806e-02, 9.0019e-01],
        [3.3016e-06, 1.0000e+00],
        [1.1793e-06, 1.0000e+00],
        [2.3987e-03, 9.9760e-01],
        [1.7211e-05, 9.9998e-01],
        [9.9952e-01, 4.7736e-04],
        [1.4486e-06, 1.0000e+00],
        [9.9505e-01, 4.9476e-03],
        [9.9975e-01, 2.5233e-04],
        [8.9883e-01, 1.0117e-01],
        [1.4371e-05, 9.9999e-01],
        [3.0073e-05, 9.9997e-01],
        [5.4020e-04, 9.9946e-01],
        [7.1706e-06, 9.9999e-01],
        [5.9151e-05, 9.9994e-01],
        [5.6458e-03, 9.9435e-01],
        [1.0809e-05, 9.9999e-01],
        [9.9987e-01, 1.2542e-04],
        [8.2344e-04, 9.9918e-01],
        [9.1556e-05, 9.9991e-01],
        [1.0601e-03, 9.9894e-01],
        [8.5122e-03, 9.9149e-01],
        [9.8126e-08, 1.0000e+00],
        [1.0432e-04, 9.9990e-01],
        [9.6638e-01, 3.3624e-02],
        [9.9971e-01, 2.8946e-04],
        [9.9979e-01, 2.0978e-04],
        [1.0720e-06, 1.0000e+00],
        [9.3262e-01, 6.7379e-02],
        [9.9857e-01, 1.4283e-03],
        [1.2978e-04, 9.9987e-01],
        [9.2853e-04, 9.9907e-01],
        [9.5012e-03, 9.9050e-01],
        [2.6309e-01, 7.3691e-01],
        [5.2732e-02, 9.4727e-01],
        [8.3286e-01, 1.6714e-01],
        [9.0734e-06, 9.9999e-01],
        [2.3253e-07, 1.0000e+00],
        [3.5958e-05, 9.9996e-01],
        [5.4702e-05, 9.9995e-01],
        [2.5634e-04, 9.9974e-01],
        [1.3598e-05, 9.9999e-01],
        [3.0207e-06, 1.0000e+00],
        [3.9519e-06, 1.0000e+00],
        [9.8523e-01, 1.4771e-02],
        [3.3653e-06, 1.0000e+00],
        [7.2621e-04, 9.9927e-01],
        [2.2134e-02, 9.7787e-01],
        [7.1204e-07, 1.0000e+00],
        [2.5972e-02, 9.7403e-01],
        [9.9882e-01, 1.1836e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9969e-01, 3.0952e-04],
        [9.9946e-01, 5.4263e-04],
        [1.3009e-04, 9.9987e-01],
        [4.4955e-05, 9.9996e-01],
        [9.4759e-05, 9.9991e-01],
        [1.0607e-03, 9.9894e-01],
        [1.6622e-03, 9.9834e-01],
        [2.1857e-03, 9.9781e-01],
        [6.9778e-05, 9.9993e-01],
        [2.4452e-03, 9.9755e-01],
        [9.5895e-07, 1.0000e+00],
        [9.9995e-01, 5.1614e-05],
        [2.4016e-05, 9.9998e-01],
        [8.2600e-07, 1.0000e+00],
        [9.9985e-01, 1.5039e-04],
        [2.9662e-05, 9.9997e-01],
        [8.2757e-03, 9.9172e-01],
        [2.6354e-06, 1.0000e+00],
        [6.1835e-08, 1.0000e+00],
        [5.6060e-06, 9.9999e-01],
        [1.7362e-07, 1.0000e+00],
        [6.4340e-07, 1.0000e+00],
        [1.4033e-05, 9.9999e-01],
        [2.9670e-04, 9.9970e-01],
        [6.9680e-03, 9.9303e-01],
        [9.9348e-01, 6.5161e-03],
        [9.9967e-01, 3.3406e-04],
        [9.9965e-01, 3.5351e-04],
        [1.3084e-03, 9.9869e-01],
        [5.9066e-07, 1.0000e+00],
        [3.4122e-06, 1.0000e+00],
        [5.5547e-05, 9.9994e-01],
        [6.2868e-02, 9.3713e-01],
        [1.6290e-03, 9.9837e-01],
        [1.3044e-05, 9.9999e-01],
        [8.6618e-07, 1.0000e+00],
        [1.0997e-03, 9.9890e-01],
        [1.3282e-06, 1.0000e+00],
        [1.9034e-03, 9.9810e-01],
        [3.1314e-02, 9.6869e-01],
        [3.5573e-03, 9.9644e-01],
        [9.9962e-01, 3.7807e-04],
        [3.5456e-05, 9.9996e-01],
        [1.7670e-05, 9.9998e-01],
        [9.9987e-01, 1.2705e-04],
        [8.3035e-05, 9.9992e-01],
        [9.9874e-01, 1.2558e-03],
        [9.9989e-01, 1.1261e-04],
        [1.1979e-02, 9.8802e-01],
        [4.7265e-06, 1.0000e+00],
        [4.6088e-01, 5.3912e-01],
        [1.7805e-04, 9.9982e-01],
        [9.9969e-01, 3.1408e-04],
        [9.8661e-01, 1.3389e-02],
        [9.9818e-01, 1.8227e-03],
        [4.5443e-05, 9.9995e-01],
        [4.4854e-03, 9.9551e-01],
        [7.5463e-01, 2.4537e-01],
        [9.9970e-01, 3.0223e-04],
        [4.1219e-03, 9.9588e-01],
        [9.9986e-01, 1.3678e-04],
        [5.2755e-02, 9.4725e-01],
        [2.1933e-04, 9.9978e-01],
        [2.3195e-04, 9.9977e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.3041e-05, 9.9997e-01],
        [4.2817e-08, 1.0000e+00],
        [3.2838e-06, 1.0000e+00],
        [1.0344e-04, 9.9990e-01],
        [9.9766e-01, 2.3429e-03],
        [1.9533e-03, 9.9805e-01],
        [4.1213e-02, 9.5879e-01],
        [7.6544e-02, 9.2346e-01],
        [1.2420e-05, 9.9999e-01],
        [1.2390e-02, 9.8761e-01],
        [9.9908e-01, 9.1693e-04],
        [2.3900e-04, 9.9976e-01],
        [9.4358e-08, 1.0000e+00],
        [9.9977e-01, 2.3014e-04],
        [1.6495e-01, 8.3505e-01],
        [1.2591e-06, 1.0000e+00],
        [9.9978e-01, 2.1632e-04],
        [1.0922e-05, 9.9999e-01],
        [1.6597e-03, 9.9834e-01],
        [7.2040e-07, 1.0000e+00],
        [1.1282e-06, 1.0000e+00],
        [1.0397e-06, 1.0000e+00],
        [4.3385e-03, 9.9566e-01],
        [5.9095e-04, 9.9941e-01],
        [9.9817e-01, 1.8315e-03],
        [2.3080e-01, 7.6920e-01],
        [9.6474e-01, 3.5261e-02],
        [9.9979e-01, 2.0929e-04],
        [9.9966e-01, 3.3529e-04],
        [3.5710e-03, 9.9643e-01],
        [8.1938e-03, 9.9181e-01],
        [8.7275e-04, 9.9913e-01],
        [1.5033e-04, 9.9985e-01],
        [1.7085e-04, 9.9983e-01],
        [1.3140e-03, 9.9869e-01],
        [9.9985e-01, 1.4985e-04],
        [9.9903e-01, 9.6547e-04],
        [9.9952e-01, 4.7828e-04],
        [4.5667e-06, 1.0000e+00],
        [3.2556e-05, 9.9997e-01],
        [1.4507e-01, 8.5493e-01],
        [9.9650e-01, 3.4978e-03],
        [9.2833e-01, 7.1670e-02],
        [1.3118e-04, 9.9987e-01],
        [9.9975e-01, 2.4584e-04],
        [2.4174e-04, 9.9976e-01],
        [9.9956e-01, 4.4383e-04],
        [1.0380e-04, 9.9990e-01],
        [8.8835e-02, 9.1117e-01],
        [1.5708e-04, 9.9984e-01],
        [9.9875e-01, 1.2477e-03],
        [1.3933e-07, 1.0000e+00],
        [2.3768e-06, 1.0000e+00],
        [8.0874e-03, 9.9191e-01],
        [6.7718e-03, 9.9323e-01],
        [2.2711e-04, 9.9977e-01],
        [2.0038e-05, 9.9998e-01],
        [2.2117e-09, 1.0000e+00],
        [6.4739e-04, 9.9935e-01],
        [9.7005e-01, 2.9954e-02],
        [2.2288e-03, 9.9777e-01],
        [5.0269e-07, 1.0000e+00],
        [9.8951e-01, 1.0490e-02],
        [9.7243e-01, 2.7568e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.4977e-05, 9.9990e-01],
        [2.1134e-04, 9.9979e-01],
        [9.9914e-01, 8.6282e-04],
        [2.4604e-04, 9.9975e-01],
        [9.9968e-01, 3.1654e-04],
        [1.0738e-06, 1.0000e+00],
        [9.9983e-01, 1.6530e-04],
        [1.3954e-04, 9.9986e-01],
        [1.0403e-05, 9.9999e-01],
        [4.5470e-04, 9.9955e-01],
        [1.5441e-04, 9.9985e-01],
        [1.1065e-03, 9.9889e-01],
        [1.2577e-05, 9.9999e-01],
        [5.8915e-07, 1.0000e+00],
        [1.0886e-04, 9.9989e-01],
        [9.9795e-01, 2.0532e-03],
        [1.3694e-01, 8.6306e-01],
        [8.6834e-01, 1.3166e-01],
        [9.9499e-01, 5.0123e-03],
        [7.0144e-05, 9.9993e-01],
        [9.0767e-03, 9.9092e-01],
        [1.6279e-05, 9.9998e-01],
        [4.1103e-06, 1.0000e+00],
        [1.1702e-01, 8.8298e-01],
        [5.6802e-07, 1.0000e+00],
        [9.9921e-01, 7.8574e-04],
        [4.0199e-08, 1.0000e+00],
        [9.9999e-01, 6.7407e-06],
        [9.9846e-01, 1.5425e-03],
        [1.4796e-04, 9.9985e-01],
        [2.7607e-05, 9.9997e-01],
        [3.6300e-06, 1.0000e+00],
        [1.2884e-04, 9.9987e-01],
        [9.9990e-01, 9.7430e-05],
        [7.4653e-03, 9.9253e-01],
        [8.7208e-05, 9.9991e-01],
        [5.2686e-03, 9.9473e-01],
        [6.2271e-01, 3.7729e-01],
        [4.1088e-05, 9.9996e-01],
        [9.9934e-01, 6.5716e-04],
        [9.9545e-01, 4.5469e-03],
        [9.9826e-01, 1.7433e-03],
        [1.9585e-02, 9.8041e-01],
        [9.9846e-01, 1.5421e-03],
        [2.5414e-06, 1.0000e+00],
        [5.1888e-06, 9.9999e-01],
        [4.6784e-14, 1.0000e+00],
        [3.6239e-03, 9.9638e-01],
        [5.2288e-04, 9.9948e-01],
        [3.1881e-04, 9.9968e-01],
        [8.9340e-01, 1.0660e-01],
        [5.1559e-04, 9.9948e-01],
        [2.2076e-07, 1.0000e+00],
        [1.1183e-03, 9.9888e-01],
        [1.1451e-08, 1.0000e+00],
        [9.5641e-05, 9.9990e-01],
        [9.9984e-01, 1.5792e-04],
        [2.5103e-06, 1.0000e+00],
        [9.9973e-01, 2.7134e-04],
        [3.2275e-06, 1.0000e+00],
        [1.1690e-07, 1.0000e+00],
        [7.5390e-07, 1.0000e+00],
        [3.4200e-04, 9.9966e-01],
        [3.4291e-04, 9.9966e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9910e-01, 9.0363e-04],
        [5.7194e-02, 9.4281e-01],
        [5.2726e-03, 9.9473e-01],
        [3.8236e-02, 9.6176e-01],
        [5.9528e-06, 9.9999e-01],
        [9.9681e-01, 3.1916e-03],
        [7.0000e-05, 9.9993e-01],
        [9.7934e-01, 2.0660e-02],
        [7.1316e-02, 9.2868e-01],
        [1.5344e-04, 9.9985e-01],
        [5.0580e-08, 1.0000e+00],
        [7.9214e-05, 9.9992e-01],
        [1.2873e-05, 9.9999e-01],
        [1.2323e-05, 9.9999e-01],
        [3.5995e-05, 9.9996e-01],
        [1.4069e-05, 9.9999e-01],
        [4.9593e-07, 1.0000e+00],
        [1.5402e-05, 9.9998e-01],
        [3.0629e-01, 6.9371e-01],
        [3.0651e-03, 9.9693e-01],
        [1.6057e-04, 9.9984e-01],
        [1.8339e-05, 9.9998e-01],
        [9.9967e-01, 3.3469e-04],
        [4.1534e-01, 5.8466e-01],
        [1.4017e-06, 1.0000e+00],
        [9.1198e-01, 8.8021e-02],
        [3.0296e-07, 1.0000e+00],
        [1.1804e-06, 1.0000e+00],
        [2.7758e-01, 7.2242e-01],
        [9.9894e-01, 1.0581e-03],
        [9.9968e-01, 3.2385e-04],
        [1.2438e-03, 9.9876e-01],
        [1.6342e-03, 9.9837e-01],
        [9.9812e-01, 1.8766e-03],
        [1.4909e-07, 1.0000e+00],
        [1.2646e-03, 9.9874e-01],
        [9.9973e-01, 2.6590e-04],
        [2.5839e-06, 1.0000e+00],
        [1.8900e-06, 1.0000e+00],
        [8.8377e-06, 9.9999e-01],
        [1.6818e-03, 9.9832e-01],
        [2.3142e-04, 9.9977e-01],
        [7.0865e-04, 9.9929e-01],
        [9.9996e-01, 4.2477e-05],
        [1.2027e-03, 9.9880e-01],
        [9.9980e-01, 2.0111e-04],
        [8.8248e-01, 1.1752e-01],
        [6.8291e-05, 9.9993e-01],
        [5.9728e-03, 9.9403e-01],
        [9.9136e-01, 8.6397e-03],
        [4.7285e-06, 1.0000e+00],
        [1.4629e-04, 9.9985e-01],
        [9.9958e-01, 4.2411e-04],
        [1.5939e-03, 9.9841e-01],
        [2.0957e-03, 9.9790e-01],
        [2.1186e-07, 1.0000e+00],
        [2.7379e-06, 1.0000e+00],
        [9.9929e-01, 7.1219e-04],
        [7.4987e-06, 9.9999e-01],
        [9.9198e-01, 8.0181e-03],
        [5.5844e-06, 9.9999e-01],
        [9.9905e-01, 9.5196e-04],
        [9.9985e-01, 1.4834e-04],
        [7.8174e-02, 9.2183e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.8626e-01, 7.1374e-01],
        [1.0394e-04, 9.9990e-01],
        [2.7741e-04, 9.9972e-01],
        [6.5391e-04, 9.9935e-01],
        [9.9944e-01, 5.5994e-04],
        [9.9955e-01, 4.5279e-04],
        [2.5848e-01, 7.4152e-01],
        [1.4595e-07, 1.0000e+00],
        [2.5616e-07, 1.0000e+00],
        [5.7344e-02, 9.4266e-01],
        [2.5043e-01, 7.4957e-01],
        [9.7136e-01, 2.8640e-02],
        [2.2889e-03, 9.9771e-01],
        [3.1707e-07, 1.0000e+00],
        [1.4228e-03, 9.9858e-01],
        [9.9391e-01, 6.0887e-03],
        [9.1494e-05, 9.9991e-01],
        [9.9965e-01, 3.5416e-04],
        [1.2718e-06, 1.0000e+00],
        [4.6109e-08, 1.0000e+00],
        [1.1591e-05, 9.9999e-01],
        [9.9964e-01, 3.6469e-04],
        [4.4346e-06, 1.0000e+00],
        [1.2736e-05, 9.9999e-01],
        [1.6003e-01, 8.3997e-01],
        [2.9254e-02, 9.7075e-01],
        [1.6419e-01, 8.3581e-01],
        [2.7812e-07, 1.0000e+00],
        [9.9952e-01, 4.8364e-04],
        [9.8513e-01, 1.4870e-02],
        [7.6951e-06, 9.9999e-01],
        [9.9966e-01, 3.4005e-04],
        [4.4948e-07, 1.0000e+00],
        [2.4798e-02, 9.7520e-01],
        [2.0793e-04, 9.9979e-01],
        [9.9838e-01, 1.6232e-03],
        [9.9936e-01, 6.4398e-04],
        [9.9987e-01, 1.3436e-04],
        [2.1461e-02, 9.7854e-01],
        [9.9693e-01, 3.0716e-03],
        [6.7041e-06, 9.9999e-01],
        [1.4750e-03, 9.9853e-01],
        [1.8918e-01, 8.1082e-01],
        [2.1756e-05, 9.9998e-01],
        [2.0290e-05, 9.9998e-01],
        [9.9801e-01, 1.9912e-03],
        [7.7527e-05, 9.9992e-01],
        [9.1299e-05, 9.9991e-01],
        [3.6849e-04, 9.9963e-01],
        [9.1010e-01, 8.9900e-02],
        [9.9956e-01, 4.4261e-04],
        [4.0491e-03, 9.9595e-01],
        [7.5422e-08, 1.0000e+00],
        [9.9969e-01, 3.0773e-04],
        [1.8995e-04, 9.9981e-01],
        [1.0409e-01, 8.9591e-01],
        [3.3666e-07, 1.0000e+00],
        [7.4381e-04, 9.9926e-01],
        [1.4584e-06, 1.0000e+00],
        [9.1158e-06, 9.9999e-01],
        [2.0598e-03, 9.9794e-01],
        [9.9700e-01, 3.0042e-03],
        [1.1652e-01, 8.8348e-01],
        [1.3072e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9990e-01, 1.0091e-04],
        [2.6635e-05, 9.9997e-01],
        [4.2712e-04, 9.9957e-01],
        [7.7236e-04, 9.9923e-01],
        [3.1917e-04, 9.9968e-01],
        [9.9219e-01, 7.8109e-03],
        [1.9601e-04, 9.9980e-01],
        [9.9920e-01, 8.0257e-04],
        [1.2329e-04, 9.9988e-01],
        [3.9686e-04, 9.9960e-01],
        [9.5102e-09, 1.0000e+00],
        [9.9967e-01, 3.2744e-04],
        [4.4165e-05, 9.9996e-01],
        [9.9917e-01, 8.3078e-04],
        [9.9713e-01, 2.8678e-03],
        [7.4245e-05, 9.9993e-01],
        [6.8785e-07, 1.0000e+00],
        [4.0789e-05, 9.9996e-01],
        [7.5849e-01, 2.4151e-01],
        [2.6815e-04, 9.9973e-01],
        [1.2398e-02, 9.8760e-01],
        [3.1266e-04, 9.9969e-01],
        [9.9884e-01, 1.1576e-03],
        [6.4514e-05, 9.9994e-01],
        [9.9800e-01, 1.9973e-03],
        [2.3961e-05, 9.9998e-01],
        [9.9866e-01, 1.3424e-03],
        [3.9057e-07, 1.0000e+00],
        [2.3743e-06, 1.0000e+00],
        [7.9388e-05, 9.9992e-01],
        [9.9900e-01, 1.0023e-03],
        [1.1965e-06, 1.0000e+00],
        [3.8541e-03, 9.9615e-01],
        [5.0381e-04, 9.9950e-01],
        [9.9962e-01, 3.7899e-04],
        [3.9682e-08, 1.0000e+00],
        [1.1505e-02, 9.8850e-01],
        [9.6925e-06, 9.9999e-01],
        [5.0884e-08, 1.0000e+00],
        [9.8621e-01, 1.3791e-02],
        [9.9943e-01, 5.7158e-04],
        [9.3766e-01, 6.2344e-02],
        [3.1238e-05, 9.9997e-01],
        [9.1755e-03, 9.9082e-01],
        [1.5662e-06, 1.0000e+00],
        [5.0797e-03, 9.9492e-01],
        [1.5883e-04, 9.9984e-01],
        [5.8692e-03, 9.9413e-01],
        [1.4142e-04, 9.9986e-01],
        [3.8311e-06, 1.0000e+00],
        [7.4500e-04, 9.9926e-01],
        [1.2858e-04, 9.9987e-01],
        [9.9971e-01, 2.9311e-04],
        [2.2026e-06, 1.0000e+00],
        [2.0728e-03, 9.9793e-01],
        [6.9473e-05, 9.9993e-01],
        [3.9703e-06, 1.0000e+00],
        [9.9814e-01, 1.8603e-03],
        [1.2951e-03, 9.9870e-01],
        [1.1139e-03, 9.9889e-01],
        [8.8349e-04, 9.9912e-01],
        [7.5640e-03, 9.9244e-01],
        [4.5189e-03, 9.9548e-01],
        [8.8495e-05, 9.9991e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2991e-05, 9.9998e-01],
        [4.8999e-07, 1.0000e+00],
        [4.5841e-05, 9.9995e-01],
        [3.8085e-02, 9.6192e-01],
        [1.0654e-04, 9.9989e-01],
        [1.2340e-05, 9.9999e-01],
        [8.2188e-05, 9.9992e-01],
        [2.4997e-04, 9.9975e-01],
        [9.9812e-01, 1.8785e-03],
        [2.2368e-05, 9.9998e-01],
        [2.5227e-06, 1.0000e+00],
        [9.9996e-01, 4.4550e-05],
        [6.2796e-08, 1.0000e+00],
        [3.9748e-04, 9.9960e-01],
        [4.6043e-07, 1.0000e+00],
        [9.9944e-01, 5.6485e-04],
        [7.8858e-05, 9.9992e-01],
        [1.4784e-06, 1.0000e+00],
        [1.2713e-05, 9.9999e-01],
        [9.2043e-01, 7.9567e-02],
        [2.1186e-05, 9.9998e-01],
        [4.5566e-05, 9.9995e-01],
        [9.9537e-01, 4.6269e-03],
        [2.0994e-03, 9.9790e-01],
        [9.8110e-07, 1.0000e+00],
        [1.2425e-05, 9.9999e-01],
        [9.9949e-01, 5.0523e-04],
        [8.1335e-03, 9.9187e-01],
        [4.1122e-06, 1.0000e+00],
        [9.8437e-01, 1.5628e-02],
        [1.3918e-06, 1.0000e+00],
        [7.2458e-06, 9.9999e-01],
        [7.9052e-06, 9.9999e-01],
        [1.8673e-02, 9.8133e-01],
        [9.9997e-01, 2.9647e-05],
        [1.4472e-01, 8.5528e-01],
        [2.3169e-03, 9.9768e-01],
        [5.1866e-03, 9.9481e-01],
        [2.8218e-06, 1.0000e+00],
        [7.2953e-06, 9.9999e-01],
        [5.9646e-07, 1.0000e+00],
        [2.1361e-06, 1.0000e+00],
        [9.9947e-01, 5.2528e-04],
        [2.7758e-08, 1.0000e+00],
        [5.3594e-06, 9.9999e-01],
        [6.4979e-04, 9.9935e-01],
        [9.5021e-01, 4.9793e-02],
        [1.3253e-03, 9.9867e-01],
        [5.5867e-06, 9.9999e-01],
        [5.3328e-03, 9.9467e-01],
        [1.0467e-02, 9.8953e-01],
        [9.9940e-01, 6.0236e-04],
        [9.9918e-01, 8.2178e-04],
        [1.3342e-08, 1.0000e+00],
        [4.6826e-06, 1.0000e+00],
        [2.3190e-02, 9.7681e-01],
        [3.5872e-06, 1.0000e+00],
        [9.9020e-01, 9.7982e-03],
        [2.0819e-06, 1.0000e+00],
        [2.4173e-04, 9.9976e-01],
        [1.3272e-06, 1.0000e+00],
        [1.1319e-07, 1.0000e+00],
        [1.0780e-01, 8.9220e-01],
        [2.3763e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5525e-05, 9.9995e-01],
        [1.2129e-04, 9.9988e-01],
        [9.9970e-01, 2.9906e-04],
        [9.9971e-01, 2.9007e-04],
        [9.9029e-01, 9.7082e-03],
        [1.1479e-07, 1.0000e+00],
        [9.9944e-01, 5.5935e-04],
        [9.9845e-01, 1.5522e-03],
        [9.7823e-01, 2.1767e-02],
        [1.3171e-03, 9.9868e-01],
        [1.0212e-06, 1.0000e+00],
        [1.6383e-03, 9.9836e-01],
        [6.7172e-06, 9.9999e-01],
        [2.2425e-08, 1.0000e+00],
        [6.2882e-05, 9.9994e-01],
        [1.4243e-05, 9.9999e-01],
        [3.6279e-05, 9.9996e-01],
        [2.1062e-04, 9.9979e-01],
        [9.9982e-01, 1.8031e-04],
        [3.1386e-03, 9.9686e-01],
        [4.5669e-05, 9.9995e-01],
        [1.0022e-06, 1.0000e+00],
        [1.4064e-04, 9.9986e-01],
        [2.6032e-05, 9.9997e-01],
        [5.5945e-04, 9.9944e-01],
        [4.4506e-04, 9.9955e-01],
        [7.6754e-07, 1.0000e+00],
        [1.6401e-03, 9.9836e-01],
        [4.9997e-05, 9.9995e-01],
        [1.8422e-03, 9.9816e-01],
        [8.0381e-06, 9.9999e-01],
        [9.9911e-01, 8.9476e-04],
        [9.9984e-01, 1.5671e-04],
        [1.9993e-05, 9.9998e-01],
        [9.8902e-01, 1.0976e-02],
        [3.9643e-01, 6.0357e-01],
        [9.9998e-01, 1.9326e-05],
        [1.1873e-02, 9.8813e-01],
        [9.9933e-01, 6.7355e-04],
        [2.1312e-06, 1.0000e+00],
        [4.6793e-05, 9.9995e-01],
        [1.9670e-06, 1.0000e+00],
        [6.8503e-04, 9.9932e-01],
        [1.6543e-06, 1.0000e+00],
        [1.1962e-05, 9.9999e-01],
        [9.3804e-01, 6.1956e-02],
        [4.2191e-06, 1.0000e+00],
        [8.1113e-01, 1.8887e-01],
        [9.8921e-01, 1.0792e-02],
        [9.8156e-01, 1.8441e-02],
        [7.1535e-08, 1.0000e+00],
        [9.6844e-01, 3.1564e-02],
        [1.4410e-05, 9.9999e-01],
        [9.9929e-01, 7.1196e-04],
        [9.9942e-01, 5.8128e-04],
        [9.7506e-05, 9.9990e-01],
        [9.9794e-01, 2.0607e-03],
        [4.4421e-05, 9.9996e-01],
        [5.2401e-01, 4.7599e-01],
        [1.4833e-04, 9.9985e-01],
        [4.1116e-07, 1.0000e+00],
        [3.8641e-06, 1.0000e+00],
        [7.1816e-04, 9.9928e-01],
        [5.7816e-05, 9.9994e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.9026e-06, 9.9999e-01],
        [9.9920e-01, 7.9510e-04],
        [9.9902e-01, 9.8377e-04],
        [9.6008e-01, 3.9921e-02],
        [5.2987e-06, 9.9999e-01],
        [1.0718e-09, 1.0000e+00],
        [9.9863e-01, 1.3701e-03],
        [2.6521e-06, 1.0000e+00],
        [1.4902e-08, 1.0000e+00],
        [1.6417e-07, 1.0000e+00],
        [9.9971e-01, 2.9005e-04],
        [6.5037e-07, 1.0000e+00],
        [5.5024e-04, 9.9945e-01],
        [2.7359e-06, 1.0000e+00],
        [2.7671e-06, 1.0000e+00],
        [7.5066e-06, 9.9999e-01],
        [1.3501e-07, 1.0000e+00],
        [1.8321e-02, 9.8168e-01],
        [5.2156e-03, 9.9478e-01],
        [9.2380e-01, 7.6202e-02],
        [2.7851e-06, 1.0000e+00],
        [1.4828e-07, 1.0000e+00],
        [3.7988e-04, 9.9962e-01],
        [7.6556e-04, 9.9923e-01],
        [2.0429e-08, 1.0000e+00],
        [1.1266e-07, 1.0000e+00],
        [2.1154e-08, 1.0000e+00],
        [3.1229e-05, 9.9997e-01],
        [5.7035e-06, 9.9999e-01],
        [4.1304e-01, 5.8696e-01],
        [1.5798e-04, 9.9984e-01],
        [3.6408e-07, 1.0000e+00],
        [9.7330e-01, 2.6699e-02],
        [9.9679e-01, 3.2057e-03],
        [9.4288e-01, 5.7120e-02],
        [9.6359e-06, 9.9999e-01],
        [1.0593e-05, 9.9999e-01],
        [9.9764e-01, 2.3616e-03],
        [1.1591e-04, 9.9988e-01],
        [1.0149e-03, 9.9899e-01],
        [4.3496e-03, 9.9565e-01],
        [9.9384e-01, 6.1566e-03],
        [9.9945e-01, 5.4839e-04],
        [9.6613e-01, 3.3870e-02],
        [2.0110e-01, 7.9890e-01],
        [4.6289e-04, 9.9954e-01],
        [7.7980e-05, 9.9992e-01],
        [9.9966e-01, 3.3756e-04],
        [9.9933e-01, 6.7067e-04],
        [9.9967e-01, 3.3379e-04],
        [1.2317e-07, 1.0000e+00],
        [9.9952e-01, 4.8375e-04],
        [8.9979e-02, 9.1002e-01],
        [9.7222e-01, 2.7776e-02],
        [4.5081e-05, 9.9995e-01],
        [2.9210e-03, 9.9708e-01],
        [4.0649e-08, 1.0000e+00],
        [1.7459e-03, 9.9825e-01],
        [1.1068e-04, 9.9989e-01],
        [9.9950e-01, 5.0468e-04],
        [9.9978e-05, 9.9990e-01],
        [1.6417e-05, 9.9998e-01],
        [4.8804e-06, 1.0000e+00],
        [1.8464e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.5537e-05, 9.9994e-01],
        [5.8159e-05, 9.9994e-01],
        [2.1689e-07, 1.0000e+00],
        [1.0094e-06, 1.0000e+00],
        [6.2040e-04, 9.9938e-01],
        [3.2998e-02, 9.6700e-01],
        [9.9830e-01, 1.7049e-03],
        [7.8248e-11, 1.0000e+00],
        [9.9407e-01, 5.9293e-03],
        [8.9240e-01, 1.0760e-01],
        [2.7437e-05, 9.9997e-01],
        [4.5653e-06, 1.0000e+00],
        [9.0995e-05, 9.9991e-01],
        [7.2884e-05, 9.9993e-01],
        [9.9972e-01, 2.7671e-04],
        [9.9944e-01, 5.6434e-04],
        [1.0756e-04, 9.9989e-01],
        [8.8836e-08, 1.0000e+00],
        [6.6136e-06, 9.9999e-01],
        [1.6314e-05, 9.9998e-01],
        [9.9953e-01, 4.7452e-04],
        [1.1786e-05, 9.9999e-01],
        [1.4596e-07, 1.0000e+00],
        [2.3473e-06, 1.0000e+00],
        [9.9956e-01, 4.3618e-04],
        [9.9591e-01, 4.0894e-03],
        [1.2199e-05, 9.9999e-01],
        [3.0150e-04, 9.9970e-01],
        [7.7627e-06, 9.9999e-01],
        [9.9808e-01, 1.9225e-03],
        [5.1121e-08, 1.0000e+00],
        [7.6821e-08, 1.0000e+00],
        [9.6287e-01, 3.7133e-02],
        [2.3370e-04, 9.9977e-01],
        [2.1960e-07, 1.0000e+00],
        [9.8875e-01, 1.1251e-02],
        [5.5639e-05, 9.9994e-01],
        [6.0111e-01, 3.9889e-01],
        [9.9911e-01, 8.9425e-04],
        [1.1532e-02, 9.8847e-01],
        [4.2699e-06, 1.0000e+00],
        [8.2011e-04, 9.9918e-01],
        [8.1948e-01, 1.8052e-01],
        [7.9164e-01, 2.0836e-01],
        [9.9958e-01, 4.2196e-04],
        [7.7735e-03, 9.9223e-01],
        [1.4377e-07, 1.0000e+00],
        [7.1096e-06, 9.9999e-01],
        [9.9960e-01, 3.9648e-04],
        [9.9290e-01, 7.0974e-03],
        [7.6111e-03, 9.9239e-01],
        [1.7646e-01, 8.2354e-01],
        [5.6399e-05, 9.9994e-01],
        [4.9348e-02, 9.5065e-01],
        [3.3464e-06, 1.0000e+00],
        [1.2344e-04, 9.9988e-01],
        [3.9386e-05, 9.9996e-01],
        [9.9860e-01, 1.4044e-03],
        [1.3172e-07, 1.0000e+00],
        [2.6998e-08, 1.0000e+00],
        [9.9889e-01, 1.1124e-03],
        [9.7971e-01, 2.0293e-02],
        [1.6548e-05, 9.9998e-01],
        [6.2017e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.6238e-06, 9.9999e-01],
        [9.9943e-01, 5.6952e-04],
        [3.1469e-03, 9.9685e-01],
        [1.4898e-05, 9.9999e-01],
        [2.1115e-06, 1.0000e+00],
        [7.4786e-08, 1.0000e+00],
        [7.8401e-07, 1.0000e+00],
        [9.8892e-01, 1.1077e-02],
        [6.5328e-05, 9.9993e-01],
        [9.9774e-01, 2.2642e-03],
        [7.1582e-04, 9.9928e-01],
        [1.3813e-01, 8.6187e-01],
        [9.9614e-01, 3.8649e-03],
        [1.4421e-03, 9.9856e-01],
        [1.5217e-07, 1.0000e+00],
        [1.3145e-05, 9.9999e-01],
        [1.2135e-05, 9.9999e-01],
        [9.4729e-05, 9.9991e-01],
        [1.4343e-07, 1.0000e+00],
        [1.1410e-04, 9.9989e-01],
        [9.6468e-05, 9.9990e-01],
        [7.3054e-02, 9.2695e-01],
        [9.9892e-01, 1.0773e-03],
        [1.0371e-06, 1.0000e+00],
        [1.0757e-07, 1.0000e+00],
        [3.3143e-04, 9.9967e-01],
        [9.9939e-01, 6.1340e-04],
        [2.4119e-04, 9.9976e-01],
        [7.4129e-06, 9.9999e-01],
        [9.9958e-01, 4.1656e-04],
        [9.9979e-01, 2.0696e-04],
        [5.1542e-07, 1.0000e+00],
        [1.2224e-06, 1.0000e+00],
        [4.0046e-09, 1.0000e+00],
        [1.4878e-04, 9.9985e-01],
        [4.7234e-02, 9.5277e-01],
        [4.6937e-03, 9.9531e-01],
        [1.6179e-06, 1.0000e+00],
        [9.6666e-01, 3.3336e-02],
        [7.0841e-05, 9.9993e-01],
        [3.4564e-08, 1.0000e+00],
        [6.0378e-07, 1.0000e+00],
        [3.3768e-05, 9.9997e-01],
        [9.9804e-01, 1.9617e-03],
        [2.8265e-04, 9.9972e-01],
        [1.8372e-06, 1.0000e+00],
        [6.7085e-06, 9.9999e-01],
        [5.0919e-05, 9.9995e-01],
        [4.4690e-06, 1.0000e+00],
        [9.9946e-01, 5.4396e-04],
        [1.1326e-03, 9.9887e-01],
        [9.9688e-01, 3.1163e-03],
        [1.6163e-06, 1.0000e+00],
        [1.8363e-04, 9.9982e-01],
        [9.9669e-10, 1.0000e+00],
        [5.2229e-08, 1.0000e+00],
        [6.7253e-02, 9.3275e-01],
        [9.7809e-01, 2.1914e-02],
        [1.2129e-06, 1.0000e+00],
        [4.2697e-06, 1.0000e+00],
        [7.7340e-04, 9.9923e-01],
        [9.9945e-01, 5.5163e-04],
        [9.9696e-01, 3.0403e-03],
        [9.9788e-01, 2.1163e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9950e-01, 4.9541e-04],
        [4.1894e-05, 9.9996e-01],
        [9.9935e-01, 6.5453e-04],
        [1.8196e-07, 1.0000e+00],
        [1.4881e-05, 9.9999e-01],
        [1.9204e-05, 9.9998e-01],
        [6.0149e-04, 9.9940e-01],
        [1.0006e-08, 1.0000e+00],
        [8.3538e-01, 1.6462e-01],
        [2.6144e-06, 1.0000e+00],
        [1.0061e-05, 9.9999e-01],
        [1.4355e-09, 1.0000e+00],
        [9.7916e-01, 2.0844e-02],
        [9.9557e-01, 4.4274e-03],
        [7.4271e-08, 1.0000e+00],
        [8.1916e-06, 9.9999e-01],
        [1.0476e-04, 9.9990e-01],
        [5.0814e-05, 9.9995e-01],
        [5.2764e-01, 4.7236e-01],
        [1.5680e-06, 1.0000e+00],
        [9.8216e-03, 9.9018e-01],
        [1.0338e-10, 1.0000e+00],
        [5.6609e-03, 9.9434e-01],
        [4.3194e-08, 1.0000e+00],
        [9.9811e-01, 1.8941e-03],
        [9.9942e-01, 5.8096e-04],
        [9.9997e-01, 3.1087e-05],
        [7.0753e-04, 9.9929e-01],
        [5.2807e-07, 1.0000e+00],
        [1.6075e-02, 9.8392e-01],
        [9.0270e-04, 9.9910e-01],
        [1.0081e-04, 9.9990e-01],
        [9.9968e-01, 3.1834e-04],
        [9.9606e-01, 3.9428e-03],
        [2.3806e-07, 1.0000e+00],
        [3.1837e-06, 1.0000e+00],
        [4.4566e-07, 1.0000e+00],
        [1.8146e-07, 1.0000e+00],
        [9.9878e-01, 1.2184e-03],
        [1.3987e-04, 9.9986e-01],
        [9.9954e-01, 4.6050e-04],
        [6.0501e-06, 9.9999e-01],
        [9.9964e-01, 3.5504e-04],
        [1.6865e-03, 9.9831e-01],
        [5.6791e-08, 1.0000e+00],
        [9.9902e-01, 9.8060e-04],
        [5.5895e-04, 9.9944e-01],
        [9.9818e-01, 1.8212e-03],
        [4.8721e-05, 9.9995e-01],
        [3.4210e-07, 1.0000e+00],
        [4.2064e-04, 9.9958e-01],
        [3.2957e-05, 9.9997e-01],
        [9.9871e-01, 1.2854e-03],
        [6.8629e-07, 1.0000e+00],
        [9.9933e-01, 6.7093e-04],
        [9.9954e-01, 4.6104e-04],
        [7.7832e-02, 9.2217e-01],
        [1.1577e-05, 9.9999e-01],
        [1.7673e-09, 1.0000e+00],
        [5.6723e-07, 1.0000e+00],
        [6.6880e-08, 1.0000e+00],
        [9.9994e-01, 6.4582e-05],
        [9.9925e-01, 7.5103e-04],
        [4.7564e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6686e-05, 9.9998e-01],
        [2.6413e-05, 9.9997e-01],
        [8.4056e-01, 1.5944e-01],
        [3.9635e-05, 9.9996e-01],
        [3.9655e-06, 1.0000e+00],
        [3.1589e-06, 1.0000e+00],
        [6.5583e-07, 1.0000e+00],
        [6.9078e-06, 9.9999e-01],
        [6.7461e-09, 1.0000e+00],
        [9.9727e-01, 2.7295e-03],
        [1.2899e-07, 1.0000e+00],
        [9.3247e-06, 9.9999e-01],
        [4.3204e-04, 9.9957e-01],
        [7.9536e-09, 1.0000e+00],
        [3.4827e-04, 9.9965e-01],
        [7.6402e-01, 2.3598e-01],
        [1.3713e-01, 8.6287e-01],
        [1.6054e-01, 8.3946e-01],
        [2.1493e-08, 1.0000e+00],
        [9.9962e-01, 3.8355e-04],
        [6.3687e-08, 1.0000e+00],
        [4.1546e-04, 9.9958e-01],
        [2.2184e-03, 9.9778e-01],
        [5.5286e-08, 1.0000e+00],
        [7.2331e-08, 1.0000e+00],
        [2.0125e-05, 9.9998e-01],
        [9.9824e-01, 1.7599e-03],
        [1.7217e-04, 9.9983e-01],
        [9.9999e-01, 5.5602e-06],
        [3.7148e-04, 9.9963e-01],
        [1.2801e-04, 9.9987e-01],
        [7.1992e-08, 1.0000e+00],
        [9.3691e-04, 9.9906e-01],
        [9.9926e-01, 7.3683e-04],
        [2.0625e-07, 1.0000e+00],
        [1.4972e-01, 8.5028e-01],
        [3.4979e-05, 9.9997e-01],
        [5.1989e-04, 9.9948e-01],
        [1.4523e-07, 1.0000e+00],
        [3.9706e-04, 9.9960e-01],
        [9.9831e-01, 1.6913e-03],
        [6.9581e-01, 3.0419e-01],
        [1.9924e-04, 9.9980e-01],
        [1.6636e-04, 9.9983e-01],
        [6.4660e-02, 9.3534e-01],
        [6.5418e-07, 1.0000e+00],
        [1.9492e-06, 1.0000e+00],
        [1.1061e-07, 1.0000e+00],
        [1.1714e-05, 9.9999e-01],
        [9.4753e-07, 1.0000e+00],
        [1.5824e-07, 1.0000e+00],
        [8.4771e-05, 9.9992e-01],
        [1.0376e-01, 8.9624e-01],
        [4.5068e-04, 9.9955e-01],
        [9.9985e-01, 1.4572e-04],
        [1.8543e-07, 1.0000e+00],
        [9.6516e-06, 9.9999e-01],
        [7.4271e-06, 9.9999e-01],
        [1.7251e-08, 1.0000e+00],
        [9.9985e-01, 1.4572e-04],
        [1.1127e-06, 1.0000e+00],
        [1.2605e-05, 9.9999e-01],
        [2.9126e-07, 1.0000e+00],
        [2.7073e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1369e-04, 9.9979e-01],
        [1.9663e-05, 9.9998e-01],
        [8.9946e-04, 9.9910e-01],
        [4.4002e-05, 9.9996e-01],
        [9.9888e-01, 1.1215e-03],
        [1.5621e-04, 9.9984e-01],
        [1.0020e-06, 1.0000e+00],
        [4.9458e-06, 1.0000e+00],
        [5.5804e-05, 9.9994e-01],
        [1.4718e-05, 9.9999e-01],
        [1.8855e-03, 9.9811e-01],
        [2.8785e-08, 1.0000e+00],
        [9.9998e-01, 2.0428e-05],
        [9.7345e-01, 2.6549e-02],
        [1.4457e-09, 1.0000e+00],
        [6.7928e-06, 9.9999e-01],
        [1.0205e-04, 9.9990e-01],
        [1.8638e-01, 8.1362e-01],
        [9.1696e-01, 8.3045e-02],
        [1.5369e-05, 9.9998e-01],
        [9.9935e-01, 6.4923e-04],
        [9.9801e-01, 1.9895e-03],
        [9.9948e-01, 5.1509e-04],
        [9.5347e-01, 4.6533e-02],
        [1.5945e-03, 9.9841e-01],
        [3.2886e-01, 6.7114e-01],
        [3.0909e-06, 1.0000e+00],
        [2.2058e-07, 1.0000e+00],
        [1.2946e-02, 9.8705e-01],
        [5.5109e-07, 1.0000e+00],
        [2.2204e-08, 1.0000e+00],
        [1.6838e-05, 9.9998e-01],
        [1.0863e-03, 9.9891e-01],
        [3.7452e-04, 9.9963e-01],
        [4.4344e-05, 9.9996e-01],
        [5.2240e-07, 1.0000e+00],
        [8.4202e-01, 1.5798e-01],
        [3.2342e-01, 6.7658e-01],
        [1.3284e-05, 9.9999e-01],
        [6.9469e-07, 1.0000e+00],
        [6.7041e-02, 9.3296e-01],
        [3.6875e-09, 1.0000e+00],
        [2.0565e-07, 1.0000e+00],
        [9.9657e-01, 3.4325e-03],
        [1.9734e-07, 1.0000e+00],
        [9.9989e-01, 1.0886e-04],
        [1.3382e-04, 9.9987e-01],
        [2.5770e-05, 9.9997e-01],
        [9.1640e-01, 8.3597e-02],
        [5.0834e-02, 9.4917e-01],
        [1.0465e-05, 9.9999e-01],
        [4.1880e-07, 1.0000e+00],
        [5.3028e-05, 9.9995e-01],
        [1.1382e-03, 9.9886e-01],
        [1.7778e-04, 9.9982e-01],
        [1.4254e-03, 9.9857e-01],
        [8.1693e-05, 9.9992e-01],
        [5.6400e-06, 9.9999e-01],
        [5.4548e-05, 9.9995e-01],
        [1.4966e-04, 9.9985e-01],
        [1.0764e-04, 9.9989e-01],
        [9.9045e-01, 9.5500e-03],
        [9.9747e-01, 2.5283e-03],
        [9.9928e-01, 7.2387e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[6.0182e-06, 9.9999e-01],
        [9.9472e-01, 5.2774e-03],
        [3.8619e-05, 9.9996e-01],
        [1.8500e-08, 1.0000e+00],
        [5.0636e-03, 9.9494e-01],
        [9.1202e-06, 9.9999e-01],
        [9.9991e-01, 8.6999e-05],
        [9.9686e-01, 3.1359e-03],
        [9.9935e-01, 6.5337e-04],
        [3.4256e-08, 1.0000e+00],
        [6.4981e-06, 9.9999e-01],
        [1.5056e-06, 1.0000e+00],
        [9.9833e-01, 1.6697e-03],
        [3.1072e-06, 1.0000e+00],
        [1.8325e-04, 9.9982e-01],
        [1.3353e-05, 9.9999e-01],
        [6.9371e-09, 1.0000e+00],
        [9.3611e-06, 9.9999e-01],
        [9.9876e-01, 1.2376e-03],
        [1.7791e-04, 9.9982e-01],
        [1.2709e-05, 9.9999e-01],
        [7.4389e-04, 9.9926e-01],
        [3.4561e-03, 9.9654e-01],
        [1.2088e-06, 1.0000e+00],
        [9.9795e-01, 2.0511e-03],
        [5.1864e-07, 1.0000e+00],
        [5.6729e-01, 4.3271e-01],
        [9.2299e-05, 9.9991e-01],
        [2.0866e-06, 1.0000e+00],
        [1.7279e-06, 1.0000e+00],
        [2.3731e-03, 9.9763e-01],
        [9.4899e-01, 5.1009e-02],
        [8.3775e-01, 1.6225e-01],
        [2.8812e-03, 9.9712e-01],
        [3.6449e-06, 1.0000e+00],
        [9.9876e-01, 1.2354e-03],
        [6.0466e-04, 9.9940e-01],
        [3.4041e-07, 1.0000e+00],
        [1.4025e-03, 9.9860e-01],
        [3.7816e-04, 9.9962e-01],
        [1.2020e-06, 1.0000e+00],
        [9.9964e-01, 3.6222e-04],
        [5.6455e-09, 1.0000e+00],
        [2.5720e-03, 9.9743e-01],
        [3.4846e-04, 9.9965e-01],
        [1.5971e-05, 9.9998e-01],
        [1.3689e-03, 9.9863e-01],
        [2.0157e-05, 9.9998e-01],
        [1.5171e-06, 1.0000e+00],
        [6.2607e-07, 1.0000e+00],
        [3.9571e-03, 9.9604e-01],
        [3.5296e-04, 9.9965e-01],
        [1.1944e-09, 1.0000e+00],
        [4.4638e-08, 1.0000e+00],
        [7.9339e-04, 9.9921e-01],
        [7.8801e-07, 1.0000e+00],
        [3.6378e-09, 1.0000e+00],
        [1.4781e-03, 9.9852e-01],
        [4.4078e-02, 9.5592e-01],
        [9.6245e-01, 3.7546e-02],
        [8.8783e-06, 9.9999e-01],
        [9.0677e-05, 9.9991e-01],
        [3.6231e-02, 9.6377e-01],
        [6.2925e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1216e-04, 9.9979e-01],
        [2.3620e-08, 1.0000e+00],
        [2.7905e-04, 9.9972e-01],
        [9.0711e-01, 9.2887e-02],
        [9.9896e-01, 1.0447e-03],
        [2.1928e-05, 9.9998e-01],
        [2.6176e-06, 1.0000e+00],
        [7.3592e-06, 9.9999e-01],
        [9.8524e-01, 1.4760e-02],
        [2.6440e-04, 9.9974e-01],
        [5.9523e-01, 4.0477e-01],
        [1.5044e-06, 1.0000e+00],
        [2.9945e-06, 1.0000e+00],
        [9.1113e-04, 9.9909e-01],
        [3.5616e-05, 9.9996e-01],
        [5.1882e-06, 9.9999e-01],
        [1.6555e-01, 8.3445e-01],
        [7.9760e-04, 9.9920e-01],
        [6.4944e-07, 1.0000e+00],
        [1.3020e-03, 9.9870e-01],
        [3.3218e-07, 1.0000e+00],
        [9.9934e-01, 6.5998e-04],
        [1.8209e-07, 1.0000e+00],
        [9.9908e-01, 9.1930e-04],
        [9.9967e-01, 3.3286e-04],
        [9.8199e-01, 1.8007e-02],
        [9.9960e-01, 4.0191e-04],
        [1.6025e-06, 1.0000e+00],
        [9.9987e-01, 1.3283e-04],
        [2.7266e-08, 1.0000e+00],
        [8.6920e-04, 9.9913e-01],
        [9.8108e-01, 1.8915e-02],
        [5.2151e-03, 9.9478e-01],
        [1.3546e-08, 1.0000e+00],
        [5.2637e-05, 9.9995e-01],
        [9.9985e-01, 1.5057e-04],
        [9.9943e-01, 5.7037e-04],
        [2.0442e-07, 1.0000e+00],
        [7.0046e-08, 1.0000e+00],
        [9.9384e-01, 6.1631e-03],
        [9.5468e-05, 9.9990e-01],
        [1.2473e-04, 9.9988e-01],
        [5.7038e-03, 9.9430e-01],
        [2.9737e-06, 1.0000e+00],
        [1.7803e-06, 1.0000e+00],
        [5.6240e-09, 1.0000e+00],
        [1.8828e-06, 1.0000e+00],
        [3.3183e-06, 1.0000e+00],
        [1.2325e-02, 9.8768e-01],
        [1.0558e-07, 1.0000e+00],
        [1.0324e-04, 9.9990e-01],
        [1.2783e-07, 1.0000e+00],
        [8.9081e-04, 9.9911e-01],
        [3.7938e-03, 9.9621e-01],
        [2.4332e-03, 9.9757e-01],
        [8.9135e-03, 9.9109e-01],
        [2.3840e-05, 9.9998e-01],
        [9.4646e-08, 1.0000e+00],
        [6.0664e-06, 9.9999e-01],
        [1.4335e-03, 9.9857e-01],
        [1.8830e-04, 9.9981e-01],
        [6.0521e-03, 9.9395e-01],
        [3.5125e-06, 1.0000e+00],
        [9.0804e-01, 9.1964e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.3288e-01, 6.7124e-02],
        [3.2645e-05, 9.9997e-01],
        [2.5495e-09, 1.0000e+00],
        [5.2356e-05, 9.9995e-01],
        [9.9876e-01, 1.2366e-03],
        [2.2214e-07, 1.0000e+00],
        [1.4626e-01, 8.5374e-01],
        [9.9673e-01, 3.2709e-03],
        [9.9993e-01, 6.7780e-05],
        [9.9961e-01, 3.8784e-04],
        [2.7192e-04, 9.9973e-01],
        [2.1728e-06, 1.0000e+00],
        [6.2826e-09, 1.0000e+00],
        [9.4930e-01, 5.0696e-02],
        [9.9963e-01, 3.6902e-04],
        [9.9958e-01, 4.1746e-04],
        [1.3027e-07, 1.0000e+00],
        [1.1174e-02, 9.8883e-01],
        [3.6785e-03, 9.9632e-01],
        [8.5877e-08, 1.0000e+00],
        [9.9583e-01, 4.1661e-03],
        [5.5696e-04, 9.9944e-01],
        [1.9882e-07, 1.0000e+00],
        [1.6057e-05, 9.9998e-01],
        [9.9975e-01, 2.5092e-04],
        [4.4638e-03, 9.9554e-01],
        [1.1527e-04, 9.9988e-01],
        [2.1294e-07, 1.0000e+00],
        [2.5201e-02, 9.7480e-01],
        [6.9637e-05, 9.9993e-01],
        [3.0457e-06, 1.0000e+00],
        [9.5567e-09, 1.0000e+00],
        [3.4655e-02, 9.6535e-01],
        [5.4214e-04, 9.9946e-01],
        [3.2780e-07, 1.0000e+00],
        [1.2043e-05, 9.9999e-01],
        [9.6339e-02, 9.0366e-01],
        [9.9967e-01, 3.2870e-04],
        [1.9502e-03, 9.9805e-01],
        [9.9954e-01, 4.6066e-04],
        [2.4356e-04, 9.9976e-01],
        [1.9427e-06, 1.0000e+00],
        [1.0362e-04, 9.9990e-01],
        [8.8156e-04, 9.9912e-01],
        [9.2866e-01, 7.1343e-02],
        [9.3446e-01, 6.5539e-02],
        [1.0482e-05, 9.9999e-01],
        [4.2066e-06, 1.0000e+00],
        [4.9191e-07, 1.0000e+00],
        [3.3460e-04, 9.9967e-01],
        [4.2413e-07, 1.0000e+00],
        [1.2350e-03, 9.9876e-01],
        [7.5438e-05, 9.9992e-01],
        [7.1218e-06, 9.9999e-01],
        [9.9981e-01, 1.8621e-04],
        [1.2130e-04, 9.9988e-01],
        [2.7492e-04, 9.9973e-01],
        [5.9008e-04, 9.9941e-01],
        [9.9989e-01, 1.0887e-04],
        [9.4524e-06, 9.9999e-01],
        [2.3136e-04, 9.9977e-01],
        [2.0141e-04, 9.9980e-01],
        [2.3734e-10, 1.0000e+00],
        [1.6887e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3762e-03, 9.9862e-01],
        [9.9988e-01, 1.2286e-04],
        [3.9667e-07, 1.0000e+00],
        [9.9882e-05, 9.9990e-01],
        [9.9941e-01, 5.8951e-04],
        [9.9937e-01, 6.2813e-04],
        [9.9940e-01, 6.0388e-04],
        [1.7949e-06, 1.0000e+00],
        [1.3153e-05, 9.9999e-01],
        [9.9999e-01, 5.5098e-06],
        [9.9831e-05, 9.9990e-01],
        [9.9896e-01, 1.0403e-03],
        [1.6968e-05, 9.9998e-01],
        [2.2499e-04, 9.9977e-01],
        [1.5127e-03, 9.9849e-01],
        [7.2719e-07, 1.0000e+00],
        [3.5440e-04, 9.9965e-01],
        [9.9949e-01, 5.1057e-04],
        [5.0564e-04, 9.9949e-01],
        [7.1524e-06, 9.9999e-01],
        [2.2982e-07, 1.0000e+00],
        [5.4972e-02, 9.4503e-01],
        [6.8445e-06, 9.9999e-01],
        [7.0546e-03, 9.9295e-01],
        [2.1924e-03, 9.9781e-01],
        [1.0006e-08, 1.0000e+00],
        [1.2757e-04, 9.9987e-01],
        [9.6083e-01, 3.9166e-02],
        [4.6837e-04, 9.9953e-01],
        [9.9834e-01, 1.6571e-03],
        [2.8913e-08, 1.0000e+00],
        [9.9517e-01, 4.8345e-03],
        [1.6915e-04, 9.9983e-01],
        [1.4386e-04, 9.9986e-01],
        [4.2663e-08, 1.0000e+00],
        [9.9918e-01, 8.1707e-04],
        [8.3506e-03, 9.9165e-01],
        [1.5800e-03, 9.9842e-01],
        [1.3842e-05, 9.9999e-01],
        [9.9847e-01, 1.5336e-03],
        [7.3986e-07, 1.0000e+00],
        [9.8746e-01, 1.2545e-02],
        [1.5240e-02, 9.8476e-01],
        [8.8921e-01, 1.1079e-01],
        [5.5263e-05, 9.9994e-01],
        [5.9012e-03, 9.9410e-01],
        [9.9977e-01, 2.3441e-04],
        [1.5159e-06, 1.0000e+00],
        [1.7949e-06, 1.0000e+00],
        [1.8938e-04, 9.9981e-01],
        [3.0182e-04, 9.9970e-01],
        [2.0856e-03, 9.9791e-01],
        [3.0394e-05, 9.9997e-01],
        [3.4480e-06, 1.0000e+00],
        [1.6408e-04, 9.9984e-01],
        [9.9746e-01, 2.5391e-03],
        [8.5667e-05, 9.9991e-01],
        [5.8995e-07, 1.0000e+00],
        [9.9990e-01, 1.0230e-04],
        [1.0553e-06, 1.0000e+00],
        [3.9667e-07, 1.0000e+00],
        [9.9613e-01, 3.8687e-03],
        [1.0000e+00, 4.8147e-06],
        [4.0239e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.4364e-07, 1.0000e+00],
        [2.0910e-05, 9.9998e-01],
        [6.0807e-08, 1.0000e+00],
        [4.6476e-04, 9.9954e-01],
        [9.0156e-01, 9.8437e-02],
        [9.4343e-07, 1.0000e+00],
        [9.9968e-01, 3.1599e-04],
        [1.5140e-05, 9.9998e-01],
        [2.3402e-05, 9.9998e-01],
        [1.8681e-06, 1.0000e+00],
        [5.7968e-06, 9.9999e-01],
        [3.4290e-03, 9.9657e-01],
        [1.9214e-04, 9.9981e-01],
        [2.3249e-04, 9.9977e-01],
        [2.7795e-03, 9.9722e-01],
        [9.9978e-01, 2.1672e-04],
        [1.6309e-05, 9.9998e-01],
        [4.2729e-05, 9.9996e-01],
        [1.2071e-04, 9.9988e-01],
        [3.7850e-06, 1.0000e+00],
        [9.9915e-01, 8.4983e-04],
        [6.9260e-06, 9.9999e-01],
        [8.3595e-04, 9.9916e-01],
        [1.8891e-07, 1.0000e+00],
        [9.9964e-01, 3.5893e-04],
        [5.1826e-06, 9.9999e-01],
        [7.8537e-01, 2.1463e-01],
        [1.3400e-03, 9.9866e-01],
        [9.9678e-01, 3.2224e-03],
        [9.9935e-01, 6.4722e-04],
        [8.0101e-01, 1.9899e-01],
        [7.1944e-04, 9.9928e-01],
        [9.9980e-01, 1.9589e-04],
        [8.0001e-01, 1.9999e-01],
        [2.0505e-03, 9.9795e-01],
        [1.8096e-05, 9.9998e-01],
        [8.1279e-07, 1.0000e+00],
        [8.1629e-05, 9.9992e-01],
        [5.4712e-07, 1.0000e+00],
        [7.3679e-01, 2.6321e-01],
        [1.6090e-06, 1.0000e+00],
        [3.1736e-08, 1.0000e+00],
        [1.9672e-02, 9.8033e-01],
        [1.0754e-06, 1.0000e+00],
        [4.3362e-05, 9.9996e-01],
        [9.9489e-01, 5.1067e-03],
        [1.5709e-04, 9.9984e-01],
        [5.6243e-07, 1.0000e+00],
        [4.3409e-07, 1.0000e+00],
        [2.8546e-05, 9.9997e-01],
        [1.2589e-02, 9.8741e-01],
        [2.9814e-04, 9.9970e-01],
        [9.9913e-01, 8.7380e-04],
        [6.8481e-06, 9.9999e-01],
        [7.0379e-06, 9.9999e-01],
        [9.7873e-01, 2.1269e-02],
        [9.3915e-01, 6.0849e-02],
        [9.9950e-01, 4.9717e-04],
        [9.4403e-01, 5.5974e-02],
        [8.1327e-05, 9.9992e-01],
        [9.9881e-01, 1.1871e-03],
        [9.9982e-01, 1.7876e-04],
        [5.4651e-05, 9.9995e-01],
        [3.0281e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9926e-01, 7.4300e-04],
        [9.9983e-01, 1.6944e-04],
        [6.4059e-05, 9.9994e-01],
        [7.9177e-04, 9.9921e-01],
        [7.3923e-01, 2.6077e-01],
        [7.2100e-05, 9.9993e-01],
        [6.0020e-05, 9.9994e-01],
        [9.8401e-01, 1.5989e-02],
        [9.6237e-01, 3.7627e-02],
        [7.8782e-05, 9.9992e-01],
        [1.1992e-04, 9.9988e-01],
        [8.8835e-05, 9.9991e-01],
        [7.3735e-05, 9.9993e-01],
        [2.4564e-03, 9.9754e-01],
        [1.6129e-04, 9.9984e-01],
        [6.9472e-02, 9.3053e-01],
        [8.7351e-05, 9.9991e-01],
        [9.8221e-01, 1.7786e-02],
        [9.9897e-01, 1.0263e-03],
        [1.0584e-05, 9.9999e-01],
        [2.9540e-07, 1.0000e+00],
        [4.6986e-03, 9.9530e-01],
        [8.0961e-05, 9.9992e-01],
        [2.8897e-04, 9.9971e-01],
        [9.7276e-07, 1.0000e+00],
        [9.9998e-01, 1.5757e-05],
        [7.4918e-03, 9.9251e-01],
        [1.1079e-03, 9.9889e-01],
        [1.2434e-08, 1.0000e+00],
        [2.8881e-05, 9.9997e-01],
        [5.2490e-04, 9.9948e-01],
        [1.2082e-02, 9.8792e-01],
        [2.9192e-07, 1.0000e+00],
        [1.2923e-04, 9.9987e-01],
        [1.0559e-05, 9.9999e-01],
        [9.9544e-01, 4.5594e-03],
        [1.7711e-04, 9.9982e-01],
        [9.9994e-01, 6.1581e-05],
        [1.1317e-04, 9.9989e-01],
        [4.3385e-05, 9.9996e-01],
        [5.5524e-07, 1.0000e+00],
        [1.4351e-02, 9.8565e-01],
        [3.3501e-03, 9.9665e-01],
        [3.7057e-03, 9.9629e-01],
        [1.1774e-03, 9.9882e-01],
        [2.4927e-05, 9.9998e-01],
        [9.9926e-01, 7.3782e-04],
        [9.9915e-01, 8.5037e-04],
        [1.4356e-01, 8.5644e-01],
        [1.1178e-05, 9.9999e-01],
        [3.7195e-04, 9.9963e-01],
        [8.5491e-05, 9.9991e-01],
        [9.9489e-01, 5.1096e-03],
        [5.7956e-05, 9.9994e-01],
        [5.2750e-08, 1.0000e+00],
        [1.7917e-07, 1.0000e+00],
        [8.5921e-06, 9.9999e-01],
        [9.9943e-01, 5.7464e-04],
        [9.9951e-01, 4.8660e-04],
        [1.1089e-05, 9.9999e-01],
        [9.9916e-01, 8.3669e-04],
        [7.3918e-05, 9.9993e-01],
        [6.2318e-08, 1.0000e+00],
        [1.6565e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5458e-08, 1.0000e+00],
        [8.5778e-01, 1.4222e-01],
        [1.1237e-05, 9.9999e-01],
        [6.0359e-08, 1.0000e+00],
        [6.7445e-01, 3.2555e-01],
        [6.9021e-06, 9.9999e-01],
        [1.0252e-03, 9.9897e-01],
        [2.0491e-04, 9.9980e-01],
        [7.2680e-06, 9.9999e-01],
        [2.6550e-02, 9.7345e-01],
        [9.9900e-01, 1.0027e-03],
        [6.0155e-05, 9.9994e-01],
        [3.0596e-05, 9.9997e-01],
        [3.9163e-01, 6.0837e-01],
        [1.8642e-04, 9.9981e-01],
        [9.9942e-01, 5.8496e-04],
        [2.2453e-06, 1.0000e+00],
        [2.3779e-04, 9.9976e-01],
        [2.5664e-06, 1.0000e+00],
        [9.9953e-01, 4.7045e-04],
        [3.5065e-04, 9.9965e-01],
        [1.6248e-05, 9.9998e-01],
        [9.7079e-03, 9.9029e-01],
        [1.7626e-03, 9.9824e-01],
        [2.8561e-02, 9.7144e-01],
        [2.0914e-01, 7.9086e-01],
        [3.2981e-05, 9.9997e-01],
        [9.9930e-01, 7.0365e-04],
        [2.2067e-04, 9.9978e-01],
        [9.9970e-01, 2.9911e-04],
        [6.8023e-07, 1.0000e+00],
        [2.2491e-05, 9.9998e-01],
        [9.9989e-01, 1.0673e-04],
        [1.9363e-04, 9.9981e-01],
        [9.9937e-01, 6.3138e-04],
        [6.0389e-06, 9.9999e-01],
        [4.8784e-01, 5.1216e-01],
        [1.6694e-01, 8.3306e-01],
        [7.8050e-04, 9.9922e-01],
        [6.6896e-06, 9.9999e-01],
        [5.9562e-05, 9.9994e-01],
        [9.9192e-01, 8.0830e-03],
        [3.5910e-06, 1.0000e+00],
        [2.5631e-04, 9.9974e-01],
        [4.9701e-05, 9.9995e-01],
        [2.1032e-06, 1.0000e+00],
        [2.9059e-04, 9.9971e-01],
        [5.2335e-03, 9.9477e-01],
        [7.1469e-06, 9.9999e-01],
        [4.0391e-03, 9.9596e-01],
        [2.8564e-06, 1.0000e+00],
        [3.0669e-05, 9.9997e-01],
        [9.9954e-01, 4.5611e-04],
        [1.2917e-04, 9.9987e-01],
        [7.9927e-06, 9.9999e-01],
        [9.9924e-01, 7.5855e-04],
        [2.4024e-06, 1.0000e+00],
        [3.5255e-03, 9.9647e-01],
        [6.1010e-04, 9.9939e-01],
        [3.6763e-05, 9.9996e-01],
        [1.4515e-05, 9.9999e-01],
        [7.7437e-06, 9.9999e-01],
        [6.2654e-04, 9.9937e-01],
        [6.3436e-02, 9.3656e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.5603e-09, 1.0000e+00],
        [1.2163e-02, 9.8784e-01],
        [9.9930e-01, 7.0451e-04],
        [2.2723e-04, 9.9977e-01],
        [3.7128e-07, 1.0000e+00],
        [1.4446e-04, 9.9986e-01],
        [1.2513e-05, 9.9999e-01],
        [5.8524e-03, 9.9415e-01],
        [1.3993e-04, 9.9986e-01],
        [4.4729e-09, 1.0000e+00],
        [1.8437e-03, 9.9816e-01],
        [1.7282e-02, 9.8272e-01],
        [1.9659e-04, 9.9980e-01],
        [1.2447e-06, 1.0000e+00],
        [9.3734e-05, 9.9991e-01],
        [1.5961e-03, 9.9840e-01],
        [2.4588e-03, 9.9754e-01],
        [1.3738e-06, 1.0000e+00],
        [7.3709e-04, 9.9926e-01],
        [2.4357e-07, 1.0000e+00],
        [3.7564e-04, 9.9962e-01],
        [2.0254e-03, 9.9797e-01],
        [9.9942e-01, 5.7760e-04],
        [9.9988e-01, 1.1565e-04],
        [1.0515e-04, 9.9989e-01],
        [6.0286e-08, 1.0000e+00],
        [1.2344e-03, 9.9877e-01],
        [9.9943e-01, 5.6869e-04],
        [5.1307e-02, 9.4869e-01],
        [4.0832e-03, 9.9592e-01],
        [9.9981e-01, 1.8562e-04],
        [1.0000e+00, 8.3055e-07],
        [2.8257e-06, 1.0000e+00],
        [9.9935e-01, 6.4641e-04],
        [2.2035e-04, 9.9978e-01],
        [2.4429e-07, 1.0000e+00],
        [9.9684e-01, 3.1637e-03],
        [9.9969e-01, 3.0949e-04],
        [1.0000e+00, 5.7253e-08],
        [1.0000e+00, 8.0556e-07],
        [2.5529e-04, 9.9974e-01],
        [3.1863e-06, 1.0000e+00],
        [1.2474e-02, 9.8753e-01],
        [2.5165e-05, 9.9997e-01],
        [9.9935e-01, 6.4559e-04],
        [3.3213e-03, 9.9668e-01],
        [3.5445e-07, 1.0000e+00],
        [1.3994e-03, 9.9860e-01],
        [4.0517e-03, 9.9595e-01],
        [9.9980e-01, 1.9919e-04],
        [9.9973e-01, 2.7105e-04],
        [2.9158e-04, 9.9971e-01],
        [8.0377e-06, 9.9999e-01],
        [1.2033e-06, 1.0000e+00],
        [2.6269e-08, 1.0000e+00],
        [1.1477e-05, 9.9999e-01],
        [9.9516e-01, 4.8385e-03],
        [9.9926e-01, 7.3554e-04],
        [1.6209e-03, 9.9838e-01],
        [8.6861e-03, 9.9131e-01],
        [9.9642e-01, 3.5780e-03],
        [7.5177e-08, 1.0000e+00],
        [1.7289e-03, 9.9827e-01],
        [8.1383e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.4054e-03, 9.9759e-01],
        [9.9974e-01, 2.6171e-04],
        [4.9300e-06, 1.0000e+00],
        [2.3610e-05, 9.9998e-01],
        [9.9905e-01, 9.5228e-04],
        [7.5251e-05, 9.9992e-01],
        [9.9946e-01, 5.3887e-04],
        [1.0728e-01, 8.9272e-01],
        [9.9964e-01, 3.5921e-04],
        [7.8186e-01, 2.1814e-01],
        [9.9910e-01, 9.0220e-04],
        [1.2179e-03, 9.9878e-01],
        [2.2626e-02, 9.7737e-01],
        [2.0883e-07, 1.0000e+00],
        [2.7569e-05, 9.9997e-01],
        [5.3494e-04, 9.9947e-01],
        [9.9974e-01, 2.6337e-04],
        [9.9958e-01, 4.2063e-04],
        [2.2014e-04, 9.9978e-01],
        [9.9978e-01, 2.1716e-04],
        [1.3070e-03, 9.9869e-01],
        [9.9971e-01, 2.8713e-04],
        [9.9954e-01, 4.6117e-04],
        [9.9656e-01, 3.4397e-03],
        [1.8609e-04, 9.9981e-01],
        [1.0992e-02, 9.8901e-01],
        [4.4348e-07, 1.0000e+00],
        [1.1050e-05, 9.9999e-01],
        [8.7644e-01, 1.2356e-01],
        [9.9971e-01, 2.9339e-04],
        [2.1495e-06, 1.0000e+00],
        [1.4981e-03, 9.9850e-01],
        [9.5540e-01, 4.4601e-02],
        [4.6507e-08, 1.0000e+00],
        [4.5452e-01, 5.4548e-01],
        [1.0213e-05, 9.9999e-01],
        [5.8617e-07, 1.0000e+00],
        [9.9970e-01, 3.0399e-04],
        [8.2881e-05, 9.9992e-01],
        [2.7976e-03, 9.9720e-01],
        [6.7166e-07, 1.0000e+00],
        [9.9915e-01, 8.4745e-04],
        [6.1210e-04, 9.9939e-01],
        [6.1489e-02, 9.3851e-01],
        [3.8766e-05, 9.9996e-01],
        [7.2923e-08, 1.0000e+00],
        [1.4004e-04, 9.9986e-01],
        [5.9356e-07, 1.0000e+00],
        [2.1426e-07, 1.0000e+00],
        [3.1479e-06, 1.0000e+00],
        [6.4225e-07, 1.0000e+00],
        [4.2826e-07, 1.0000e+00],
        [9.9929e-01, 7.0591e-04],
        [1.2691e-04, 9.9987e-01],
        [1.8639e-03, 9.9814e-01],
        [9.8400e-01, 1.5999e-02],
        [9.9891e-01, 1.0907e-03],
        [1.2008e-06, 1.0000e+00],
        [9.9915e-01, 8.4875e-04],
        [1.0985e-05, 9.9999e-01],
        [1.4142e-04, 9.9986e-01],
        [7.6013e-05, 9.9992e-01],
        [6.8969e-04, 9.9931e-01],
        [1.0839e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.1945e-02, 9.5806e-01],
        [1.1889e-03, 9.9881e-01],
        [6.2233e-03, 9.9378e-01],
        [9.1485e-05, 9.9991e-01],
        [9.7060e-01, 2.9400e-02],
        [9.7864e-03, 9.9021e-01],
        [2.3835e-05, 9.9998e-01],
        [9.9973e-01, 2.6665e-04],
        [1.7562e-08, 1.0000e+00],
        [1.6489e-04, 9.9984e-01],
        [5.4770e-04, 9.9945e-01],
        [2.7975e-04, 9.9972e-01],
        [9.9944e-01, 5.6396e-04],
        [1.6601e-04, 9.9983e-01],
        [1.1823e-06, 1.0000e+00],
        [9.6688e-05, 9.9990e-01],
        [7.5940e-04, 9.9924e-01],
        [1.2303e-06, 1.0000e+00],
        [7.7783e-05, 9.9992e-01],
        [2.3319e-04, 9.9977e-01],
        [9.9505e-01, 4.9482e-03],
        [1.0061e-04, 9.9990e-01],
        [8.7838e-05, 9.9991e-01],
        [6.2445e-05, 9.9994e-01],
        [5.4566e-04, 9.9945e-01],
        [5.3225e-05, 9.9995e-01],
        [7.6932e-01, 2.3068e-01],
        [9.9418e-01, 5.8193e-03],
        [1.2232e-03, 9.9878e-01],
        [8.1213e-05, 9.9992e-01],
        [9.9946e-01, 5.3974e-04],
        [3.4602e-05, 9.9997e-01],
        [1.3614e-06, 1.0000e+00],
        [9.9991e-01, 8.7237e-05],
        [8.2067e-01, 1.7933e-01],
        [9.9979e-01, 2.0697e-04],
        [9.5798e-01, 4.2016e-02],
        [9.9930e-01, 7.0224e-04],
        [1.0025e-04, 9.9990e-01],
        [1.2413e-03, 9.9876e-01],
        [9.9995e-01, 4.6365e-05],
        [9.9992e-01, 7.5651e-05],
        [9.7554e-01, 2.4455e-02],
        [3.4505e-03, 9.9655e-01],
        [1.7246e-03, 9.9828e-01],
        [7.3354e-08, 1.0000e+00],
        [1.2084e-05, 9.9999e-01],
        [3.0778e-02, 9.6922e-01],
        [3.2385e-05, 9.9997e-01],
        [5.8024e-03, 9.9420e-01],
        [4.8988e-04, 9.9951e-01],
        [3.7685e-06, 1.0000e+00],
        [2.0024e-03, 9.9800e-01],
        [2.2901e-01, 7.7099e-01],
        [2.5071e-04, 9.9975e-01],
        [9.1150e-04, 9.9909e-01],
        [6.5813e-05, 9.9993e-01],
        [9.4038e-06, 9.9999e-01],
        [6.7266e-08, 1.0000e+00],
        [2.1593e-02, 9.7841e-01],
        [7.8013e-08, 1.0000e+00],
        [9.5680e-01, 4.3199e-02],
        [3.6866e-05, 9.9996e-01],
        [6.1802e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.5305e-01, 4.6954e-02],
        [9.9891e-01, 1.0910e-03],
        [4.5982e-05, 9.9995e-01],
        [8.6877e-04, 9.9913e-01],
        [8.5880e-05, 9.9991e-01],
        [2.1795e-02, 9.7821e-01],
        [6.8085e-04, 9.9932e-01],
        [1.2339e-06, 1.0000e+00],
        [1.9725e-05, 9.9998e-01],
        [2.9734e-06, 1.0000e+00],
        [6.5922e-04, 9.9934e-01],
        [8.3285e-09, 1.0000e+00],
        [3.1870e-08, 1.0000e+00],
        [5.2496e-03, 9.9475e-01],
        [9.1098e-07, 1.0000e+00],
        [9.9977e-01, 2.2954e-04],
        [2.2329e-07, 1.0000e+00],
        [1.0660e-06, 1.0000e+00],
        [2.9910e-04, 9.9970e-01],
        [2.8888e-05, 9.9997e-01],
        [9.9984e-01, 1.5902e-04],
        [9.9875e-01, 1.2462e-03],
        [1.2407e-07, 1.0000e+00],
        [2.5764e-03, 9.9742e-01],
        [4.3379e-07, 1.0000e+00],
        [9.3799e-05, 9.9991e-01],
        [3.5368e-04, 9.9965e-01],
        [9.5925e-05, 9.9990e-01],
        [3.5522e-06, 1.0000e+00],
        [9.7745e-01, 2.2552e-02],
        [5.1114e-07, 1.0000e+00],
        [2.1396e-03, 9.9786e-01],
        [5.4139e-04, 9.9946e-01],
        [9.9520e-01, 4.7982e-03],
        [9.9884e-01, 1.1578e-03],
        [2.8426e-05, 9.9997e-01],
        [9.9501e-01, 4.9928e-03],
        [2.3159e-03, 9.9768e-01],
        [3.3254e-06, 1.0000e+00],
        [5.9038e-04, 9.9941e-01],
        [1.4408e-05, 9.9999e-01],
        [1.0385e-07, 1.0000e+00],
        [1.5109e-03, 9.9849e-01],
        [8.5150e-01, 1.4850e-01],
        [1.4882e-04, 9.9985e-01],
        [1.6309e-04, 9.9984e-01],
        [9.9998e-01, 2.3842e-05],
        [9.9959e-01, 4.0950e-04],
        [1.6628e-05, 9.9998e-01],
        [6.4363e-03, 9.9356e-01],
        [1.2454e-02, 9.8755e-01],
        [2.1122e-04, 9.9979e-01],
        [9.9994e-01, 6.1300e-05],
        [1.4643e-03, 9.9854e-01],
        [9.3776e-05, 9.9991e-01],
        [9.9675e-01, 3.2514e-03],
        [1.8255e-04, 9.9982e-01],
        [9.9946e-01, 5.3583e-04],
        [3.0943e-05, 9.9997e-01],
        [9.9961e-01, 3.8726e-04],
        [9.9386e-01, 6.1432e-03],
        [4.6983e-05, 9.9995e-01],
        [8.7919e-07, 1.0000e+00],
        [4.5036e-04, 9.9955e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8327e-04, 9.9902e-01],
        [9.9965e-01, 3.4909e-04],
        [2.2967e-02, 9.7703e-01],
        [5.6021e-08, 1.0000e+00],
        [6.2253e-03, 9.9377e-01],
        [5.9368e-05, 9.9994e-01],
        [8.2340e-02, 9.1766e-01],
        [4.2477e-03, 9.9575e-01],
        [4.7810e-08, 1.0000e+00],
        [1.8039e-04, 9.9982e-01],
        [1.9435e-07, 1.0000e+00],
        [9.9966e-01, 3.4194e-04],
        [9.9943e-01, 5.7103e-04],
        [7.5614e-04, 9.9924e-01],
        [5.0758e-04, 9.9949e-01],
        [9.1685e-02, 9.0832e-01],
        [7.2922e-08, 1.0000e+00],
        [7.9457e-01, 2.0543e-01],
        [1.3721e-07, 1.0000e+00],
        [1.4753e-03, 9.9852e-01],
        [9.9971e-01, 2.9348e-04],
        [2.8071e-07, 1.0000e+00],
        [4.6670e-02, 9.5333e-01],
        [8.9972e-06, 9.9999e-01],
        [7.0831e-03, 9.9292e-01],
        [1.0076e-09, 1.0000e+00],
        [9.4277e-01, 5.7227e-02],
        [5.6485e-08, 1.0000e+00],
        [1.1292e-05, 9.9999e-01],
        [5.2523e-06, 9.9999e-01],
        [4.2268e-07, 1.0000e+00],
        [4.1657e-04, 9.9958e-01],
        [9.9857e-01, 1.4323e-03],
        [4.6243e-07, 1.0000e+00],
        [2.2840e-06, 1.0000e+00],
        [2.7551e-02, 9.7245e-01],
        [6.3918e-11, 1.0000e+00],
        [9.9594e-01, 4.0610e-03],
        [7.8291e-06, 9.9999e-01],
        [4.3437e-02, 9.5656e-01],
        [9.6480e-01, 3.5202e-02],
        [2.6652e-05, 9.9997e-01],
        [2.1287e-04, 9.9979e-01],
        [3.5153e-04, 9.9965e-01],
        [5.7855e-08, 1.0000e+00],
        [9.9885e-01, 1.1545e-03],
        [1.5265e-03, 9.9847e-01],
        [4.2955e-05, 9.9996e-01],
        [1.0445e-01, 8.9555e-01],
        [8.1794e-03, 9.9182e-01],
        [9.9992e-01, 8.4864e-05],
        [1.4431e-05, 9.9999e-01],
        [2.0721e-03, 9.9793e-01],
        [3.4511e-09, 1.0000e+00],
        [1.8276e-07, 1.0000e+00],
        [1.2284e-04, 9.9988e-01],
        [1.1211e-06, 1.0000e+00],
        [9.9868e-01, 1.3199e-03],
        [2.6657e-04, 9.9973e-01],
        [7.9238e-07, 1.0000e+00],
        [8.4380e-05, 9.9992e-01],
        [8.4359e-08, 1.0000e+00],
        [9.9522e-01, 4.7794e-03],
        [8.2383e-04, 9.9918e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9722e-03, 9.9803e-01],
        [1.8498e-04, 9.9982e-01],
        [7.2824e-09, 1.0000e+00],
        [1.1881e-05, 9.9999e-01],
        [9.9967e-01, 3.2699e-04],
        [1.9215e-03, 9.9808e-01],
        [8.2415e-01, 1.7585e-01],
        [5.4497e-03, 9.9455e-01],
        [8.2771e-05, 9.9992e-01],
        [9.9983e-01, 1.7447e-04],
        [9.9993e-01, 6.8116e-05],
        [2.9677e-07, 1.0000e+00],
        [6.1635e-08, 1.0000e+00],
        [1.0435e-02, 9.8956e-01],
        [8.4398e-04, 9.9916e-01],
        [6.7541e-06, 9.9999e-01],
        [6.2435e-04, 9.9938e-01],
        [4.7630e-06, 1.0000e+00],
        [2.4893e-02, 9.7511e-01],
        [2.2115e-06, 1.0000e+00],
        [8.3500e-06, 9.9999e-01],
        [3.8831e-06, 1.0000e+00],
        [9.4250e-07, 1.0000e+00],
        [1.5570e-06, 1.0000e+00],
        [2.3399e-03, 9.9766e-01],
        [9.9383e-01, 6.1718e-03],
        [3.5951e-07, 1.0000e+00],
        [2.2333e-05, 9.9998e-01],
        [2.3682e-06, 1.0000e+00],
        [9.9974e-01, 2.5845e-04],
        [2.2345e-07, 1.0000e+00],
        [9.9797e-01, 2.0337e-03],
        [1.7062e-06, 1.0000e+00],
        [5.0838e-04, 9.9949e-01],
        [4.5256e-05, 9.9995e-01],
        [5.9259e-05, 9.9994e-01],
        [3.6595e-05, 9.9996e-01],
        [9.0906e-01, 9.0938e-02],
        [5.0120e-04, 9.9950e-01],
        [1.3187e-02, 9.8681e-01],
        [3.4494e-04, 9.9966e-01],
        [9.8983e-01, 1.0171e-02],
        [8.0311e-08, 1.0000e+00],
        [1.3523e-06, 1.0000e+00],
        [1.5675e-04, 9.9984e-01],
        [4.6805e-07, 1.0000e+00],
        [3.2732e-10, 1.0000e+00],
        [2.2481e-08, 1.0000e+00],
        [4.3932e-06, 1.0000e+00],
        [5.3713e-01, 4.6287e-01],
        [9.6435e-01, 3.5653e-02],
        [8.6817e-06, 9.9999e-01],
        [6.2335e-06, 9.9999e-01],
        [7.5914e-04, 9.9924e-01],
        [1.5377e-04, 9.9985e-01],
        [9.9508e-01, 4.9198e-03],
        [6.8825e-01, 3.1175e-01],
        [1.7668e-04, 9.9982e-01],
        [9.9620e-01, 3.7972e-03],
        [3.3807e-05, 9.9997e-01],
        [1.9400e-05, 9.9998e-01],
        [9.9986e-01, 1.4460e-04],
        [2.7437e-05, 9.9997e-01],
        [1.1874e-03, 9.9881e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9958e-01, 4.2307e-04],
        [9.9458e-01, 5.4175e-03],
        [1.2010e-04, 9.9988e-01],
        [1.5144e-07, 1.0000e+00],
        [9.9961e-01, 3.8940e-04],
        [5.1418e-06, 9.9999e-01],
        [6.4428e-05, 9.9994e-01],
        [6.7165e-03, 9.9328e-01],
        [9.9986e-01, 1.3546e-04],
        [8.3677e-01, 1.6323e-01],
        [1.3254e-04, 9.9987e-01],
        [5.9966e-06, 9.9999e-01],
        [6.2496e-05, 9.9994e-01],
        [3.3592e-05, 9.9997e-01],
        [9.9957e-01, 4.3474e-04],
        [2.8261e-05, 9.9997e-01],
        [1.2980e-04, 9.9987e-01],
        [3.0709e-06, 1.0000e+00],
        [1.0730e-04, 9.9989e-01],
        [9.9899e-01, 1.0139e-03],
        [3.7509e-04, 9.9962e-01],
        [2.2800e-03, 9.9772e-01],
        [1.2761e-05, 9.9999e-01],
        [3.3160e-02, 9.6684e-01],
        [2.0342e-01, 7.9658e-01],
        [4.2378e-04, 9.9958e-01],
        [4.8475e-06, 1.0000e+00],
        [3.5026e-04, 9.9965e-01],
        [1.1028e-02, 9.8897e-01],
        [2.4726e-04, 9.9975e-01],
        [3.8899e-03, 9.9611e-01],
        [1.0453e-03, 9.9895e-01],
        [3.5087e-02, 9.6491e-01],
        [3.2973e-07, 1.0000e+00],
        [5.9416e-05, 9.9994e-01],
        [3.3766e-05, 9.9997e-01],
        [5.5141e-03, 9.9449e-01],
        [6.0373e-07, 1.0000e+00],
        [4.5924e-03, 9.9541e-01],
        [5.2403e-05, 9.9995e-01],
        [2.8111e-08, 1.0000e+00],
        [8.9161e-01, 1.0839e-01],
        [6.0006e-02, 9.3999e-01],
        [3.2987e-07, 1.0000e+00],
        [1.9525e-04, 9.9980e-01],
        [3.3297e-03, 9.9667e-01],
        [6.3159e-06, 9.9999e-01],
        [7.9635e-08, 1.0000e+00],
        [1.0518e-03, 9.9895e-01],
        [9.9890e-01, 1.0968e-03],
        [6.5042e-08, 1.0000e+00],
        [1.6123e-07, 1.0000e+00],
        [9.9947e-01, 5.3140e-04],
        [3.5451e-02, 9.6455e-01],
        [2.3499e-07, 1.0000e+00],
        [1.4548e-10, 1.0000e+00],
        [2.9939e-04, 9.9970e-01],
        [1.5580e-04, 9.9984e-01],
        [1.5178e-08, 1.0000e+00],
        [1.1800e-06, 1.0000e+00],
        [9.9145e-01, 8.5533e-03],
        [1.4177e-04, 9.9986e-01],
        [9.9971e-01, 2.9307e-04],
        [7.5709e-04, 9.9924e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9996e-01, 3.5935e-05],
        [2.5011e-03, 9.9750e-01],
        [9.9969e-03, 9.9000e-01],
        [9.9981e-01, 1.8746e-04],
        [9.9651e-01, 3.4907e-03],
        [9.9944e-01, 5.5771e-04],
        [1.7014e-03, 9.9830e-01],
        [1.8484e-02, 9.8152e-01],
        [3.2958e-02, 9.6704e-01],
        [8.0816e-01, 1.9184e-01],
        [3.6483e-07, 1.0000e+00],
        [2.7509e-05, 9.9997e-01],
        [1.4338e-02, 9.8566e-01],
        [3.9895e-06, 1.0000e+00],
        [9.9972e-01, 2.7883e-04],
        [3.2340e-04, 9.9968e-01],
        [9.2417e-07, 1.0000e+00],
        [8.7206e-06, 9.9999e-01],
        [1.7764e-06, 1.0000e+00],
        [9.9971e-01, 2.9494e-04],
        [9.2396e-05, 9.9991e-01],
        [5.1985e-04, 9.9948e-01],
        [1.7805e-05, 9.9998e-01],
        [1.4688e-06, 1.0000e+00],
        [7.6221e-02, 9.2378e-01],
        [9.9912e-01, 8.8489e-04],
        [9.8907e-01, 1.0929e-02],
        [1.2962e-06, 1.0000e+00],
        [4.3433e-06, 1.0000e+00],
        [6.2543e-07, 1.0000e+00],
        [2.4992e-05, 9.9997e-01],
        [8.4305e-06, 9.9999e-01],
        [5.9243e-06, 9.9999e-01],
        [2.7892e-05, 9.9997e-01],
        [9.9683e-01, 3.1665e-03],
        [9.7732e-01, 2.2679e-02],
        [9.9624e-01, 3.7607e-03],
        [9.0174e-01, 9.8263e-02],
        [1.6878e-03, 9.9831e-01],
        [1.0277e-04, 9.9990e-01],
        [2.4596e-05, 9.9998e-01],
        [1.8428e-07, 1.0000e+00],
        [3.5956e-05, 9.9996e-01],
        [5.3916e-07, 1.0000e+00],
        [2.5314e-05, 9.9997e-01],
        [9.2985e-03, 9.9070e-01],
        [7.6297e-06, 9.9999e-01],
        [8.5924e-08, 1.0000e+00],
        [3.4773e-07, 1.0000e+00],
        [9.9935e-01, 6.5379e-04],
        [5.6079e-03, 9.9439e-01],
        [9.6172e-01, 3.8277e-02],
        [1.1515e-07, 1.0000e+00],
        [9.0064e-06, 9.9999e-01],
        [4.7505e-06, 1.0000e+00],
        [9.9909e-01, 9.1305e-04],
        [5.9067e-05, 9.9994e-01],
        [3.9235e-07, 1.0000e+00],
        [1.7205e-05, 9.9998e-01],
        [6.2299e-08, 1.0000e+00],
        [9.2805e-01, 7.1948e-02],
        [9.6626e-01, 3.3738e-02],
        [4.8148e-08, 1.0000e+00],
        [3.7630e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5712e-02, 9.8429e-01],
        [1.0940e-03, 9.9891e-01],
        [9.9629e-01, 3.7055e-03],
        [2.4362e-07, 1.0000e+00],
        [9.9945e-01, 5.5501e-04],
        [5.5863e-06, 9.9999e-01],
        [4.2427e-09, 1.0000e+00],
        [4.7019e-05, 9.9995e-01],
        [9.9868e-01, 1.3168e-03],
        [9.9969e-01, 3.1050e-04],
        [1.0893e-05, 9.9999e-01],
        [1.0861e-07, 1.0000e+00],
        [9.9974e-01, 2.6089e-04],
        [1.1250e-05, 9.9999e-01],
        [2.8593e-06, 1.0000e+00],
        [2.3453e-03, 9.9765e-01],
        [1.8275e-05, 9.9998e-01],
        [2.6652e-08, 1.0000e+00],
        [1.7608e-06, 1.0000e+00],
        [1.3461e-04, 9.9987e-01],
        [4.4513e-04, 9.9955e-01],
        [2.7136e-04, 9.9973e-01],
        [5.7325e-06, 9.9999e-01],
        [4.0048e-08, 1.0000e+00],
        [9.9983e-01, 1.7284e-04],
        [9.2338e-01, 7.6621e-02],
        [2.8445e-06, 1.0000e+00],
        [1.3461e-04, 9.9987e-01],
        [1.1996e-05, 9.9999e-01],
        [9.9950e-01, 4.9773e-04],
        [9.9447e-01, 5.5309e-03],
        [3.4016e-05, 9.9997e-01],
        [3.6449e-04, 9.9964e-01],
        [8.1397e-06, 9.9999e-01],
        [1.1471e-03, 9.9885e-01],
        [1.2281e-03, 9.9877e-01],
        [1.4291e-06, 1.0000e+00],
        [3.4256e-05, 9.9997e-01],
        [1.1304e-07, 1.0000e+00],
        [9.9954e-01, 4.5698e-04],
        [1.5281e-06, 1.0000e+00],
        [5.2691e-04, 9.9947e-01],
        [4.3215e-06, 1.0000e+00],
        [2.7887e-06, 1.0000e+00],
        [1.9737e-05, 9.9998e-01],
        [4.1205e-05, 9.9996e-01],
        [8.1139e-05, 9.9992e-01],
        [4.4471e-01, 5.5529e-01],
        [3.1186e-07, 1.0000e+00],
        [8.8533e-07, 1.0000e+00],
        [3.0063e-02, 9.6994e-01],
        [9.9947e-01, 5.2729e-04],
        [9.9994e-01, 5.9882e-05],
        [2.9932e-03, 9.9701e-01],
        [3.4256e-05, 9.9997e-01],
        [9.9976e-01, 2.4471e-04],
        [1.8683e-05, 9.9998e-01],
        [5.4401e-07, 1.0000e+00],
        [3.3453e-06, 1.0000e+00],
        [9.9665e-01, 3.3499e-03],
        [1.4869e-09, 1.0000e+00],
        [7.1092e-07, 1.0000e+00],
        [1.9741e-06, 1.0000e+00],
        [1.7429e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9945e-03, 9.9801e-01],
        [1.7665e-04, 9.9982e-01],
        [9.9971e-01, 2.8508e-04],
        [8.9803e-04, 9.9910e-01],
        [4.3552e-06, 1.0000e+00],
        [1.5230e-03, 9.9848e-01],
        [2.0343e-05, 9.9998e-01],
        [5.7974e-01, 4.2026e-01],
        [6.8609e-02, 9.3139e-01],
        [2.6276e-04, 9.9974e-01],
        [1.6295e-07, 1.0000e+00],
        [4.7307e-08, 1.0000e+00],
        [9.9825e-01, 1.7451e-03],
        [1.7028e-01, 8.2972e-01],
        [1.3181e-06, 1.0000e+00],
        [5.3550e-04, 9.9946e-01],
        [7.2398e-03, 9.9276e-01],
        [9.9965e-01, 3.4833e-04],
        [9.9445e-01, 5.5483e-03],
        [5.3415e-04, 9.9947e-01],
        [4.7208e-04, 9.9953e-01],
        [9.9961e-01, 3.9253e-04],
        [3.8952e-07, 1.0000e+00],
        [7.6976e-09, 1.0000e+00],
        [9.9958e-01, 4.1628e-04],
        [8.9743e-04, 9.9910e-01],
        [7.6147e-07, 1.0000e+00],
        [5.4451e-09, 1.0000e+00],
        [9.1108e-01, 8.8915e-02],
        [9.9922e-01, 7.7521e-04],
        [6.4621e-04, 9.9935e-01],
        [1.8000e-07, 1.0000e+00],
        [7.9004e-08, 1.0000e+00],
        [9.9821e-01, 1.7900e-03],
        [5.4193e-04, 9.9946e-01],
        [1.5045e-04, 9.9985e-01],
        [1.6188e-07, 1.0000e+00],
        [9.3242e-01, 6.7583e-02],
        [9.4159e-05, 9.9991e-01],
        [6.1428e-09, 1.0000e+00],
        [9.7921e-01, 2.0791e-02],
        [3.7367e-05, 9.9996e-01],
        [6.8581e-04, 9.9931e-01],
        [5.2002e-01, 4.7998e-01],
        [1.6177e-06, 1.0000e+00],
        [5.2968e-03, 9.9470e-01],
        [3.2749e-05, 9.9997e-01],
        [6.4855e-08, 1.0000e+00],
        [7.9194e-05, 9.9992e-01],
        [5.3754e-03, 9.9462e-01],
        [9.9962e-01, 3.7960e-04],
        [1.3007e-07, 1.0000e+00],
        [1.2414e-07, 1.0000e+00],
        [1.2402e-04, 9.9988e-01],
        [9.9582e-01, 4.1763e-03],
        [2.6278e-06, 1.0000e+00],
        [9.9926e-01, 7.4234e-04],
        [3.1713e-07, 1.0000e+00],
        [2.5097e-01, 7.4903e-01],
        [1.1907e-03, 9.9881e-01],
        [9.8759e-05, 9.9990e-01],
        [4.7643e-06, 1.0000e+00],
        [8.4368e-06, 9.9999e-01],
        [8.9302e-04, 9.9911e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5283e-04, 9.9985e-01],
        [2.9829e-07, 1.0000e+00],
        [2.7817e-04, 9.9972e-01],
        [5.4763e-05, 9.9995e-01],
        [9.5795e-07, 1.0000e+00],
        [1.1214e-07, 1.0000e+00],
        [4.9950e-05, 9.9995e-01],
        [3.8598e-07, 1.0000e+00],
        [9.9989e-01, 1.0749e-04],
        [2.0365e-07, 1.0000e+00],
        [9.9961e-01, 3.8809e-04],
        [3.4153e-07, 1.0000e+00],
        [9.8441e-01, 1.5591e-02],
        [9.9487e-01, 5.1348e-03],
        [4.1243e-03, 9.9588e-01],
        [9.0094e-01, 9.9064e-02],
        [9.5739e-01, 4.2608e-02],
        [1.4858e-01, 8.5142e-01],
        [1.4219e-08, 1.0000e+00],
        [9.9995e-01, 4.5488e-05],
        [5.4941e-01, 4.5059e-01],
        [9.9541e-01, 4.5870e-03],
        [1.9051e-07, 1.0000e+00],
        [9.9989e-01, 1.1262e-04],
        [7.7314e-07, 1.0000e+00],
        [2.2527e-03, 9.9775e-01],
        [2.4952e-09, 1.0000e+00],
        [3.6420e-06, 1.0000e+00],
        [5.9837e-06, 9.9999e-01],
        [1.4256e-04, 9.9986e-01],
        [1.2666e-06, 1.0000e+00],
        [9.2736e-01, 7.2642e-02],
        [7.6794e-06, 9.9999e-01],
        [9.9940e-01, 6.0419e-04],
        [1.2644e-07, 1.0000e+00],
        [6.3675e-04, 9.9936e-01],
        [1.2130e-04, 9.9988e-01],
        [2.7954e-08, 1.0000e+00],
        [1.0000e+00, 3.0239e-06],
        [4.8078e-02, 9.5192e-01],
        [2.6262e-02, 9.7374e-01],
        [3.1707e-04, 9.9968e-01],
        [9.9983e-01, 1.6753e-04],
        [9.8838e-01, 1.1622e-02],
        [2.5426e-03, 9.9746e-01],
        [7.5364e-08, 1.0000e+00],
        [5.8576e-05, 9.9994e-01],
        [2.2509e-07, 1.0000e+00],
        [9.9961e-01, 3.9121e-04],
        [1.9083e-06, 1.0000e+00],
        [1.2445e-03, 9.9876e-01],
        [7.7910e-04, 9.9922e-01],
        [1.0247e-06, 1.0000e+00],
        [1.3934e-06, 1.0000e+00],
        [2.6058e-08, 1.0000e+00],
        [9.9919e-01, 8.1319e-04],
        [3.4748e-06, 1.0000e+00],
        [5.7561e-02, 9.4244e-01],
        [4.2069e-04, 9.9958e-01],
        [7.0776e-06, 9.9999e-01],
        [2.0852e-06, 1.0000e+00],
        [1.4694e-02, 9.8531e-01],
        [9.9994e-01, 6.4915e-05],
        [2.5701e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9970e-01, 2.9943e-04],
        [7.6856e-09, 1.0000e+00],
        [9.3412e-01, 6.5875e-02],
        [2.6416e-04, 9.9974e-01],
        [1.4666e-05, 9.9999e-01],
        [9.9995e-01, 4.7847e-05],
        [1.0080e-05, 9.9999e-01],
        [2.9686e-05, 9.9997e-01],
        [1.2264e-05, 9.9999e-01],
        [3.9776e-05, 9.9996e-01],
        [1.3147e-06, 1.0000e+00],
        [6.9763e-05, 9.9993e-01],
        [5.2463e-07, 1.0000e+00],
        [7.0911e-03, 9.9291e-01],
        [4.0946e-04, 9.9959e-01],
        [1.7949e-07, 1.0000e+00],
        [9.4398e-04, 9.9906e-01],
        [2.3734e-05, 9.9998e-01],
        [7.1175e-06, 9.9999e-01],
        [9.5008e-02, 9.0499e-01],
        [1.2613e-04, 9.9987e-01],
        [4.2958e-03, 9.9570e-01],
        [9.9925e-01, 7.4751e-04],
        [2.2047e-06, 1.0000e+00],
        [6.8655e-08, 1.0000e+00],
        [5.6301e-03, 9.9437e-01],
        [9.9985e-01, 1.4678e-04],
        [3.2309e-05, 9.9997e-01],
        [9.7857e-01, 2.1427e-02],
        [8.5455e-06, 9.9999e-01],
        [9.9524e-01, 4.7626e-03],
        [1.8574e-05, 9.9998e-01],
        [1.9552e-03, 9.9804e-01],
        [2.0747e-07, 1.0000e+00],
        [3.0156e-06, 1.0000e+00],
        [1.3147e-05, 9.9999e-01],
        [1.6254e-08, 1.0000e+00],
        [5.3261e-04, 9.9947e-01],
        [5.6334e-04, 9.9944e-01],
        [1.6460e-08, 1.0000e+00],
        [5.1818e-03, 9.9482e-01],
        [9.0839e-01, 9.1610e-02],
        [8.5850e-03, 9.9141e-01],
        [9.7735e-05, 9.9990e-01],
        [9.9382e-01, 6.1807e-03],
        [2.2192e-06, 1.0000e+00],
        [9.9879e-01, 1.2120e-03],
        [4.1595e-05, 9.9996e-01],
        [1.4349e-03, 9.9857e-01],
        [1.2439e-05, 9.9999e-01],
        [1.9865e-06, 1.0000e+00],
        [7.9659e-06, 9.9999e-01],
        [9.9880e-01, 1.1957e-03],
        [9.9945e-01, 5.5088e-04],
        [6.0447e-04, 9.9940e-01],
        [9.9814e-01, 1.8633e-03],
        [2.5397e-09, 1.0000e+00],
        [4.3436e-06, 1.0000e+00],
        [1.3874e-01, 8.6126e-01],
        [1.0327e-05, 9.9999e-01],
        [1.5382e-06, 1.0000e+00],
        [6.2940e-03, 9.9371e-01],
        [1.7786e-05, 9.9998e-01],
        [6.6589e-01, 3.3411e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9806e-01, 1.9355e-03],
        [2.1508e-03, 9.9785e-01],
        [9.0584e-05, 9.9991e-01],
        [4.9055e-01, 5.0945e-01],
        [6.0437e-05, 9.9994e-01],
        [8.0785e-04, 9.9919e-01],
        [3.0675e-04, 9.9969e-01],
        [1.3682e-04, 9.9986e-01],
        [3.0947e-07, 1.0000e+00],
        [2.6148e-06, 1.0000e+00],
        [1.6763e-08, 1.0000e+00],
        [7.0585e-07, 1.0000e+00],
        [3.7738e-07, 1.0000e+00],
        [6.6209e-07, 1.0000e+00],
        [3.7738e-07, 1.0000e+00],
        [9.9958e-01, 4.2200e-04],
        [9.9978e-01, 2.1517e-04],
        [1.5830e-05, 9.9998e-01],
        [2.0917e-04, 9.9979e-01],
        [1.5048e-02, 9.8495e-01],
        [1.6101e-04, 9.9984e-01],
        [4.1978e-06, 1.0000e+00],
        [2.7316e-07, 1.0000e+00],
        [2.5365e-07, 1.0000e+00],
        [1.1842e-05, 9.9999e-01],
        [3.1374e-06, 1.0000e+00],
        [9.9936e-01, 6.4223e-04],
        [9.9195e-01, 8.0497e-03],
        [1.0100e-07, 1.0000e+00],
        [7.1523e-01, 2.8477e-01],
        [4.5756e-06, 1.0000e+00],
        [2.9904e-07, 1.0000e+00],
        [8.4640e-05, 9.9992e-01],
        [1.3791e-04, 9.9986e-01],
        [9.9973e-01, 2.7072e-04],
        [9.9915e-01, 8.5122e-04],
        [3.2762e-04, 9.9967e-01],
        [4.8450e-04, 9.9952e-01],
        [2.1344e-02, 9.7866e-01],
        [1.0486e-08, 1.0000e+00],
        [1.1636e-03, 9.9884e-01],
        [1.8268e-08, 1.0000e+00],
        [1.4349e-02, 9.8565e-01],
        [9.8129e-01, 1.8707e-02],
        [2.2690e-04, 9.9977e-01],
        [9.1479e-01, 8.5206e-02],
        [1.7371e-05, 9.9998e-01],
        [9.9921e-01, 7.9167e-04],
        [1.3455e-02, 9.8654e-01],
        [1.4189e-03, 9.9858e-01],
        [9.5052e-01, 4.9483e-02],
        [2.0457e-06, 1.0000e+00],
        [6.0843e-05, 9.9994e-01],
        [2.7233e-05, 9.9997e-01],
        [5.4342e-08, 1.0000e+00],
        [4.6654e-07, 1.0000e+00],
        [9.9935e-01, 6.5089e-04],
        [1.0329e-05, 9.9999e-01],
        [9.9974e-01, 2.6410e-04],
        [6.9839e-07, 1.0000e+00],
        [8.5846e-04, 9.9914e-01],
        [9.9918e-01, 8.2452e-04],
        [9.9927e-01, 7.2751e-04],
        [9.9955e-01, 4.5470e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[8.5224e-01, 1.4776e-01],
        [9.9935e-01, 6.5469e-04],
        [9.9986e-01, 1.3645e-04],
        [1.0657e-07, 1.0000e+00],
        [3.8097e-04, 9.9962e-01],
        [9.9975e-01, 2.5402e-04],
        [9.9921e-01, 7.8640e-04],
        [3.7584e-03, 9.9624e-01],
        [1.5152e-06, 1.0000e+00],
        [4.7887e-01, 5.2113e-01],
        [1.7691e-03, 9.9823e-01],
        [2.1739e-02, 9.7826e-01],
        [9.9983e-01, 1.6991e-04],
        [6.0056e-04, 9.9940e-01],
        [3.2110e-05, 9.9997e-01],
        [1.0388e-03, 9.9896e-01],
        [9.5471e-01, 4.5288e-02],
        [8.4471e-06, 9.9999e-01],
        [2.0588e-07, 1.0000e+00],
        [6.2771e-05, 9.9994e-01],
        [3.2793e-08, 1.0000e+00],
        [3.3003e-05, 9.9997e-01],
        [4.9799e-06, 9.9999e-01],
        [9.2406e-01, 7.5943e-02],
        [4.3671e-05, 9.9996e-01],
        [2.6922e-07, 1.0000e+00],
        [4.7140e-04, 9.9953e-01],
        [9.9991e-01, 8.7034e-05],
        [1.0629e-06, 1.0000e+00],
        [1.0105e-07, 1.0000e+00],
        [6.1779e-06, 9.9999e-01],
        [1.3667e-08, 1.0000e+00],
        [1.0425e-05, 9.9999e-01],
        [7.5124e-07, 1.0000e+00],
        [1.7305e-05, 9.9998e-01],
        [3.3432e-04, 9.9967e-01],
        [9.9410e-04, 9.9901e-01],
        [3.6810e-05, 9.9996e-01],
        [2.2743e-05, 9.9998e-01],
        [4.2882e-06, 1.0000e+00],
        [2.6124e-05, 9.9997e-01],
        [5.2282e-08, 1.0000e+00],
        [5.6711e-08, 1.0000e+00],
        [3.1173e-01, 6.8827e-01],
        [4.0614e-05, 9.9996e-01],
        [2.2303e-09, 1.0000e+00],
        [7.1474e-07, 1.0000e+00],
        [9.9369e-01, 6.3084e-03],
        [2.2832e-04, 9.9977e-01],
        [9.2362e-04, 9.9908e-01],
        [9.9912e-01, 8.7615e-04],
        [8.9594e-05, 9.9991e-01],
        [6.8265e-08, 1.0000e+00],
        [5.4244e-01, 4.5756e-01],
        [9.9983e-01, 1.7214e-04],
        [4.5264e-05, 9.9995e-01],
        [5.0981e-07, 1.0000e+00],
        [2.6057e-06, 1.0000e+00],
        [9.0649e-05, 9.9991e-01],
        [1.3105e-04, 9.9987e-01],
        [1.4567e-06, 1.0000e+00],
        [2.0214e-06, 1.0000e+00],
        [9.9904e-01, 9.6437e-04],
        [4.7983e-03, 9.9520e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1259e-06, 1.0000e+00],
        [2.4366e-04, 9.9976e-01],
        [9.9983e-01, 1.7166e-04],
        [5.1522e-04, 9.9948e-01],
        [3.9676e-07, 1.0000e+00],
        [9.9942e-01, 5.7563e-04],
        [4.1114e-04, 9.9959e-01],
        [1.4428e-02, 9.8557e-01],
        [3.5274e-06, 1.0000e+00],
        [2.1881e-05, 9.9998e-01],
        [8.3457e-06, 9.9999e-01],
        [2.4038e-05, 9.9998e-01],
        [6.2029e-06, 9.9999e-01],
        [9.8355e-01, 1.6448e-02],
        [1.0352e-05, 9.9999e-01],
        [1.4789e-05, 9.9999e-01],
        [7.9200e-07, 1.0000e+00],
        [1.9360e-03, 9.9806e-01],
        [5.2920e-02, 9.4708e-01],
        [9.9994e-01, 6.0365e-05],
        [4.3673e-05, 9.9996e-01],
        [4.7816e-09, 1.0000e+00],
        [9.9952e-01, 4.8069e-04],
        [1.1683e-04, 9.9988e-01],
        [6.8070e-04, 9.9932e-01],
        [9.2167e-04, 9.9908e-01],
        [8.5348e-07, 1.0000e+00],
        [2.2344e-06, 1.0000e+00],
        [9.6688e-01, 3.3125e-02],
        [2.8533e-05, 9.9997e-01],
        [9.9913e-01, 8.6743e-04],
        [4.0229e-05, 9.9996e-01],
        [9.9773e-01, 2.2737e-03],
        [3.8718e-05, 9.9996e-01],
        [9.9586e-01, 4.1415e-03],
        [6.5693e-05, 9.9993e-01],
        [2.9191e-06, 1.0000e+00],
        [2.1242e-07, 1.0000e+00],
        [1.1420e-04, 9.9989e-01],
        [1.1864e-05, 9.9999e-01],
        [9.9944e-01, 5.6123e-04],
        [1.9616e-04, 9.9980e-01],
        [8.6518e-09, 1.0000e+00],
        [1.1362e-05, 9.9999e-01],
        [2.4616e-07, 1.0000e+00],
        [4.0300e-03, 9.9597e-01],
        [6.1528e-07, 1.0000e+00],
        [9.9835e-03, 9.9002e-01],
        [9.9315e-01, 6.8536e-03],
        [4.9183e-03, 9.9508e-01],
        [1.9196e-05, 9.9998e-01],
        [2.5642e-06, 1.0000e+00],
        [9.9498e-01, 5.0178e-03],
        [9.9783e-01, 2.1741e-03],
        [9.9956e-01, 4.4058e-04],
        [9.9847e-01, 1.5271e-03],
        [1.1755e-01, 8.8245e-01],
        [3.4428e-05, 9.9997e-01],
        [1.4187e-05, 9.9999e-01],
        [9.9950e-01, 5.0051e-04],
        [1.2524e-02, 9.8748e-01],
        [1.9919e-07, 1.0000e+00],
        [7.2350e-08, 1.0000e+00],
        [7.9658e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0355e-04, 9.9990e-01],
        [4.8077e-02, 9.5192e-01],
        [9.9891e-01, 1.0893e-03],
        [3.2977e-04, 9.9967e-01],
        [1.0389e-07, 1.0000e+00],
        [1.0245e-06, 1.0000e+00],
        [9.9927e-01, 7.3460e-04],
        [4.9364e-05, 9.9995e-01],
        [9.9984e-01, 1.5874e-04],
        [9.8985e-01, 1.0155e-02],
        [8.7651e-07, 1.0000e+00],
        [3.9599e-07, 1.0000e+00],
        [1.8188e-02, 9.8181e-01],
        [1.2768e-05, 9.9999e-01],
        [8.5575e-01, 1.4425e-01],
        [1.2619e-03, 9.9874e-01],
        [3.4291e-08, 1.0000e+00],
        [1.6248e-05, 9.9998e-01],
        [9.9967e-01, 3.3437e-04],
        [5.0123e-05, 9.9995e-01],
        [9.9273e-01, 7.2745e-03],
        [9.9990e-01, 9.9673e-05],
        [8.4009e-04, 9.9916e-01],
        [4.0062e-04, 9.9960e-01],
        [9.4974e-01, 5.0261e-02],
        [5.1100e-04, 9.9949e-01],
        [4.6768e-04, 9.9953e-01],
        [1.5644e-03, 9.9844e-01],
        [9.9583e-01, 4.1713e-03],
        [9.3880e-01, 6.1201e-02],
        [9.9816e-01, 1.8446e-03],
        [9.9976e-01, 2.4223e-04],
        [8.6943e-05, 9.9991e-01],
        [9.9954e-01, 4.5877e-04],
        [2.4383e-03, 9.9756e-01],
        [1.7057e-04, 9.9983e-01],
        [5.5731e-07, 1.0000e+00],
        [4.0849e-04, 9.9959e-01],
        [9.9978e-01, 2.1784e-04],
        [9.8027e-07, 1.0000e+00],
        [2.7028e-08, 1.0000e+00],
        [9.9954e-01, 4.6090e-04],
        [7.8428e-11, 1.0000e+00],
        [1.6969e-07, 1.0000e+00],
        [1.6820e-04, 9.9983e-01],
        [4.7478e-06, 1.0000e+00],
        [2.4470e-04, 9.9976e-01],
        [2.3079e-04, 9.9977e-01],
        [6.5694e-08, 1.0000e+00],
        [4.6891e-05, 9.9995e-01],
        [3.2154e-07, 1.0000e+00],
        [5.8691e-08, 1.0000e+00],
        [2.2731e-08, 1.0000e+00],
        [9.9575e-01, 4.2505e-03],
        [1.1042e-05, 9.9999e-01],
        [5.1446e-06, 9.9999e-01],
        [3.0487e-04, 9.9970e-01],
        [5.2075e-08, 1.0000e+00],
        [1.4020e-04, 9.9986e-01],
        [9.9662e-01, 3.3814e-03],
        [1.5352e-03, 9.9846e-01],
        [2.1068e-05, 9.9998e-01],
        [9.9939e-01, 6.0554e-04],
        [3.0718e-04, 9.9969e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0623e-05, 9.9999e-01],
        [9.9976e-01, 2.3769e-04],
        [8.5194e-06, 9.9999e-01],
        [9.9974e-01, 2.5811e-04],
        [9.9635e-01, 3.6546e-03],
        [5.0297e-05, 9.9995e-01],
        [1.5983e-06, 1.0000e+00],
        [9.9991e-01, 9.1875e-05],
        [1.8179e-07, 1.0000e+00],
        [2.0100e-08, 1.0000e+00],
        [5.2273e-06, 9.9999e-01],
        [5.7187e-04, 9.9943e-01],
        [9.8686e-01, 1.3145e-02],
        [3.1222e-05, 9.9997e-01],
        [4.6782e-09, 1.0000e+00],
        [1.0821e-06, 1.0000e+00],
        [2.5994e-07, 1.0000e+00],
        [3.4966e-05, 9.9997e-01],
        [3.8089e-06, 1.0000e+00],
        [9.8225e-07, 1.0000e+00],
        [2.5452e-06, 1.0000e+00],
        [3.7538e-05, 9.9996e-01],
        [2.2572e-08, 1.0000e+00],
        [8.5937e-06, 9.9999e-01],
        [3.5749e-03, 9.9643e-01],
        [8.2389e-06, 9.9999e-01],
        [2.7432e-02, 9.7257e-01],
        [9.3911e-01, 6.0887e-02],
        [1.1648e-02, 9.8835e-01],
        [9.9974e-01, 2.6273e-04],
        [6.9835e-05, 9.9993e-01],
        [9.9567e-01, 4.3346e-03],
        [2.1703e-05, 9.9998e-01],
        [9.9964e-01, 3.5709e-04],
        [4.8378e-05, 9.9995e-01],
        [8.6989e-01, 1.3011e-01],
        [3.9906e-05, 9.9996e-01],
        [1.3197e-04, 9.9987e-01],
        [5.5505e-08, 1.0000e+00],
        [3.5268e-05, 9.9996e-01],
        [4.7318e-06, 1.0000e+00],
        [3.4056e-05, 9.9997e-01],
        [4.8197e-05, 9.9995e-01],
        [7.7247e-08, 1.0000e+00],
        [3.2693e-05, 9.9997e-01],
        [1.3872e-02, 9.8613e-01],
        [9.9925e-01, 7.5371e-04],
        [9.0107e-05, 9.9991e-01],
        [6.6887e-03, 9.9331e-01],
        [1.6437e-08, 1.0000e+00],
        [9.9906e-01, 9.3976e-04],
        [9.3272e-04, 9.9907e-01],
        [1.6256e-06, 1.0000e+00],
        [1.6126e-04, 9.9984e-01],
        [1.2264e-08, 1.0000e+00],
        [4.2389e-03, 9.9576e-01],
        [5.7049e-04, 9.9943e-01],
        [1.9070e-04, 9.9981e-01],
        [5.1586e-08, 1.0000e+00],
        [8.7904e-07, 1.0000e+00],
        [2.0721e-04, 9.9979e-01],
        [1.0694e-04, 9.9989e-01],
        [7.6472e-06, 9.9999e-01],
        [1.0816e-04, 9.9989e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0801e-07, 1.0000e+00],
        [9.9814e-01, 1.8581e-03],
        [1.1475e-05, 9.9999e-01],
        [9.9921e-01, 7.9409e-04],
        [9.9987e-01, 1.3022e-04],
        [2.6847e-03, 9.9732e-01],
        [3.5187e-03, 9.9648e-01],
        [2.4343e-03, 9.9757e-01],
        [1.2957e-01, 8.7043e-01],
        [9.9177e-01, 8.2348e-03],
        [2.2985e-06, 1.0000e+00],
        [8.7547e-07, 1.0000e+00],
        [4.5499e-05, 9.9995e-01],
        [7.9882e-02, 9.2012e-01],
        [1.7078e-08, 1.0000e+00],
        [1.5023e-03, 9.9850e-01],
        [6.4920e-06, 9.9999e-01],
        [5.6409e-07, 1.0000e+00],
        [3.0225e-05, 9.9997e-01],
        [5.1126e-03, 9.9489e-01],
        [3.6273e-04, 9.9964e-01],
        [7.6188e-08, 1.0000e+00],
        [9.9978e-01, 2.1677e-04],
        [3.3477e-08, 1.0000e+00],
        [8.1424e-01, 1.8576e-01],
        [9.9982e-01, 1.8489e-04],
        [5.9878e-03, 9.9401e-01],
        [2.6494e-04, 9.9974e-01],
        [6.0784e-03, 9.9392e-01],
        [4.8275e-01, 5.1725e-01],
        [2.0356e-02, 9.7964e-01],
        [1.9084e-05, 9.9998e-01],
        [5.3921e-03, 9.9461e-01],
        [4.2908e-04, 9.9957e-01],
        [1.2296e-06, 1.0000e+00],
        [3.1999e-03, 9.9680e-01],
        [2.2032e-03, 9.9780e-01],
        [7.1833e-06, 9.9999e-01],
        [1.7813e-06, 1.0000e+00],
        [9.9843e-01, 1.5696e-03],
        [1.4025e-05, 9.9999e-01],
        [6.1124e-07, 1.0000e+00],
        [5.4423e-07, 1.0000e+00],
        [9.9982e-01, 1.8210e-04],
        [3.4086e-06, 1.0000e+00],
        [4.0399e-04, 9.9960e-01],
        [1.8484e-03, 9.9815e-01],
        [9.8202e-01, 1.7978e-02],
        [7.0203e-08, 1.0000e+00],
        [9.9939e-01, 6.0572e-04],
        [7.5997e-04, 9.9924e-01],
        [5.6793e-06, 9.9999e-01],
        [7.5033e-01, 2.4967e-01],
        [9.7970e-03, 9.9020e-01],
        [7.6187e-07, 1.0000e+00],
        [9.5188e-04, 9.9905e-01],
        [2.6752e-04, 9.9973e-01],
        [9.9939e-01, 6.1043e-04],
        [9.9987e-01, 1.2785e-04],
        [9.9972e-01, 2.8200e-04],
        [9.8528e-05, 9.9990e-01],
        [9.4815e-01, 5.1852e-02],
        [2.8368e-06, 1.0000e+00],
        [8.5108e-04, 9.9915e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8101e-05, 9.9998e-01],
        [1.6235e-03, 9.9838e-01],
        [8.1622e-06, 9.9999e-01],
        [5.0243e-04, 9.9950e-01],
        [3.5656e-05, 9.9996e-01],
        [3.0287e-04, 9.9970e-01],
        [9.9977e-01, 2.3392e-04],
        [1.2716e-09, 1.0000e+00],
        [3.5615e-05, 9.9996e-01],
        [2.3751e-05, 9.9998e-01],
        [9.8005e-01, 1.9952e-02],
        [7.6256e-07, 1.0000e+00],
        [3.7740e-04, 9.9962e-01],
        [2.0660e-06, 1.0000e+00],
        [2.5679e-02, 9.7432e-01],
        [5.1527e-03, 9.9485e-01],
        [1.9406e-04, 9.9981e-01],
        [6.4677e-06, 9.9999e-01],
        [9.9856e-01, 1.4427e-03],
        [3.0742e-05, 9.9997e-01],
        [5.7841e-01, 4.2159e-01],
        [9.9978e-01, 2.2361e-04],
        [2.1628e-04, 9.9978e-01],
        [9.9964e-01, 3.5633e-04],
        [6.6283e-04, 9.9934e-01],
        [9.9965e-01, 3.4768e-04],
        [3.9843e-09, 1.0000e+00],
        [8.5232e-05, 9.9991e-01],
        [2.7213e-03, 9.9728e-01],
        [9.9953e-01, 4.6769e-04],
        [5.4779e-02, 9.4522e-01],
        [9.9964e-01, 3.6242e-04],
        [1.4277e-05, 9.9999e-01],
        [8.7263e-01, 1.2737e-01],
        [9.9964e-01, 3.6354e-04],
        [2.8768e-01, 7.1232e-01],
        [4.3841e-03, 9.9562e-01],
        [8.6704e-04, 9.9913e-01],
        [9.7568e-04, 9.9902e-01],
        [1.5730e-03, 9.9843e-01],
        [7.1485e-01, 2.8515e-01],
        [3.2195e-02, 9.6781e-01],
        [1.0870e-03, 9.9891e-01],
        [4.1693e-01, 5.8307e-01],
        [7.0691e-07, 1.0000e+00],
        [9.8950e-01, 1.0498e-02],
        [7.0812e-03, 9.9292e-01],
        [5.6408e-01, 4.3592e-01],
        [9.4373e-02, 9.0563e-01],
        [8.7134e-01, 1.2866e-01],
        [3.3461e-04, 9.9967e-01],
        [3.2453e-07, 1.0000e+00],
        [8.6208e-06, 9.9999e-01],
        [3.3860e-06, 1.0000e+00],
        [2.4861e-02, 9.7514e-01],
        [1.7966e-03, 9.9820e-01],
        [1.0364e-04, 9.9990e-01],
        [3.8632e-02, 9.6137e-01],
        [9.9751e-01, 2.4948e-03],
        [1.1354e-07, 1.0000e+00],
        [5.3844e-07, 1.0000e+00],
        [9.9991e-01, 9.4041e-05],
        [9.1045e-07, 1.0000e+00],
        [2.6216e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1957e-04, 9.9988e-01],
        [7.2689e-05, 9.9993e-01],
        [6.6366e-03, 9.9336e-01],
        [2.8677e-07, 1.0000e+00],
        [4.6358e-05, 9.9995e-01],
        [2.5747e-05, 9.9997e-01],
        [1.0305e-01, 8.9695e-01],
        [9.9840e-01, 1.6027e-03],
        [9.9748e-01, 2.5163e-03],
        [2.9804e-03, 9.9702e-01],
        [3.6685e-05, 9.9996e-01],
        [7.0351e-01, 2.9649e-01],
        [1.3792e-06, 1.0000e+00],
        [1.3364e-05, 9.9999e-01],
        [4.8971e-06, 1.0000e+00],
        [8.3621e-05, 9.9992e-01],
        [2.4762e-04, 9.9975e-01],
        [4.2864e-09, 1.0000e+00],
        [4.2470e-07, 1.0000e+00],
        [9.9922e-01, 7.8035e-04],
        [1.4845e-06, 1.0000e+00],
        [9.9989e-01, 1.1114e-04],
        [3.7939e-05, 9.9996e-01],
        [1.3721e-05, 9.9999e-01],
        [1.6142e-03, 9.9839e-01],
        [9.9955e-01, 4.5167e-04],
        [1.9375e-04, 9.9981e-01],
        [2.3897e-04, 9.9976e-01],
        [4.6980e-07, 1.0000e+00],
        [1.1068e-06, 1.0000e+00],
        [3.3704e-03, 9.9663e-01],
        [9.2431e-01, 7.5686e-02],
        [8.8514e-01, 1.1486e-01],
        [4.4576e-02, 9.5542e-01],
        [9.9577e-01, 4.2338e-03],
        [5.8807e-07, 1.0000e+00],
        [1.1553e-06, 1.0000e+00],
        [2.1313e-06, 1.0000e+00],
        [2.7648e-07, 1.0000e+00],
        [1.7123e-05, 9.9998e-01],
        [6.3272e-05, 9.9994e-01],
        [4.0473e-08, 1.0000e+00],
        [9.8734e-01, 1.2662e-02],
        [4.9381e-03, 9.9506e-01],
        [9.9876e-01, 1.2359e-03],
        [1.0768e-04, 9.9989e-01],
        [9.9315e-03, 9.9007e-01],
        [5.2567e-03, 9.9474e-01],
        [8.5361e-01, 1.4639e-01],
        [6.7846e-06, 9.9999e-01],
        [6.8232e-09, 1.0000e+00],
        [1.2710e-04, 9.9987e-01],
        [9.6030e-04, 9.9904e-01],
        [6.7038e-05, 9.9993e-01],
        [9.2720e-08, 1.0000e+00],
        [8.8284e-07, 1.0000e+00],
        [6.4892e-05, 9.9994e-01],
        [9.9747e-01, 2.5346e-03],
        [1.1333e-02, 9.8867e-01],
        [9.9948e-01, 5.1650e-04],
        [5.7527e-02, 9.4247e-01],
        [9.9911e-01, 8.9065e-04],
        [9.9979e-01, 2.1173e-04],
        [7.0050e-03, 9.9300e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8987e-03, 9.9010e-01],
        [1.3802e-05, 9.9999e-01],
        [1.7320e-05, 9.9998e-01],
        [1.5236e-06, 1.0000e+00],
        [3.6993e-05, 9.9996e-01],
        [3.0543e-03, 9.9695e-01],
        [1.3392e-01, 8.6608e-01],
        [9.9949e-01, 5.0852e-04],
        [3.3442e-06, 1.0000e+00],
        [2.0315e-05, 9.9998e-01],
        [9.9985e-01, 1.5046e-04],
        [4.3730e-07, 1.0000e+00],
        [7.8110e-05, 9.9992e-01],
        [5.8015e-03, 9.9420e-01],
        [9.9545e-01, 4.5524e-03],
        [5.6145e-05, 9.9994e-01],
        [9.9849e-01, 1.5087e-03],
        [1.5573e-02, 9.8443e-01],
        [4.4767e-04, 9.9955e-01],
        [1.0000e+00, 1.9736e-06],
        [9.3538e-02, 9.0646e-01],
        [6.7736e-07, 1.0000e+00],
        [9.9973e-01, 2.6667e-04],
        [2.2870e-06, 1.0000e+00],
        [6.7232e-04, 9.9933e-01],
        [9.9947e-01, 5.3041e-04],
        [2.0099e-03, 9.9799e-01],
        [2.9885e-06, 1.0000e+00],
        [1.3712e-02, 9.8629e-01],
        [1.3265e-04, 9.9987e-01],
        [1.2148e-04, 9.9988e-01],
        [2.0296e-06, 1.0000e+00],
        [9.9947e-01, 5.3127e-04],
        [1.3549e-02, 9.8645e-01],
        [1.4255e-02, 9.8574e-01],
        [2.8032e-03, 9.9720e-01],
        [2.7601e-06, 1.0000e+00],
        [3.6939e-03, 9.9631e-01],
        [6.6147e-07, 1.0000e+00],
        [8.6557e-05, 9.9991e-01],
        [2.5254e-05, 9.9997e-01],
        [9.9943e-01, 5.6773e-04],
        [3.2554e-06, 1.0000e+00],
        [2.7733e-06, 1.0000e+00],
        [9.9979e-01, 2.0640e-04],
        [2.4952e-04, 9.9975e-01],
        [4.6017e-04, 9.9954e-01],
        [3.7954e-06, 1.0000e+00],
        [4.6989e-05, 9.9995e-01],
        [2.3609e-02, 9.7639e-01],
        [3.6019e-05, 9.9996e-01],
        [7.0800e-05, 9.9993e-01],
        [2.1307e-04, 9.9979e-01],
        [2.5283e-04, 9.9975e-01],
        [6.1882e-07, 1.0000e+00],
        [9.9929e-01, 7.1060e-04],
        [9.3158e-06, 9.9999e-01],
        [9.9883e-01, 1.1683e-03],
        [1.2541e-07, 1.0000e+00],
        [2.9341e-03, 9.9707e-01],
        [5.3731e-07, 1.0000e+00],
        [2.6509e-04, 9.9973e-01],
        [9.9991e-01, 9.4588e-05],
        [8.6807e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.7462e-08, 1.0000e+00],
        [2.9577e-07, 1.0000e+00],
        [6.0262e-05, 9.9994e-01],
        [2.7071e-02, 9.7293e-01],
        [5.5613e-08, 1.0000e+00],
        [1.2186e-05, 9.9999e-01],
        [2.0101e-04, 9.9980e-01],
        [9.9879e-01, 1.2147e-03],
        [1.3121e-04, 9.9987e-01],
        [3.6947e-06, 1.0000e+00],
        [1.9676e-05, 9.9998e-01],
        [1.1297e-02, 9.8870e-01],
        [7.3927e-05, 9.9993e-01],
        [3.0476e-05, 9.9997e-01],
        [1.6611e-01, 8.3389e-01],
        [1.4908e-07, 1.0000e+00],
        [9.9945e-01, 5.4887e-04],
        [1.1769e-08, 1.0000e+00],
        [7.8842e-01, 2.1158e-01],
        [3.1317e-01, 6.8683e-01],
        [1.0250e-04, 9.9990e-01],
        [2.3893e-08, 1.0000e+00],
        [3.3778e-08, 1.0000e+00],
        [2.9674e-08, 1.0000e+00],
        [2.9854e-05, 9.9997e-01],
        [5.3881e-05, 9.9995e-01],
        [9.0103e-04, 9.9910e-01],
        [1.3879e-06, 1.0000e+00],
        [9.9998e-01, 2.3038e-05],
        [6.1022e-09, 1.0000e+00],
        [9.8382e-01, 1.6179e-02],
        [9.9378e-01, 6.2196e-03],
        [7.2318e-11, 1.0000e+00],
        [9.2406e-08, 1.0000e+00],
        [4.2764e-02, 9.5724e-01],
        [2.1546e-03, 9.9785e-01],
        [1.9924e-07, 1.0000e+00],
        [5.1974e-10, 1.0000e+00],
        [3.1944e-05, 9.9997e-01],
        [3.9682e-04, 9.9960e-01],
        [9.9983e-01, 1.7044e-04],
        [1.2349e-05, 9.9999e-01],
        [7.9854e-06, 9.9999e-01],
        [1.8136e-07, 1.0000e+00],
        [4.9735e-08, 1.0000e+00],
        [9.0958e-06, 9.9999e-01],
        [7.3525e-04, 9.9926e-01],
        [4.5753e-08, 1.0000e+00],
        [9.6092e-06, 9.9999e-01],
        [1.1894e-05, 9.9999e-01],
        [2.2778e-05, 9.9998e-01],
        [2.3659e-06, 1.0000e+00],
        [1.9813e-06, 1.0000e+00],
        [1.8032e-05, 9.9998e-01],
        [9.9969e-01, 3.0745e-04],
        [7.8554e-02, 9.2145e-01],
        [2.9168e-05, 9.9997e-01],
        [1.6484e-05, 9.9998e-01],
        [3.7591e-08, 1.0000e+00],
        [5.4474e-07, 1.0000e+00],
        [1.0173e-05, 9.9999e-01],
        [3.9554e-05, 9.9996e-01],
        [2.3460e-03, 9.9765e-01],
        [2.6309e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.2836e-01, 4.7164e-01],
        [1.3613e-06, 1.0000e+00],
        [2.3810e-04, 9.9976e-01],
        [8.4889e-06, 9.9999e-01],
        [1.0279e-03, 9.9897e-01],
        [1.3220e-06, 1.0000e+00],
        [2.5956e-06, 1.0000e+00],
        [3.4846e-11, 1.0000e+00],
        [9.9909e-01, 9.1223e-04],
        [1.3014e-06, 1.0000e+00],
        [9.9931e-01, 6.9153e-04],
        [9.9969e-01, 3.1297e-04],
        [2.0573e-04, 9.9979e-01],
        [9.8822e-01, 1.1781e-02],
        [3.2524e-04, 9.9967e-01],
        [3.7581e-04, 9.9962e-01],
        [9.9881e-01, 1.1899e-03],
        [2.9791e-05, 9.9997e-01],
        [7.0115e-08, 1.0000e+00],
        [8.1973e-04, 9.9918e-01],
        [9.9143e-01, 8.5710e-03],
        [9.9981e-01, 1.8538e-04],
        [9.9996e-01, 4.3669e-05],
        [1.4783e-02, 9.8522e-01],
        [3.3087e-07, 1.0000e+00],
        [6.3228e-05, 9.9994e-01],
        [9.9942e-01, 5.7610e-04],
        [4.0751e-02, 9.5925e-01],
        [2.5971e-05, 9.9997e-01],
        [1.8353e-08, 1.0000e+00],
        [1.6626e-06, 1.0000e+00],
        [2.4572e-04, 9.9975e-01],
        [9.9657e-01, 3.4301e-03],
        [9.8579e-01, 1.4212e-02],
        [9.7003e-01, 2.9965e-02],
        [2.8887e-06, 1.0000e+00],
        [1.8547e-04, 9.9981e-01],
        [2.1053e-05, 9.9998e-01],
        [1.1607e-03, 9.9884e-01],
        [6.2165e-03, 9.9378e-01],
        [2.2802e-04, 9.9977e-01],
        [9.9946e-01, 5.3614e-04],
        [9.9949e-01, 5.0605e-04],
        [2.3887e-04, 9.9976e-01],
        [9.2124e-07, 1.0000e+00],
        [2.1948e-06, 1.0000e+00],
        [9.9934e-01, 6.6467e-04],
        [9.9980e-01, 2.0099e-04],
        [8.1060e-01, 1.8940e-01],
        [1.1304e-03, 9.9887e-01],
        [1.4107e-02, 9.8589e-01],
        [8.6522e-08, 1.0000e+00],
        [8.6074e-07, 1.0000e+00],
        [6.6001e-08, 1.0000e+00],
        [5.4988e-09, 1.0000e+00],
        [3.5127e-04, 9.9965e-01],
        [9.9957e-01, 4.2629e-04],
        [6.0954e-05, 9.9994e-01],
        [1.4015e-05, 9.9999e-01],
        [7.5316e-06, 9.9999e-01],
        [9.9924e-01, 7.6488e-04],
        [9.9980e-01, 2.0112e-04],
        [9.9990e-01, 1.0156e-04],
        [1.5685e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.9679e-01, 2.0321e-01],
        [3.1468e-10, 1.0000e+00],
        [9.9851e-01, 1.4918e-03],
        [1.3981e-06, 1.0000e+00],
        [1.2626e-01, 8.7374e-01],
        [1.0787e-04, 9.9989e-01],
        [7.6162e-05, 9.9992e-01],
        [1.6606e-08, 1.0000e+00],
        [5.4558e-06, 9.9999e-01],
        [5.0749e-04, 9.9949e-01],
        [9.9926e-01, 7.3859e-04],
        [9.9272e-01, 7.2758e-03],
        [3.1450e-02, 9.6855e-01],
        [2.4429e-08, 1.0000e+00],
        [1.6763e-03, 9.9832e-01],
        [2.7865e-04, 9.9972e-01],
        [2.2625e-03, 9.9774e-01],
        [1.8663e-03, 9.9813e-01],
        [3.8141e-07, 1.0000e+00],
        [7.3462e-05, 9.9993e-01],
        [1.1145e-06, 1.0000e+00],
        [7.4667e-04, 9.9925e-01],
        [7.8829e-05, 9.9992e-01],
        [1.5336e-08, 1.0000e+00],
        [9.7906e-04, 9.9902e-01],
        [2.9349e-07, 1.0000e+00],
        [3.8104e-01, 6.1896e-01],
        [8.7596e-06, 9.9999e-01],
        [1.0535e-03, 9.9895e-01],
        [2.9426e-02, 9.7057e-01],
        [4.0135e-01, 5.9865e-01],
        [7.3048e-02, 9.2695e-01],
        [4.3237e-05, 9.9996e-01],
        [1.3297e-04, 9.9987e-01],
        [9.9804e-01, 1.9576e-03],
        [1.8509e-05, 9.9998e-01],
        [8.7687e-08, 1.0000e+00],
        [4.4814e-08, 1.0000e+00],
        [3.3558e-02, 9.6644e-01],
        [9.2921e-01, 7.0793e-02],
        [1.0453e-03, 9.9895e-01],
        [1.7988e-05, 9.9998e-01],
        [8.8717e-08, 1.0000e+00],
        [9.6844e-01, 3.1561e-02],
        [9.9942e-01, 5.7554e-04],
        [1.7604e-05, 9.9998e-01],
        [1.3502e-07, 1.0000e+00],
        [9.1673e-06, 9.9999e-01],
        [9.6979e-01, 3.0212e-02],
        [8.6980e-05, 9.9991e-01],
        [1.2541e-07, 1.0000e+00],
        [1.5434e-05, 9.9998e-01],
        [9.8869e-05, 9.9990e-01],
        [9.9964e-01, 3.6432e-04],
        [2.9852e-04, 9.9970e-01],
        [9.9927e-01, 7.3075e-04],
        [9.1839e-03, 9.9082e-01],
        [8.6039e-07, 1.0000e+00],
        [7.7033e-04, 9.9923e-01],
        [8.3092e-06, 9.9999e-01],
        [1.9530e-04, 9.9980e-01],
        [7.2241e-08, 1.0000e+00],
        [5.1258e-04, 9.9949e-01],
        [5.2540e-04, 9.9947e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.3005e-01, 6.9954e-02],
        [6.1161e-10, 1.0000e+00],
        [5.8767e-08, 1.0000e+00],
        [3.8719e-06, 1.0000e+00],
        [1.5787e-04, 9.9984e-01],
        [5.3139e-08, 1.0000e+00],
        [3.1002e-01, 6.8998e-01],
        [4.9740e-04, 9.9950e-01],
        [1.4637e-02, 9.8536e-01],
        [1.5519e-04, 9.9984e-01],
        [1.0000e+00, 4.5038e-06],
        [9.9836e-01, 1.6421e-03],
        [6.0533e-01, 3.9467e-01],
        [3.8014e-05, 9.9996e-01],
        [9.9846e-01, 1.5419e-03],
        [1.1725e-05, 9.9999e-01],
        [8.8320e-06, 9.9999e-01],
        [8.7233e-09, 1.0000e+00],
        [9.9973e-01, 2.7473e-04],
        [7.2571e-03, 9.9274e-01],
        [9.9783e-01, 2.1672e-03],
        [4.7398e-07, 1.0000e+00],
        [3.7124e-10, 1.0000e+00],
        [1.9379e-05, 9.9998e-01],
        [3.0697e-05, 9.9997e-01],
        [2.7258e-01, 7.2742e-01],
        [2.0464e-02, 9.7954e-01],
        [1.3558e-08, 1.0000e+00],
        [2.1993e-01, 7.8007e-01],
        [1.8683e-06, 1.0000e+00],
        [1.1711e-06, 1.0000e+00],
        [4.4893e-05, 9.9996e-01],
        [1.5108e-07, 1.0000e+00],
        [9.4176e-01, 5.8240e-02],
        [5.0465e-07, 1.0000e+00],
        [5.5443e-04, 9.9945e-01],
        [5.9393e-09, 1.0000e+00],
        [8.7601e-03, 9.9124e-01],
        [3.7426e-08, 1.0000e+00],
        [3.4457e-08, 1.0000e+00],
        [1.9464e-04, 9.9981e-01],
        [5.5108e-06, 9.9999e-01],
        [9.9910e-01, 9.0350e-04],
        [4.6612e-05, 9.9995e-01],
        [9.9904e-01, 9.5847e-04],
        [4.3053e-08, 1.0000e+00],
        [9.9125e-01, 8.7535e-03],
        [9.9534e-01, 4.6616e-03],
        [5.9364e-09, 1.0000e+00],
        [9.6216e-01, 3.7844e-02],
        [9.9942e-01, 5.8456e-04],
        [9.9991e-01, 8.9055e-05],
        [2.2802e-05, 9.9998e-01],
        [1.1902e-08, 1.0000e+00],
        [2.2058e-05, 9.9998e-01],
        [5.8715e-11, 1.0000e+00],
        [9.9862e-01, 1.3768e-03],
        [9.9998e-01, 2.2838e-05],
        [8.5688e-07, 1.0000e+00],
        [6.6076e-04, 9.9934e-01],
        [5.1293e-05, 9.9995e-01],
        [9.5810e-01, 4.1898e-02],
        [9.7565e-01, 2.4350e-02],
        [8.5002e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2065e-09, 1.0000e+00],
        [1.5774e-06, 1.0000e+00],
        [1.0477e-07, 1.0000e+00],
        [2.6627e-07, 1.0000e+00],
        [1.7728e-09, 1.0000e+00],
        [2.3128e-06, 1.0000e+00],
        [7.9197e-07, 1.0000e+00],
        [9.1074e-06, 9.9999e-01],
        [1.2736e-07, 1.0000e+00],
        [5.3369e-07, 1.0000e+00],
        [2.1190e-05, 9.9998e-01],
        [4.9845e-08, 1.0000e+00],
        [9.9644e-01, 3.5623e-03],
        [3.7408e-03, 9.9626e-01],
        [3.8934e-06, 1.0000e+00],
        [6.2652e-06, 9.9999e-01],
        [1.1207e-08, 1.0000e+00],
        [2.2728e-04, 9.9977e-01],
        [9.7667e-06, 9.9999e-01],
        [3.1164e-05, 9.9997e-01],
        [9.4642e-01, 5.3583e-02],
        [1.9658e-06, 1.0000e+00],
        [4.9056e-06, 1.0000e+00],
        [7.5913e-05, 9.9992e-01],
        [1.7879e-05, 9.9998e-01],
        [6.2084e-09, 1.0000e+00],
        [9.9939e-01, 6.0901e-04],
        [1.0886e-05, 9.9999e-01],
        [9.9978e-01, 2.2218e-04],
        [5.0116e-01, 4.9884e-01],
        [9.3952e-01, 6.0477e-02],
        [1.1791e-07, 1.0000e+00],
        [2.3017e-07, 1.0000e+00],
        [1.3067e-04, 9.9987e-01],
        [3.6164e-05, 9.9996e-01],
        [8.0448e-07, 1.0000e+00],
        [8.1329e-06, 9.9999e-01],
        [1.4540e-06, 1.0000e+00],
        [9.9977e-01, 2.3274e-04],
        [2.5045e-05, 9.9997e-01],
        [2.6783e-07, 1.0000e+00],
        [9.9966e-01, 3.3871e-04],
        [5.2174e-03, 9.9478e-01],
        [1.1459e-08, 1.0000e+00],
        [4.9549e-09, 1.0000e+00],
        [1.3465e-05, 9.9999e-01],
        [9.9901e-01, 9.9109e-04],
        [9.9979e-01, 2.1423e-04],
        [5.6218e-07, 1.0000e+00],
        [2.3229e-04, 9.9977e-01],
        [5.5120e-08, 1.0000e+00],
        [4.7686e-08, 1.0000e+00],
        [1.7966e-04, 9.9982e-01],
        [9.9908e-01, 9.2409e-04],
        [4.1050e-08, 1.0000e+00],
        [1.4347e-08, 1.0000e+00],
        [3.8298e-06, 1.0000e+00],
        [9.6554e-01, 3.4459e-02],
        [2.0542e-07, 1.0000e+00],
        [3.8030e-01, 6.1970e-01],
        [9.9948e-01, 5.1581e-04],
        [9.9823e-01, 1.7658e-03],
        [1.3031e-02, 9.8697e-01],
        [1.8485e-01, 8.1515e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5511e-03, 9.9845e-01],
        [1.6473e-03, 9.9835e-01],
        [9.4952e-05, 9.9990e-01],
        [9.9961e-01, 3.8932e-04],
        [9.9877e-01, 1.2348e-03],
        [2.3600e-07, 1.0000e+00],
        [1.3981e-02, 9.8602e-01],
        [6.1788e-05, 9.9994e-01],
        [1.6950e-02, 9.8305e-01],
        [1.7105e-03, 9.9829e-01],
        [3.4405e-06, 1.0000e+00],
        [4.3247e-04, 9.9957e-01],
        [2.6713e-05, 9.9997e-01],
        [6.2952e-03, 9.9370e-01],
        [1.4505e-04, 9.9985e-01],
        [1.0529e-04, 9.9989e-01],
        [5.2519e-08, 1.0000e+00],
        [9.9951e-01, 4.8784e-04],
        [1.4231e-06, 1.0000e+00],
        [1.6680e-02, 9.8332e-01],
        [1.5009e-02, 9.8499e-01],
        [2.1602e-10, 1.0000e+00],
        [3.2808e-07, 1.0000e+00],
        [2.0141e-06, 1.0000e+00],
        [4.6689e-04, 9.9953e-01],
        [2.1696e-08, 1.0000e+00],
        [1.6019e-07, 1.0000e+00],
        [9.9875e-01, 1.2494e-03],
        [2.8120e-04, 9.9972e-01],
        [2.7288e-06, 1.0000e+00],
        [9.9833e-01, 1.6738e-03],
        [4.6983e-04, 9.9953e-01],
        [6.7219e-04, 9.9933e-01],
        [4.7265e-07, 1.0000e+00],
        [8.3884e-06, 9.9999e-01],
        [1.7882e-07, 1.0000e+00],
        [3.5521e-02, 9.6448e-01],
        [3.0511e-07, 1.0000e+00],
        [6.7152e-05, 9.9993e-01],
        [8.2056e-06, 9.9999e-01],
        [4.8762e-04, 9.9951e-01],
        [9.9971e-01, 2.9349e-04],
        [9.9835e-01, 1.6500e-03],
        [4.1277e-07, 1.0000e+00],
        [5.7321e-05, 9.9994e-01],
        [1.5399e-04, 9.9985e-01],
        [1.1423e-07, 1.0000e+00],
        [4.5844e-09, 1.0000e+00],
        [3.5211e-01, 6.4789e-01],
        [9.9972e-01, 2.7813e-04],
        [4.6192e-04, 9.9954e-01],
        [9.9862e-01, 1.3821e-03],
        [6.4378e-07, 1.0000e+00],
        [1.3364e-09, 1.0000e+00],
        [2.6135e-03, 9.9739e-01],
        [1.9434e-05, 9.9998e-01],
        [2.8439e-09, 1.0000e+00],
        [1.2597e-08, 1.0000e+00],
        [8.9954e-05, 9.9991e-01],
        [5.0074e-07, 1.0000e+00],
        [6.5462e-06, 9.9999e-01],
        [9.9610e-01, 3.9022e-03],
        [9.9261e-01, 7.3880e-03],
        [2.9171e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9875e-01, 1.2519e-03],
        [2.8694e-06, 1.0000e+00],
        [3.6178e-06, 1.0000e+00],
        [8.7744e-06, 9.9999e-01],
        [1.2455e-07, 1.0000e+00],
        [1.4238e-02, 9.8576e-01],
        [3.8469e-06, 1.0000e+00],
        [6.0604e-04, 9.9939e-01],
        [1.4845e-05, 9.9999e-01],
        [9.9813e-01, 1.8657e-03],
        [9.9923e-01, 7.7242e-04],
        [5.9095e-09, 1.0000e+00],
        [8.0388e-05, 9.9992e-01],
        [9.9990e-01, 1.0398e-04],
        [9.9141e-01, 8.5878e-03],
        [9.8577e-01, 1.4232e-02],
        [2.3719e-05, 9.9998e-01],
        [6.0598e-07, 1.0000e+00],
        [3.8930e-07, 1.0000e+00],
        [1.3111e-04, 9.9987e-01],
        [9.9951e-01, 4.8785e-04],
        [9.7103e-01, 2.8968e-02],
        [9.8391e-01, 1.6092e-02],
        [3.1760e-06, 1.0000e+00],
        [1.5742e-08, 1.0000e+00],
        [2.2919e-06, 1.0000e+00],
        [9.9952e-01, 4.8172e-04],
        [1.1665e-04, 9.9988e-01],
        [4.4357e-04, 9.9956e-01],
        [4.7657e-06, 1.0000e+00],
        [6.9129e-05, 9.9993e-01],
        [8.8677e-05, 9.9991e-01],
        [4.6973e-01, 5.3027e-01],
        [7.9442e-01, 2.0558e-01],
        [8.1765e-01, 1.8235e-01],
        [9.9879e-07, 1.0000e+00],
        [8.6610e-01, 1.3390e-01],
        [4.0284e-04, 9.9960e-01],
        [9.9926e-01, 7.4389e-04],
        [9.9994e-01, 6.1995e-05],
        [9.4908e-01, 5.0921e-02],
        [7.4034e-01, 2.5966e-01],
        [3.6298e-06, 1.0000e+00],
        [1.1730e-03, 9.9883e-01],
        [9.9595e-01, 4.0523e-03],
        [1.2053e-06, 1.0000e+00],
        [9.9328e-01, 6.7158e-03],
        [6.2951e-04, 9.9937e-01],
        [9.9883e-01, 1.1680e-03],
        [9.8431e-04, 9.9902e-01],
        [1.0539e-05, 9.9999e-01],
        [9.9986e-01, 1.4495e-04],
        [9.8970e-01, 1.0296e-02],
        [8.3614e-06, 9.9999e-01],
        [1.0000e+00, 3.5841e-06],
        [1.6644e-05, 9.9998e-01],
        [2.7464e-07, 1.0000e+00],
        [7.0458e-05, 9.9993e-01],
        [9.9366e-01, 6.3423e-03],
        [4.5372e-04, 9.9955e-01],
        [8.7247e-08, 1.0000e+00],
        [9.9768e-01, 2.3208e-03],
        [4.7696e-07, 1.0000e+00],
        [9.2415e-01, 7.5851e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[3.1059e-04, 9.9969e-01],
        [4.4306e-07, 1.0000e+00],
        [9.9941e-01, 5.9300e-04],
        [4.7667e-09, 1.0000e+00],
        [5.9427e-07, 1.0000e+00],
        [9.9741e-01, 2.5914e-03],
        [4.3151e-04, 9.9957e-01],
        [1.1350e-06, 1.0000e+00],
        [2.2830e-06, 1.0000e+00],
        [3.2054e-08, 1.0000e+00],
        [9.9739e-01, 2.6090e-03],
        [9.5852e-01, 4.1478e-02],
        [9.3391e-01, 6.6090e-02],
        [1.8875e-08, 1.0000e+00],
        [7.5305e-02, 9.2469e-01],
        [1.0934e-07, 1.0000e+00],
        [8.4113e-04, 9.9916e-01],
        [9.9964e-01, 3.6088e-04],
        [1.8080e-07, 1.0000e+00],
        [6.0754e-06, 9.9999e-01],
        [5.9740e-07, 1.0000e+00],
        [7.1241e-07, 1.0000e+00],
        [2.3045e-02, 9.7696e-01],
        [2.5124e-02, 9.7488e-01],
        [3.1689e-02, 9.6831e-01],
        [9.0082e-07, 1.0000e+00],
        [9.9680e-01, 3.1984e-03],
        [2.9086e-04, 9.9971e-01],
        [9.2958e-08, 1.0000e+00],
        [8.3485e-01, 1.6515e-01],
        [9.9965e-01, 3.5164e-04],
        [9.9956e-01, 4.4453e-04],
        [9.9973e-01, 2.6942e-04],
        [3.7158e-09, 1.0000e+00],
        [8.0846e-07, 1.0000e+00],
        [1.8741e-06, 1.0000e+00],
        [7.2238e-05, 9.9993e-01],
        [9.9971e-01, 2.9462e-04],
        [8.4263e-10, 1.0000e+00],
        [6.4142e-04, 9.9936e-01],
        [3.3563e-06, 1.0000e+00],
        [6.4343e-07, 1.0000e+00],
        [9.9945e-01, 5.5468e-04],
        [6.8255e-05, 9.9993e-01],
        [7.9576e-01, 2.0424e-01],
        [3.0097e-04, 9.9970e-01],
        [2.7867e-08, 1.0000e+00],
        [3.0334e-04, 9.9970e-01],
        [1.0652e-08, 1.0000e+00],
        [8.0243e-09, 1.0000e+00],
        [1.0758e-01, 8.9242e-01],
        [1.1961e-06, 1.0000e+00],
        [9.9903e-01, 9.6895e-04],
        [7.4611e-07, 1.0000e+00],
        [1.1401e-06, 1.0000e+00],
        [2.4107e-06, 1.0000e+00],
        [1.3935e-04, 9.9986e-01],
        [9.8832e-01, 1.1680e-02],
        [4.4691e-06, 1.0000e+00],
        [1.3545e-03, 9.9865e-01],
        [1.1305e-05, 9.9999e-01],
        [7.7386e-01, 2.2614e-01],
        [2.2560e-03, 9.9774e-01],
        [5.2158e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9951e-01, 4.8989e-04],
        [9.9624e-01, 3.7586e-03],
        [7.5928e-09, 1.0000e+00],
        [7.1740e-06, 9.9999e-01],
        [3.9370e-09, 1.0000e+00],
        [5.7770e-06, 9.9999e-01],
        [2.4262e-06, 1.0000e+00],
        [6.1162e-08, 1.0000e+00],
        [3.3096e-05, 9.9997e-01],
        [1.3671e-05, 9.9999e-01],
        [3.2811e-05, 9.9997e-01],
        [8.0840e-09, 1.0000e+00],
        [1.4143e-08, 1.0000e+00],
        [3.9789e-04, 9.9960e-01],
        [2.2175e-08, 1.0000e+00],
        [3.6375e-07, 1.0000e+00],
        [9.9985e-01, 1.5069e-04],
        [1.6566e-05, 9.9998e-01],
        [9.7277e-01, 2.7232e-02],
        [2.9739e-08, 1.0000e+00],
        [5.9216e-04, 9.9941e-01],
        [1.4613e-07, 1.0000e+00],
        [1.6591e-07, 1.0000e+00],
        [2.8361e-05, 9.9997e-01],
        [6.3714e-07, 1.0000e+00],
        [9.9806e-01, 1.9361e-03],
        [3.1576e-04, 9.9968e-01],
        [3.9154e-08, 1.0000e+00],
        [1.0137e-01, 8.9863e-01],
        [2.1504e-06, 1.0000e+00],
        [3.4910e-04, 9.9965e-01],
        [9.9954e-01, 4.6031e-04],
        [9.6737e-06, 9.9999e-01],
        [1.9358e-06, 1.0000e+00],
        [5.2431e-05, 9.9995e-01],
        [4.0111e-07, 1.0000e+00],
        [9.6618e-01, 3.3820e-02],
        [1.6156e-06, 1.0000e+00],
        [9.9950e-01, 5.0272e-04],
        [1.5520e-04, 9.9984e-01],
        [3.2824e-10, 1.0000e+00],
        [8.7857e-01, 1.2143e-01],
        [1.3826e-03, 9.9862e-01],
        [9.8794e-01, 1.2062e-02],
        [3.4212e-07, 1.0000e+00],
        [2.4219e-03, 9.9758e-01],
        [7.7055e-07, 1.0000e+00],
        [3.8160e-05, 9.9996e-01],
        [2.3395e-06, 1.0000e+00],
        [8.8439e-09, 1.0000e+00],
        [9.9149e-01, 8.5119e-03],
        [1.0055e-04, 9.9990e-01],
        [1.4636e-08, 1.0000e+00],
        [2.6169e-01, 7.3831e-01],
        [1.8800e-07, 1.0000e+00],
        [9.0683e-07, 1.0000e+00],
        [1.6453e-04, 9.9984e-01],
        [9.8000e-04, 9.9902e-01],
        [9.9301e-01, 6.9934e-03],
        [5.3006e-05, 9.9995e-01],
        [4.5887e-07, 1.0000e+00],
        [5.8598e-06, 9.9999e-01],
        [2.7934e-02, 9.7207e-01],
        [3.5528e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1067e-05, 9.9998e-01],
        [1.1153e-05, 9.9999e-01],
        [1.4527e-06, 1.0000e+00],
        [2.3544e-08, 1.0000e+00],
        [2.0278e-08, 1.0000e+00],
        [1.6639e-06, 1.0000e+00],
        [9.9979e-01, 2.1113e-04],
        [7.7732e-01, 2.2268e-01],
        [9.9917e-01, 8.2544e-04],
        [2.2182e-07, 1.0000e+00],
        [9.9978e-01, 2.2185e-04],
        [2.1744e-06, 1.0000e+00],
        [7.1671e-01, 2.8329e-01],
        [1.2361e-06, 1.0000e+00],
        [7.9306e-01, 2.0694e-01],
        [2.6350e-06, 1.0000e+00],
        [1.0599e-06, 1.0000e+00],
        [1.8777e-04, 9.9981e-01],
        [3.6315e-05, 9.9996e-01],
        [2.9761e-07, 1.0000e+00],
        [1.2399e-05, 9.9999e-01],
        [3.4058e-07, 1.0000e+00],
        [9.9904e-01, 9.6379e-04],
        [9.9842e-01, 1.5764e-03],
        [5.7642e-04, 9.9942e-01],
        [5.9261e-07, 1.0000e+00],
        [8.9425e-04, 9.9911e-01],
        [3.1820e-05, 9.9997e-01],
        [9.9925e-01, 7.5132e-04],
        [4.4122e-07, 1.0000e+00],
        [3.7021e-05, 9.9996e-01],
        [4.4997e-09, 1.0000e+00],
        [3.1935e-06, 1.0000e+00],
        [9.9192e-05, 9.9990e-01],
        [4.9484e-04, 9.9951e-01],
        [2.7253e-05, 9.9997e-01],
        [9.9842e-01, 1.5777e-03],
        [1.0150e-02, 9.8985e-01],
        [1.7972e-05, 9.9998e-01],
        [4.8026e-04, 9.9952e-01],
        [5.5086e-03, 9.9449e-01],
        [9.8396e-06, 9.9999e-01],
        [9.5583e-05, 9.9990e-01],
        [9.9960e-01, 4.0045e-04],
        [7.5460e-07, 1.0000e+00],
        [9.8676e-01, 1.3240e-02],
        [9.9981e-01, 1.8556e-04],
        [1.5590e-01, 8.4410e-01],
        [1.3743e-07, 1.0000e+00],
        [4.8393e-09, 1.0000e+00],
        [2.2509e-04, 9.9977e-01],
        [6.7562e-07, 1.0000e+00],
        [2.9398e-05, 9.9997e-01],
        [8.7958e-05, 9.9991e-01],
        [7.7873e-04, 9.9922e-01],
        [1.4250e-05, 9.9999e-01],
        [3.6143e-05, 9.9996e-01],
        [4.5076e-03, 9.9549e-01],
        [2.1428e-05, 9.9998e-01],
        [8.1402e-05, 9.9992e-01],
        [9.8581e-01, 1.4194e-02],
        [1.0002e-04, 9.9990e-01],
        [2.6918e-04, 9.9973e-01],
        [1.7284e-02, 9.8272e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.3286e-07, 1.0000e+00],
        [1.0954e-02, 9.8905e-01],
        [4.9984e-08, 1.0000e+00],
        [9.9225e-01, 7.7482e-03],
        [3.7298e-04, 9.9963e-01],
        [1.5450e-08, 1.0000e+00],
        [9.9991e-01, 8.6509e-05],
        [2.7549e-06, 1.0000e+00],
        [4.0265e-06, 1.0000e+00],
        [9.9906e-01, 9.4392e-04],
        [6.5368e-03, 9.9346e-01],
        [1.7644e-04, 9.9982e-01],
        [1.0753e-05, 9.9999e-01],
        [9.9931e-01, 6.8874e-04],
        [1.6585e-06, 1.0000e+00],
        [2.4436e-04, 9.9976e-01],
        [4.3546e-10, 1.0000e+00],
        [7.9654e-08, 1.0000e+00],
        [5.6682e-05, 9.9994e-01],
        [9.9993e-01, 7.4189e-05],
        [1.3645e-06, 1.0000e+00],
        [3.8343e-05, 9.9996e-01],
        [3.5758e-07, 1.0000e+00],
        [8.8672e-10, 1.0000e+00],
        [9.9040e-08, 1.0000e+00],
        [9.9791e-01, 2.0894e-03],
        [7.3943e-08, 1.0000e+00],
        [4.3739e-06, 1.0000e+00],
        [1.8723e-05, 9.9998e-01],
        [1.2652e-06, 1.0000e+00],
        [9.9431e-01, 5.6934e-03],
        [1.6155e-08, 1.0000e+00],
        [3.6842e-05, 9.9996e-01],
        [4.9845e-04, 9.9950e-01],
        [9.9972e-01, 2.8428e-04],
        [2.1907e-07, 1.0000e+00],
        [8.7149e-01, 1.2851e-01],
        [9.9887e-01, 1.1310e-03],
        [3.7374e-03, 9.9626e-01],
        [1.3946e-04, 9.9986e-01],
        [9.9971e-01, 2.8689e-04],
        [1.5403e-06, 1.0000e+00],
        [5.5834e-07, 1.0000e+00],
        [3.4169e-06, 1.0000e+00],
        [3.1820e-06, 1.0000e+00],
        [2.3752e-08, 1.0000e+00],
        [3.9081e-03, 9.9609e-01],
        [9.9943e-01, 5.7359e-04],
        [1.0382e-03, 9.9896e-01],
        [9.9693e-01, 3.0704e-03],
        [1.3120e-04, 9.9987e-01],
        [2.5923e-08, 1.0000e+00],
        [2.1609e-04, 9.9978e-01],
        [9.9972e-01, 2.7528e-04],
        [1.0389e-02, 9.8961e-01],
        [9.7196e-01, 2.8040e-02],
        [6.7053e-07, 1.0000e+00],
        [2.1332e-08, 1.0000e+00],
        [1.2540e-04, 9.9987e-01],
        [7.4370e-07, 1.0000e+00],
        [3.5941e-10, 1.0000e+00],
        [1.0155e-04, 9.9990e-01],
        [7.0854e-05, 9.9993e-01],
        [9.7335e-01, 2.6650e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[6.3340e-03, 9.9367e-01],
        [1.1790e-04, 9.9988e-01],
        [1.0874e-04, 9.9989e-01],
        [1.6300e-07, 1.0000e+00],
        [6.9011e-01, 3.0989e-01],
        [2.8938e-03, 9.9711e-01],
        [7.9376e-07, 1.0000e+00],
        [9.9933e-01, 6.7329e-04],
        [9.9920e-01, 8.0228e-04],
        [9.6991e-05, 9.9990e-01],
        [1.4529e-03, 9.9855e-01],
        [2.4393e-06, 1.0000e+00],
        [9.9980e-01, 1.9941e-04],
        [5.1019e-05, 9.9995e-01],
        [9.9979e-01, 2.0920e-04],
        [9.6182e-05, 9.9990e-01],
        [4.5101e-02, 9.5490e-01],
        [7.5584e-04, 9.9924e-01],
        [2.6266e-04, 9.9974e-01],
        [6.9467e-10, 1.0000e+00],
        [2.5530e-07, 1.0000e+00],
        [9.9886e-05, 9.9990e-01],
        [7.7363e-04, 9.9923e-01],
        [6.5813e-04, 9.9934e-01],
        [1.1073e-05, 9.9999e-01],
        [3.3889e-04, 9.9966e-01],
        [2.9329e-03, 9.9707e-01],
        [5.0193e-09, 1.0000e+00],
        [3.1601e-06, 1.0000e+00],
        [6.4881e-06, 9.9999e-01],
        [3.4779e-04, 9.9965e-01],
        [4.9722e-05, 9.9995e-01],
        [9.9919e-01, 8.1477e-04],
        [9.7193e-05, 9.9990e-01],
        [1.7975e-06, 1.0000e+00],
        [4.1701e-02, 9.5830e-01],
        [9.9936e-01, 6.4100e-04],
        [9.9997e-01, 2.9625e-05],
        [9.9825e-01, 1.7528e-03],
        [8.2977e-05, 9.9992e-01],
        [9.2090e-09, 1.0000e+00],
        [9.8369e-01, 1.6310e-02],
        [9.9963e-01, 3.7179e-04],
        [1.5427e-05, 9.9998e-01],
        [2.3270e-06, 1.0000e+00],
        [9.5127e-05, 9.9990e-01],
        [1.0636e-05, 9.9999e-01],
        [1.0338e-04, 9.9990e-01],
        [9.7677e-01, 2.3227e-02],
        [3.2689e-05, 9.9997e-01],
        [1.0268e-01, 8.9732e-01],
        [5.9267e-02, 9.4073e-01],
        [1.8280e-06, 1.0000e+00],
        [9.4330e-01, 5.6697e-02],
        [6.9301e-01, 3.0699e-01],
        [9.2774e-08, 1.0000e+00],
        [8.4292e-04, 9.9916e-01],
        [1.9139e-08, 1.0000e+00],
        [1.0319e-05, 9.9999e-01],
        [3.5800e-08, 1.0000e+00],
        [4.2727e-03, 9.9573e-01],
        [5.7266e-05, 9.9994e-01],
        [1.7120e-03, 9.9829e-01],
        [7.3440e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1283e-04, 9.9979e-01],
        [3.9259e-05, 9.9996e-01],
        [2.9798e-02, 9.7020e-01],
        [8.8835e-04, 9.9911e-01],
        [8.6541e-04, 9.9913e-01],
        [8.9093e-05, 9.9991e-01],
        [9.9069e-01, 9.3080e-03],
        [4.7765e-05, 9.9995e-01],
        [1.5102e-06, 1.0000e+00],
        [1.0181e-02, 9.8982e-01],
        [8.7017e-03, 9.9130e-01],
        [9.9983e-01, 1.6871e-04],
        [2.4314e-04, 9.9976e-01],
        [5.9627e-06, 9.9999e-01],
        [3.9342e-05, 9.9996e-01],
        [2.2153e-03, 9.9778e-01],
        [6.2633e-06, 9.9999e-01],
        [2.9740e-09, 1.0000e+00],
        [1.6263e-04, 9.9984e-01],
        [1.3702e-03, 9.9863e-01],
        [6.0793e-05, 9.9994e-01],
        [3.2367e-04, 9.9968e-01],
        [5.3748e-07, 1.0000e+00],
        [5.9670e-06, 9.9999e-01],
        [9.9928e-01, 7.2352e-04],
        [9.2770e-04, 9.9907e-01],
        [9.9995e-01, 5.4405e-05],
        [1.2406e-04, 9.9988e-01],
        [1.6541e-08, 1.0000e+00],
        [5.6839e-05, 9.9994e-01],
        [2.3958e-08, 1.0000e+00],
        [7.0801e-08, 1.0000e+00],
        [8.4177e-04, 9.9916e-01],
        [5.7874e-04, 9.9942e-01],
        [9.8493e-01, 1.5071e-02],
        [6.1368e-08, 1.0000e+00],
        [3.4898e-05, 9.9997e-01],
        [2.4238e-03, 9.9758e-01],
        [3.2489e-08, 1.0000e+00],
        [5.9446e-03, 9.9406e-01],
        [1.0256e-01, 8.9744e-01],
        [4.7938e-07, 1.0000e+00],
        [8.5011e-01, 1.4989e-01],
        [1.5237e-06, 1.0000e+00],
        [9.9557e-01, 4.4318e-03],
        [7.4723e-06, 9.9999e-01],
        [4.7959e-08, 1.0000e+00],
        [9.9930e-01, 6.9505e-04],
        [7.5765e-07, 1.0000e+00],
        [4.1007e-02, 9.5899e-01],
        [9.9939e-01, 6.1299e-04],
        [1.9172e-02, 9.8083e-01],
        [6.4872e-05, 9.9994e-01],
        [7.1618e-07, 1.0000e+00],
        [1.4694e-04, 9.9985e-01],
        [8.1789e-03, 9.9182e-01],
        [2.4895e-03, 9.9751e-01],
        [3.1930e-03, 9.9681e-01],
        [3.1210e-04, 9.9969e-01],
        [9.9986e-01, 1.3762e-04],
        [7.5942e-06, 9.9999e-01],
        [1.4849e-03, 9.9852e-01],
        [2.1826e-04, 9.9978e-01],
        [5.4207e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.8506e-06, 9.9999e-01],
        [6.0285e-05, 9.9994e-01],
        [3.8681e-07, 1.0000e+00],
        [4.3855e-08, 1.0000e+00],
        [1.5444e-03, 9.9846e-01],
        [9.9966e-01, 3.4125e-04],
        [9.9737e-01, 2.6306e-03],
        [1.9138e-12, 1.0000e+00],
        [3.2264e-03, 9.9677e-01],
        [9.9909e-01, 9.0985e-04],
        [9.9808e-01, 1.9174e-03],
        [3.1091e-01, 6.8909e-01],
        [1.5066e-05, 9.9998e-01],
        [6.6779e-05, 9.9993e-01],
        [9.9988e-01, 1.1676e-04],
        [2.0341e-03, 9.9797e-01],
        [9.9983e-01, 1.6832e-04],
        [6.8375e-07, 1.0000e+00],
        [3.2122e-05, 9.9997e-01],
        [9.0103e-06, 9.9999e-01],
        [5.9764e-05, 9.9994e-01],
        [2.1403e-07, 1.0000e+00],
        [5.9562e-01, 4.0438e-01],
        [9.5586e-07, 1.0000e+00],
        [7.7581e-05, 9.9992e-01],
        [9.9985e-01, 1.5273e-04],
        [1.0047e-03, 9.9900e-01],
        [9.9992e-01, 8.1139e-05],
        [6.3264e-06, 9.9999e-01],
        [2.1149e-07, 1.0000e+00],
        [3.5693e-07, 1.0000e+00],
        [9.9988e-01, 1.2360e-04],
        [7.2651e-08, 1.0000e+00],
        [1.2029e-08, 1.0000e+00],
        [9.9968e-01, 3.1799e-04],
        [1.2249e-03, 9.9878e-01],
        [4.1949e-08, 1.0000e+00],
        [2.2530e-08, 1.0000e+00],
        [2.0960e-05, 9.9998e-01],
        [5.3922e-03, 9.9461e-01],
        [3.9704e-05, 9.9996e-01],
        [9.9969e-01, 3.0765e-04],
        [9.9989e-01, 1.0900e-04],
        [3.3791e-06, 1.0000e+00],
        [1.1711e-06, 1.0000e+00],
        [5.1423e-08, 1.0000e+00],
        [4.8800e-08, 1.0000e+00],
        [1.7435e-06, 1.0000e+00],
        [8.9081e-08, 1.0000e+00],
        [9.9997e-01, 2.9323e-05],
        [1.1046e-02, 9.8895e-01],
        [2.7356e-05, 9.9997e-01],
        [4.3713e-07, 1.0000e+00],
        [1.2527e-07, 1.0000e+00],
        [9.9939e-01, 6.1340e-04],
        [1.4863e-02, 9.8514e-01],
        [1.8112e-03, 9.9819e-01],
        [8.6752e-04, 9.9913e-01],
        [1.0749e-03, 9.9893e-01],
        [1.6071e-07, 1.0000e+00],
        [9.9983e-01, 1.7193e-04],
        [5.1288e-04, 9.9949e-01],
        [3.7214e-04, 9.9963e-01],
        [9.2089e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1123e-06, 1.0000e+00],
        [4.8057e-08, 1.0000e+00],
        [5.1012e-03, 9.9490e-01],
        [1.4901e-03, 9.9851e-01],
        [3.3412e-02, 9.6659e-01],
        [3.6227e-02, 9.6377e-01],
        [5.5591e-04, 9.9944e-01],
        [2.1990e-03, 9.9780e-01],
        [3.6193e-05, 9.9996e-01],
        [9.9963e-01, 3.7173e-04],
        [1.3867e-05, 9.9999e-01],
        [9.6210e-10, 1.0000e+00],
        [9.9962e-01, 3.8482e-04],
        [7.4068e-04, 9.9926e-01],
        [9.9960e-01, 4.0464e-04],
        [9.9962e-01, 3.7979e-04],
        [9.9609e-01, 3.9123e-03],
        [1.1482e-05, 9.9999e-01],
        [1.0965e-05, 9.9999e-01],
        [7.6036e-04, 9.9924e-01],
        [3.9649e-02, 9.6035e-01],
        [9.5723e-07, 1.0000e+00],
        [8.8835e-05, 9.9991e-01],
        [1.3862e-02, 9.8614e-01],
        [5.2861e-02, 9.4714e-01],
        [3.5059e-02, 9.6494e-01],
        [3.9701e-02, 9.6030e-01],
        [9.1558e-04, 9.9908e-01],
        [1.1238e-06, 1.0000e+00],
        [9.9980e-01, 2.0421e-04],
        [3.5635e-06, 1.0000e+00],
        [7.6409e-03, 9.9236e-01],
        [5.9918e-01, 4.0082e-01],
        [9.4857e-04, 9.9905e-01],
        [9.9966e-01, 3.4364e-04],
        [9.9959e-01, 4.1113e-04],
        [7.9297e-07, 1.0000e+00],
        [3.3585e-03, 9.9664e-01],
        [3.0116e-02, 9.6988e-01],
        [4.1500e-06, 1.0000e+00],
        [2.4986e-05, 9.9997e-01],
        [3.1399e-07, 1.0000e+00],
        [6.0642e-04, 9.9939e-01],
        [9.7110e-01, 2.8901e-02],
        [8.2499e-06, 9.9999e-01],
        [6.6594e-07, 1.0000e+00],
        [9.9950e-01, 5.0025e-04],
        [5.2824e-06, 9.9999e-01],
        [5.4969e-04, 9.9945e-01],
        [6.4701e-08, 1.0000e+00],
        [3.8254e-04, 9.9962e-01],
        [9.9986e-01, 1.3747e-04],
        [9.8770e-01, 1.2299e-02],
        [1.0823e-05, 9.9999e-01],
        [9.9250e-08, 1.0000e+00],
        [1.8576e-04, 9.9981e-01],
        [8.3968e-01, 1.6032e-01],
        [1.2722e-07, 1.0000e+00],
        [2.7159e-02, 9.7284e-01],
        [4.0503e-05, 9.9996e-01],
        [3.1536e-06, 1.0000e+00],
        [1.3467e-07, 1.0000e+00],
        [1.1281e-06, 1.0000e+00],
        [5.4466e-04, 9.9946e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1337e-03, 9.9887e-01],
        [9.6083e-09, 1.0000e+00],
        [3.4282e-05, 9.9997e-01],
        [5.6661e-02, 9.4334e-01],
        [2.8959e-05, 9.9997e-01],
        [4.6460e-04, 9.9954e-01],
        [3.2446e-08, 1.0000e+00],
        [8.3327e-05, 9.9992e-01],
        [9.9987e-01, 1.3032e-04],
        [9.9987e-01, 1.3007e-04],
        [1.3962e-04, 9.9986e-01],
        [2.3744e-06, 1.0000e+00],
        [2.7407e-02, 9.7259e-01],
        [2.3394e-05, 9.9998e-01],
        [9.9983e-01, 1.6914e-04],
        [3.2357e-05, 9.9997e-01],
        [3.9206e-03, 9.9608e-01],
        [4.8962e-06, 1.0000e+00],
        [9.9971e-01, 2.8809e-04],
        [2.4127e-04, 9.9976e-01],
        [2.7412e-07, 1.0000e+00],
        [1.6907e-08, 1.0000e+00],
        [1.1590e-04, 9.9988e-01],
        [6.7535e-05, 9.9993e-01],
        [5.3857e-04, 9.9946e-01],
        [2.0232e-02, 9.7977e-01],
        [9.9943e-01, 5.6709e-04],
        [9.9955e-01, 4.4893e-04],
        [1.1227e-03, 9.9888e-01],
        [2.0609e-03, 9.9794e-01],
        [3.3876e-04, 9.9966e-01],
        [2.6199e-02, 9.7380e-01],
        [9.9983e-01, 1.7079e-04],
        [7.5700e-05, 9.9992e-01],
        [9.9707e-01, 2.9337e-03],
        [1.6459e-02, 9.8354e-01],
        [2.0358e-03, 9.9796e-01],
        [8.3991e-01, 1.6009e-01],
        [6.6124e-05, 9.9993e-01],
        [9.9956e-01, 4.4301e-04],
        [1.1496e-01, 8.8504e-01],
        [4.3251e-04, 9.9957e-01],
        [1.3901e-07, 1.0000e+00],
        [9.8290e-01, 1.7103e-02],
        [1.8619e-08, 1.0000e+00],
        [3.3432e-05, 9.9997e-01],
        [6.5148e-01, 3.4852e-01],
        [6.8750e-06, 9.9999e-01],
        [1.0614e-07, 1.0000e+00],
        [4.5240e-05, 9.9995e-01],
        [1.5844e-05, 9.9998e-01],
        [3.1193e-04, 9.9969e-01],
        [9.2718e-01, 7.2816e-02],
        [3.1527e-04, 9.9968e-01],
        [3.8360e-01, 6.1640e-01],
        [5.0456e-08, 1.0000e+00],
        [9.9973e-01, 2.6907e-04],
        [1.8153e-04, 9.9982e-01],
        [3.4915e-09, 1.0000e+00],
        [9.9738e-01, 2.6166e-03],
        [1.7063e-07, 1.0000e+00],
        [9.6279e-01, 3.7215e-02],
        [1.5394e-03, 9.9846e-01],
        [9.9973e-01, 2.7399e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2218e-07, 1.0000e+00],
        [9.9952e-01, 4.7944e-04],
        [1.1567e-04, 9.9988e-01],
        [1.7715e-03, 9.9823e-01],
        [4.9975e-05, 9.9995e-01],
        [2.0724e-03, 9.9793e-01],
        [7.0449e-08, 1.0000e+00],
        [5.9080e-03, 9.9409e-01],
        [3.4170e-02, 9.6583e-01],
        [2.0510e-06, 1.0000e+00],
        [1.3476e-06, 1.0000e+00],
        [4.1575e-05, 9.9996e-01],
        [7.4627e-07, 1.0000e+00],
        [9.7668e-01, 2.3325e-02],
        [7.7823e-04, 9.9922e-01],
        [9.9980e-01, 2.0009e-04],
        [6.4966e-04, 9.9935e-01],
        [8.8381e-06, 9.9999e-01],
        [2.9800e-06, 1.0000e+00],
        [2.3838e-07, 1.0000e+00],
        [2.4011e-04, 9.9976e-01],
        [1.0634e-04, 9.9989e-01],
        [9.6978e-01, 3.0215e-02],
        [1.1202e-06, 1.0000e+00],
        [1.0882e-04, 9.9989e-01],
        [8.4315e-07, 1.0000e+00],
        [8.6418e-01, 1.3582e-01],
        [5.4964e-06, 9.9999e-01],
        [6.5792e-07, 1.0000e+00],
        [2.1969e-03, 9.9780e-01],
        [1.7047e-01, 8.2953e-01],
        [9.9992e-01, 7.8921e-05],
        [1.4793e-03, 9.9852e-01],
        [6.8534e-01, 3.1466e-01],
        [3.7061e-05, 9.9996e-01],
        [4.5661e-05, 9.9995e-01],
        [9.9972e-01, 2.8256e-04],
        [9.9976e-01, 2.3996e-04],
        [2.2308e-04, 9.9978e-01],
        [1.0208e-05, 9.9999e-01],
        [9.9874e-01, 1.2632e-03],
        [6.4620e-08, 1.0000e+00],
        [9.9932e-01, 6.8348e-04],
        [4.8809e-07, 1.0000e+00],
        [1.2884e-08, 1.0000e+00],
        [6.7795e-05, 9.9993e-01],
        [9.9961e-01, 3.9168e-04],
        [5.0323e-03, 9.9497e-01],
        [1.7330e-06, 1.0000e+00],
        [9.9966e-01, 3.4001e-04],
        [2.3636e-03, 9.9764e-01],
        [5.7314e-04, 9.9943e-01],
        [2.0640e-03, 9.9794e-01],
        [9.9730e-01, 2.6960e-03],
        [2.4514e-07, 1.0000e+00],
        [1.6350e-06, 1.0000e+00],
        [6.0485e-03, 9.9395e-01],
        [4.1605e-03, 9.9584e-01],
        [8.5306e-06, 9.9999e-01],
        [4.4530e-05, 9.9996e-01],
        [9.9957e-01, 4.3217e-04],
        [2.7651e-05, 9.9997e-01],
        [1.9900e-05, 9.9998e-01],
        [6.6190e-03, 9.9338e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9732e-01, 2.6848e-03],
        [5.7461e-09, 1.0000e+00],
        [9.9936e-01, 6.4301e-04],
        [5.6779e-08, 1.0000e+00],
        [3.2587e-04, 9.9967e-01],
        [6.7006e-04, 9.9933e-01],
        [3.6258e-05, 9.9996e-01],
        [3.6284e-07, 1.0000e+00],
        [1.1868e-01, 8.8132e-01],
        [3.1625e-04, 9.9968e-01],
        [5.2870e-09, 1.0000e+00],
        [1.3424e-04, 9.9987e-01],
        [9.9932e-01, 6.8074e-04],
        [9.9978e-01, 2.2007e-04],
        [9.9956e-01, 4.4016e-04],
        [4.6536e-03, 9.9535e-01],
        [1.1917e-04, 9.9988e-01],
        [6.9722e-07, 1.0000e+00],
        [5.6821e-07, 1.0000e+00],
        [1.4154e-08, 1.0000e+00],
        [5.9856e-07, 1.0000e+00],
        [1.2164e-06, 1.0000e+00],
        [1.4766e-02, 9.8523e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.7571e-03, 9.9524e-01],
        [9.9977e-01, 2.2871e-04],
        [4.7502e-08, 1.0000e+00],
        [2.6410e-05, 9.9997e-01],
        [6.0775e-06, 9.9999e-01],
        [1.2373e-03, 9.9876e-01],
        [9.8321e-05, 9.9990e-01],
        [4.9225e-05, 9.9995e-01],
        [2.3134e-03, 9.9769e-01],
        [2.5447e-07, 1.0000e+00],
        [7.9054e-01, 2.0946e-01],
        [2.2236e-06, 1.0000e+00],
        [9.9921e-01, 7.9285e-04],
        [3.7432e-04, 9.9963e-01],
        [9.9921e-01, 7.9085e-04],
        [9.9980e-01, 1.9898e-04],
        [9.9947e-01, 5.2722e-04],
        [2.6964e-09, 1.0000e+00],
        [5.8671e-04, 9.9941e-01],
        [1.3318e-04, 9.9987e-01],
        [1.8583e-07, 1.0000e+00],
        [7.9077e-04, 9.9921e-01],
        [6.8044e-07, 1.0000e+00],
        [2.3764e-05, 9.9998e-01],
        [4.6919e-05, 9.9995e-01],
        [3.9282e-03, 9.9607e-01],
        [1.9402e-06, 1.0000e+00],
        [1.7299e-03, 9.9827e-01],
        [6.9574e-05, 9.9993e-01],
        [1.5628e-05, 9.9998e-01],
        [2.5459e-04, 9.9975e-01],
        [6.3460e-08, 1.0000e+00],
        [9.9988e-01, 1.1724e-04],
        [9.7378e-01, 2.6219e-02],
        [6.8755e-07, 1.0000e+00],
        [3.9854e-03, 9.9601e-01],
        [9.9242e-01, 7.5834e-03],
        [9.9992e-01, 7.7524e-05],
        [2.4300e-08, 1.0000e+00],
        [1.1444e-06, 1.0000e+00],
        [9.9999e-01, 1.2246e-05],
        [4.6510e-05, 9.9995e-01],
        [4.3858e-04, 9.9956e-01],
        [3.9890e-08, 1.0000e+00],
        [1.0081e-01, 8.9919e-01],
        [3.3590e-02, 9.6641e-01],
        [9.7870e-01, 2.1300e-02],
        [1.9042e-03, 9.9810e-01],
        [9.9997e-01, 3.0471e-05],
        [1.1799e-09, 1.0000e+00],
        [9.9919e-01, 8.0575e-04],
        [2.7973e-05, 9.9997e-01],
        [1.6309e-06, 1.0000e+00],
        [9.9863e-01, 1.3680e-03],
        [8.9369e-07, 1.0000e+00],
        [9.9928e-01, 7.1601e-04],
        [1.3680e-06, 1.0000e+00],
        [9.8043e-03, 9.9020e-01],
        [7.3894e-06, 9.9999e-01],
        [9.9952e-01, 4.8288e-04],
        [2.9794e-06, 1.0000e+00],
        [4.0497e-06, 1.0000e+00],
        [9.9981e-01, 1.9128e-04],
        [9.2049e-01, 7.9507e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.3625e-07, 1.0000e+00],
        [2.9934e-05, 9.9997e-01],
        [1.0301e-07, 1.0000e+00],
        [7.6480e-04, 9.9924e-01],
        [5.3421e-08, 1.0000e+00],
        [1.2689e-01, 8.7311e-01],
        [2.5587e-05, 9.9997e-01],
        [9.9915e-01, 8.4675e-04],
        [9.9936e-01, 6.3899e-04],
        [1.0945e-04, 9.9989e-01],
        [3.5283e-07, 1.0000e+00],
        [1.9425e-03, 9.9806e-01],
        [2.8073e-05, 9.9997e-01],
        [2.3896e-09, 1.0000e+00],
        [9.9977e-01, 2.3459e-04],
        [1.0729e-08, 1.0000e+00],
        [1.5475e-04, 9.9985e-01],
        [6.3635e-03, 9.9364e-01],
        [6.4579e-05, 9.9994e-01],
        [9.9056e-01, 9.4432e-03],
        [1.5786e-04, 9.9984e-01],
        [4.4021e-05, 9.9996e-01],
        [1.1728e-05, 9.9999e-01],
        [9.9978e-01, 2.1535e-04],
        [3.1209e-04, 9.9969e-01],
        [1.4191e-05, 9.9999e-01],
        [9.9844e-01, 1.5572e-03],
        [7.4758e-01, 2.5242e-01],
        [3.0574e-06, 1.0000e+00],
        [2.3896e-09, 1.0000e+00],
        [9.9894e-01, 1.0642e-03],
        [9.5944e-04, 9.9904e-01],
        [1.0158e-05, 9.9999e-01],
        [1.2362e-02, 9.8764e-01],
        [2.5931e-05, 9.9997e-01],
        [9.9831e-01, 1.6931e-03],
        [4.5461e-05, 9.9995e-01],
        [7.4610e-05, 9.9993e-01],
        [2.2827e-07, 1.0000e+00],
        [4.5367e-09, 1.0000e+00],
        [9.0154e-08, 1.0000e+00],
        [1.9129e-05, 9.9998e-01],
        [3.5230e-05, 9.9996e-01],
        [1.3876e-02, 9.8612e-01],
        [1.9326e-05, 9.9998e-01],
        [9.9985e-01, 1.4629e-04],
        [1.6703e-02, 9.8330e-01],
        [5.6022e-05, 9.9994e-01],
        [6.3951e-02, 9.3605e-01],
        [2.7437e-08, 1.0000e+00],
        [1.0757e-03, 9.9892e-01],
        [8.6488e-05, 9.9991e-01],
        [3.1926e-01, 6.8074e-01],
        [8.3932e-09, 1.0000e+00],
        [9.9023e-01, 9.7750e-03],
        [9.9946e-01, 5.3834e-04],
        [2.6369e-06, 1.0000e+00],
        [2.4147e-04, 9.9976e-01],
        [9.9971e-01, 2.8588e-04],
        [1.7256e-08, 1.0000e+00],
        [9.9969e-01, 3.0631e-04],
        [9.9999e-01, 1.2998e-05],
        [9.9988e-01, 1.2115e-04],
        [7.7369e-04, 9.9923e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5780e-02, 9.5422e-01],
        [4.8762e-04, 9.9951e-01],
        [6.3119e-09, 1.0000e+00],
        [9.9921e-01, 7.8953e-04],
        [2.5287e-06, 1.0000e+00],
        [6.2198e-03, 9.9378e-01],
        [2.9186e-05, 9.9997e-01],
        [3.1917e-06, 1.0000e+00],
        [7.7067e-01, 2.2933e-01],
        [6.2679e-03, 9.9373e-01],
        [9.9150e-01, 8.5027e-03],
        [4.9423e-04, 9.9951e-01],
        [1.3252e-05, 9.9999e-01],
        [1.4845e-05, 9.9999e-01],
        [1.7948e-06, 1.0000e+00],
        [6.2354e-04, 9.9938e-01],
        [9.9796e-01, 2.0373e-03],
        [9.9898e-01, 1.0204e-03],
        [1.9971e-05, 9.9998e-01],
        [2.0124e-04, 9.9980e-01],
        [2.9358e-06, 1.0000e+00],
        [2.2065e-07, 1.0000e+00],
        [5.2369e-05, 9.9995e-01],
        [9.9899e-01, 1.0149e-03],
        [2.4894e-06, 1.0000e+00],
        [9.8088e-01, 1.9118e-02],
        [8.4986e-05, 9.9992e-01],
        [2.0491e-07, 1.0000e+00],
        [9.2568e-01, 7.4317e-02],
        [9.9970e-01, 2.9717e-04],
        [2.3627e-09, 1.0000e+00],
        [9.0362e-01, 9.6378e-02],
        [9.9912e-01, 8.8049e-04],
        [4.8264e-05, 9.9995e-01],
        [2.6674e-08, 1.0000e+00],
        [3.9786e-03, 9.9602e-01],
        [1.6452e-08, 1.0000e+00],
        [7.5208e-06, 9.9999e-01],
        [7.9330e-04, 9.9921e-01],
        [7.4047e-04, 9.9926e-01],
        [9.8962e-01, 1.0377e-02],
        [7.0149e-04, 9.9930e-01],
        [9.6434e-07, 1.0000e+00],
        [1.6180e-05, 9.9998e-01],
        [1.0510e-03, 9.9895e-01],
        [4.7948e-06, 1.0000e+00],
        [3.9717e-07, 1.0000e+00],
        [5.0505e-08, 1.0000e+00],
        [4.6242e-05, 9.9995e-01],
        [2.5824e-05, 9.9997e-01],
        [3.8280e-04, 9.9962e-01],
        [1.5109e-05, 9.9998e-01],
        [9.9994e-01, 6.0403e-05],
        [5.0331e-02, 9.4967e-01],
        [3.9257e-07, 1.0000e+00],
        [5.6700e-03, 9.9433e-01],
        [2.8619e-05, 9.9997e-01],
        [2.4665e-04, 9.9975e-01],
        [2.5736e-09, 1.0000e+00],
        [8.9888e-05, 9.9991e-01],
        [1.1479e-02, 9.8852e-01],
        [6.3013e-05, 9.9994e-01],
        [9.4758e-02, 9.0524e-01],
        [7.1324e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9833e-06, 1.0000e+00],
        [2.3354e-06, 1.0000e+00],
        [9.9961e-01, 3.8888e-04],
        [2.6647e-08, 1.0000e+00],
        [1.4128e-06, 1.0000e+00],
        [9.7845e-01, 2.1548e-02],
        [9.9316e-07, 1.0000e+00],
        [8.2850e-08, 1.0000e+00],
        [5.6242e-03, 9.9438e-01],
        [6.9827e-06, 9.9999e-01],
        [3.5928e-07, 1.0000e+00],
        [9.9894e-01, 1.0573e-03],
        [9.2786e-06, 9.9999e-01],
        [1.0000e+00, 3.3270e-06],
        [2.1238e-06, 1.0000e+00],
        [8.9932e-09, 1.0000e+00],
        [1.2194e-07, 1.0000e+00],
        [4.1400e-08, 1.0000e+00],
        [4.3207e-03, 9.9568e-01],
        [3.6736e-06, 1.0000e+00],
        [1.5292e-02, 9.8471e-01],
        [9.9969e-01, 3.0703e-04],
        [7.3766e-05, 9.9993e-01],
        [5.6748e-04, 9.9943e-01],
        [9.9981e-01, 1.9463e-04],
        [1.9119e-05, 9.9998e-01],
        [9.9663e-01, 3.3727e-03],
        [4.1094e-08, 1.0000e+00],
        [9.9959e-01, 4.0682e-04],
        [9.1417e-01, 8.5830e-02],
        [9.9998e-01, 2.1541e-05],
        [2.1002e-05, 9.9998e-01],
        [7.4077e-02, 9.2592e-01],
        [9.9954e-01, 4.5705e-04],
        [8.4548e-01, 1.5452e-01],
        [4.4420e-07, 1.0000e+00],
        [3.0473e-01, 6.9527e-01],
        [2.3043e-03, 9.9770e-01],
        [1.7939e-02, 9.8206e-01],
        [1.4765e-04, 9.9985e-01],
        [1.0212e-05, 9.9999e-01],
        [5.7366e-06, 9.9999e-01],
        [9.9959e-01, 4.1111e-04],
        [1.7715e-06, 1.0000e+00],
        [8.3651e-09, 1.0000e+00],
        [1.2853e-06, 1.0000e+00],
        [9.2712e-01, 7.2877e-02],
        [9.9909e-01, 9.1221e-04],
        [3.4081e-06, 1.0000e+00],
        [9.4806e-03, 9.9052e-01],
        [1.1076e-05, 9.9999e-01],
        [1.8364e-08, 1.0000e+00],
        [8.7305e-04, 9.9913e-01],
        [3.3623e-09, 1.0000e+00],
        [4.6036e-05, 9.9995e-01],
        [1.8720e-05, 9.9998e-01],
        [9.9964e-01, 3.5690e-04],
        [1.3835e-04, 9.9986e-01],
        [5.9764e-07, 1.0000e+00],
        [1.4301e-07, 1.0000e+00],
        [2.6208e-08, 1.0000e+00],
        [1.2645e-05, 9.9999e-01],
        [5.9482e-06, 9.9999e-01],
        [3.9009e-05, 9.9996e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.7242e-01, 2.7580e-02],
        [1.3913e-07, 1.0000e+00],
        [4.6634e-06, 1.0000e+00],
        [9.9980e-01, 1.9588e-04],
        [7.0431e-05, 9.9993e-01],
        [3.5801e-06, 1.0000e+00],
        [2.9453e-07, 1.0000e+00],
        [9.9971e-01, 2.9205e-04],
        [1.5578e-08, 1.0000e+00],
        [2.5533e-02, 9.7447e-01],
        [3.0730e-01, 6.9270e-01],
        [4.0354e-06, 1.0000e+00],
        [1.5903e-04, 9.9984e-01],
        [9.9245e-01, 7.5509e-03],
        [1.7498e-08, 1.0000e+00],
        [5.4201e-03, 9.9458e-01],
        [3.6967e-06, 1.0000e+00],
        [1.5983e-05, 9.9998e-01],
        [1.2775e-06, 1.0000e+00],
        [1.3932e-05, 9.9999e-01],
        [8.1231e-05, 9.9992e-01],
        [1.0789e-06, 1.0000e+00],
        [9.9956e-01, 4.3620e-04],
        [1.9369e-07, 1.0000e+00],
        [9.9992e-01, 7.9228e-05],
        [9.0211e-05, 9.9991e-01],
        [4.6213e-08, 1.0000e+00],
        [5.3721e-09, 1.0000e+00],
        [2.5662e-02, 9.7434e-01],
        [6.5548e-05, 9.9993e-01],
        [9.7028e-01, 2.9723e-02],
        [4.2953e-08, 1.0000e+00],
        [5.2826e-07, 1.0000e+00],
        [8.8868e-01, 1.1132e-01],
        [5.8770e-06, 9.9999e-01],
        [9.9841e-01, 1.5910e-03],
        [3.2211e-06, 1.0000e+00],
        [3.3609e-06, 1.0000e+00],
        [9.9985e-01, 1.4519e-04],
        [2.6533e-03, 9.9735e-01],
        [4.5818e-08, 1.0000e+00],
        [3.6888e-08, 1.0000e+00],
        [1.3510e-05, 9.9999e-01],
        [2.8433e-02, 9.7157e-01],
        [3.8391e-07, 1.0000e+00],
        [4.6531e-07, 1.0000e+00],
        [9.9884e-01, 1.1632e-03],
        [9.9919e-01, 8.0545e-04],
        [9.9970e-01, 2.9842e-04],
        [6.9069e-06, 9.9999e-01],
        [1.9031e-03, 9.9810e-01],
        [1.2087e-06, 1.0000e+00],
        [1.4722e-05, 9.9999e-01],
        [1.3258e-07, 1.0000e+00],
        [8.6155e-06, 9.9999e-01],
        [9.2920e-01, 7.0798e-02],
        [5.8300e-05, 9.9994e-01],
        [1.4920e-07, 1.0000e+00],
        [1.5419e-07, 1.0000e+00],
        [1.3561e-07, 1.0000e+00],
        [2.5584e-02, 9.7442e-01],
        [1.5888e-01, 8.4112e-01],
        [2.0521e-06, 1.0000e+00],
        [9.9993e-01, 6.6700e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[5.2587e-06, 9.9999e-01],
        [9.9996e-01, 4.2195e-05],
        [9.9840e-01, 1.5964e-03],
        [4.1890e-05, 9.9996e-01],
        [1.0875e-04, 9.9989e-01],
        [1.6702e-05, 9.9998e-01],
        [1.1155e-06, 1.0000e+00],
        [3.5827e-07, 1.0000e+00],
        [7.6969e-08, 1.0000e+00],
        [4.8527e-05, 9.9995e-01],
        [2.4174e-08, 1.0000e+00],
        [9.9868e-01, 1.3190e-03],
        [2.1723e-03, 9.9783e-01],
        [6.1238e-04, 9.9939e-01],
        [7.9512e-08, 1.0000e+00],
        [5.2090e-03, 9.9479e-01],
        [9.0752e-05, 9.9991e-01],
        [1.4551e-04, 9.9985e-01],
        [6.3211e-03, 9.9368e-01],
        [8.2099e-09, 1.0000e+00],
        [9.0401e-07, 1.0000e+00],
        [1.8205e-01, 8.1795e-01],
        [7.3607e-08, 1.0000e+00],
        [5.2422e-07, 1.0000e+00],
        [1.5799e-04, 9.9984e-01],
        [9.9855e-01, 1.4526e-03],
        [9.8917e-06, 9.9999e-01],
        [2.7891e-05, 9.9997e-01],
        [5.3121e-05, 9.9995e-01],
        [9.9884e-01, 1.1634e-03],
        [3.7041e-03, 9.9630e-01],
        [7.4253e-09, 1.0000e+00],
        [1.1351e-05, 9.9999e-01],
        [3.6015e-04, 9.9964e-01],
        [1.4730e-04, 9.9985e-01],
        [1.0042e-03, 9.9900e-01],
        [3.1088e-04, 9.9969e-01],
        [2.8105e-08, 1.0000e+00],
        [6.5093e-05, 9.9993e-01],
        [1.8673e-06, 1.0000e+00],
        [6.2812e-06, 9.9999e-01],
        [7.8920e-01, 2.1080e-01],
        [1.6950e-05, 9.9998e-01],
        [3.0064e-07, 1.0000e+00],
        [1.0972e-02, 9.8903e-01],
        [2.9665e-04, 9.9970e-01],
        [6.0282e-05, 9.9994e-01],
        [1.9289e-05, 9.9998e-01],
        [6.9145e-06, 9.9999e-01],
        [9.9971e-01, 2.9230e-04],
        [1.7127e-06, 1.0000e+00],
        [9.0309e-09, 1.0000e+00],
        [5.2004e-07, 1.0000e+00],
        [9.9856e-01, 1.4354e-03],
        [2.0573e-05, 9.9998e-01],
        [1.8404e-05, 9.9998e-01],
        [1.4700e-01, 8.5300e-01],
        [2.7687e-07, 1.0000e+00],
        [1.5071e-08, 1.0000e+00],
        [2.9179e-07, 1.0000e+00],
        [2.4525e-09, 1.0000e+00],
        [1.2955e-09, 1.0000e+00],
        [9.9216e-01, 7.8385e-03],
        [3.3133e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0263e-02, 9.8974e-01],
        [2.1535e-03, 9.9785e-01],
        [9.9264e-01, 7.3572e-03],
        [2.3723e-07, 1.0000e+00],
        [2.0659e-05, 9.9998e-01],
        [9.3401e-08, 1.0000e+00],
        [9.9108e-07, 1.0000e+00],
        [2.8291e-09, 1.0000e+00],
        [2.8275e-08, 1.0000e+00],
        [4.7703e-06, 1.0000e+00],
        [7.1178e-06, 9.9999e-01],
        [1.6121e-09, 1.0000e+00],
        [1.0504e-06, 1.0000e+00],
        [9.9789e-01, 2.1092e-03],
        [7.6598e-05, 9.9992e-01],
        [1.1915e-05, 9.9999e-01],
        [9.5860e-01, 4.1402e-02],
        [1.0329e-05, 9.9999e-01],
        [4.6350e-08, 1.0000e+00],
        [9.9949e-01, 5.0590e-04],
        [8.0747e-01, 1.9253e-01],
        [6.5820e-05, 9.9993e-01],
        [9.9529e-01, 4.7131e-03],
        [9.4340e-01, 5.6603e-02],
        [9.9642e-01, 3.5789e-03],
        [2.9988e-08, 1.0000e+00],
        [2.5463e-07, 1.0000e+00],
        [9.9993e-01, 6.5282e-05],
        [5.9302e-06, 9.9999e-01],
        [2.0901e-06, 1.0000e+00],
        [1.9630e-06, 1.0000e+00],
        [3.1844e-08, 1.0000e+00],
        [9.9765e-01, 2.3475e-03],
        [9.7919e-08, 1.0000e+00],
        [4.0294e-05, 9.9996e-01],
        [9.4376e-07, 1.0000e+00],
        [1.4186e-03, 9.9858e-01],
        [5.4919e-06, 9.9999e-01],
        [1.3059e-07, 1.0000e+00],
        [1.1401e-06, 1.0000e+00],
        [6.4358e-05, 9.9994e-01],
        [1.0721e-04, 9.9989e-01],
        [1.0780e-04, 9.9989e-01],
        [1.4926e-07, 1.0000e+00],
        [3.6980e-07, 1.0000e+00],
        [5.7995e-07, 1.0000e+00],
        [9.9800e-01, 1.9955e-03],
        [5.8290e-05, 9.9994e-01],
        [2.1792e-04, 9.9978e-01],
        [6.2216e-07, 1.0000e+00],
        [8.8361e-01, 1.1639e-01],
        [4.2182e-04, 9.9958e-01],
        [9.8609e-01, 1.3906e-02],
        [2.4131e-05, 9.9998e-01],
        [1.2957e-03, 9.9870e-01],
        [7.3796e-02, 9.2620e-01],
        [9.9488e-01, 5.1217e-03],
        [2.3224e-02, 9.7678e-01],
        [1.1217e-02, 9.8878e-01],
        [9.9742e-01, 2.5811e-03],
        [1.2965e-06, 1.0000e+00],
        [2.7199e-07, 1.0000e+00],
        [9.9976e-01, 2.4225e-04],
        [2.0825e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.8758e-03, 9.9512e-01],
        [9.9471e-01, 5.2855e-03],
        [2.0709e-03, 9.9793e-01],
        [8.6620e-07, 1.0000e+00],
        [2.3428e-05, 9.9998e-01],
        [3.9235e-06, 1.0000e+00],
        [8.4714e-01, 1.5286e-01],
        [5.6472e-08, 1.0000e+00],
        [2.2630e-08, 1.0000e+00],
        [5.4493e-08, 1.0000e+00],
        [9.9998e-01, 2.2659e-05],
        [2.6147e-07, 1.0000e+00],
        [9.9973e-01, 2.6954e-04],
        [7.4727e-06, 9.9999e-01],
        [1.4517e-06, 1.0000e+00],
        [1.2741e-04, 9.9987e-01],
        [1.2921e-03, 9.9871e-01],
        [1.6006e-07, 1.0000e+00],
        [6.0646e-07, 1.0000e+00],
        [1.4595e-08, 1.0000e+00],
        [1.0167e-03, 9.9898e-01],
        [9.9955e-01, 4.5349e-04],
        [9.8149e-01, 1.8514e-02],
        [9.9212e-01, 7.8771e-03],
        [1.9205e-05, 9.9998e-01],
        [3.1514e-06, 1.0000e+00],
        [1.2462e-08, 1.0000e+00],
        [8.2426e-08, 1.0000e+00],
        [4.9479e-06, 9.9999e-01],
        [9.9897e-01, 1.0266e-03],
        [9.9933e-01, 6.7175e-04],
        [1.7999e-06, 1.0000e+00],
        [1.4890e-06, 1.0000e+00],
        [8.0847e-05, 9.9992e-01],
        [8.8101e-08, 1.0000e+00],
        [3.0632e-05, 9.9997e-01],
        [9.9902e-01, 9.7868e-04],
        [8.1479e-02, 9.1852e-01],
        [2.1200e-02, 9.7880e-01],
        [3.7362e-07, 1.0000e+00],
        [1.1471e-06, 1.0000e+00],
        [9.9005e-01, 9.9547e-03],
        [9.9940e-01, 5.9631e-04],
        [3.3972e-10, 1.0000e+00],
        [1.6162e-04, 9.9984e-01],
        [1.4782e-06, 1.0000e+00],
        [6.1721e-09, 1.0000e+00],
        [1.7543e-07, 1.0000e+00],
        [1.5530e-01, 8.4470e-01],
        [1.1354e-04, 9.9989e-01],
        [2.9608e-08, 1.0000e+00],
        [4.1111e-09, 1.0000e+00],
        [5.6843e-04, 9.9943e-01],
        [6.4587e-07, 1.0000e+00],
        [9.9940e-01, 6.0334e-04],
        [1.8468e-07, 1.0000e+00],
        [6.0656e-04, 9.9939e-01],
        [9.9967e-01, 3.2595e-04],
        [9.9959e-01, 4.1362e-04],
        [3.5960e-05, 9.9996e-01],
        [2.5103e-06, 1.0000e+00],
        [9.6857e-06, 9.9999e-01],
        [2.4701e-04, 9.9975e-01],
        [1.2048e-04, 9.9988e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.2587e-07, 1.0000e+00],
        [1.0871e-02, 9.8913e-01],
        [6.3550e-07, 1.0000e+00],
        [9.9988e-01, 1.2067e-04],
        [9.6574e-01, 3.4259e-02],
        [9.9974e-01, 2.6201e-04],
        [8.5786e-07, 1.0000e+00],
        [9.9963e-01, 3.6673e-04],
        [3.2528e-07, 1.0000e+00],
        [5.5409e-07, 1.0000e+00],
        [4.0492e-08, 1.0000e+00],
        [9.7612e-01, 2.3877e-02],
        [8.5154e-07, 1.0000e+00],
        [9.9966e-01, 3.3513e-04],
        [1.4019e-05, 9.9999e-01],
        [9.8428e-01, 1.5717e-02],
        [1.1192e-04, 9.9989e-01],
        [4.7055e-08, 1.0000e+00],
        [5.1216e-07, 1.0000e+00],
        [6.7446e-07, 1.0000e+00],
        [4.9899e-03, 9.9501e-01],
        [9.9927e-01, 7.3377e-04],
        [2.0727e-08, 1.0000e+00],
        [2.8010e-05, 9.9997e-01],
        [9.9962e-01, 3.7800e-04],
        [4.0178e-04, 9.9960e-01],
        [6.6644e-06, 9.9999e-01],
        [9.7402e-01, 2.5981e-02],
        [3.6837e-07, 1.0000e+00],
        [1.8113e-04, 9.9982e-01],
        [1.0605e-05, 9.9999e-01],
        [6.5063e-05, 9.9993e-01],
        [5.8545e-09, 1.0000e+00],
        [9.9959e-01, 4.1360e-04],
        [1.7389e-07, 1.0000e+00],
        [4.0792e-09, 1.0000e+00],
        [3.4234e-04, 9.9966e-01],
        [1.6838e-05, 9.9998e-01],
        [5.4259e-08, 1.0000e+00],
        [1.6955e-05, 9.9998e-01],
        [9.9749e-01, 2.5094e-03],
        [1.8656e-02, 9.8134e-01],
        [1.0000e+00, 2.4638e-06],
        [8.9211e-01, 1.0789e-01],
        [1.4547e-05, 9.9999e-01],
        [8.5837e-07, 1.0000e+00],
        [9.9256e-01, 7.4364e-03],
        [9.9928e-01, 7.2497e-04],
        [8.9547e-07, 1.0000e+00],
        [1.7079e-07, 1.0000e+00],
        [1.4872e-08, 1.0000e+00],
        [1.3365e-03, 9.9866e-01],
        [1.2749e-06, 1.0000e+00],
        [5.1969e-04, 9.9948e-01],
        [9.9987e-01, 1.2780e-04],
        [1.5732e-08, 1.0000e+00],
        [2.4942e-04, 9.9975e-01],
        [5.4359e-06, 9.9999e-01],
        [9.5655e-01, 4.3448e-02],
        [8.8860e-07, 1.0000e+00],
        [9.1123e-01, 8.8769e-02],
        [1.0644e-05, 9.9999e-01],
        [9.9950e-01, 5.0021e-04],
        [8.1502e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.8658e-06, 1.0000e+00],
        [1.2150e-10, 1.0000e+00],
        [7.8599e-05, 9.9992e-01],
        [3.5879e-04, 9.9964e-01],
        [1.2821e-03, 9.9872e-01],
        [4.1877e-07, 1.0000e+00],
        [9.6967e-01, 3.0331e-02],
        [9.9832e-01, 1.6777e-03],
        [2.1860e-04, 9.9978e-01],
        [6.0811e-05, 9.9994e-01],
        [1.2726e-02, 9.8727e-01],
        [2.8448e-06, 1.0000e+00],
        [9.9970e-01, 3.0471e-04],
        [2.3178e-05, 9.9998e-01],
        [9.9157e-01, 8.4349e-03],
        [2.3301e-06, 1.0000e+00],
        [6.4389e-04, 9.9936e-01],
        [2.7050e-06, 1.0000e+00],
        [1.4801e-08, 1.0000e+00],
        [5.7356e-06, 9.9999e-01],
        [8.6866e-10, 1.0000e+00],
        [4.9197e-04, 9.9951e-01],
        [2.0003e-08, 1.0000e+00],
        [1.3905e-04, 9.9986e-01],
        [3.6312e-04, 9.9964e-01],
        [9.8073e-05, 9.9990e-01],
        [4.6354e-07, 1.0000e+00],
        [1.7911e-05, 9.9998e-01],
        [8.5543e-05, 9.9991e-01],
        [3.9084e-07, 1.0000e+00],
        [2.0503e-06, 1.0000e+00],
        [2.1564e-10, 1.0000e+00],
        [3.9631e-05, 9.9996e-01],
        [7.6985e-06, 9.9999e-01],
        [3.9624e-06, 1.0000e+00],
        [3.2248e-05, 9.9997e-01],
        [9.6157e-06, 9.9999e-01],
        [3.5655e-04, 9.9964e-01],
        [2.4904e-06, 1.0000e+00],
        [9.6389e-05, 9.9990e-01],
        [3.9233e-01, 6.0767e-01],
        [9.7651e-01, 2.3492e-02],
        [6.7431e-07, 1.0000e+00],
        [1.8150e-05, 9.9998e-01],
        [4.0073e-08, 1.0000e+00],
        [9.9948e-01, 5.1981e-04],
        [1.4124e-06, 1.0000e+00],
        [1.0909e-06, 1.0000e+00],
        [1.3086e-09, 1.0000e+00],
        [9.8995e-01, 1.0051e-02],
        [1.8977e-07, 1.0000e+00],
        [9.7212e-01, 2.7876e-02],
        [4.8956e-04, 9.9951e-01],
        [7.1515e-05, 9.9993e-01],
        [9.9940e-01, 6.0459e-04],
        [9.4296e-08, 1.0000e+00],
        [1.4073e-07, 1.0000e+00],
        [3.1445e-03, 9.9686e-01],
        [8.2265e-01, 1.7735e-01],
        [3.1429e-05, 9.9997e-01],
        [9.9968e-01, 3.1702e-04],
        [9.5490e-08, 1.0000e+00],
        [1.6267e-02, 9.8373e-01],
        [9.9996e-01, 3.9434e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9942e-01, 5.7713e-04],
        [6.7644e-04, 9.9932e-01],
        [1.1753e-09, 1.0000e+00],
        [1.6363e-07, 1.0000e+00],
        [2.7992e-08, 1.0000e+00],
        [2.3298e-05, 9.9998e-01],
        [6.2008e-05, 9.9994e-01],
        [4.2824e-06, 1.0000e+00],
        [9.9548e-01, 4.5200e-03],
        [9.9918e-01, 8.1846e-04],
        [2.1232e-05, 9.9998e-01],
        [7.2412e-04, 9.9928e-01],
        [3.2079e-05, 9.9997e-01],
        [1.0000e+00, 4.2029e-06],
        [1.4595e-06, 1.0000e+00],
        [9.9910e-01, 9.0177e-04],
        [9.9848e-01, 1.5208e-03],
        [1.9428e-05, 9.9998e-01],
        [2.0548e-05, 9.9998e-01],
        [3.1709e-04, 9.9968e-01],
        [1.8057e-03, 9.9819e-01],
        [7.6557e-07, 1.0000e+00],
        [9.9586e-03, 9.9004e-01],
        [2.3321e-04, 9.9977e-01],
        [1.2534e-03, 9.9875e-01],
        [3.6180e-03, 9.9638e-01],
        [2.4602e-02, 9.7540e-01],
        [2.4653e-08, 1.0000e+00],
        [1.8844e-04, 9.9981e-01],
        [1.9847e-02, 9.8015e-01],
        [5.3787e-09, 1.0000e+00],
        [6.3369e-05, 9.9994e-01],
        [2.7182e-04, 9.9973e-01],
        [4.9617e-08, 1.0000e+00],
        [6.5110e-09, 1.0000e+00],
        [9.9977e-01, 2.2673e-04],
        [9.9566e-01, 4.3399e-03],
        [9.9981e-01, 1.9210e-04],
        [1.5936e-06, 1.0000e+00],
        [9.9383e-01, 6.1690e-03],
        [8.3972e-09, 1.0000e+00],
        [9.9981e-01, 1.8560e-04],
        [8.1641e-05, 9.9992e-01],
        [4.3757e-04, 9.9956e-01],
        [1.7834e-06, 1.0000e+00],
        [2.4438e-04, 9.9976e-01],
        [9.9691e-01, 3.0911e-03],
        [6.0556e-04, 9.9939e-01],
        [8.6619e-06, 9.9999e-01],
        [9.9970e-01, 2.9838e-04],
        [1.4387e-03, 9.9856e-01],
        [9.2651e-05, 9.9991e-01],
        [3.6260e-04, 9.9964e-01],
        [1.5680e-03, 9.9843e-01],
        [9.9939e-01, 6.1251e-04],
        [1.3584e-03, 9.9864e-01],
        [7.3047e-04, 9.9927e-01],
        [2.4184e-05, 9.9998e-01],
        [9.8949e-01, 1.0515e-02],
        [9.9449e-01, 5.5104e-03],
        [1.1128e-05, 9.9999e-01],
        [2.8206e-05, 9.9997e-01],
        [2.9756e-05, 9.9997e-01],
        [9.9932e-01, 6.8113e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3397e-03, 9.9766e-01],
        [1.8563e-09, 1.0000e+00],
        [9.0977e-03, 9.9090e-01],
        [9.9891e-01, 1.0941e-03],
        [7.5247e-07, 1.0000e+00],
        [1.0153e-03, 9.9898e-01],
        [1.6637e-05, 9.9998e-01],
        [3.9136e-04, 9.9961e-01],
        [9.9949e-01, 5.0963e-04],
        [7.4430e-03, 9.9256e-01],
        [1.2067e-08, 1.0000e+00],
        [6.5343e-11, 1.0000e+00],
        [1.1561e-03, 9.9884e-01],
        [1.0421e-02, 9.8958e-01],
        [1.8010e-05, 9.9998e-01],
        [1.9646e-04, 9.9980e-01],
        [5.4198e-08, 1.0000e+00],
        [2.6646e-07, 1.0000e+00],
        [7.7395e-01, 2.2605e-01],
        [5.3354e-04, 9.9947e-01],
        [1.3457e-05, 9.9999e-01],
        [3.1239e-02, 9.6876e-01],
        [4.7465e-05, 9.9995e-01],
        [6.4343e-03, 9.9357e-01],
        [8.1792e-01, 1.8208e-01],
        [9.9955e-01, 4.4996e-04],
        [2.8966e-06, 1.0000e+00],
        [8.7578e-09, 1.0000e+00],
        [1.2358e-10, 1.0000e+00],
        [8.7411e-06, 9.9999e-01],
        [5.0413e-06, 9.9999e-01],
        [3.3246e-03, 9.9668e-01],
        [8.5327e-08, 1.0000e+00],
        [6.6759e-06, 9.9999e-01],
        [4.7749e-04, 9.9952e-01],
        [2.4214e-05, 9.9998e-01],
        [5.9632e-03, 9.9404e-01],
        [9.9981e-01, 1.8717e-04],
        [1.8307e-06, 1.0000e+00],
        [9.0703e-01, 9.2970e-02],
        [1.8665e-06, 1.0000e+00],
        [3.6123e-03, 9.9639e-01],
        [7.3857e-09, 1.0000e+00],
        [4.7470e-04, 9.9953e-01],
        [4.4818e-08, 1.0000e+00],
        [6.7736e-04, 9.9932e-01],
        [9.9974e-01, 2.6073e-04],
        [2.5054e-09, 1.0000e+00],
        [5.6400e-06, 9.9999e-01],
        [2.2152e-05, 9.9998e-01],
        [1.1778e-08, 1.0000e+00],
        [3.4205e-07, 1.0000e+00],
        [2.2970e-03, 9.9770e-01],
        [5.6579e-05, 9.9994e-01],
        [7.2381e-04, 9.9928e-01],
        [9.9960e-01, 4.0253e-04],
        [2.4563e-08, 1.0000e+00],
        [4.4371e-09, 1.0000e+00],
        [2.1365e-08, 1.0000e+00],
        [1.0076e-05, 9.9999e-01],
        [4.9183e-03, 9.9508e-01],
        [3.4067e-05, 9.9997e-01],
        [1.6238e-08, 1.0000e+00],
        [7.8003e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4698e-05, 9.9999e-01],
        [5.3759e-07, 1.0000e+00],
        [4.0246e-05, 9.9996e-01],
        [9.9935e-01, 6.4739e-04],
        [1.1355e-05, 9.9999e-01],
        [5.6734e-06, 9.9999e-01],
        [5.6228e-02, 9.4377e-01],
        [7.1847e-02, 9.2815e-01],
        [6.8003e-05, 9.9993e-01],
        [2.6225e-01, 7.3775e-01],
        [9.9984e-01, 1.5915e-04],
        [9.9957e-01, 4.2841e-04],
        [1.7672e-07, 1.0000e+00],
        [7.2287e-09, 1.0000e+00],
        [2.1662e-08, 1.0000e+00],
        [2.4208e-04, 9.9976e-01],
        [9.9989e-01, 1.1153e-04],
        [9.9041e-01, 9.5949e-03],
        [9.9973e-01, 2.6995e-04],
        [3.1220e-04, 9.9969e-01],
        [9.9973e-01, 2.6995e-04],
        [9.9979e-01, 2.1266e-04],
        [9.9920e-01, 7.9812e-04],
        [9.9866e-01, 1.3443e-03],
        [2.9891e-07, 1.0000e+00],
        [3.4526e-04, 9.9965e-01],
        [9.9964e-01, 3.5608e-04],
        [1.6506e-06, 1.0000e+00],
        [1.2653e-06, 1.0000e+00],
        [2.4988e-07, 1.0000e+00],
        [2.0223e-04, 9.9980e-01],
        [1.8997e-05, 9.9998e-01],
        [2.2269e-07, 1.0000e+00],
        [9.9932e-01, 6.8246e-04],
        [9.9664e-01, 3.3577e-03],
        [6.8215e-02, 9.3179e-01],
        [9.9358e-01, 6.4200e-03],
        [6.3346e-08, 1.0000e+00],
        [5.6807e-04, 9.9943e-01],
        [4.1615e-06, 1.0000e+00],
        [1.1047e-03, 9.9890e-01],
        [3.4104e-01, 6.5896e-01],
        [7.1191e-04, 9.9929e-01],
        [2.1370e-06, 1.0000e+00],
        [9.3051e-01, 6.9494e-02],
        [2.1470e-04, 9.9979e-01],
        [3.1315e-03, 9.9687e-01],
        [9.9991e-01, 8.6967e-05],
        [1.0406e-06, 1.0000e+00],
        [2.4880e-09, 1.0000e+00],
        [4.7141e-09, 1.0000e+00],
        [4.4624e-08, 1.0000e+00],
        [3.1848e-03, 9.9682e-01],
        [1.1479e-05, 9.9999e-01],
        [1.5673e-05, 9.9998e-01],
        [5.1397e-08, 1.0000e+00],
        [3.6064e-04, 9.9964e-01],
        [9.7481e-01, 2.5188e-02],
        [5.7787e-07, 1.0000e+00],
        [4.7876e-07, 1.0000e+00],
        [9.9907e-01, 9.2920e-04],
        [1.9738e-08, 1.0000e+00],
        [1.3723e-05, 9.9999e-01],
        [3.5013e-02, 9.6499e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9952e-01, 4.8004e-04],
        [9.9968e-01, 3.2086e-04],
        [7.2669e-06, 9.9999e-01],
        [1.4743e-04, 9.9985e-01],
        [6.5520e-05, 9.9993e-01],
        [1.6033e-04, 9.9984e-01],
        [4.1570e-05, 9.9996e-01],
        [9.8263e-01, 1.7365e-02],
        [8.3835e-09, 1.0000e+00],
        [8.4829e-04, 9.9915e-01],
        [2.5561e-05, 9.9997e-01],
        [9.5973e-01, 4.0274e-02],
        [9.9532e-01, 4.6751e-03],
        [9.7616e-01, 2.3841e-02],
        [8.7403e-01, 1.2597e-01],
        [1.6417e-08, 1.0000e+00],
        [9.6171e-01, 3.8290e-02],
        [6.8101e-05, 9.9993e-01],
        [1.1865e-01, 8.8135e-01],
        [3.8123e-06, 1.0000e+00],
        [6.8396e-06, 9.9999e-01],
        [1.0910e-05, 9.9999e-01],
        [1.9043e-07, 1.0000e+00],
        [6.9741e-07, 1.0000e+00],
        [9.9757e-01, 2.4299e-03],
        [1.0536e-07, 1.0000e+00],
        [3.3348e-07, 1.0000e+00],
        [1.9693e-05, 9.9998e-01],
        [1.3634e-06, 1.0000e+00],
        [9.7702e-02, 9.0230e-01],
        [1.0330e-05, 9.9999e-01],
        [2.4495e-05, 9.9998e-01],
        [9.9986e-01, 1.3739e-04],
        [9.9989e-01, 1.0593e-04],
        [1.5893e-03, 9.9841e-01],
        [1.0842e-06, 1.0000e+00],
        [6.0429e-08, 1.0000e+00],
        [3.5266e-07, 1.0000e+00],
        [9.9971e-01, 2.9379e-04],
        [4.2921e-11, 1.0000e+00],
        [9.9988e-01, 1.1982e-04],
        [1.7662e-05, 9.9998e-01],
        [2.2371e-03, 9.9776e-01],
        [9.6189e-06, 9.9999e-01],
        [7.8789e-03, 9.9212e-01],
        [9.9513e-01, 4.8736e-03],
        [2.4822e-11, 1.0000e+00],
        [9.9979e-01, 2.1488e-04],
        [9.9912e-01, 8.7680e-04],
        [1.0358e-02, 9.8964e-01],
        [1.6511e-05, 9.9998e-01],
        [2.8258e-05, 9.9997e-01],
        [3.3329e-02, 9.6667e-01],
        [1.2872e-03, 9.9871e-01],
        [1.0682e-04, 9.9989e-01],
        [6.8679e-01, 3.1321e-01],
        [3.3877e-08, 1.0000e+00],
        [7.7984e-06, 9.9999e-01],
        [1.9445e-04, 9.9981e-01],
        [1.2891e-06, 1.0000e+00],
        [1.7133e-08, 1.0000e+00],
        [9.8970e-01, 1.0295e-02],
        [2.8344e-05, 9.9997e-01],
        [5.2572e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9991e-01, 8.7256e-05],
        [1.6343e-06, 1.0000e+00],
        [9.6790e-01, 3.2105e-02],
        [9.9968e-01, 3.1911e-04],
        [1.0029e-05, 9.9999e-01],
        [2.8625e-06, 1.0000e+00],
        [9.5581e-07, 1.0000e+00],
        [9.9986e-01, 1.4043e-04],
        [6.5252e-03, 9.9347e-01],
        [9.1352e-07, 1.0000e+00],
        [1.5869e-04, 9.9984e-01],
        [8.5858e-04, 9.9914e-01],
        [3.2498e-06, 1.0000e+00],
        [3.6954e-03, 9.9630e-01],
        [9.9367e-01, 6.3350e-03],
        [9.2584e-01, 7.4158e-02],
        [1.3604e-05, 9.9999e-01],
        [1.0033e-03, 9.9900e-01],
        [9.9624e-07, 1.0000e+00],
        [9.9836e-06, 9.9999e-01],
        [4.4515e-04, 9.9955e-01],
        [2.7206e-05, 9.9997e-01],
        [9.9994e-01, 5.8211e-05],
        [5.5738e-06, 9.9999e-01],
        [1.1000e-03, 9.9890e-01],
        [4.9783e-07, 1.0000e+00],
        [1.4597e-04, 9.9985e-01],
        [6.8660e-05, 9.9993e-01],
        [6.8849e-07, 1.0000e+00],
        [7.6647e-05, 9.9992e-01],
        [9.9955e-01, 4.5132e-04],
        [6.2619e-07, 1.0000e+00],
        [7.2894e-03, 9.9271e-01],
        [1.2844e-06, 1.0000e+00],
        [8.6355e-05, 9.9991e-01],
        [9.9971e-01, 2.9015e-04],
        [1.2029e-08, 1.0000e+00],
        [1.4777e-02, 9.8522e-01],
        [1.9164e-04, 9.9981e-01],
        [4.8145e-04, 9.9952e-01],
        [2.0629e-09, 1.0000e+00],
        [1.3075e-05, 9.9999e-01],
        [1.4863e-06, 1.0000e+00],
        [8.1599e-09, 1.0000e+00],
        [9.3460e-01, 6.5402e-02],
        [3.6872e-05, 9.9996e-01],
        [9.9986e-01, 1.3853e-04],
        [1.0460e-03, 9.9895e-01],
        [1.0000e+00, 2.2757e-06],
        [4.7290e-10, 1.0000e+00],
        [3.9534e-05, 9.9996e-01],
        [1.3075e-05, 9.9999e-01],
        [2.7652e-03, 9.9723e-01],
        [9.9979e-01, 2.0711e-04],
        [9.9962e-01, 3.8451e-04],
        [2.5558e-04, 9.9974e-01],
        [9.4226e-01, 5.7738e-02],
        [1.2411e-05, 9.9999e-01],
        [1.8925e-03, 9.9811e-01],
        [7.3618e-07, 1.0000e+00],
        [1.7511e-03, 9.9825e-01],
        [2.9533e-03, 9.9705e-01],
        [7.8822e-07, 1.0000e+00],
        [1.4935e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3936e-04, 9.9986e-01],
        [9.9201e-01, 7.9902e-03],
        [4.6320e-06, 1.0000e+00],
        [9.9655e-01, 3.4537e-03],
        [9.9974e-01, 2.6255e-04],
        [9.9965e-01, 3.5135e-04],
        [3.4519e-04, 9.9965e-01],
        [2.7451e-03, 9.9725e-01],
        [4.8742e-06, 1.0000e+00],
        [1.8314e-02, 9.8169e-01],
        [7.6706e-06, 9.9999e-01],
        [6.9034e-04, 9.9931e-01],
        [2.4735e-06, 1.0000e+00],
        [2.6884e-05, 9.9997e-01],
        [9.9883e-01, 1.1708e-03],
        [6.2479e-04, 9.9938e-01],
        [9.9981e-01, 1.8937e-04],
        [6.6196e-10, 1.0000e+00],
        [1.0577e-02, 9.8942e-01],
        [3.0781e-05, 9.9997e-01],
        [3.2633e-06, 1.0000e+00],
        [1.2964e-06, 1.0000e+00],
        [9.9780e-01, 2.1991e-03],
        [4.7674e-08, 1.0000e+00],
        [9.9972e-01, 2.8270e-04],
        [1.4666e-03, 9.9853e-01],
        [6.9112e-01, 3.0888e-01],
        [6.4335e-03, 9.9357e-01],
        [2.0041e-02, 9.7996e-01],
        [9.9818e-01, 1.8212e-03],
        [2.7886e-06, 1.0000e+00],
        [9.9978e-01, 2.2065e-04],
        [3.6752e-07, 1.0000e+00],
        [9.6955e-01, 3.0447e-02],
        [3.3847e-02, 9.6615e-01],
        [9.9975e-01, 2.4581e-04],
        [1.0064e-04, 9.9990e-01],
        [6.4510e-08, 1.0000e+00],
        [7.1899e-05, 9.9993e-01],
        [2.9716e-08, 1.0000e+00],
        [1.0289e-04, 9.9990e-01],
        [2.6116e-01, 7.3884e-01],
        [9.9967e-01, 3.2800e-04],
        [5.7921e-05, 9.9994e-01],
        [1.8155e-03, 9.9818e-01],
        [1.2464e-06, 1.0000e+00],
        [9.6568e-01, 3.4318e-02],
        [5.7868e-06, 9.9999e-01],
        [8.9374e-01, 1.0626e-01],
        [9.1362e-08, 1.0000e+00],
        [1.6542e-02, 9.8346e-01],
        [2.7348e-06, 1.0000e+00],
        [2.4555e-08, 1.0000e+00],
        [7.2291e-03, 9.9277e-01],
        [9.9970e-01, 2.9947e-04],
        [4.2281e-06, 1.0000e+00],
        [8.3233e-03, 9.9168e-01],
        [1.3670e-03, 9.9863e-01],
        [1.1083e-07, 1.0000e+00],
        [2.1203e-04, 9.9979e-01],
        [3.1705e-01, 6.8295e-01],
        [8.7359e-03, 9.9126e-01],
        [2.6350e-04, 9.9974e-01],
        [9.9996e-01, 4.4646e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9978e-01, 2.2351e-04],
        [9.2235e-01, 7.7653e-02],
        [2.1751e-05, 9.9998e-01],
        [9.9022e-01, 9.7776e-03],
        [8.0790e-06, 9.9999e-01],
        [3.2510e-03, 9.9675e-01],
        [1.7074e-04, 9.9983e-01],
        [4.9963e-05, 9.9995e-01],
        [2.2479e-02, 9.7752e-01],
        [8.2254e-07, 1.0000e+00],
        [7.9978e-01, 2.0022e-01],
        [6.0655e-03, 9.9393e-01],
        [1.2691e-03, 9.9873e-01],
        [9.9981e-01, 1.8940e-04],
        [1.8346e-07, 1.0000e+00],
        [7.6223e-04, 9.9924e-01],
        [5.6855e-06, 9.9999e-01],
        [5.3844e-06, 9.9999e-01],
        [9.2020e-05, 9.9991e-01],
        [2.1071e-05, 9.9998e-01],
        [3.2277e-06, 1.0000e+00],
        [2.8068e-06, 1.0000e+00],
        [3.9376e-04, 9.9961e-01],
        [5.0468e-04, 9.9950e-01],
        [9.9958e-01, 4.2260e-04],
        [1.2643e-04, 9.9987e-01],
        [9.4788e-07, 1.0000e+00],
        [9.9646e-01, 3.5350e-03],
        [1.8778e-06, 1.0000e+00],
        [2.1502e-02, 9.7850e-01],
        [3.0938e-02, 9.6906e-01],
        [5.3716e-04, 9.9946e-01],
        [9.9977e-01, 2.3147e-04],
        [3.7636e-01, 6.2364e-01],
        [2.4943e-04, 9.9975e-01],
        [1.1351e-04, 9.9989e-01],
        [2.0881e-03, 9.9791e-01],
        [8.5197e-01, 1.4803e-01],
        [9.9968e-01, 3.2281e-04],
        [8.2858e-06, 9.9999e-01],
        [9.9419e-01, 5.8066e-03],
        [5.1569e-02, 9.4843e-01],
        [8.3932e-01, 1.6068e-01],
        [6.3325e-05, 9.9994e-01],
        [3.5540e-05, 9.9996e-01],
        [1.8052e-06, 1.0000e+00],
        [9.9997e-01, 2.7872e-05],
        [2.2833e-04, 9.9977e-01],
        [9.6628e-01, 3.3721e-02],
        [1.3400e-07, 1.0000e+00],
        [9.9997e-01, 3.4574e-05],
        [2.4713e-04, 9.9975e-01],
        [3.2332e-05, 9.9997e-01],
        [9.9312e-01, 6.8796e-03],
        [2.3675e-01, 7.6325e-01],
        [9.9961e-01, 3.8564e-04],
        [2.3373e-04, 9.9977e-01],
        [1.4002e-02, 9.8600e-01],
        [6.9899e-02, 9.3010e-01],
        [1.5110e-03, 9.9849e-01],
        [2.6801e-05, 9.9997e-01],
        [1.0647e-06, 1.0000e+00],
        [9.8708e-01, 1.2916e-02],
        [1.2969e-04, 9.9987e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.6184e-08, 1.0000e+00],
        [3.4978e-06, 1.0000e+00],
        [1.8043e-03, 9.9820e-01],
        [1.8430e-03, 9.9816e-01],
        [6.9175e-04, 9.9931e-01],
        [9.9903e-01, 9.7074e-04],
        [8.4548e-07, 1.0000e+00],
        [2.0243e-03, 9.9798e-01],
        [2.3854e-02, 9.7615e-01],
        [9.9921e-01, 7.9246e-04],
        [4.3843e-04, 9.9956e-01],
        [5.2152e-04, 9.9948e-01],
        [8.6248e-04, 9.9914e-01],
        [2.3369e-06, 1.0000e+00],
        [9.9811e-01, 1.8893e-03],
        [1.1915e-06, 1.0000e+00],
        [4.8335e-05, 9.9995e-01],
        [8.4210e-05, 9.9992e-01],
        [1.3478e-02, 9.8652e-01],
        [5.4799e-06, 9.9999e-01],
        [1.3855e-05, 9.9999e-01],
        [9.9998e-01, 2.1923e-05],
        [9.9986e-01, 1.3674e-04],
        [1.0708e-04, 9.9989e-01],
        [3.2014e-08, 1.0000e+00],
        [4.3828e-06, 1.0000e+00],
        [8.4790e-01, 1.5210e-01],
        [4.4551e-04, 9.9955e-01],
        [1.3710e-04, 9.9986e-01],
        [3.2570e-07, 1.0000e+00],
        [2.6220e-03, 9.9738e-01],
        [9.6427e-04, 9.9904e-01],
        [9.9953e-01, 4.6817e-04],
        [4.3149e-04, 9.9957e-01],
        [8.9858e-01, 1.0142e-01],
        [1.0981e-06, 1.0000e+00],
        [1.5865e-03, 9.9841e-01],
        [1.3855e-05, 9.9999e-01],
        [9.9980e-01, 2.0132e-04],
        [8.2170e-07, 1.0000e+00],
        [5.1614e-08, 1.0000e+00],
        [1.4415e-02, 9.8559e-01],
        [1.1293e-06, 1.0000e+00],
        [6.6045e-09, 1.0000e+00],
        [1.9590e-01, 8.0410e-01],
        [9.9572e-01, 4.2826e-03],
        [9.9842e-01, 1.5840e-03],
        [9.9975e-01, 2.4508e-04],
        [9.8008e-01, 1.9917e-02],
        [1.6732e-05, 9.9998e-01],
        [3.0035e-05, 9.9997e-01],
        [6.1189e-04, 9.9939e-01],
        [9.9992e-01, 7.9017e-05],
        [4.4161e-05, 9.9996e-01],
        [9.8833e-01, 1.1674e-02],
        [6.6798e-06, 9.9999e-01],
        [3.8339e-06, 1.0000e+00],
        [2.4830e-08, 1.0000e+00],
        [2.6656e-04, 9.9973e-01],
        [9.6427e-04, 9.9904e-01],
        [2.0423e-04, 9.9980e-01],
        [9.0791e-03, 9.9092e-01],
        [1.6335e-06, 1.0000e+00],
        [9.9952e-01, 4.7660e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3365e-06, 1.0000e+00],
        [7.2896e-03, 9.9271e-01],
        [9.9904e-01, 9.6218e-04],
        [1.5245e-03, 9.9848e-01],
        [4.6590e-01, 5.3410e-01],
        [9.9924e-01, 7.6491e-04],
        [9.9971e-01, 2.8773e-04],
        [4.0667e-02, 9.5933e-01],
        [9.0917e-04, 9.9909e-01],
        [4.0055e-09, 1.0000e+00],
        [1.1930e-03, 9.9881e-01],
        [1.0216e-05, 9.9999e-01],
        [9.9985e-01, 1.5071e-04],
        [1.9459e-02, 9.8054e-01],
        [7.2017e-02, 9.2798e-01],
        [7.5860e-05, 9.9992e-01],
        [1.8435e-04, 9.9982e-01],
        [9.9675e-01, 3.2534e-03],
        [8.4595e-03, 9.9154e-01],
        [2.2515e-05, 9.9998e-01],
        [8.7987e-07, 1.0000e+00],
        [9.9892e-01, 1.0778e-03],
        [2.6297e-05, 9.9997e-01],
        [2.2883e-04, 9.9977e-01],
        [3.5821e-05, 9.9996e-01],
        [9.1626e-06, 9.9999e-01],
        [1.2682e-02, 9.8732e-01],
        [2.5894e-01, 7.4106e-01],
        [9.9933e-01, 6.7444e-04],
        [6.0462e-04, 9.9940e-01],
        [9.9977e-01, 2.2664e-04],
        [9.9953e-01, 4.7159e-04],
        [5.3514e-08, 1.0000e+00],
        [1.0688e-03, 9.9893e-01],
        [9.7873e-01, 2.1273e-02],
        [1.5435e-04, 9.9985e-01],
        [7.9180e-03, 9.9208e-01],
        [9.9963e-01, 3.7104e-04],
        [2.0957e-04, 9.9979e-01],
        [1.4107e-04, 9.9986e-01],
        [1.4054e-03, 9.9859e-01],
        [9.9956e-01, 4.4063e-04],
        [8.9652e-05, 9.9991e-01],
        [1.4583e-06, 1.0000e+00],
        [3.1489e-06, 1.0000e+00],
        [4.6756e-05, 9.9995e-01],
        [9.9986e-01, 1.3897e-04],
        [5.9308e-04, 9.9941e-01],
        [1.3776e-03, 9.9862e-01],
        [1.9370e-03, 9.9806e-01],
        [1.0699e-05, 9.9999e-01],
        [4.4962e-07, 1.0000e+00],
        [8.6808e-03, 9.9132e-01],
        [1.0791e-03, 9.9892e-01],
        [9.9915e-01, 8.5146e-04],
        [1.3635e-04, 9.9986e-01],
        [2.9811e-01, 7.0189e-01],
        [1.0173e-06, 1.0000e+00],
        [9.9985e-01, 1.5307e-04],
        [1.0022e-06, 1.0000e+00],
        [2.1698e-05, 9.9998e-01],
        [8.0735e-07, 1.0000e+00],
        [3.0063e-03, 9.9699e-01],
        [1.2604e-04, 9.9987e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.7759e-04, 9.9972e-01],
        [4.5952e-07, 1.0000e+00],
        [3.6503e-01, 6.3497e-01],
        [6.9950e-02, 9.3005e-01],
        [2.8242e-07, 1.0000e+00],
        [2.2023e-01, 7.7977e-01],
        [8.5869e-02, 9.1413e-01],
        [8.2801e-06, 9.9999e-01],
        [2.5980e-05, 9.9997e-01],
        [5.4467e-05, 9.9995e-01],
        [2.6318e-05, 9.9997e-01],
        [2.1118e-06, 1.0000e+00],
        [9.9949e-01, 5.0649e-04],
        [7.6121e-04, 9.9924e-01],
        [1.7535e-07, 1.0000e+00],
        [4.4436e-03, 9.9556e-01],
        [9.9888e-01, 1.1239e-03],
        [2.6645e-03, 9.9734e-01],
        [5.3335e-02, 9.4666e-01],
        [1.3914e-01, 8.6086e-01],
        [4.4681e-03, 9.9553e-01],
        [7.0057e-05, 9.9993e-01],
        [6.5206e-09, 1.0000e+00],
        [9.6255e-04, 9.9904e-01],
        [9.9978e-01, 2.2233e-04],
        [9.9988e-01, 1.2238e-04],
        [1.8755e-06, 1.0000e+00],
        [6.6105e-05, 9.9993e-01],
        [2.0622e-03, 9.9794e-01],
        [9.9967e-01, 3.3405e-04],
        [9.8901e-01, 1.0987e-02],
        [2.3029e-04, 9.9977e-01],
        [9.9986e-01, 1.3845e-04],
        [7.6634e-01, 2.3366e-01],
        [1.9632e-06, 1.0000e+00],
        [9.9983e-01, 1.6994e-04],
        [9.9997e-01, 3.0770e-05],
        [9.5724e-01, 4.2758e-02],
        [1.3221e-08, 1.0000e+00],
        [2.6526e-07, 1.0000e+00],
        [1.3014e-03, 9.9870e-01],
        [9.9965e-01, 3.4616e-04],
        [5.1411e-03, 9.9486e-01],
        [1.5184e-06, 1.0000e+00],
        [3.0294e-03, 9.9697e-01],
        [2.9729e-03, 9.9703e-01],
        [5.4495e-03, 9.9455e-01],
        [9.9975e-01, 2.5378e-04],
        [6.4657e-03, 9.9353e-01],
        [5.6608e-05, 9.9994e-01],
        [1.5954e-05, 9.9998e-01],
        [8.0570e-02, 9.1943e-01],
        [4.5987e-07, 1.0000e+00],
        [1.2242e-06, 1.0000e+00],
        [5.7356e-08, 1.0000e+00],
        [2.4374e-04, 9.9976e-01],
        [6.6331e-01, 3.3669e-01],
        [5.6408e-06, 9.9999e-01],
        [7.6894e-06, 9.9999e-01],
        [1.4260e-06, 1.0000e+00],
        [1.5205e-05, 9.9998e-01],
        [2.4270e-04, 9.9976e-01],
        [3.3851e-01, 6.6149e-01],
        [9.9992e-01, 8.0432e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9967e-01, 3.3227e-04],
        [7.3197e-05, 9.9993e-01],
        [3.9292e-07, 1.0000e+00],
        [7.7541e-03, 9.9225e-01],
        [9.9985e-01, 1.4781e-04],
        [2.0375e-06, 1.0000e+00],
        [9.9973e-01, 2.6900e-04],
        [2.9243e-04, 9.9971e-01],
        [7.4283e-06, 9.9999e-01],
        [8.4860e-03, 9.9151e-01],
        [5.6895e-05, 9.9994e-01],
        [2.7744e-07, 1.0000e+00],
        [2.8641e-04, 9.9971e-01],
        [1.0433e-10, 1.0000e+00],
        [6.4742e-01, 3.5258e-01],
        [2.7945e-06, 1.0000e+00],
        [1.1882e-04, 9.9988e-01],
        [1.2005e-02, 9.8799e-01],
        [1.9142e-07, 1.0000e+00],
        [2.3931e-02, 9.7607e-01],
        [9.9980e-01, 1.9952e-04],
        [6.5535e-07, 1.0000e+00],
        [1.5495e-03, 9.9845e-01],
        [8.2820e-05, 9.9992e-01],
        [1.2826e-06, 1.0000e+00],
        [9.9555e-03, 9.9004e-01],
        [2.0788e-07, 1.0000e+00],
        [9.4634e-04, 9.9905e-01],
        [9.9953e-01, 4.7097e-04],
        [1.2676e-08, 1.0000e+00],
        [2.4789e-07, 1.0000e+00],
        [4.5616e-03, 9.9544e-01],
        [9.8506e-01, 1.4937e-02],
        [9.9896e-01, 1.0432e-03],
        [9.9757e-01, 2.4339e-03],
        [1.7622e-03, 9.9824e-01],
        [1.9950e-05, 9.9998e-01],
        [1.0556e-06, 1.0000e+00],
        [2.5810e-05, 9.9997e-01],
        [9.9754e-01, 2.4634e-03],
        [6.8629e-03, 9.9314e-01],
        [4.6401e-04, 9.9954e-01],
        [2.1500e-06, 1.0000e+00],
        [3.6964e-05, 9.9996e-01],
        [1.0624e-05, 9.9999e-01],
        [1.2676e-08, 1.0000e+00],
        [9.9784e-01, 2.1561e-03],
        [9.9991e-01, 8.5995e-05],
        [4.1656e-05, 9.9996e-01],
        [7.0703e-03, 9.9293e-01],
        [6.8050e-02, 9.3195e-01],
        [9.9934e-01, 6.5792e-04],
        [1.4252e-07, 1.0000e+00],
        [2.4929e-05, 9.9998e-01],
        [4.5530e-03, 9.9545e-01],
        [1.3805e-04, 9.9986e-01],
        [1.5329e-07, 1.0000e+00],
        [1.4888e-03, 9.9851e-01],
        [1.0082e-05, 9.9999e-01],
        [9.2218e-04, 9.9908e-01],
        [1.3539e-02, 9.8646e-01],
        [3.3538e-03, 9.9665e-01],
        [3.1796e-02, 9.6820e-01],
        [1.4031e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3288e-05, 9.9998e-01],
        [3.6102e-04, 9.9964e-01],
        [4.9492e-03, 9.9505e-01],
        [6.7763e-05, 9.9993e-01],
        [5.4365e-04, 9.9946e-01],
        [1.0561e-01, 8.9439e-01],
        [4.9905e-01, 5.0095e-01],
        [9.9958e-01, 4.2271e-04],
        [2.7978e-04, 9.9972e-01],
        [1.8797e-08, 1.0000e+00],
        [8.4750e-01, 1.5250e-01],
        [3.1021e-08, 1.0000e+00],
        [4.2971e-09, 1.0000e+00],
        [3.4606e-06, 1.0000e+00],
        [6.2064e-04, 9.9938e-01],
        [2.5240e-07, 1.0000e+00],
        [2.4569e-03, 9.9754e-01],
        [9.9994e-01, 5.9328e-05],
        [9.9889e-01, 1.1144e-03],
        [7.5582e-08, 1.0000e+00],
        [5.8967e-01, 4.1033e-01],
        [9.9992e-01, 8.3393e-05],
        [5.2074e-06, 9.9999e-01],
        [8.2258e-01, 1.7742e-01],
        [1.5897e-05, 9.9998e-01],
        [6.0143e-03, 9.9399e-01],
        [8.4259e-04, 9.9916e-01],
        [7.9111e-04, 9.9921e-01],
        [3.7642e-07, 1.0000e+00],
        [9.3068e-01, 6.9323e-02],
        [5.0799e-02, 9.4920e-01],
        [1.5049e-05, 9.9998e-01],
        [9.9995e-01, 5.3831e-05],
        [2.1583e-02, 9.7842e-01],
        [4.6008e-05, 9.9995e-01],
        [9.9977e-01, 2.3020e-04],
        [9.3613e-03, 9.9064e-01],
        [1.3044e-05, 9.9999e-01],
        [9.9972e-01, 2.8289e-04],
        [6.3798e-06, 9.9999e-01],
        [6.1853e-07, 1.0000e+00],
        [2.0556e-06, 1.0000e+00],
        [6.9891e-04, 9.9930e-01],
        [9.4582e-01, 5.4182e-02],
        [9.9987e-01, 1.2630e-04],
        [4.5847e-02, 9.5415e-01],
        [8.6365e-05, 9.9991e-01],
        [1.6973e-03, 9.9830e-01],
        [9.9894e-01, 1.0587e-03],
        [7.2696e-04, 9.9927e-01],
        [3.5653e-03, 9.9643e-01],
        [9.9996e-01, 4.2014e-05],
        [4.1005e-03, 9.9590e-01],
        [4.9067e-05, 9.9995e-01],
        [7.1698e-07, 1.0000e+00],
        [9.9984e-01, 1.6308e-04],
        [9.9947e-01, 5.3285e-04],
        [6.4447e-03, 9.9356e-01],
        [9.9213e-01, 7.8691e-03],
        [4.2670e-04, 9.9957e-01],
        [9.9978e-01, 2.2220e-04],
        [9.7220e-01, 2.7803e-02],
        [4.0933e-07, 1.0000e+00],
        [4.2037e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3610e-05, 9.9999e-01],
        [9.9945e-01, 5.5494e-04],
        [7.8282e-04, 9.9922e-01],
        [9.9965e-01, 3.5450e-04],
        [5.9733e-06, 9.9999e-01],
        [9.9972e-01, 2.7743e-04],
        [9.9983e-01, 1.7219e-04],
        [3.0282e-06, 1.0000e+00],
        [2.6018e-09, 1.0000e+00],
        [2.1865e-05, 9.9998e-01],
        [9.9972e-01, 2.7760e-04],
        [1.3349e-03, 9.9867e-01],
        [2.4598e-06, 1.0000e+00],
        [2.9557e-03, 9.9704e-01],
        [9.4189e-05, 9.9991e-01],
        [3.0820e-08, 1.0000e+00],
        [1.1583e-03, 9.9884e-01],
        [7.1095e-04, 9.9929e-01],
        [1.2806e-03, 9.9872e-01],
        [5.2762e-03, 9.9472e-01],
        [4.8857e-04, 9.9951e-01],
        [6.2778e-08, 1.0000e+00],
        [3.1825e-05, 9.9997e-01],
        [9.9962e-01, 3.7852e-04],
        [5.3520e-02, 9.4648e-01],
        [2.2245e-05, 9.9998e-01],
        [1.4966e-07, 1.0000e+00],
        [1.6180e-02, 9.8382e-01],
        [6.1682e-11, 1.0000e+00],
        [9.9174e-01, 8.2583e-03],
        [2.5176e-04, 9.9975e-01],
        [1.2509e-08, 1.0000e+00],
        [9.9980e-01, 1.9910e-04],
        [9.9983e-01, 1.6509e-04],
        [9.9964e-01, 3.6374e-04],
        [1.9146e-04, 9.9981e-01],
        [1.0741e-07, 1.0000e+00],
        [4.3429e-08, 1.0000e+00],
        [4.8731e-07, 1.0000e+00],
        [5.0194e-04, 9.9950e-01],
        [3.0005e-07, 1.0000e+00],
        [9.9926e-01, 7.3715e-04],
        [1.4328e-05, 9.9999e-01],
        [1.7102e-09, 1.0000e+00],
        [1.2668e-05, 9.9999e-01],
        [2.1629e-05, 9.9998e-01],
        [6.0969e-05, 9.9994e-01],
        [6.4546e-04, 9.9935e-01],
        [9.9037e-01, 9.6262e-03],
        [6.8687e-06, 9.9999e-01],
        [6.3174e-06, 9.9999e-01],
        [1.0172e-03, 9.9898e-01],
        [3.5345e-05, 9.9996e-01],
        [1.1167e-03, 9.9888e-01],
        [1.0216e-06, 1.0000e+00],
        [9.9298e-01, 7.0194e-03],
        [6.1819e-04, 9.9938e-01],
        [9.9737e-04, 9.9900e-01],
        [1.3428e-04, 9.9987e-01],
        [5.8291e-06, 9.9999e-01],
        [3.4211e-08, 1.0000e+00],
        [1.7869e-01, 8.2131e-01],
        [1.4987e-04, 9.9985e-01],
        [2.2842e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.7098e-08, 1.0000e+00],
        [1.5824e-05, 9.9998e-01],
        [9.8380e-01, 1.6198e-02],
        [1.3541e-03, 9.9865e-01],
        [1.6031e-06, 1.0000e+00],
        [5.4037e-03, 9.9460e-01],
        [5.2775e-06, 9.9999e-01],
        [4.5346e-04, 9.9955e-01],
        [7.1850e-05, 9.9993e-01],
        [3.8133e-04, 9.9962e-01],
        [3.0279e-08, 1.0000e+00],
        [2.6689e-06, 1.0000e+00],
        [4.6778e-03, 9.9532e-01],
        [2.8697e-02, 9.7130e-01],
        [8.5579e-06, 9.9999e-01],
        [2.2009e-04, 9.9978e-01],
        [2.0632e-05, 9.9998e-01],
        [4.8249e-05, 9.9995e-01],
        [4.4378e-08, 1.0000e+00],
        [4.2793e-02, 9.5721e-01],
        [4.0321e-03, 9.9597e-01],
        [6.7898e-08, 1.0000e+00],
        [9.9954e-01, 4.5694e-04],
        [9.9957e-01, 4.3019e-04],
        [1.1457e-06, 1.0000e+00],
        [6.8431e-05, 9.9993e-01],
        [1.7722e-08, 1.0000e+00],
        [6.7599e-07, 1.0000e+00],
        [6.5956e-06, 9.9999e-01],
        [9.9925e-01, 7.4554e-04],
        [1.1220e-03, 9.9888e-01],
        [9.9977e-01, 2.3266e-04],
        [1.5006e-09, 1.0000e+00],
        [7.1823e-01, 2.8177e-01],
        [2.1667e-06, 1.0000e+00],
        [4.5036e-06, 1.0000e+00],
        [1.0000e+00, 1.8813e-06],
        [9.9976e-01, 2.3902e-04],
        [2.3929e-04, 9.9976e-01],
        [1.2351e-03, 9.9876e-01],
        [9.9944e-01, 5.6115e-04],
        [9.9336e-01, 6.6362e-03],
        [2.1954e-06, 1.0000e+00],
        [1.7847e-04, 9.9982e-01],
        [3.2866e-01, 6.7134e-01],
        [6.4068e-07, 1.0000e+00],
        [2.0989e-05, 9.9998e-01],
        [9.8409e-01, 1.5914e-02],
        [4.9366e-06, 1.0000e+00],
        [2.4736e-05, 9.9998e-01],
        [1.5608e-06, 1.0000e+00],
        [9.9938e-01, 6.2266e-04],
        [9.9232e-01, 7.6791e-03],
        [2.3564e-05, 9.9998e-01],
        [9.8084e-05, 9.9990e-01],
        [1.3921e-05, 9.9999e-01],
        [3.1663e-05, 9.9997e-01],
        [9.4430e-08, 1.0000e+00],
        [1.3963e-03, 9.9860e-01],
        [9.9768e-01, 2.3170e-03],
        [9.2614e-05, 9.9991e-01],
        [1.0500e-06, 1.0000e+00],
        [8.0889e-09, 1.0000e+00],
        [9.9943e-01, 5.7068e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[5.1529e-04, 9.9948e-01],
        [6.1617e-08, 1.0000e+00],
        [5.2081e-07, 1.0000e+00],
        [9.9972e-01, 2.8073e-04],
        [9.9998e-01, 1.7871e-05],
        [4.1590e-05, 9.9996e-01],
        [1.4467e-05, 9.9999e-01],
        [6.3761e-07, 1.0000e+00],
        [2.4776e-05, 9.9998e-01],
        [9.9996e-01, 3.8234e-05],
        [9.7946e-01, 2.0544e-02],
        [3.1635e-04, 9.9968e-01],
        [7.2716e-03, 9.9273e-01],
        [9.8322e-08, 1.0000e+00],
        [4.9782e-06, 9.9999e-01],
        [9.3099e-04, 9.9907e-01],
        [1.4833e-03, 9.9852e-01],
        [9.9634e-01, 3.6558e-03],
        [2.1806e-06, 1.0000e+00],
        [2.4462e-07, 1.0000e+00],
        [1.2201e-01, 8.7799e-01],
        [5.9928e-01, 4.0072e-01],
        [7.0142e-06, 9.9999e-01],
        [4.1806e-09, 1.0000e+00],
        [9.9667e-01, 3.3275e-03],
        [4.3858e-04, 9.9956e-01],
        [1.3883e-06, 1.0000e+00],
        [2.2817e-08, 1.0000e+00],
        [3.1494e-05, 9.9997e-01],
        [1.5595e-05, 9.9998e-01],
        [2.8597e-09, 1.0000e+00],
        [9.9957e-01, 4.3252e-04],
        [1.7135e-03, 9.9829e-01],
        [1.0946e-03, 9.9891e-01],
        [6.1405e-08, 1.0000e+00],
        [6.0383e-07, 1.0000e+00],
        [2.1815e-07, 1.0000e+00],
        [5.4566e-08, 1.0000e+00],
        [9.9999e-01, 6.0083e-06],
        [1.3671e-06, 1.0000e+00],
        [9.9950e-01, 5.0434e-04],
        [1.1095e-02, 9.8891e-01],
        [1.8893e-05, 9.9998e-01],
        [1.5199e-03, 9.9848e-01],
        [7.2869e-08, 1.0000e+00],
        [5.7099e-05, 9.9994e-01],
        [4.0315e-07, 1.0000e+00],
        [4.2229e-06, 1.0000e+00],
        [9.8851e-01, 1.1490e-02],
        [9.5654e-08, 1.0000e+00],
        [5.9028e-04, 9.9941e-01],
        [2.6473e-06, 1.0000e+00],
        [2.1998e-05, 9.9998e-01],
        [9.9971e-01, 2.9142e-04],
        [3.1060e-06, 1.0000e+00],
        [1.7434e-06, 1.0000e+00],
        [1.6832e-07, 1.0000e+00],
        [9.6386e-06, 9.9999e-01],
        [7.3346e-01, 2.6654e-01],
        [1.2443e-04, 9.9988e-01],
        [4.6782e-10, 1.0000e+00],
        [9.9882e-01, 1.1785e-03],
        [1.1454e-08, 1.0000e+00],
        [2.8951e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.2031e-02, 9.2797e-01],
        [7.6401e-05, 9.9992e-01],
        [8.2727e-07, 1.0000e+00],
        [4.9869e-09, 1.0000e+00],
        [1.5011e-05, 9.9998e-01],
        [3.6746e-07, 1.0000e+00],
        [1.8886e-02, 9.8111e-01],
        [2.2546e-01, 7.7454e-01],
        [2.7887e-04, 9.9972e-01],
        [9.9999e-01, 6.0116e-06],
        [9.8945e-01, 1.0555e-02],
        [4.0210e-11, 1.0000e+00],
        [5.7276e-12, 1.0000e+00],
        [5.7326e-06, 9.9999e-01],
        [3.6339e-03, 9.9637e-01],
        [1.4616e-07, 1.0000e+00],
        [5.3430e-06, 9.9999e-01],
        [1.7930e-06, 1.0000e+00],
        [1.4855e-01, 8.5145e-01],
        [9.9942e-01, 5.7609e-04],
        [5.9283e-06, 9.9999e-01],
        [1.0656e-03, 9.9893e-01],
        [2.2140e-03, 9.9779e-01],
        [1.0834e-07, 1.0000e+00],
        [2.6043e-04, 9.9974e-01],
        [8.7046e-04, 9.9913e-01],
        [3.1661e-03, 9.9683e-01],
        [5.7086e-04, 9.9943e-01],
        [9.9887e-01, 1.1274e-03],
        [8.2305e-06, 9.9999e-01],
        [6.4101e-02, 9.3590e-01],
        [7.4386e-10, 1.0000e+00],
        [9.6173e-11, 1.0000e+00],
        [3.1843e-01, 6.8157e-01],
        [5.8037e-08, 1.0000e+00],
        [8.6630e-06, 9.9999e-01],
        [6.6944e-09, 1.0000e+00],
        [9.1188e-01, 8.8124e-02],
        [4.5849e-06, 1.0000e+00],
        [9.9878e-01, 1.2212e-03],
        [9.9219e-01, 7.8081e-03],
        [4.3323e-06, 1.0000e+00],
        [2.0787e-07, 1.0000e+00],
        [3.4954e-10, 1.0000e+00],
        [7.0239e-08, 1.0000e+00],
        [9.9942e-01, 5.7716e-04],
        [3.7574e-09, 1.0000e+00],
        [1.8712e-06, 1.0000e+00],
        [1.5524e-03, 9.9845e-01],
        [3.6679e-06, 1.0000e+00],
        [1.5571e-02, 9.8443e-01],
        [7.6636e-07, 1.0000e+00],
        [9.9943e-01, 5.7099e-04],
        [1.4849e-05, 9.9999e-01],
        [6.7446e-07, 1.0000e+00],
        [3.7222e-04, 9.9963e-01],
        [1.5090e-08, 1.0000e+00],
        [8.7495e-01, 1.2505e-01],
        [9.9934e-01, 6.6118e-04],
        [1.9429e-09, 1.0000e+00],
        [3.4482e-06, 1.0000e+00],
        [7.8811e-08, 1.0000e+00],
        [9.0132e-02, 9.0987e-01],
        [9.9982e-01, 1.8299e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2691e-06, 1.0000e+00],
        [2.9425e-02, 9.7057e-01],
        [1.6134e-10, 1.0000e+00],
        [4.0961e-01, 5.9039e-01],
        [1.6617e-06, 1.0000e+00],
        [2.7555e-08, 1.0000e+00],
        [9.9906e-01, 9.4333e-04],
        [8.6502e-05, 9.9991e-01],
        [9.7308e-01, 2.6924e-02],
        [1.1325e-06, 1.0000e+00],
        [6.6848e-07, 1.0000e+00],
        [9.9425e-01, 5.7542e-03],
        [9.1193e-07, 1.0000e+00],
        [1.2816e-07, 1.0000e+00],
        [3.8889e-05, 9.9996e-01],
        [7.8111e-05, 9.9992e-01],
        [2.4085e-07, 1.0000e+00],
        [9.6442e-01, 3.5580e-02],
        [1.4576e-10, 1.0000e+00],
        [1.7316e-04, 9.9983e-01],
        [3.4873e-11, 1.0000e+00],
        [3.3292e-06, 1.0000e+00],
        [5.5096e-04, 9.9945e-01],
        [3.9731e-10, 1.0000e+00],
        [1.6914e-04, 9.9983e-01],
        [8.1921e-06, 9.9999e-01],
        [1.9577e-05, 9.9998e-01],
        [1.3543e-08, 1.0000e+00],
        [9.9933e-01, 6.6625e-04],
        [1.4836e-05, 9.9999e-01],
        [2.3683e-10, 1.0000e+00],
        [4.4152e-08, 1.0000e+00],
        [9.9700e-01, 3.0047e-03],
        [5.6588e-02, 9.4341e-01],
        [1.1680e-04, 9.9988e-01],
        [5.3196e-10, 1.0000e+00],
        [2.0463e-05, 9.9998e-01],
        [1.5935e-07, 1.0000e+00],
        [2.1405e-06, 1.0000e+00],
        [9.6215e-08, 1.0000e+00],
        [1.0789e-07, 1.0000e+00],
        [1.8420e-05, 9.9998e-01],
        [7.8127e-01, 2.1873e-01],
        [1.1764e-06, 1.0000e+00],
        [1.5214e-08, 1.0000e+00],
        [1.7462e-05, 9.9998e-01],
        [9.6522e-01, 3.4775e-02],
        [1.3476e-07, 1.0000e+00],
        [5.4871e-09, 1.0000e+00],
        [1.0000e+00, 2.3744e-06],
        [8.0401e-03, 9.9196e-01],
        [2.1339e-05, 9.9998e-01],
        [1.2043e-03, 9.9880e-01],
        [4.8341e-04, 9.9952e-01],
        [9.9536e-01, 4.6431e-03],
        [6.8814e-06, 9.9999e-01],
        [2.0640e-06, 1.0000e+00],
        [9.9644e-01, 3.5568e-03],
        [7.4487e-07, 1.0000e+00],
        [9.9974e-01, 2.5665e-04],
        [2.9336e-05, 9.9997e-01],
        [1.0547e-06, 1.0000e+00],
        [1.6664e-05, 9.9998e-01],
        [9.9682e-01, 3.1812e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[5.7338e-08, 1.0000e+00],
        [1.4524e-07, 1.0000e+00],
        [8.9208e-04, 9.9911e-01],
        [9.9956e-01, 4.3775e-04],
        [8.5027e-06, 9.9999e-01],
        [1.0868e-06, 1.0000e+00],
        [4.0268e-05, 9.9996e-01],
        [3.3315e-01, 6.6685e-01],
        [2.8251e-06, 1.0000e+00],
        [1.9996e-04, 9.9980e-01],
        [2.2796e-03, 9.9772e-01],
        [1.2505e-04, 9.9987e-01],
        [9.9542e-01, 4.5790e-03],
        [2.1072e-05, 9.9998e-01],
        [5.3146e-06, 9.9999e-01],
        [1.7338e-05, 9.9998e-01],
        [3.3838e-05, 9.9997e-01],
        [1.7418e-06, 1.0000e+00],
        [4.8641e-06, 1.0000e+00],
        [8.7575e-06, 9.9999e-01],
        [1.5811e-07, 1.0000e+00],
        [6.0265e-09, 1.0000e+00],
        [1.5811e-07, 1.0000e+00],
        [9.5851e-07, 1.0000e+00],
        [9.9177e-01, 8.2322e-03],
        [9.9938e-01, 6.2259e-04],
        [4.9198e-04, 9.9951e-01],
        [9.9981e-01, 1.8916e-04],
        [1.2249e-04, 9.9988e-01],
        [2.8132e-06, 1.0000e+00],
        [3.1934e-06, 1.0000e+00],
        [2.3450e-04, 9.9977e-01],
        [8.4771e-04, 9.9915e-01],
        [1.0000e+00, 1.2729e-06],
        [5.0161e-05, 9.9995e-01],
        [9.5983e-09, 1.0000e+00],
        [9.9338e-01, 6.6232e-03],
        [1.3682e-07, 1.0000e+00],
        [9.9680e-01, 3.1991e-03],
        [1.6347e-07, 1.0000e+00],
        [1.9653e-05, 9.9998e-01],
        [4.5854e-09, 1.0000e+00],
        [2.0484e-01, 7.9516e-01],
        [9.8834e-01, 1.1665e-02],
        [2.5405e-05, 9.9997e-01],
        [9.9186e-01, 8.1358e-03],
        [3.8957e-04, 9.9961e-01],
        [9.9646e-01, 3.5431e-03],
        [2.4316e-07, 1.0000e+00],
        [1.7834e-05, 9.9998e-01],
        [3.2998e-01, 6.7002e-01],
        [2.1883e-07, 1.0000e+00],
        [1.0840e-06, 1.0000e+00],
        [7.4032e-07, 1.0000e+00],
        [3.3821e-08, 1.0000e+00],
        [3.1868e-07, 1.0000e+00],
        [4.0012e-07, 1.0000e+00],
        [2.9355e-05, 9.9997e-01],
        [4.7914e-07, 1.0000e+00],
        [5.8862e-01, 4.1138e-01],
        [5.7530e-08, 1.0000e+00],
        [7.5338e-04, 9.9925e-01],
        [5.9984e-09, 1.0000e+00],
        [1.6297e-03, 9.9837e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9114e-03, 9.9809e-01],
        [7.4363e-07, 1.0000e+00],
        [9.9122e-01, 8.7820e-03],
        [2.6119e-09, 1.0000e+00],
        [8.5415e-08, 1.0000e+00],
        [4.8560e-07, 1.0000e+00],
        [9.9952e-01, 4.7613e-04],
        [2.8655e-07, 1.0000e+00],
        [5.0669e-07, 1.0000e+00],
        [8.4062e-04, 9.9916e-01],
        [3.1614e-04, 9.9968e-01],
        [5.0883e-01, 4.9117e-01],
        [2.3116e-07, 1.0000e+00],
        [2.1268e-08, 1.0000e+00],
        [4.3976e-06, 1.0000e+00],
        [6.6321e-05, 9.9993e-01],
        [1.3226e-05, 9.9999e-01],
        [8.3844e-03, 9.9162e-01],
        [5.8418e-06, 9.9999e-01],
        [5.3480e-03, 9.9465e-01],
        [9.9906e-01, 9.3744e-04],
        [9.9816e-01, 1.8415e-03],
        [1.4467e-04, 9.9986e-01],
        [9.5825e-01, 4.1755e-02],
        [1.5011e-06, 1.0000e+00],
        [9.9840e-01, 1.5982e-03],
        [3.0907e-07, 1.0000e+00],
        [1.9879e-05, 9.9998e-01],
        [6.4799e-04, 9.9935e-01],
        [3.4153e-14, 1.0000e+00],
        [3.0107e-07, 1.0000e+00],
        [9.9947e-01, 5.3213e-04],
        [1.1007e-04, 9.9989e-01],
        [9.8094e-08, 1.0000e+00],
        [3.7843e-04, 9.9962e-01],
        [1.0178e-03, 9.9898e-01],
        [2.5317e-01, 7.4683e-01],
        [3.1063e-09, 1.0000e+00],
        [9.8627e-06, 9.9999e-01],
        [4.7775e-09, 1.0000e+00],
        [5.7499e-03, 9.9425e-01],
        [1.6943e-03, 9.9831e-01],
        [5.6502e-07, 1.0000e+00],
        [2.5717e-07, 1.0000e+00],
        [9.9685e-01, 3.1521e-03],
        [2.7508e-08, 1.0000e+00],
        [6.7473e-01, 3.2527e-01],
        [3.8195e-08, 1.0000e+00],
        [4.8187e-06, 1.0000e+00],
        [3.1472e-04, 9.9969e-01],
        [4.7027e-04, 9.9953e-01],
        [9.9972e-01, 2.8343e-04],
        [5.0457e-04, 9.9950e-01],
        [1.8057e-09, 1.0000e+00],
        [1.2174e-06, 1.0000e+00],
        [9.9976e-01, 2.3676e-04],
        [9.9911e-01, 8.8941e-04],
        [7.2605e-08, 1.0000e+00],
        [1.1233e-08, 1.0000e+00],
        [7.2598e-06, 9.9999e-01],
        [9.9917e-01, 8.2968e-04],
        [3.5387e-08, 1.0000e+00],
        [4.2942e-08, 1.0000e+00],
        [1.0030e-04, 9.9990e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3478e-02, 9.8652e-01],
        [3.1148e-08, 1.0000e+00],
        [5.9743e-06, 9.9999e-01],
        [9.9939e-01, 6.1411e-04],
        [9.0580e-04, 9.9909e-01],
        [1.1558e-04, 9.9988e-01],
        [3.8854e-07, 1.0000e+00],
        [9.9821e-01, 1.7882e-03],
        [9.0193e-07, 1.0000e+00],
        [4.2155e-07, 1.0000e+00],
        [2.4441e-07, 1.0000e+00],
        [1.8253e-08, 1.0000e+00],
        [1.8346e-07, 1.0000e+00],
        [9.9476e-01, 5.2375e-03],
        [2.9322e-10, 1.0000e+00],
        [1.2600e-03, 9.9874e-01],
        [5.6404e-05, 9.9994e-01],
        [3.8387e-09, 1.0000e+00],
        [1.4245e-08, 1.0000e+00],
        [9.9933e-01, 6.7042e-04],
        [7.0712e-06, 9.9999e-01],
        [2.4717e-04, 9.9975e-01],
        [1.8051e-08, 1.0000e+00],
        [2.5496e-08, 1.0000e+00],
        [3.6803e-08, 1.0000e+00],
        [1.1115e-05, 9.9999e-01],
        [2.4610e-07, 1.0000e+00],
        [8.5809e-07, 1.0000e+00],
        [7.3201e-07, 1.0000e+00],
        [1.2082e-04, 9.9988e-01],
        [9.9146e-01, 8.5420e-03],
        [3.9502e-10, 1.0000e+00],
        [4.3046e-07, 1.0000e+00],
        [5.3503e-07, 1.0000e+00],
        [7.1321e-03, 9.9287e-01],
        [6.6459e-07, 1.0000e+00],
        [1.1060e-08, 1.0000e+00],
        [5.6786e-08, 1.0000e+00],
        [5.7252e-06, 9.9999e-01],
        [1.0185e-08, 1.0000e+00],
        [5.5304e-04, 9.9945e-01],
        [2.7695e-05, 9.9997e-01],
        [5.1655e-07, 1.0000e+00],
        [4.4808e-05, 9.9996e-01],
        [1.4888e-04, 9.9985e-01],
        [3.7513e-08, 1.0000e+00],
        [3.1623e-04, 9.9968e-01],
        [1.2816e-05, 9.9999e-01],
        [1.7819e-08, 1.0000e+00],
        [4.2182e-04, 9.9958e-01],
        [1.4222e-08, 1.0000e+00],
        [2.2148e-02, 9.7785e-01],
        [1.6055e-06, 1.0000e+00],
        [9.9923e-01, 7.7437e-04],
        [6.7689e-04, 9.9932e-01],
        [9.9989e-01, 1.1050e-04],
        [2.5856e-07, 1.0000e+00],
        [2.0535e-03, 9.9795e-01],
        [1.9823e-04, 9.9980e-01],
        [4.9385e-06, 1.0000e+00],
        [9.9917e-01, 8.3329e-04],
        [1.2048e-10, 1.0000e+00],
        [9.9903e-01, 9.6519e-04],
        [3.2779e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.8903e-08, 1.0000e+00],
        [1.0389e-02, 9.8961e-01],
        [7.8168e-07, 1.0000e+00],
        [3.8361e-10, 1.0000e+00],
        [1.0690e-06, 1.0000e+00],
        [1.6053e-05, 9.9998e-01],
        [3.9430e-05, 9.9996e-01],
        [9.0772e-06, 9.9999e-01],
        [2.9049e-03, 9.9710e-01],
        [4.5016e-02, 9.5498e-01],
        [9.9971e-01, 2.9261e-04],
        [4.0329e-09, 1.0000e+00],
        [9.9915e-01, 8.4554e-04],
        [9.9025e-01, 9.7547e-03],
        [2.4841e-01, 7.5159e-01],
        [1.0448e-05, 9.9999e-01],
        [9.9964e-01, 3.5607e-04],
        [6.3908e-06, 9.9999e-01],
        [1.1579e-07, 1.0000e+00],
        [9.9870e-01, 1.3034e-03],
        [8.5363e-10, 1.0000e+00],
        [9.7038e-01, 2.9621e-02],
        [2.4689e-08, 1.0000e+00],
        [5.3609e-09, 1.0000e+00],
        [2.3911e-07, 1.0000e+00],
        [6.8380e-04, 9.9932e-01],
        [2.7324e-07, 1.0000e+00],
        [2.3339e-08, 1.0000e+00],
        [4.7535e-06, 1.0000e+00],
        [4.0660e-07, 1.0000e+00],
        [5.1704e-08, 1.0000e+00],
        [1.0824e-05, 9.9999e-01],
        [9.9680e-01, 3.2025e-03],
        [9.0212e-01, 9.7879e-02],
        [9.9187e-01, 8.1320e-03],
        [2.2833e-04, 9.9977e-01],
        [2.6269e-07, 1.0000e+00],
        [6.2770e-06, 9.9999e-01],
        [1.1287e-06, 1.0000e+00],
        [1.0148e-09, 1.0000e+00],
        [8.6818e-01, 1.3182e-01],
        [9.9969e-01, 3.1432e-04],
        [5.4761e-08, 1.0000e+00],
        [3.7617e-04, 9.9962e-01],
        [9.9268e-01, 7.3184e-03],
        [3.7153e-09, 1.0000e+00],
        [6.6534e-06, 9.9999e-01],
        [4.6330e-08, 1.0000e+00],
        [3.4135e-07, 1.0000e+00],
        [9.9868e-01, 1.3196e-03],
        [7.4299e-09, 1.0000e+00],
        [8.3093e-05, 9.9992e-01],
        [3.7844e-03, 9.9622e-01],
        [1.3005e-06, 1.0000e+00],
        [3.4042e-05, 9.9997e-01],
        [9.9953e-01, 4.7404e-04],
        [6.0586e-04, 9.9939e-01],
        [2.1971e-05, 9.9998e-01],
        [1.0000e+00, 5.9642e-07],
        [3.0559e-08, 1.0000e+00],
        [8.0122e-05, 9.9992e-01],
        [1.0580e-06, 1.0000e+00],
        [1.9984e-05, 9.9998e-01],
        [5.9582e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3969e-05, 9.9999e-01],
        [2.9684e-06, 1.0000e+00],
        [6.9147e-07, 1.0000e+00],
        [7.5627e-06, 9.9999e-01],
        [9.8886e-01, 1.1139e-02],
        [2.3561e-05, 9.9998e-01],
        [8.6427e-01, 1.3573e-01],
        [2.4789e-08, 1.0000e+00],
        [9.9955e-01, 4.4751e-04],
        [1.4268e-06, 1.0000e+00],
        [1.8822e-05, 9.9998e-01],
        [1.8858e-05, 9.9998e-01],
        [2.9367e-07, 1.0000e+00],
        [9.1717e-01, 8.2829e-02],
        [1.8885e-06, 1.0000e+00],
        [1.2893e-04, 9.9987e-01],
        [4.1391e-04, 9.9959e-01],
        [1.7430e-07, 1.0000e+00],
        [9.9461e-01, 5.3862e-03],
        [4.5240e-06, 1.0000e+00],
        [5.5446e-02, 9.4455e-01],
        [4.7416e-04, 9.9953e-01],
        [5.8504e-07, 1.0000e+00],
        [3.3688e-04, 9.9966e-01],
        [1.4032e-03, 9.9860e-01],
        [4.4483e-08, 1.0000e+00],
        [6.5069e-04, 9.9935e-01],
        [1.4125e-09, 1.0000e+00],
        [9.9997e-01, 3.1805e-05],
        [9.9321e-01, 6.7925e-03],
        [7.2976e-05, 9.9993e-01],
        [3.4407e-05, 9.9997e-01],
        [2.4376e-05, 9.9998e-01],
        [5.0721e-07, 1.0000e+00],
        [1.5153e-06, 1.0000e+00],
        [1.0152e-04, 9.9990e-01],
        [1.5026e-09, 1.0000e+00],
        [5.8700e-01, 4.1300e-01],
        [1.1386e-08, 1.0000e+00],
        [2.4723e-06, 1.0000e+00],
        [9.3366e-04, 9.9907e-01],
        [1.9528e-07, 1.0000e+00],
        [2.3699e-04, 9.9976e-01],
        [2.6865e-07, 1.0000e+00],
        [9.9788e-01, 2.1195e-03],
        [1.4497e-02, 9.8550e-01],
        [9.8744e-01, 1.2563e-02],
        [6.3819e-05, 9.9994e-01],
        [9.9943e-01, 5.7095e-04],
        [3.2604e-04, 9.9967e-01],
        [4.5386e-10, 1.0000e+00],
        [2.7808e-06, 1.0000e+00],
        [5.1098e-01, 4.8902e-01],
        [9.6608e-01, 3.3920e-02],
        [2.0584e-06, 1.0000e+00],
        [2.6206e-05, 9.9997e-01],
        [2.9019e-08, 1.0000e+00],
        [9.9974e-01, 2.6122e-04],
        [9.1508e-05, 9.9991e-01],
        [7.8112e-01, 2.1888e-01],
        [3.1393e-07, 1.0000e+00],
        [1.3893e-07, 1.0000e+00],
        [2.9875e-08, 1.0000e+00],
        [1.1774e-04, 9.9988e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9951e-01, 4.9383e-04],
        [1.8307e-09, 1.0000e+00],
        [1.4975e-03, 9.9850e-01],
        [9.9990e-01, 9.8956e-05],
        [2.0235e-05, 9.9998e-01],
        [5.8152e-06, 9.9999e-01],
        [1.2273e-05, 9.9999e-01],
        [3.5353e-05, 9.9996e-01],
        [4.6201e-07, 1.0000e+00],
        [4.0574e-07, 1.0000e+00],
        [5.7303e-01, 4.2697e-01],
        [9.2915e-08, 1.0000e+00],
        [1.2566e-06, 1.0000e+00],
        [9.9935e-01, 6.5202e-04],
        [1.0992e-03, 9.9890e-01],
        [9.7298e-01, 2.7016e-02],
        [1.2198e-07, 1.0000e+00],
        [9.9964e-01, 3.6184e-04],
        [3.1248e-05, 9.9997e-01],
        [6.8594e-09, 1.0000e+00],
        [7.1616e-05, 9.9993e-01],
        [9.9934e-01, 6.6371e-04],
        [9.9897e-01, 1.0270e-03],
        [2.1307e-04, 9.9979e-01],
        [9.9948e-01, 5.1594e-04],
        [3.2831e-08, 1.0000e+00],
        [2.1384e-04, 9.9979e-01],
        [9.0089e-06, 9.9999e-01],
        [8.7324e-01, 1.2676e-01],
        [4.7032e-07, 1.0000e+00],
        [9.9957e-01, 4.2963e-04],
        [3.7076e-05, 9.9996e-01],
        [1.1426e-05, 9.9999e-01],
        [2.6313e-06, 1.0000e+00],
        [9.8303e-01, 1.6974e-02],
        [9.9536e-01, 4.6378e-03],
        [2.7096e-06, 1.0000e+00],
        [1.0316e-05, 9.9999e-01],
        [3.4130e-06, 1.0000e+00],
        [5.4752e-02, 9.4525e-01],
        [1.1336e-05, 9.9999e-01],
        [5.2077e-05, 9.9995e-01],
        [2.6049e-06, 1.0000e+00],
        [9.5663e-01, 4.3373e-02],
        [2.6247e-04, 9.9974e-01],
        [5.7831e-08, 1.0000e+00],
        [2.5713e-07, 1.0000e+00],
        [4.5073e-05, 9.9995e-01],
        [4.0281e-06, 1.0000e+00],
        [9.9991e-01, 9.0991e-05],
        [9.9996e-01, 4.2654e-05],
        [1.2897e-08, 1.0000e+00],
        [2.7086e-03, 9.9729e-01],
        [9.0685e-01, 9.3154e-02],
        [6.0020e-08, 1.0000e+00],
        [1.1841e-04, 9.9988e-01],
        [5.9463e-01, 4.0537e-01],
        [2.8652e-10, 1.0000e+00],
        [2.2404e-06, 1.0000e+00],
        [7.1087e-07, 1.0000e+00],
        [6.7632e-02, 9.3237e-01],
        [7.7568e-05, 9.9992e-01],
        [3.9882e-05, 9.9996e-01],
        [5.0886e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9594e-06, 1.0000e+00],
        [9.9999e-01, 8.9944e-06],
        [2.2718e-09, 1.0000e+00],
        [5.5844e-08, 1.0000e+00],
        [8.6072e-02, 9.1393e-01],
        [1.4181e-02, 9.8582e-01],
        [1.6539e-08, 1.0000e+00],
        [9.9123e-01, 8.7739e-03],
        [7.6844e-08, 1.0000e+00],
        [8.4608e-04, 9.9915e-01],
        [1.4001e-03, 9.9860e-01],
        [9.9967e-01, 3.3236e-04],
        [9.9939e-01, 6.0900e-04],
        [2.7790e-05, 9.9997e-01],
        [1.0680e-04, 9.9989e-01],
        [9.8745e-01, 1.2550e-02],
        [6.7894e-08, 1.0000e+00],
        [8.6318e-09, 1.0000e+00],
        [3.5463e-05, 9.9996e-01],
        [1.9420e-04, 9.9981e-01],
        [5.3007e-08, 1.0000e+00],
        [9.9948e-01, 5.2043e-04],
        [9.9988e-01, 1.1533e-04],
        [2.8829e-03, 9.9712e-01],
        [9.9980e-01, 1.9977e-04],
        [9.2420e-05, 9.9991e-01],
        [4.4575e-03, 9.9554e-01],
        [6.3907e-06, 9.9999e-01],
        [5.9971e-02, 9.4003e-01],
        [1.9425e-06, 1.0000e+00],
        [3.5427e-11, 1.0000e+00],
        [1.9746e-05, 9.9998e-01],
        [4.0488e-09, 1.0000e+00],
        [1.0198e-08, 1.0000e+00],
        [1.9292e-05, 9.9998e-01],
        [1.1145e-06, 1.0000e+00],
        [1.2363e-04, 9.9988e-01],
        [1.7995e-05, 9.9998e-01],
        [5.4354e-04, 9.9946e-01],
        [2.2988e-07, 1.0000e+00],
        [1.3651e-03, 9.9863e-01],
        [4.3518e-05, 9.9996e-01],
        [9.8688e-01, 1.3116e-02],
        [4.1836e-07, 1.0000e+00],
        [7.5732e-06, 9.9999e-01],
        [9.9987e-01, 1.3158e-04],
        [7.9534e-01, 2.0466e-01],
        [4.2201e-02, 9.5780e-01],
        [1.4250e-05, 9.9999e-01],
        [1.2240e-07, 1.0000e+00],
        [9.9990e-01, 1.0446e-04],
        [9.9987e-01, 1.2718e-04],
        [7.7950e-04, 9.9922e-01],
        [9.9949e-01, 5.0959e-04],
        [9.9952e-01, 4.7667e-04],
        [9.9962e-01, 3.8417e-04],
        [3.9350e-04, 9.9961e-01],
        [9.5090e-01, 4.9103e-02],
        [8.7797e-04, 9.9912e-01],
        [6.0542e-07, 1.0000e+00],
        [2.0765e-03, 9.9792e-01],
        [9.8330e-01, 1.6695e-02],
        [3.4376e-05, 9.9997e-01],
        [1.8579e-04, 9.9981e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8645e-01, 1.3549e-02],
        [9.0067e-07, 1.0000e+00],
        [8.2236e-06, 9.9999e-01],
        [9.9648e-01, 3.5158e-03],
        [9.9999e-01, 6.7713e-06],
        [8.5195e-03, 9.9148e-01],
        [8.3621e-01, 1.6379e-01],
        [3.9680e-08, 1.0000e+00],
        [9.9963e-01, 3.7061e-04],
        [5.5022e-02, 9.4498e-01],
        [5.7938e-03, 9.9421e-01],
        [1.9766e-05, 9.9998e-01],
        [9.1552e-06, 9.9999e-01],
        [1.3405e-09, 1.0000e+00],
        [5.3084e-04, 9.9947e-01],
        [9.9951e-01, 4.8630e-04],
        [3.0891e-09, 1.0000e+00],
        [9.9955e-01, 4.4598e-04],
        [7.0545e-05, 9.9993e-01],
        [2.3570e-05, 9.9998e-01],
        [3.6637e-09, 1.0000e+00],
        [2.7817e-05, 9.9997e-01],
        [9.8158e-01, 1.8420e-02],
        [7.8938e-06, 9.9999e-01],
        [8.5403e-01, 1.4597e-01],
        [2.8273e-05, 9.9997e-01],
        [4.6029e-05, 9.9995e-01],
        [9.9514e-06, 9.9999e-01],
        [9.9981e-01, 1.8552e-04],
        [3.8581e-06, 1.0000e+00],
        [6.0861e-08, 1.0000e+00],
        [3.5976e-06, 1.0000e+00],
        [1.6731e-07, 1.0000e+00],
        [1.4735e-04, 9.9985e-01],
        [2.9756e-05, 9.9997e-01],
        [3.3757e-02, 9.6624e-01],
        [7.8381e-03, 9.9216e-01],
        [8.0965e-07, 1.0000e+00],
        [8.1908e-07, 1.0000e+00],
        [8.8770e-09, 1.0000e+00],
        [1.4837e-04, 9.9985e-01],
        [1.3822e-02, 9.8618e-01],
        [1.2890e-02, 9.8711e-01],
        [3.8875e-07, 1.0000e+00],
        [9.9953e-01, 4.6714e-04],
        [3.9180e-06, 1.0000e+00],
        [9.9930e-01, 7.0046e-04],
        [9.6165e-09, 1.0000e+00],
        [9.9969e-01, 3.1194e-04],
        [3.0569e-04, 9.9969e-01],
        [9.9942e-01, 5.7520e-04],
        [3.6892e-07, 1.0000e+00],
        [2.0801e-07, 1.0000e+00],
        [1.3004e-05, 9.9999e-01],
        [8.5362e-05, 9.9991e-01],
        [1.3774e-06, 1.0000e+00],
        [1.0211e-05, 9.9999e-01],
        [1.7119e-01, 8.2881e-01],
        [5.2802e-06, 9.9999e-01],
        [2.6354e-03, 9.9736e-01],
        [9.9964e-01, 3.5908e-04],
        [9.9955e-01, 4.4818e-04],
        [4.9914e-06, 9.9999e-01],
        [1.2295e-04, 9.9988e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.7408e-04, 9.9943e-01],
        [9.9956e-01, 4.3633e-04],
        [1.1877e-05, 9.9999e-01],
        [1.0423e-06, 1.0000e+00],
        [1.9884e-07, 1.0000e+00],
        [2.9211e-05, 9.9997e-01],
        [5.7289e-05, 9.9994e-01],
        [9.3939e-06, 9.9999e-01],
        [9.6062e-01, 3.9383e-02],
        [2.1005e-06, 1.0000e+00],
        [2.0005e-05, 9.9998e-01],
        [1.4918e-04, 9.9985e-01],
        [9.9972e-01, 2.8487e-04],
        [2.7004e-05, 9.9997e-01],
        [4.5014e-06, 1.0000e+00],
        [1.7625e-03, 9.9824e-01],
        [7.7889e-07, 1.0000e+00],
        [9.9945e-01, 5.5117e-04],
        [3.0132e-07, 1.0000e+00],
        [3.1941e-04, 9.9968e-01],
        [9.9992e-01, 7.7778e-05],
        [9.9957e-01, 4.2881e-04],
        [1.3830e-02, 9.8617e-01],
        [9.5684e-01, 4.3164e-02],
        [4.0495e-05, 9.9996e-01],
        [4.0084e-09, 1.0000e+00],
        [3.1971e-05, 9.9997e-01],
        [1.0210e-01, 8.9790e-01],
        [2.2723e-06, 1.0000e+00],
        [4.5461e-05, 9.9995e-01],
        [2.0052e-05, 9.9998e-01],
        [9.9928e-01, 7.1993e-04],
        [6.3896e-06, 9.9999e-01],
        [1.7191e-01, 8.2809e-01],
        [1.4943e-04, 9.9985e-01],
        [2.4996e-06, 1.0000e+00],
        [2.7114e-07, 1.0000e+00],
        [6.0563e-03, 9.9394e-01],
        [1.5598e-09, 1.0000e+00],
        [2.0359e-03, 9.9796e-01],
        [9.8961e-01, 1.0387e-02],
        [4.7157e-08, 1.0000e+00],
        [6.1852e-01, 3.8148e-01],
        [2.8769e-05, 9.9997e-01],
        [8.1337e-01, 1.8663e-01],
        [9.9938e-01, 6.2255e-04],
        [1.6859e-01, 8.3141e-01],
        [1.5350e-04, 9.9985e-01],
        [8.9989e-03, 9.9100e-01],
        [9.9976e-01, 2.4347e-04],
        [2.4596e-06, 1.0000e+00],
        [1.9151e-05, 9.9998e-01],
        [1.1048e-04, 9.9989e-01],
        [9.1130e-01, 8.8698e-02],
        [9.4234e-06, 9.9999e-01],
        [2.3841e-08, 1.0000e+00],
        [3.2446e-06, 1.0000e+00],
        [9.9957e-01, 4.3411e-04],
        [5.6039e-06, 9.9999e-01],
        [9.9991e-01, 8.5170e-05],
        [5.0176e-06, 9.9999e-01],
        [1.2059e-03, 9.9879e-01],
        [1.7028e-05, 9.9998e-01],
        [5.1689e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9971e-01, 2.8931e-04],
        [1.3642e-06, 1.0000e+00],
        [2.9466e-03, 9.9705e-01],
        [1.2585e-03, 9.9874e-01],
        [7.9449e-05, 9.9992e-01],
        [6.2942e-05, 9.9994e-01],
        [9.9993e-01, 7.3236e-05],
        [9.9983e-01, 1.7103e-04],
        [5.8394e-05, 9.9994e-01],
        [3.9750e-07, 1.0000e+00],
        [9.9984e-01, 1.6439e-04],
        [9.1013e-04, 9.9909e-01],
        [9.9822e-01, 1.7786e-03],
        [1.9279e-07, 1.0000e+00],
        [9.9932e-01, 6.7648e-04],
        [5.8988e-05, 9.9994e-01],
        [3.3861e-07, 1.0000e+00],
        [3.0650e-02, 9.6935e-01],
        [3.0903e-03, 9.9691e-01],
        [1.3609e-02, 9.8639e-01],
        [1.4232e-02, 9.8577e-01],
        [3.9398e-05, 9.9996e-01],
        [1.5382e-03, 9.9846e-01],
        [2.3156e-06, 1.0000e+00],
        [2.4241e-05, 9.9998e-01],
        [5.1312e-07, 1.0000e+00],
        [1.0455e-05, 9.9999e-01],
        [1.0856e-05, 9.9999e-01],
        [9.0850e-04, 9.9909e-01],
        [9.9467e-01, 5.3269e-03],
        [9.9987e-01, 1.2897e-04],
        [1.5873e-06, 1.0000e+00],
        [9.7489e-01, 2.5114e-02],
        [3.8742e-03, 9.9613e-01],
        [1.0233e-06, 1.0000e+00],
        [9.9961e-01, 3.8603e-04],
        [1.5582e-04, 9.9984e-01],
        [5.3844e-04, 9.9946e-01],
        [9.9946e-01, 5.3803e-04],
        [1.0725e-05, 9.9999e-01],
        [2.3023e-04, 9.9977e-01],
        [1.3347e-04, 9.9987e-01],
        [2.2496e-03, 9.9775e-01],
        [7.8193e-06, 9.9999e-01],
        [1.9054e-08, 1.0000e+00],
        [6.9866e-04, 9.9930e-01],
        [2.8968e-06, 1.0000e+00],
        [2.5487e-05, 9.9997e-01],
        [4.2903e-07, 1.0000e+00],
        [3.5847e-07, 1.0000e+00],
        [9.1050e-01, 8.9496e-02],
        [1.0340e-08, 1.0000e+00],
        [2.5916e-07, 1.0000e+00],
        [6.7593e-07, 1.0000e+00],
        [1.7840e-05, 9.9998e-01],
        [9.9961e-01, 3.8558e-04],
        [2.6045e-08, 1.0000e+00],
        [1.2626e-02, 9.8737e-01],
        [1.7186e-09, 1.0000e+00],
        [5.7175e-04, 9.9943e-01],
        [7.3863e-01, 2.6137e-01],
        [9.9165e-01, 8.3506e-03],
        [6.6510e-06, 9.9999e-01],
        [1.2000e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.9475e-05, 9.9994e-01],
        [1.7173e-02, 9.8283e-01],
        [9.7293e-01, 2.7072e-02],
        [1.1831e-03, 9.9882e-01],
        [6.4159e-04, 9.9936e-01],
        [4.3668e-01, 5.6332e-01],
        [5.7590e-05, 9.9994e-01],
        [1.3641e-06, 1.0000e+00],
        [2.4523e-03, 9.9755e-01],
        [6.5630e-08, 1.0000e+00],
        [2.3776e-04, 9.9976e-01],
        [2.9324e-05, 9.9997e-01],
        [9.7396e-01, 2.6036e-02],
        [9.9946e-01, 5.3760e-04],
        [9.2965e-07, 1.0000e+00],
        [2.4422e-03, 9.9756e-01],
        [9.9853e-01, 1.4656e-03],
        [6.6355e-06, 9.9999e-01],
        [4.4822e-07, 1.0000e+00],
        [1.7922e-07, 1.0000e+00],
        [2.5915e-06, 1.0000e+00],
        [6.1652e-05, 9.9994e-01],
        [4.8647e-05, 9.9995e-01],
        [9.9955e-01, 4.4736e-04],
        [9.3126e-01, 6.8744e-02],
        [9.9746e-01, 2.5376e-03],
        [6.7972e-02, 9.3203e-01],
        [8.1702e-11, 1.0000e+00],
        [5.7681e-01, 4.2319e-01],
        [4.3368e-06, 1.0000e+00],
        [8.5106e-01, 1.4894e-01],
        [3.7658e-06, 1.0000e+00],
        [1.5030e-07, 1.0000e+00],
        [9.9979e-01, 2.0987e-04],
        [9.9886e-01, 1.1431e-03],
        [1.9262e-05, 9.9998e-01],
        [9.9954e-01, 4.5817e-04],
        [3.0875e-06, 1.0000e+00],
        [1.4436e-04, 9.9986e-01],
        [2.2816e-09, 1.0000e+00],
        [9.5516e-04, 9.9904e-01],
        [1.7229e-02, 9.8277e-01],
        [1.2460e-03, 9.9875e-01],
        [1.7610e-03, 9.9824e-01],
        [2.3506e-04, 9.9976e-01],
        [9.9599e-01, 4.0084e-03],
        [4.6651e-04, 9.9953e-01],
        [4.4837e-06, 1.0000e+00],
        [1.4868e-04, 9.9985e-01],
        [3.1106e-04, 9.9969e-01],
        [1.3843e-07, 1.0000e+00],
        [2.5340e-07, 1.0000e+00],
        [9.9541e-01, 4.5911e-03],
        [6.1247e-08, 1.0000e+00],
        [5.0981e-08, 1.0000e+00],
        [9.9939e-01, 6.1322e-04],
        [1.8647e-06, 1.0000e+00],
        [2.0110e-06, 1.0000e+00],
        [7.8501e-09, 1.0000e+00],
        [6.7021e-06, 9.9999e-01],
        [1.2356e-02, 9.8764e-01],
        [9.8627e-01, 1.3731e-02],
        [6.7027e-08, 1.0000e+00],
        [1.5331e-04, 9.9985e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.1472e-03, 9.9385e-01],
        [5.4350e-05, 9.9995e-01],
        [2.9620e-02, 9.7038e-01],
        [5.8485e-02, 9.4152e-01],
        [4.9329e-06, 1.0000e+00],
        [9.9768e-01, 2.3249e-03],
        [5.0384e-06, 9.9999e-01],
        [9.9803e-01, 1.9667e-03],
        [9.9080e-01, 9.2002e-03],
        [2.1930e-02, 9.7807e-01],
        [7.9580e-06, 9.9999e-01],
        [3.0221e-07, 1.0000e+00],
        [1.2552e-08, 1.0000e+00],
        [1.0000e+00, 4.6413e-06],
        [6.1159e-03, 9.9388e-01],
        [8.8493e-05, 9.9991e-01],
        [2.8713e-01, 7.1287e-01],
        [9.9929e-01, 7.1301e-04],
        [2.9715e-07, 1.0000e+00],
        [3.9103e-07, 1.0000e+00],
        [4.1682e-05, 9.9996e-01],
        [3.6420e-04, 9.9964e-01],
        [1.5827e-07, 1.0000e+00],
        [1.2374e-01, 8.7626e-01],
        [6.0139e-05, 9.9994e-01],
        [9.9972e-01, 2.7744e-04],
        [1.0000e+00, 3.1385e-08],
        [2.2471e-01, 7.7529e-01],
        [1.3126e-05, 9.9999e-01],
        [7.4815e-07, 1.0000e+00],
        [3.2807e-03, 9.9672e-01],
        [8.7361e-02, 9.1264e-01],
        [2.3459e-02, 9.7654e-01],
        [3.2625e-04, 9.9967e-01],
        [8.5900e-08, 1.0000e+00],
        [9.5511e-01, 4.4890e-02],
        [1.1629e-05, 9.9999e-01],
        [4.6435e-06, 1.0000e+00],
        [9.5328e-05, 9.9990e-01],
        [2.0998e-07, 1.0000e+00],
        [9.9976e-01, 2.3760e-04],
        [3.6363e-07, 1.0000e+00],
        [1.0475e-07, 1.0000e+00],
        [9.9817e-01, 1.8277e-03],
        [2.1819e-04, 9.9978e-01],
        [2.7482e-06, 1.0000e+00],
        [3.7330e-01, 6.2670e-01],
        [9.9868e-01, 1.3150e-03],
        [6.3579e-08, 1.0000e+00],
        [9.9900e-01, 1.0020e-03],
        [7.8879e-06, 9.9999e-01],
        [7.8386e-04, 9.9922e-01],
        [9.9057e-06, 9.9999e-01],
        [4.3697e-06, 1.0000e+00],
        [3.0137e-04, 9.9970e-01],
        [4.1414e-04, 9.9959e-01],
        [1.7250e-06, 1.0000e+00],
        [9.9990e-01, 9.7321e-05],
        [4.3271e-04, 9.9957e-01],
        [9.9978e-01, 2.2013e-04],
        [1.9953e-03, 9.9800e-01],
        [1.2538e-02, 9.8746e-01],
        [1.9096e-03, 9.9809e-01],
        [1.4925e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.5685e-06, 9.9999e-01],
        [6.7456e-03, 9.9325e-01],
        [3.2876e-05, 9.9997e-01],
        [3.4885e-06, 1.0000e+00],
        [7.0641e-03, 9.9294e-01],
        [9.9946e-01, 5.4420e-04],
        [5.3488e-04, 9.9947e-01],
        [1.4968e-05, 9.9998e-01],
        [3.1471e-02, 9.6853e-01],
        [9.9780e-01, 2.2041e-03],
        [7.2047e-04, 9.9928e-01],
        [1.4203e-08, 1.0000e+00],
        [2.4290e-09, 1.0000e+00],
        [2.5990e-04, 9.9974e-01],
        [9.9984e-01, 1.6292e-04],
        [9.2324e-06, 9.9999e-01],
        [5.4913e-05, 9.9995e-01],
        [3.1957e-07, 1.0000e+00],
        [5.8982e-08, 1.0000e+00],
        [1.5650e-07, 1.0000e+00],
        [7.3982e-03, 9.9260e-01],
        [1.6135e-06, 1.0000e+00],
        [9.7214e-01, 2.7861e-02],
        [3.6768e-05, 9.9996e-01],
        [4.9187e-06, 1.0000e+00],
        [4.3633e-06, 1.0000e+00],
        [1.8926e-07, 1.0000e+00],
        [9.9974e-01, 2.5870e-04],
        [9.9926e-06, 9.9999e-01],
        [3.2299e-06, 1.0000e+00],
        [1.5179e-08, 1.0000e+00],
        [1.6721e-04, 9.9983e-01],
        [1.3337e-04, 9.9987e-01],
        [9.7623e-05, 9.9990e-01],
        [1.7573e-03, 9.9824e-01],
        [1.9863e-02, 9.8014e-01],
        [9.8177e-01, 1.8231e-02],
        [9.9994e-01, 5.5213e-05],
        [9.9983e-01, 1.7242e-04],
        [3.7747e-05, 9.9996e-01],
        [9.9966e-01, 3.3916e-04],
        [8.8669e-04, 9.9911e-01],
        [1.1946e-06, 1.0000e+00],
        [9.9982e-01, 1.7593e-04],
        [9.7143e-01, 2.8568e-02],
        [2.3302e-03, 9.9767e-01],
        [2.6661e-04, 9.9973e-01],
        [1.0802e-06, 1.0000e+00],
        [9.9157e-01, 8.4309e-03],
        [6.0003e-02, 9.4000e-01],
        [9.9965e-01, 3.4877e-04],
        [2.7596e-04, 9.9972e-01],
        [9.9970e-01, 3.0481e-04],
        [9.9997e-01, 2.8405e-05],
        [5.4320e-05, 9.9995e-01],
        [6.4329e-03, 9.9357e-01],
        [9.9966e-01, 3.4076e-04],
        [3.0747e-07, 1.0000e+00],
        [1.8458e-05, 9.9998e-01],
        [9.0001e-08, 1.0000e+00],
        [6.2288e-07, 1.0000e+00],
        [9.9940e-01, 5.9890e-04],
        [9.5667e-04, 9.9904e-01],
        [9.9750e-01, 2.5045e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[6.1469e-05, 9.9994e-01],
        [2.4223e-01, 7.5777e-01],
        [2.8125e-03, 9.9719e-01],
        [9.9453e-01, 5.4681e-03],
        [1.0910e-07, 1.0000e+00],
        [3.2333e-06, 1.0000e+00],
        [9.9943e-01, 5.6794e-04],
        [3.8090e-04, 9.9962e-01],
        [5.3040e-03, 9.9470e-01],
        [8.1154e-03, 9.9188e-01],
        [9.9988e-01, 1.2477e-04],
        [3.6426e-04, 9.9964e-01],
        [6.3551e-05, 9.9994e-01],
        [2.8937e-03, 9.9711e-01],
        [4.4540e-06, 1.0000e+00],
        [1.0652e-07, 1.0000e+00],
        [9.9977e-01, 2.3235e-04],
        [1.7504e-04, 9.9982e-01],
        [9.1816e-04, 9.9908e-01],
        [3.1139e-06, 1.0000e+00],
        [1.0647e-04, 9.9989e-01],
        [3.1963e-06, 1.0000e+00],
        [2.4878e-08, 1.0000e+00],
        [3.1918e-09, 1.0000e+00],
        [9.9969e-01, 3.0615e-04],
        [1.9449e-05, 9.9998e-01],
        [2.5214e-02, 9.7479e-01],
        [2.6740e-03, 9.9733e-01],
        [3.2992e-03, 9.9670e-01],
        [1.6588e-06, 1.0000e+00],
        [1.4054e-06, 1.0000e+00],
        [6.7100e-01, 3.2900e-01],
        [9.9990e-01, 9.5492e-05],
        [9.9994e-01, 6.5020e-05],
        [5.0936e-03, 9.9491e-01],
        [1.7718e-03, 9.9823e-01],
        [6.2264e-04, 9.9938e-01],
        [9.9959e-01, 4.0554e-04],
        [9.7886e-05, 9.9990e-01],
        [9.9921e-01, 7.8828e-04],
        [7.5047e-01, 2.4953e-01],
        [5.7375e-03, 9.9426e-01],
        [2.0318e-07, 1.0000e+00],
        [3.7138e-04, 9.9963e-01],
        [8.1375e-02, 9.1862e-01],
        [6.8494e-05, 9.9993e-01],
        [9.5073e-01, 4.9268e-02],
        [4.9713e-01, 5.0287e-01],
        [1.9642e-03, 9.9804e-01],
        [3.8463e-07, 1.0000e+00],
        [9.9948e-01, 5.1762e-04],
        [9.9984e-01, 1.5775e-04],
        [1.4130e-03, 9.9859e-01],
        [3.1502e-03, 9.9685e-01],
        [3.9679e-05, 9.9996e-01],
        [7.6902e-04, 9.9923e-01],
        [9.3083e-06, 9.9999e-01],
        [2.0098e-05, 9.9998e-01],
        [9.9967e-01, 3.3060e-04],
        [1.4802e-03, 9.9852e-01],
        [1.2064e-04, 9.9988e-01],
        [2.0163e-05, 9.9998e-01],
        [2.8996e-05, 9.9997e-01],
        [2.5215e-03, 9.9748e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5405e-03, 9.9846e-01],
        [7.8959e-01, 2.1041e-01],
        [4.4800e-04, 9.9955e-01],
        [1.0005e-05, 9.9999e-01],
        [9.9899e-01, 1.0093e-03],
        [8.0486e-08, 1.0000e+00],
        [7.8367e-05, 9.9992e-01],
        [1.5072e-04, 9.9985e-01],
        [9.9814e-01, 1.8624e-03],
        [2.8551e-06, 1.0000e+00],
        [1.5496e-09, 1.0000e+00],
        [2.4040e-03, 9.9760e-01],
        [5.9146e-04, 9.9941e-01],
        [9.9979e-01, 2.0837e-04],
        [2.1446e-04, 9.9979e-01],
        [2.8357e-02, 9.7164e-01],
        [1.2396e-05, 9.9999e-01],
        [9.9943e-01, 5.6574e-04],
        [3.1502e-03, 9.9685e-01],
        [2.6007e-03, 9.9740e-01],
        [1.0892e-04, 9.9989e-01],
        [5.7512e-08, 1.0000e+00],
        [8.0526e-07, 1.0000e+00],
        [9.9993e-01, 6.6234e-05],
        [9.9718e-01, 2.8225e-03],
        [2.3313e-07, 1.0000e+00],
        [9.6622e-01, 3.3783e-02],
        [5.9741e-05, 9.9994e-01],
        [3.1945e-06, 1.0000e+00],
        [5.2880e-05, 9.9995e-01],
        [9.9985e-01, 1.4859e-04],
        [2.2862e-03, 9.9771e-01],
        [5.4203e-08, 1.0000e+00],
        [1.1520e-03, 9.9885e-01],
        [4.4316e-07, 1.0000e+00],
        [2.0164e-07, 1.0000e+00],
        [1.6793e-04, 9.9983e-01],
        [1.8610e-08, 1.0000e+00],
        [3.8814e-08, 1.0000e+00],
        [9.9989e-01, 1.0710e-04],
        [9.9967e-01, 3.3114e-04],
        [2.7771e-07, 1.0000e+00],
        [3.7087e-05, 9.9996e-01],
        [9.9989e-01, 1.0963e-04],
        [9.1861e-05, 9.9991e-01],
        [9.9967e-01, 3.3344e-04],
        [1.5727e-03, 9.9843e-01],
        [2.5289e-08, 1.0000e+00],
        [5.9608e-06, 9.9999e-01],
        [8.4757e-01, 1.5243e-01],
        [1.6636e-04, 9.9983e-01],
        [1.0721e-06, 1.0000e+00],
        [6.4251e-06, 9.9999e-01],
        [4.2086e-03, 9.9579e-01],
        [3.6644e-03, 9.9634e-01],
        [1.0792e-07, 1.0000e+00],
        [5.9905e-05, 9.9994e-01],
        [8.9663e-03, 9.9103e-01],
        [2.2382e-04, 9.9978e-01],
        [2.4222e-05, 9.9998e-01],
        [5.1799e-09, 1.0000e+00],
        [3.8871e-01, 6.1129e-01],
        [1.0397e-05, 9.9999e-01],
        [9.9992e-01, 7.9473e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9985e-01, 1.4930e-04],
        [9.9805e-01, 1.9473e-03],
        [1.3919e-08, 1.0000e+00],
        [4.4225e-05, 9.9996e-01],
        [2.6569e-07, 1.0000e+00],
        [9.9757e-01, 2.4284e-03],
        [9.3000e-07, 1.0000e+00],
        [1.2687e-04, 9.9987e-01],
        [5.3319e-02, 9.4668e-01],
        [9.9993e-01, 6.7219e-05],
        [4.7980e-06, 1.0000e+00],
        [1.1242e-04, 9.9989e-01],
        [3.6056e-04, 9.9964e-01],
        [9.8704e-01, 1.2963e-02],
        [4.4714e-04, 9.9955e-01],
        [1.5121e-04, 9.9985e-01],
        [1.8188e-04, 9.9982e-01],
        [6.0257e-05, 9.9994e-01],
        [9.9972e-01, 2.8072e-04],
        [3.4687e-06, 1.0000e+00],
        [8.4100e-04, 9.9916e-01],
        [1.7614e-04, 9.9982e-01],
        [9.9969e-01, 3.1493e-04],
        [5.3335e-05, 9.9995e-01],
        [3.0884e-04, 9.9969e-01],
        [9.9905e-01, 9.4928e-04],
        [4.8941e-02, 9.5106e-01],
        [4.7243e-08, 1.0000e+00],
        [2.7343e-03, 9.9727e-01],
        [2.6306e-07, 1.0000e+00],
        [1.4431e-05, 9.9999e-01],
        [2.9748e-06, 1.0000e+00],
        [9.9989e-01, 1.0664e-04],
        [8.8479e-07, 1.0000e+00],
        [5.1304e-05, 9.9995e-01],
        [9.9989e-01, 1.1314e-04],
        [9.5828e-01, 4.1716e-02],
        [2.6101e-01, 7.3899e-01],
        [1.1578e-06, 1.0000e+00],
        [6.9100e-06, 9.9999e-01],
        [9.9693e-01, 3.0662e-03],
        [9.9985e-01, 1.4603e-04],
        [3.9770e-04, 9.9960e-01],
        [5.6207e-04, 9.9944e-01],
        [4.1103e-05, 9.9996e-01],
        [8.1466e-04, 9.9919e-01],
        [1.2288e-05, 9.9999e-01],
        [9.9988e-01, 1.2360e-04],
        [1.6363e-05, 9.9998e-01],
        [1.4722e-03, 9.9853e-01],
        [9.9919e-01, 8.1501e-04],
        [9.9973e-01, 2.6901e-04],
        [9.7752e-08, 1.0000e+00],
        [2.2159e-06, 1.0000e+00],
        [2.3680e-02, 9.7632e-01],
        [2.8334e-03, 9.9717e-01],
        [2.1791e-08, 1.0000e+00],
        [3.7038e-04, 9.9963e-01],
        [7.0016e-05, 9.9993e-01],
        [9.9989e-01, 1.1190e-04],
        [9.9255e-01, 7.4468e-03],
        [1.2386e-02, 9.8761e-01],
        [3.5180e-05, 9.9996e-01],
        [6.1526e-04, 9.9938e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9997e-01, 2.5968e-05],
        [6.8414e-07, 1.0000e+00],
        [9.9954e-01, 4.5584e-04],
        [3.2512e-04, 9.9967e-01],
        [5.9729e-04, 9.9940e-01],
        [9.5985e-07, 1.0000e+00],
        [9.9981e-01, 1.8956e-04],
        [4.4216e-04, 9.9956e-01],
        [2.5036e-03, 9.9750e-01],
        [8.1341e-07, 1.0000e+00],
        [6.8297e-03, 9.9317e-01],
        [3.7044e-05, 9.9996e-01],
        [6.5784e-08, 1.0000e+00],
        [9.5745e-01, 4.2547e-02],
        [5.1967e-08, 1.0000e+00],
        [9.9552e-01, 4.4802e-03],
        [7.9383e-01, 2.0617e-01],
        [3.0485e-06, 1.0000e+00],
        [6.7277e-05, 9.9993e-01],
        [1.8339e-05, 9.9998e-01],
        [2.1177e-05, 9.9998e-01],
        [9.9986e-01, 1.4444e-04],
        [2.7102e-03, 9.9729e-01],
        [1.7520e-08, 1.0000e+00],
        [7.9665e-05, 9.9992e-01],
        [9.7870e-06, 9.9999e-01],
        [9.5401e-01, 4.5990e-02],
        [2.4476e-06, 1.0000e+00],
        [2.9091e-03, 9.9709e-01],
        [2.2487e-05, 9.9998e-01],
        [2.5807e-04, 9.9974e-01],
        [5.4945e-07, 1.0000e+00],
        [1.0107e-03, 9.9899e-01],
        [9.9985e-01, 1.4911e-04],
        [1.1603e-07, 1.0000e+00],
        [6.4560e-01, 3.5440e-01],
        [9.9824e-01, 1.7641e-03],
        [8.1073e-05, 9.9992e-01],
        [1.4014e-06, 1.0000e+00],
        [9.9206e-01, 7.9407e-03],
        [9.9985e-01, 1.4523e-04],
        [9.9970e-01, 3.0423e-04],
        [1.4926e-07, 1.0000e+00],
        [1.9806e-04, 9.9980e-01],
        [2.3862e-05, 9.9998e-01],
        [7.8056e-04, 9.9922e-01],
        [9.9963e-01, 3.6739e-04],
        [1.8841e-07, 1.0000e+00],
        [2.2574e-06, 1.0000e+00],
        [7.2305e-01, 2.7695e-01],
        [3.6034e-06, 1.0000e+00],
        [8.6463e-02, 9.1354e-01],
        [8.3034e-06, 9.9999e-01],
        [9.9974e-01, 2.5795e-04],
        [1.8303e-08, 1.0000e+00],
        [8.8775e-04, 9.9911e-01],
        [9.9977e-01, 2.2542e-04],
        [8.6789e-01, 1.3211e-01],
        [1.3123e-10, 1.0000e+00],
        [9.9971e-01, 2.9076e-04],
        [2.4308e-04, 9.9976e-01],
        [9.9658e-01, 3.4206e-03],
        [5.1141e-05, 9.9995e-01],
        [2.1464e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.2671e-05, 9.9995e-01],
        [6.3838e-05, 9.9994e-01],
        [4.9727e-07, 1.0000e+00],
        [1.4473e-05, 9.9999e-01],
        [5.5214e-06, 9.9999e-01],
        [4.5494e-07, 1.0000e+00],
        [2.8626e-01, 7.1374e-01],
        [3.9208e-05, 9.9996e-01],
        [9.9902e-01, 9.7642e-04],
        [1.2592e-04, 9.9987e-01],
        [9.9946e-01, 5.3560e-04],
        [1.2574e-08, 1.0000e+00],
        [8.8295e-04, 9.9912e-01],
        [2.1793e-03, 9.9782e-01],
        [1.9849e-01, 8.0151e-01],
        [1.2564e-03, 9.9874e-01],
        [9.9962e-01, 3.7640e-04],
        [4.3618e-03, 9.9564e-01],
        [1.3982e-07, 1.0000e+00],
        [2.8073e-04, 9.9972e-01],
        [9.8973e-01, 1.0266e-02],
        [9.9904e-01, 9.6314e-04],
        [2.4977e-05, 9.9997e-01],
        [1.2248e-06, 1.0000e+00],
        [9.9737e-01, 2.6343e-03],
        [9.9996e-01, 3.5581e-05],
        [9.7130e-06, 9.9999e-01],
        [1.6339e-04, 9.9984e-01],
        [1.6884e-07, 1.0000e+00],
        [1.5313e-03, 9.9847e-01],
        [9.7356e-01, 2.6437e-02],
        [5.4421e-04, 9.9946e-01],
        [9.9928e-01, 7.2314e-04],
        [9.0709e-01, 9.2907e-02],
        [1.5545e-01, 8.4455e-01],
        [1.8348e-07, 1.0000e+00],
        [5.1518e-02, 9.4848e-01],
        [3.4503e-06, 1.0000e+00],
        [9.9893e-01, 1.0730e-03],
        [8.8997e-01, 1.1003e-01],
        [1.1030e-04, 9.9989e-01],
        [3.7799e-04, 9.9962e-01],
        [3.3694e-08, 1.0000e+00],
        [2.8517e-04, 9.9971e-01],
        [6.4535e-03, 9.9355e-01],
        [9.9952e-01, 4.8246e-04],
        [1.4722e-05, 9.9999e-01],
        [1.4449e-04, 9.9986e-01],
        [4.9736e-03, 9.9503e-01],
        [2.4202e-05, 9.9998e-01],
        [3.7769e-06, 1.0000e+00],
        [1.6556e-07, 1.0000e+00],
        [9.9968e-01, 3.2209e-04],
        [9.4137e-06, 9.9999e-01],
        [1.0757e-04, 9.9989e-01],
        [9.9985e-01, 1.4713e-04],
        [6.1817e-07, 1.0000e+00],
        [5.0561e-05, 9.9995e-01],
        [3.3547e-04, 9.9966e-01],
        [3.5014e-07, 1.0000e+00],
        [9.9896e-01, 1.0385e-03],
        [4.4929e-02, 9.5507e-01],
        [9.9985e-01, 1.4643e-04],
        [9.3918e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.1251e-02, 9.6875e-01],
        [7.6812e-07, 1.0000e+00],
        [2.0402e-08, 1.0000e+00],
        [3.1881e-03, 9.9681e-01],
        [5.8058e-04, 9.9942e-01],
        [1.6728e-06, 1.0000e+00],
        [9.5821e-03, 9.9042e-01],
        [7.6735e-10, 1.0000e+00],
        [9.9963e-01, 3.6595e-04],
        [1.6691e-03, 9.9833e-01],
        [1.2211e-05, 9.9999e-01],
        [1.2642e-07, 1.0000e+00],
        [8.1517e-02, 9.1848e-01],
        [4.8238e-08, 1.0000e+00],
        [1.8143e-05, 9.9998e-01],
        [2.3054e-09, 1.0000e+00],
        [2.0537e-01, 7.9463e-01],
        [2.6544e-05, 9.9997e-01],
        [1.9154e-03, 9.9808e-01],
        [9.9956e-01, 4.4397e-04],
        [1.2711e-02, 9.8729e-01],
        [8.6502e-01, 1.3498e-01],
        [9.9995e-01, 4.6796e-05],
        [8.4789e-06, 9.9999e-01],
        [9.9953e-01, 4.6791e-04],
        [1.6815e-06, 1.0000e+00],
        [6.0527e-05, 9.9994e-01],
        [1.1404e-04, 9.9989e-01],
        [3.7061e-08, 1.0000e+00],
        [7.4737e-03, 9.9253e-01],
        [6.9533e-03, 9.9305e-01],
        [2.3352e-04, 9.9977e-01],
        [8.2799e-01, 1.7201e-01],
        [1.1246e-03, 9.9888e-01],
        [1.6509e-01, 8.3491e-01],
        [1.8486e-05, 9.9998e-01],
        [2.6011e-06, 1.0000e+00],
        [4.2154e-08, 1.0000e+00],
        [2.6831e-04, 9.9973e-01],
        [3.4716e-02, 9.6528e-01],
        [9.9967e-01, 3.3471e-04],
        [5.4410e-07, 1.0000e+00],
        [1.5779e-05, 9.9998e-01],
        [3.6564e-04, 9.9963e-01],
        [1.8570e-04, 9.9981e-01],
        [5.6859e-05, 9.9994e-01],
        [8.7533e-02, 9.1247e-01],
        [1.3747e-03, 9.9863e-01],
        [7.9319e-04, 9.9921e-01],
        [8.4385e-03, 9.9156e-01],
        [5.0728e-06, 9.9999e-01],
        [2.8188e-06, 1.0000e+00],
        [3.8264e-07, 1.0000e+00],
        [3.9283e-07, 1.0000e+00],
        [9.9778e-01, 2.2200e-03],
        [2.3595e-07, 1.0000e+00],
        [2.2461e-05, 9.9998e-01],
        [3.9562e-05, 9.9996e-01],
        [4.2839e-03, 9.9572e-01],
        [2.1975e-03, 9.9780e-01],
        [1.1801e-03, 9.9882e-01],
        [6.2153e-03, 9.9378e-01],
        [4.0097e-06, 1.0000e+00],
        [3.6373e-03, 9.9636e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9985e-01, 1.4818e-04],
        [9.8055e-06, 9.9999e-01],
        [9.9966e-01, 3.3763e-04],
        [7.7859e-04, 9.9922e-01],
        [4.5377e-08, 1.0000e+00],
        [3.3026e-03, 9.9670e-01],
        [3.1249e-05, 9.9997e-01],
        [7.1844e-08, 1.0000e+00],
        [4.2797e-04, 9.9957e-01],
        [3.7645e-02, 9.6236e-01],
        [2.4632e-03, 9.9754e-01],
        [3.5320e-07, 1.0000e+00],
        [8.1406e-01, 1.8594e-01],
        [1.4371e-04, 9.9986e-01],
        [5.9953e-09, 1.0000e+00],
        [4.6305e-04, 9.9954e-01],
        [1.3386e-02, 9.8661e-01],
        [9.9978e-01, 2.1723e-04],
        [8.2601e-01, 1.7399e-01],
        [3.0040e-03, 9.9700e-01],
        [1.9682e-03, 9.9803e-01],
        [1.4990e-04, 9.9985e-01],
        [5.0212e-03, 9.9498e-01],
        [7.9103e-01, 2.0897e-01],
        [9.9956e-01, 4.4225e-04],
        [3.8433e-07, 1.0000e+00],
        [1.5583e-02, 9.8442e-01],
        [2.3182e-05, 9.9998e-01],
        [6.3210e-01, 3.6790e-01],
        [1.4398e-08, 1.0000e+00],
        [1.1116e-05, 9.9999e-01],
        [7.9145e-08, 1.0000e+00],
        [4.1759e-04, 9.9958e-01],
        [9.9500e-01, 5.0037e-03],
        [5.4226e-05, 9.9995e-01],
        [9.4191e-01, 5.8090e-02],
        [2.3514e-07, 1.0000e+00],
        [1.1440e-06, 1.0000e+00],
        [1.4211e-02, 9.8579e-01],
        [9.9998e-01, 2.4231e-05],
        [5.3303e-07, 1.0000e+00],
        [2.0608e-01, 7.9392e-01],
        [1.0486e-03, 9.9895e-01],
        [7.6420e-09, 1.0000e+00],
        [5.9048e-03, 9.9410e-01],
        [7.3906e-02, 9.2609e-01],
        [2.8447e-05, 9.9997e-01],
        [7.6766e-05, 9.9992e-01],
        [3.0389e-06, 1.0000e+00],
        [3.0897e-05, 9.9997e-01],
        [9.8558e-01, 1.4418e-02],
        [8.3174e-09, 1.0000e+00],
        [8.2280e-02, 9.1772e-01],
        [2.2459e-05, 9.9998e-01],
        [3.3265e-05, 9.9997e-01],
        [5.9750e-06, 9.9999e-01],
        [8.1474e-08, 1.0000e+00],
        [9.9988e-01, 1.1758e-04],
        [1.7447e-03, 9.9826e-01],
        [2.0207e-03, 9.9798e-01],
        [2.6675e-07, 1.0000e+00],
        [1.3056e-06, 1.0000e+00],
        [4.7306e-06, 1.0000e+00],
        [8.2582e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5526e-05, 9.9995e-01],
        [1.9103e-04, 9.9981e-01],
        [3.5984e-05, 9.9996e-01],
        [9.8840e-01, 1.1597e-02],
        [9.9982e-01, 1.8364e-04],
        [3.9479e-08, 1.0000e+00],
        [3.4040e-05, 9.9997e-01],
        [2.1700e-04, 9.9978e-01],
        [3.4840e-07, 1.0000e+00],
        [6.3718e-07, 1.0000e+00],
        [4.3530e-04, 9.9956e-01],
        [2.2020e-05, 9.9998e-01],
        [1.8889e-08, 1.0000e+00],
        [1.6641e-04, 9.9983e-01],
        [1.7628e-04, 9.9982e-01],
        [9.9956e-01, 4.3668e-04],
        [2.4280e-05, 9.9998e-01],
        [9.9985e-01, 1.4916e-04],
        [1.2394e-04, 9.9988e-01],
        [9.8539e-04, 9.9901e-01],
        [1.3220e-02, 9.8678e-01],
        [9.9599e-01, 4.0056e-03],
        [9.9675e-01, 3.2542e-03],
        [2.4307e-05, 9.9998e-01],
        [9.8527e-04, 9.9901e-01],
        [3.6006e-06, 1.0000e+00],
        [2.4362e-06, 1.0000e+00],
        [1.5209e-02, 9.8479e-01],
        [2.1080e-08, 1.0000e+00],
        [9.9797e-01, 2.0304e-03],
        [4.0773e-03, 9.9592e-01],
        [1.1091e-01, 8.8909e-01],
        [5.3104e-06, 9.9999e-01],
        [5.6695e-04, 9.9943e-01],
        [3.4581e-04, 9.9965e-01],
        [9.9958e-01, 4.1653e-04],
        [4.6051e-08, 1.0000e+00],
        [6.4725e-06, 9.9999e-01],
        [9.7615e-01, 2.3851e-02],
        [8.9809e-04, 9.9910e-01],
        [1.4212e-02, 9.8579e-01],
        [3.1755e-06, 1.0000e+00],
        [8.5352e-06, 9.9999e-01],
        [4.2541e-06, 1.0000e+00],
        [6.9402e-03, 9.9306e-01],
        [2.3300e-07, 1.0000e+00],
        [8.4482e-04, 9.9916e-01],
        [1.3924e-02, 9.8608e-01],
        [9.9705e-01, 2.9512e-03],
        [5.4344e-04, 9.9946e-01],
        [7.5550e-07, 1.0000e+00],
        [5.9060e-04, 9.9941e-01],
        [7.3516e-06, 9.9999e-01],
        [9.8696e-01, 1.3041e-02],
        [1.0865e-04, 9.9989e-01],
        [9.6093e-03, 9.9039e-01],
        [5.8456e-07, 1.0000e+00],
        [1.0341e-02, 9.8966e-01],
        [1.4572e-04, 9.9985e-01],
        [9.1360e-04, 9.9909e-01],
        [7.0430e-04, 9.9930e-01],
        [3.2603e-04, 9.9967e-01],
        [6.8180e-01, 3.1820e-01],
        [3.2244e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2891e-05, 9.9998e-01],
        [5.9672e-04, 9.9940e-01],
        [1.0934e-06, 1.0000e+00],
        [9.9913e-01, 8.7375e-04],
        [9.9963e-01, 3.6545e-04],
        [2.3572e-05, 9.9998e-01],
        [1.7759e-04, 9.9982e-01],
        [9.9853e-01, 1.4709e-03],
        [9.9938e-01, 6.1642e-04],
        [7.5255e-09, 1.0000e+00],
        [9.5602e-01, 4.3983e-02],
        [8.5089e-01, 1.4911e-01],
        [4.6334e-03, 9.9537e-01],
        [4.5234e-01, 5.4766e-01],
        [9.9979e-01, 2.1229e-04],
        [8.6115e-05, 9.9991e-01],
        [1.3515e-04, 9.9986e-01],
        [1.8722e-05, 9.9998e-01],
        [9.7943e-01, 2.0575e-02],
        [1.6744e-07, 1.0000e+00],
        [2.2375e-05, 9.9998e-01],
        [3.0337e-02, 9.6966e-01],
        [6.7301e-06, 9.9999e-01],
        [3.8331e-04, 9.9962e-01],
        [3.5199e-08, 1.0000e+00],
        [9.9980e-01, 1.9657e-04],
        [4.1673e-08, 1.0000e+00],
        [2.1422e-04, 9.9979e-01],
        [8.9104e-07, 1.0000e+00],
        [2.9671e-03, 9.9703e-01],
        [3.4045e-05, 9.9997e-01],
        [3.9529e-04, 9.9960e-01],
        [3.8120e-05, 9.9996e-01],
        [9.9964e-01, 3.5777e-04],
        [3.3310e-03, 9.9667e-01],
        [6.0248e-04, 9.9940e-01],
        [9.9633e-01, 3.6714e-03],
        [4.6075e-05, 9.9995e-01],
        [1.7796e-01, 8.2204e-01],
        [2.2951e-07, 1.0000e+00],
        [1.9169e-07, 1.0000e+00],
        [1.9538e-06, 1.0000e+00],
        [9.7966e-01, 2.0344e-02],
        [1.4170e-05, 9.9999e-01],
        [2.1454e-06, 1.0000e+00],
        [1.0654e-04, 9.9989e-01],
        [9.9983e-01, 1.7030e-04],
        [1.3345e-05, 9.9999e-01],
        [3.7614e-05, 9.9996e-01],
        [6.2787e-02, 9.3721e-01],
        [6.3102e-06, 9.9999e-01],
        [6.3745e-04, 9.9936e-01],
        [2.5228e-06, 1.0000e+00],
        [5.0736e-08, 1.0000e+00],
        [9.9788e-01, 2.1170e-03],
        [1.1716e-04, 9.9988e-01],
        [4.9732e-04, 9.9950e-01],
        [2.6296e-05, 9.9997e-01],
        [9.9029e-01, 9.7131e-03],
        [3.3627e-05, 9.9997e-01],
        [9.9912e-01, 8.7546e-04],
        [7.9059e-09, 1.0000e+00],
        [1.2739e-08, 1.0000e+00],
        [9.8975e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0386e-07, 1.0000e+00],
        [3.6688e-08, 1.0000e+00],
        [5.8971e-07, 1.0000e+00],
        [3.3859e-07, 1.0000e+00],
        [5.0149e-07, 1.0000e+00],
        [3.0006e-04, 9.9970e-01],
        [9.9997e-01, 2.6271e-05],
        [1.0404e-05, 9.9999e-01],
        [6.9275e-04, 9.9931e-01],
        [1.1746e-06, 1.0000e+00],
        [1.8670e-06, 1.0000e+00],
        [5.2371e-04, 9.9948e-01],
        [9.9973e-01, 2.7439e-04],
        [8.1309e-06, 9.9999e-01],
        [9.8077e-01, 1.9231e-02],
        [9.9976e-01, 2.4412e-04],
        [6.8655e-09, 1.0000e+00],
        [4.5173e-04, 9.9955e-01],
        [1.0335e-06, 1.0000e+00],
        [2.8387e-05, 9.9997e-01],
        [2.0223e-07, 1.0000e+00],
        [9.9952e-01, 4.7664e-04],
        [1.1900e-04, 9.9988e-01],
        [1.0318e-04, 9.9990e-01],
        [9.1273e-02, 9.0873e-01],
        [3.4608e-03, 9.9654e-01],
        [4.4198e-05, 9.9996e-01],
        [9.0710e-05, 9.9991e-01],
        [8.5797e-04, 9.9914e-01],
        [1.9648e-06, 1.0000e+00],
        [9.9960e-01, 3.9946e-04],
        [2.2686e-06, 1.0000e+00],
        [9.9374e-01, 6.2559e-03],
        [9.2344e-04, 9.9908e-01],
        [1.5104e-05, 9.9998e-01],
        [9.9979e-01, 2.1368e-04],
        [9.9875e-01, 1.2543e-03],
        [1.7185e-03, 9.9828e-01],
        [4.9798e-08, 1.0000e+00],
        [9.9917e-01, 8.2894e-04],
        [1.2166e-08, 1.0000e+00],
        [1.0903e-03, 9.9891e-01],
        [4.2278e-03, 9.9577e-01],
        [3.2296e-07, 1.0000e+00],
        [1.1713e-03, 9.9883e-01],
        [9.9960e-01, 3.9989e-04],
        [9.9995e-01, 4.5712e-05],
        [2.8966e-05, 9.9997e-01],
        [2.9073e-03, 9.9709e-01],
        [3.2168e-09, 1.0000e+00],
        [9.9814e-01, 1.8596e-03],
        [4.3149e-04, 9.9957e-01],
        [1.3611e-03, 9.9864e-01],
        [9.9972e-01, 2.7698e-04],
        [9.9974e-01, 2.6321e-04],
        [4.6391e-05, 9.9995e-01],
        [9.3676e-06, 9.9999e-01],
        [9.9157e-01, 8.4337e-03],
        [9.9970e-01, 2.9731e-04],
        [9.9951e-01, 4.8546e-04],
        [5.8092e-06, 9.9999e-01],
        [4.0782e-06, 1.0000e+00],
        [9.9960e-01, 4.0019e-04],
        [4.9676e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7444e-05, 9.9998e-01],
        [3.3130e-07, 1.0000e+00],
        [1.5232e-06, 1.0000e+00],
        [9.9983e-01, 1.6957e-04],
        [9.2986e-01, 7.0142e-02],
        [5.6901e-09, 1.0000e+00],
        [5.3887e-05, 9.9995e-01],
        [7.4324e-04, 9.9926e-01],
        [9.9968e-01, 3.2148e-04],
        [1.5694e-05, 9.9998e-01],
        [9.7876e-05, 9.9990e-01],
        [6.1515e-04, 9.9938e-01],
        [3.8380e-05, 9.9996e-01],
        [1.6807e-08, 1.0000e+00],
        [9.9984e-01, 1.5957e-04],
        [1.6733e-07, 1.0000e+00],
        [3.6121e-03, 9.9639e-01],
        [9.9941e-01, 5.9340e-04],
        [4.4196e-03, 9.9558e-01],
        [2.5367e-08, 1.0000e+00],
        [6.3588e-01, 3.6412e-01],
        [1.6075e-04, 9.9984e-01],
        [6.7553e-04, 9.9932e-01],
        [9.9879e-01, 1.2058e-03],
        [9.9959e-01, 4.1464e-04],
        [1.5024e-03, 9.9850e-01],
        [9.9971e-01, 2.8770e-04],
        [7.5781e-05, 9.9992e-01],
        [9.2271e-01, 7.7287e-02],
        [9.9982e-01, 1.8402e-04],
        [2.5832e-03, 9.9742e-01],
        [7.2145e-04, 9.9928e-01],
        [9.9989e-01, 1.0705e-04],
        [5.9325e-08, 1.0000e+00],
        [3.2495e-03, 9.9675e-01],
        [2.8828e-07, 1.0000e+00],
        [4.4938e-06, 1.0000e+00],
        [6.5719e-05, 9.9993e-01],
        [3.2332e-06, 1.0000e+00],
        [9.9982e-01, 1.7606e-04],
        [6.7636e-06, 9.9999e-01],
        [1.0259e-06, 1.0000e+00],
        [2.1941e-07, 1.0000e+00],
        [9.9963e-01, 3.7404e-04],
        [9.9917e-01, 8.2869e-04],
        [8.9795e-09, 1.0000e+00],
        [8.3874e-02, 9.1613e-01],
        [7.7250e-04, 9.9923e-01],
        [9.6071e-06, 9.9999e-01],
        [9.9989e-01, 1.0537e-04],
        [9.9977e-01, 2.2785e-04],
        [9.9717e-01, 2.8266e-03],
        [1.2886e-04, 9.9987e-01],
        [9.9991e-01, 8.6891e-05],
        [9.9975e-01, 2.4628e-04],
        [9.9959e-01, 4.0731e-04],
        [1.1747e-07, 1.0000e+00],
        [6.5576e-02, 9.3442e-01],
        [9.8911e-01, 1.0890e-02],
        [9.9226e-01, 7.7395e-03],
        [1.5022e-09, 1.0000e+00],
        [2.7494e-09, 1.0000e+00],
        [1.6770e-08, 1.0000e+00],
        [6.0012e-03, 9.9400e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5136e-06, 1.0000e+00],
        [1.1425e-07, 1.0000e+00],
        [9.9948e-01, 5.1631e-04],
        [3.4881e-08, 1.0000e+00],
        [2.2197e-07, 1.0000e+00],
        [1.6289e-05, 9.9998e-01],
        [2.2200e-05, 9.9998e-01],
        [4.3121e-05, 9.9996e-01],
        [1.0860e-02, 9.8914e-01],
        [1.8595e-06, 1.0000e+00],
        [1.0000e+00, 1.8014e-06],
        [9.9976e-01, 2.4410e-04],
        [3.9343e-03, 9.9607e-01],
        [4.1598e-07, 1.0000e+00],
        [1.8452e-07, 1.0000e+00],
        [3.0062e-05, 9.9997e-01],
        [4.0821e-08, 1.0000e+00],
        [9.9867e-01, 1.3257e-03],
        [6.4869e-06, 9.9999e-01],
        [1.0381e-05, 9.9999e-01],
        [4.1318e-07, 1.0000e+00],
        [2.2679e-02, 9.7732e-01],
        [4.0143e-07, 1.0000e+00],
        [2.3747e-05, 9.9998e-01],
        [1.1080e-06, 1.0000e+00],
        [4.2402e-06, 1.0000e+00],
        [3.3166e-03, 9.9668e-01],
        [6.1444e-03, 9.9386e-01],
        [1.8222e-05, 9.9998e-01],
        [1.7345e-04, 9.9983e-01],
        [9.9993e-01, 7.3238e-05],
        [9.9042e-01, 9.5762e-03],
        [9.9979e-01, 2.1285e-04],
        [9.8142e-01, 1.8580e-02],
        [9.9983e-01, 1.6801e-04],
        [1.9926e-04, 9.9980e-01],
        [1.0182e-07, 1.0000e+00],
        [2.1013e-07, 1.0000e+00],
        [2.0259e-04, 9.9980e-01],
        [8.6189e-08, 1.0000e+00],
        [3.2518e-04, 9.9967e-01],
        [1.5370e-04, 9.9985e-01],
        [2.6518e-05, 9.9997e-01],
        [8.0908e-05, 9.9992e-01],
        [9.9878e-01, 1.2170e-03],
        [1.7975e-07, 1.0000e+00],
        [8.3561e-07, 1.0000e+00],
        [2.5769e-04, 9.9974e-01],
        [9.8109e-01, 1.8906e-02],
        [9.9934e-01, 6.5890e-04],
        [4.1196e-06, 1.0000e+00],
        [6.9839e-03, 9.9302e-01],
        [9.9976e-01, 2.3665e-04],
        [2.9247e-02, 9.7075e-01],
        [7.2426e-04, 9.9928e-01],
        [9.9932e-01, 6.7529e-04],
        [1.7390e-04, 9.9983e-01],
        [9.9586e-01, 4.1448e-03],
        [6.0817e-03, 9.9392e-01],
        [9.1868e-03, 9.9081e-01],
        [9.9959e-01, 4.1365e-04],
        [1.3259e-07, 1.0000e+00],
        [2.4729e-03, 9.9753e-01],
        [8.1299e-04, 9.9919e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0036e-07, 1.0000e+00],
        [3.2003e-08, 1.0000e+00],
        [9.9693e-01, 3.0664e-03],
        [1.1990e-03, 9.9880e-01],
        [1.5110e-04, 9.9985e-01],
        [9.9889e-01, 1.1097e-03],
        [3.4795e-04, 9.9965e-01],
        [1.5990e-04, 9.9984e-01],
        [5.7208e-08, 1.0000e+00],
        [8.2825e-01, 1.7175e-01],
        [9.9978e-01, 2.1903e-04],
        [2.1775e-05, 9.9998e-01],
        [2.5478e-07, 1.0000e+00],
        [1.2177e-06, 1.0000e+00],
        [2.1133e-07, 1.0000e+00],
        [3.7355e-02, 9.6265e-01],
        [5.0008e-06, 9.9999e-01],
        [1.4454e-03, 9.9855e-01],
        [4.5846e-06, 1.0000e+00],
        [9.9511e-01, 4.8928e-03],
        [9.9976e-01, 2.3681e-04],
        [5.6592e-08, 1.0000e+00],
        [9.9990e-01, 1.0405e-04],
        [9.9946e-01, 5.4416e-04],
        [9.5193e-01, 4.8072e-02],
        [9.9929e-01, 7.0520e-04],
        [5.8109e-05, 9.9994e-01],
        [9.9966e-01, 3.4132e-04],
        [1.9563e-08, 1.0000e+00],
        [2.5442e-03, 9.9746e-01],
        [9.9963e-01, 3.6977e-04],
        [6.3265e-04, 9.9937e-01],
        [7.7026e-07, 1.0000e+00],
        [2.0956e-06, 1.0000e+00],
        [1.3314e-07, 1.0000e+00],
        [9.6085e-01, 3.9151e-02],
        [2.2068e-03, 9.9779e-01],
        [9.8422e-03, 9.9016e-01],
        [6.1225e-06, 9.9999e-01],
        [7.6545e-07, 1.0000e+00],
        [3.7619e-07, 1.0000e+00],
        [5.3599e-06, 9.9999e-01],
        [1.0536e-08, 1.0000e+00],
        [2.2976e-03, 9.9770e-01],
        [2.2527e-04, 9.9977e-01],
        [2.8465e-03, 9.9715e-01],
        [3.3564e-05, 9.9997e-01],
        [1.0267e-06, 1.0000e+00],
        [9.9627e-01, 3.7316e-03],
        [2.9204e-03, 9.9708e-01],
        [1.0204e-06, 1.0000e+00],
        [8.1691e-04, 9.9918e-01],
        [2.5874e-05, 9.9997e-01],
        [3.7385e-07, 1.0000e+00],
        [1.4745e-08, 1.0000e+00],
        [1.8852e-04, 9.9981e-01],
        [9.9893e-01, 1.0716e-03],
        [3.8220e-06, 1.0000e+00],
        [2.5621e-05, 9.9997e-01],
        [1.0560e-03, 9.9894e-01],
        [2.5442e-03, 9.9746e-01],
        [4.0716e-03, 9.9593e-01],
        [1.2139e-04, 9.9988e-01],
        [3.1617e-04, 9.9968e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1079e-03, 9.9789e-01],
        [1.4129e-04, 9.9986e-01],
        [4.4537e-05, 9.9996e-01],
        [2.9797e-02, 9.7020e-01],
        [3.9076e-01, 6.0924e-01],
        [4.8971e-05, 9.9995e-01],
        [9.7754e-07, 1.0000e+00],
        [9.8095e-01, 1.9046e-02],
        [2.0715e-03, 9.9793e-01],
        [1.3710e-09, 1.0000e+00],
        [3.5403e-06, 1.0000e+00],
        [6.3958e-01, 3.6042e-01],
        [2.1326e-01, 7.8674e-01],
        [5.6605e-02, 9.4339e-01],
        [1.1222e-04, 9.9989e-01],
        [1.0717e-06, 1.0000e+00],
        [9.6293e-05, 9.9990e-01],
        [9.9990e-01, 1.0257e-04],
        [1.4291e-06, 1.0000e+00],
        [7.5649e-06, 9.9999e-01],
        [2.9696e-04, 9.9970e-01],
        [9.9959e-01, 4.1003e-04],
        [1.5055e-03, 9.9849e-01],
        [1.4178e-05, 9.9999e-01],
        [9.9880e-01, 1.2025e-03],
        [9.9485e-01, 5.1482e-03],
        [4.3561e-04, 9.9956e-01],
        [1.4719e-05, 9.9999e-01],
        [9.9993e-01, 7.2806e-05],
        [3.1726e-05, 9.9997e-01],
        [1.1971e-03, 9.9880e-01],
        [7.1775e-04, 9.9928e-01],
        [1.7918e-06, 1.0000e+00],
        [1.9169e-05, 9.9998e-01],
        [9.9931e-01, 6.8981e-04],
        [9.9744e-01, 2.5557e-03],
        [6.8636e-05, 9.9993e-01],
        [1.9298e-08, 1.0000e+00],
        [2.1397e-02, 9.7860e-01],
        [9.5331e-01, 4.6695e-02],
        [9.9948e-01, 5.1660e-04],
        [3.3806e-07, 1.0000e+00],
        [6.0710e-06, 9.9999e-01],
        [1.8477e-03, 9.9815e-01],
        [1.6948e-01, 8.3052e-01],
        [9.9969e-01, 3.0683e-04],
        [1.5068e-03, 9.9849e-01],
        [1.3531e-04, 9.9986e-01],
        [9.9984e-01, 1.5893e-04],
        [9.9961e-01, 3.9308e-04],
        [1.8328e-06, 1.0000e+00],
        [1.5644e-07, 1.0000e+00],
        [7.7113e-04, 9.9923e-01],
        [1.6082e-05, 9.9998e-01],
        [9.9983e-01, 1.6862e-04],
        [9.9807e-01, 1.9259e-03],
        [1.8134e-06, 1.0000e+00],
        [1.5026e-05, 9.9998e-01],
        [3.7064e-05, 9.9996e-01],
        [9.9970e-01, 2.9749e-04],
        [6.4887e-07, 1.0000e+00],
        [1.1004e-03, 9.9890e-01],
        [2.9072e-06, 1.0000e+00],
        [1.4274e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.7192e-07, 1.0000e+00],
        [3.0420e-08, 1.0000e+00],
        [1.8294e-03, 9.9817e-01],
        [1.7739e-05, 9.9998e-01],
        [9.8499e-01, 1.5009e-02],
        [6.1689e-05, 9.9994e-01],
        [9.3527e-01, 6.4729e-02],
        [9.9946e-01, 5.3922e-04],
        [1.5428e-06, 1.0000e+00],
        [7.9650e-07, 1.0000e+00],
        [9.9971e-01, 2.9475e-04],
        [9.0342e-08, 1.0000e+00],
        [9.9982e-01, 1.8058e-04],
        [2.4962e-06, 1.0000e+00],
        [9.9983e-01, 1.6506e-04],
        [2.3820e-02, 9.7618e-01],
        [1.4188e-04, 9.9986e-01],
        [2.2442e-05, 9.9998e-01],
        [2.2209e-04, 9.9978e-01],
        [2.5855e-07, 1.0000e+00],
        [2.9718e-05, 9.9997e-01],
        [1.4918e-09, 1.0000e+00],
        [9.9978e-01, 2.1612e-04],
        [3.4154e-05, 9.9997e-01],
        [5.4177e-08, 1.0000e+00],
        [4.7729e-07, 1.0000e+00],
        [9.9857e-01, 1.4308e-03],
        [9.9598e-01, 4.0197e-03],
        [3.0366e-01, 6.9634e-01],
        [6.5200e-06, 9.9999e-01],
        [7.2328e-04, 9.9928e-01],
        [9.9979e-01, 2.1023e-04],
        [1.4704e-02, 9.8530e-01],
        [2.0476e-04, 9.9980e-01],
        [9.9958e-01, 4.2205e-04],
        [9.9947e-01, 5.2875e-04],
        [9.4908e-01, 5.0924e-02],
        [9.9087e-01, 9.1267e-03],
        [9.9983e-01, 1.7191e-04],
        [1.0299e-06, 1.0000e+00],
        [9.9809e-01, 1.9093e-03],
        [9.6740e-08, 1.0000e+00],
        [1.6161e-09, 1.0000e+00],
        [9.9998e-01, 1.8139e-05],
        [4.3120e-07, 1.0000e+00],
        [9.9983e-01, 1.7039e-04],
        [2.0397e-07, 1.0000e+00],
        [9.9969e-01, 3.1157e-04],
        [8.1571e-02, 9.1843e-01],
        [1.2432e-03, 9.9876e-01],
        [2.3561e-06, 1.0000e+00],
        [4.8283e-04, 9.9952e-01],
        [1.6871e-05, 9.9998e-01],
        [6.9765e-04, 9.9930e-01],
        [1.6680e-03, 9.9833e-01],
        [7.1004e-10, 1.0000e+00],
        [4.1497e-02, 9.5850e-01],
        [5.0394e-07, 1.0000e+00],
        [2.8846e-05, 9.9997e-01],
        [3.7950e-04, 9.9962e-01],
        [6.0913e-05, 9.9994e-01],
        [9.9974e-01, 2.6145e-04],
        [9.9967e-01, 3.2920e-04],
        [5.9532e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.5544e-04, 9.9924e-01],
        [3.2532e-05, 9.9997e-01],
        [9.9912e-01, 8.7595e-04],
        [1.0349e-05, 9.9999e-01],
        [4.2995e-03, 9.9570e-01],
        [1.6597e-04, 9.9983e-01],
        [1.7269e-04, 9.9983e-01],
        [1.7212e-01, 8.2788e-01],
        [9.9026e-01, 9.7443e-03],
        [3.2037e-03, 9.9680e-01],
        [8.9691e-03, 9.9103e-01],
        [3.2634e-03, 9.9674e-01],
        [2.5018e-06, 1.0000e+00],
        [9.9969e-01, 3.0881e-04],
        [9.9961e-01, 3.9289e-04],
        [1.5814e-06, 1.0000e+00],
        [1.6943e-04, 9.9983e-01],
        [6.7854e-04, 9.9932e-01],
        [9.9985e-01, 1.4501e-04],
        [1.5507e-06, 1.0000e+00],
        [4.1908e-04, 9.9958e-01],
        [1.0454e-04, 9.9990e-01],
        [7.8860e-05, 9.9992e-01],
        [3.5662e-04, 9.9964e-01],
        [9.9963e-01, 3.6968e-04],
        [1.5007e-04, 9.9985e-01],
        [6.6625e-08, 1.0000e+00],
        [7.2237e-05, 9.9993e-01],
        [5.2807e-06, 9.9999e-01],
        [9.9985e-01, 1.5180e-04],
        [3.6002e-03, 9.9640e-01],
        [9.9954e-01, 4.6108e-04],
        [2.4565e-05, 9.9998e-01],
        [2.2037e-03, 9.9780e-01],
        [5.5870e-06, 9.9999e-01],
        [4.9488e-05, 9.9995e-01],
        [9.9769e-01, 2.3124e-03],
        [9.9989e-01, 1.1335e-04],
        [5.1718e-05, 9.9995e-01],
        [9.9898e-01, 1.0156e-03],
        [4.5814e-05, 9.9995e-01],
        [1.0569e-05, 9.9999e-01],
        [9.9951e-01, 4.8965e-04],
        [2.7170e-06, 1.0000e+00],
        [2.0494e-04, 9.9980e-01],
        [2.5544e-04, 9.9974e-01],
        [4.8468e-01, 5.1532e-01],
        [3.2860e-04, 9.9967e-01],
        [5.1316e-08, 1.0000e+00],
        [1.1365e-03, 9.9886e-01],
        [9.9819e-01, 1.8131e-03],
        [6.6700e-03, 9.9333e-01],
        [1.1605e-05, 9.9999e-01],
        [2.1410e-03, 9.9786e-01],
        [8.6066e-05, 9.9991e-01],
        [1.4592e-03, 9.9854e-01],
        [4.7826e-05, 9.9995e-01],
        [1.5182e-01, 8.4818e-01],
        [5.5102e-08, 1.0000e+00],
        [1.4553e-03, 9.9854e-01],
        [9.8429e-06, 9.9999e-01],
        [6.7172e-02, 9.3283e-01],
        [4.2472e-05, 9.9996e-01],
        [5.3584e-02, 9.4642e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9962e-01, 3.8119e-04],
        [1.2324e-08, 1.0000e+00],
        [9.9978e-01, 2.2300e-04],
        [6.6206e-07, 1.0000e+00],
        [7.7542e-04, 9.9922e-01],
        [1.1991e-06, 1.0000e+00],
        [8.9500e-07, 1.0000e+00],
        [2.6045e-08, 1.0000e+00],
        [1.2477e-04, 9.9988e-01],
        [9.9936e-01, 6.4432e-04],
        [2.2825e-01, 7.7175e-01],
        [8.1370e-01, 1.8630e-01],
        [4.7222e-05, 9.9995e-01],
        [1.5249e-03, 9.9848e-01],
        [9.9935e-01, 6.4635e-04],
        [2.0006e-01, 7.9994e-01],
        [4.1887e-03, 9.9581e-01],
        [2.2015e-05, 9.9998e-01],
        [9.9789e-01, 2.1104e-03],
        [9.8731e-01, 1.2686e-02],
        [1.3006e-07, 1.0000e+00],
        [3.4518e-02, 9.6548e-01],
        [3.8752e-05, 9.9996e-01],
        [1.2510e-05, 9.9999e-01],
        [9.4545e-05, 9.9991e-01],
        [3.0712e-03, 9.9693e-01],
        [6.1421e-06, 9.9999e-01],
        [9.4143e-01, 5.8570e-02],
        [1.0098e-01, 8.9902e-01],
        [9.9985e-01, 1.4629e-04],
        [5.3223e-08, 1.0000e+00],
        [2.0282e-05, 9.9998e-01],
        [9.9994e-01, 6.2450e-05],
        [8.4841e-04, 9.9915e-01],
        [1.3811e-03, 9.9862e-01],
        [8.3903e-01, 1.6097e-01],
        [3.5312e-03, 9.9647e-01],
        [2.9664e-01, 7.0336e-01],
        [4.2904e-02, 9.5710e-01],
        [2.9906e-06, 1.0000e+00],
        [1.4614e-03, 9.9854e-01],
        [4.1312e-05, 9.9996e-01],
        [4.9331e-04, 9.9951e-01],
        [1.8552e-06, 1.0000e+00],
        [9.9947e-01, 5.2970e-04],
        [6.6471e-07, 1.0000e+00],
        [1.4233e-04, 9.9986e-01],
        [9.9982e-01, 1.7920e-04],
        [1.9216e-06, 1.0000e+00],
        [6.6493e-05, 9.9993e-01],
        [9.9728e-01, 2.7179e-03],
        [9.9835e-01, 1.6500e-03],
        [9.9981e-01, 1.9197e-04],
        [4.8303e-04, 9.9952e-01],
        [2.4125e-06, 1.0000e+00],
        [9.8661e-01, 1.3389e-02],
        [2.4276e-05, 9.9998e-01],
        [9.9738e-01, 2.6168e-03],
        [1.0374e-04, 9.9990e-01],
        [7.4327e-04, 9.9926e-01],
        [2.3589e-02, 9.7641e-01],
        [4.3373e-10, 1.0000e+00],
        [4.9500e-05, 9.9995e-01],
        [7.4234e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9898e-01, 1.0214e-03],
        [9.6554e-03, 9.9034e-01],
        [3.4749e-05, 9.9997e-01],
        [3.2803e-04, 9.9967e-01],
        [5.7606e-03, 9.9424e-01],
        [7.9937e-08, 1.0000e+00],
        [4.1316e-08, 1.0000e+00],
        [2.8609e-04, 9.9971e-01],
        [1.8118e-06, 1.0000e+00],
        [9.9982e-01, 1.7672e-04],
        [1.0768e-05, 9.9999e-01],
        [2.9353e-09, 1.0000e+00],
        [9.9975e-01, 2.5387e-04],
        [4.1931e-04, 9.9958e-01],
        [9.9990e-01, 9.9702e-05],
        [9.9927e-01, 7.3128e-04],
        [9.9984e-01, 1.6115e-04],
        [4.4575e-06, 1.0000e+00],
        [2.6150e-04, 9.9974e-01],
        [1.7707e-03, 9.9823e-01],
        [4.2261e-06, 1.0000e+00],
        [7.3936e-05, 9.9993e-01],
        [1.3974e-06, 1.0000e+00],
        [3.2272e-07, 1.0000e+00],
        [9.9992e-01, 7.6165e-05],
        [1.8751e-06, 1.0000e+00],
        [8.7384e-05, 9.9991e-01],
        [9.9912e-01, 8.7996e-04],
        [9.7450e-06, 9.9999e-01],
        [7.5596e-03, 9.9244e-01],
        [9.8026e-05, 9.9990e-01],
        [1.4151e-05, 9.9999e-01],
        [9.9728e-01, 2.7165e-03],
        [9.9962e-01, 3.8270e-04],
        [3.0510e-07, 1.0000e+00],
        [1.1432e-04, 9.9989e-01],
        [9.9979e-01, 2.1071e-04],
        [8.0277e-01, 1.9723e-01],
        [3.5046e-06, 1.0000e+00],
        [9.9977e-01, 2.2578e-04],
        [9.9973e-01, 2.6669e-04],
        [1.1674e-06, 1.0000e+00],
        [1.5729e-04, 9.9984e-01],
        [7.4903e-05, 9.9993e-01],
        [2.2215e-07, 1.0000e+00],
        [1.6170e-01, 8.3830e-01],
        [9.4360e-01, 5.6401e-02],
        [1.3243e-08, 1.0000e+00],
        [1.0612e-02, 9.8939e-01],
        [9.9970e-01, 3.0089e-04],
        [1.1489e-05, 9.9999e-01],
        [2.5587e-06, 1.0000e+00],
        [3.0494e-06, 1.0000e+00],
        [2.2886e-05, 9.9998e-01],
        [1.7614e-03, 9.9824e-01],
        [4.1088e-07, 1.0000e+00],
        [1.2402e-02, 9.8760e-01],
        [1.4559e-05, 9.9999e-01],
        [1.3474e-06, 1.0000e+00],
        [6.1641e-03, 9.9384e-01],
        [9.9988e-01, 1.1972e-04],
        [9.9999e-01, 9.9281e-06],
        [9.9986e-01, 1.4004e-04],
        [9.8122e-01, 1.8780e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9989e-01, 1.1075e-04],
        [2.4135e-08, 1.0000e+00],
        [1.7183e-06, 1.0000e+00],
        [8.2050e-01, 1.7950e-01],
        [6.8979e-05, 9.9993e-01],
        [9.9513e-01, 4.8670e-03],
        [2.6035e-07, 1.0000e+00],
        [2.6513e-04, 9.9973e-01],
        [4.0037e-07, 1.0000e+00],
        [9.9950e-01, 4.9823e-04],
        [2.2034e-05, 9.9998e-01],
        [9.1546e-04, 9.9908e-01],
        [3.1119e-02, 9.6888e-01],
        [9.1081e-05, 9.9991e-01],
        [1.8379e-02, 9.8162e-01],
        [4.9930e-01, 5.0070e-01],
        [3.3456e-06, 1.0000e+00],
        [9.8295e-01, 1.7048e-02],
        [9.7383e-01, 2.6166e-02],
        [9.9930e-01, 7.0131e-04],
        [2.9571e-08, 1.0000e+00],
        [9.9982e-01, 1.7984e-04],
        [9.3385e-07, 1.0000e+00],
        [1.1091e-03, 9.9889e-01],
        [6.6034e-05, 9.9993e-01],
        [8.1720e-07, 1.0000e+00],
        [1.1458e-03, 9.9885e-01],
        [5.4372e-07, 1.0000e+00],
        [2.6032e-07, 1.0000e+00],
        [9.9985e-01, 1.5217e-04],
        [1.5218e-05, 9.9998e-01],
        [1.4055e-04, 9.9986e-01],
        [9.8001e-01, 1.9986e-02],
        [1.1531e-08, 1.0000e+00],
        [9.9926e-01, 7.4271e-04],
        [9.9978e-01, 2.2136e-04],
        [8.4148e-04, 9.9916e-01],
        [2.9497e-09, 1.0000e+00],
        [9.8489e-01, 1.5110e-02],
        [9.4196e-08, 1.0000e+00],
        [4.4208e-04, 9.9956e-01],
        [1.2731e-04, 9.9987e-01],
        [1.4236e-07, 1.0000e+00],
        [9.9991e-01, 8.7374e-05],
        [7.1159e-04, 9.9929e-01],
        [2.8364e-05, 9.9997e-01],
        [4.0289e-06, 1.0000e+00],
        [1.1828e-09, 1.0000e+00],
        [9.9989e-01, 1.0737e-04],
        [8.3355e-09, 1.0000e+00],
        [1.0170e-04, 9.9990e-01],
        [1.7892e-03, 9.9821e-01],
        [1.5607e-06, 1.0000e+00],
        [2.4279e-05, 9.9998e-01],
        [5.4696e-06, 9.9999e-01],
        [1.0000e+00, 8.0501e-07],
        [4.1601e-08, 1.0000e+00],
        [5.5757e-08, 1.0000e+00],
        [4.2470e-06, 1.0000e+00],
        [3.3749e-06, 1.0000e+00],
        [9.9133e-01, 8.6669e-03],
        [1.9319e-08, 1.0000e+00],
        [7.2908e-08, 1.0000e+00],
        [1.1748e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0978e-05, 9.9999e-01],
        [9.9957e-01, 4.3037e-04],
        [9.5685e-05, 9.9990e-01],
        [9.9986e-01, 1.4126e-04],
        [7.4560e-08, 1.0000e+00],
        [9.3952e-01, 6.0482e-02],
        [1.4495e-04, 9.9986e-01],
        [4.0584e-07, 1.0000e+00],
        [3.4430e-06, 1.0000e+00],
        [1.1360e-03, 9.9886e-01],
        [3.2880e-07, 1.0000e+00],
        [1.2570e-02, 9.8743e-01],
        [1.4581e-05, 9.9999e-01],
        [1.5617e-05, 9.9998e-01],
        [1.7455e-04, 9.9983e-01],
        [9.9684e-01, 3.1598e-03],
        [1.1152e-06, 1.0000e+00],
        [9.7768e-07, 1.0000e+00],
        [1.6362e-05, 9.9998e-01],
        [9.9968e-01, 3.2144e-04],
        [1.2736e-06, 1.0000e+00],
        [6.8722e-07, 1.0000e+00],
        [6.9371e-06, 9.9999e-01],
        [9.9612e-01, 3.8847e-03],
        [5.9577e-10, 1.0000e+00],
        [5.5052e-02, 9.4495e-01],
        [1.6185e-01, 8.3815e-01],
        [8.6442e-05, 9.9991e-01],
        [1.5260e-05, 9.9998e-01],
        [4.5981e-05, 9.9995e-01],
        [9.9982e-01, 1.7550e-04],
        [3.5786e-07, 1.0000e+00],
        [9.9963e-01, 3.7182e-04],
        [9.9996e-01, 4.1276e-05],
        [9.9998e-01, 1.7474e-05],
        [7.5071e-07, 1.0000e+00],
        [2.4198e-06, 1.0000e+00],
        [5.7541e-07, 1.0000e+00],
        [1.2295e-06, 1.0000e+00],
        [4.4290e-05, 9.9996e-01],
        [8.7786e-09, 1.0000e+00],
        [3.6597e-05, 9.9996e-01],
        [4.5051e-01, 5.4949e-01],
        [1.7859e-07, 1.0000e+00],
        [1.7322e-03, 9.9827e-01],
        [1.6088e-03, 9.9839e-01],
        [9.9985e-01, 1.5329e-04],
        [2.7355e-07, 1.0000e+00],
        [9.9989e-01, 1.1238e-04],
        [9.4394e-07, 1.0000e+00],
        [1.3404e-07, 1.0000e+00],
        [1.5458e-04, 9.9985e-01],
        [1.8989e-08, 1.0000e+00],
        [6.1860e-08, 1.0000e+00],
        [7.4997e-07, 1.0000e+00],
        [9.7161e-07, 1.0000e+00],
        [9.9995e-01, 4.8533e-05],
        [3.5877e-10, 1.0000e+00],
        [9.9985e-01, 1.4779e-04],
        [1.6203e-02, 9.8380e-01],
        [1.9916e-06, 1.0000e+00],
        [9.9650e-01, 3.5003e-03],
        [7.1085e-02, 9.2891e-01],
        [2.6679e-04, 9.9973e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4612e-06, 1.0000e+00],
        [9.9585e-01, 4.1515e-03],
        [1.1423e-07, 1.0000e+00],
        [5.3602e-04, 9.9946e-01],
        [1.9710e-03, 9.9803e-01],
        [9.9977e-01, 2.2983e-04],
        [8.4404e-03, 9.9156e-01],
        [7.2248e-07, 1.0000e+00],
        [9.8707e-01, 1.2934e-02],
        [9.9948e-01, 5.1926e-04],
        [6.0708e-09, 1.0000e+00],
        [1.1108e-08, 1.0000e+00],
        [4.7894e-05, 9.9995e-01],
        [9.4655e-04, 9.9905e-01],
        [2.6356e-05, 9.9997e-01],
        [1.6520e-05, 9.9998e-01],
        [9.9892e-01, 1.0751e-03],
        [2.4422e-04, 9.9976e-01],
        [1.3966e-06, 1.0000e+00],
        [7.0675e-03, 9.9293e-01],
        [1.2021e-04, 9.9988e-01],
        [7.5627e-05, 9.9992e-01],
        [9.9944e-01, 5.5614e-04],
        [1.3133e-06, 1.0000e+00],
        [3.7798e-04, 9.9962e-01],
        [2.8038e-08, 1.0000e+00],
        [9.9592e-01, 4.0842e-03],
        [1.0751e-08, 1.0000e+00],
        [1.6158e-06, 1.0000e+00],
        [2.5180e-06, 1.0000e+00],
        [9.4813e-04, 9.9905e-01],
        [5.4955e-03, 9.9450e-01],
        [1.4475e-03, 9.9855e-01],
        [1.1255e-08, 1.0000e+00],
        [1.6366e-04, 9.9984e-01],
        [1.6796e-06, 1.0000e+00],
        [5.7959e-08, 1.0000e+00],
        [6.1538e-06, 9.9999e-01],
        [9.8308e-01, 1.6922e-02],
        [1.0000e+00, 3.7526e-06],
        [9.9788e-01, 2.1169e-03],
        [1.2323e-07, 1.0000e+00],
        [4.6174e-05, 9.9995e-01],
        [1.8946e-03, 9.9811e-01],
        [9.9986e-01, 1.4104e-04],
        [1.4810e-08, 1.0000e+00],
        [9.9956e-01, 4.4065e-04],
        [1.3822e-02, 9.8618e-01],
        [6.0239e-06, 9.9999e-01],
        [1.0168e-06, 1.0000e+00],
        [3.3886e-07, 1.0000e+00],
        [1.2853e-01, 8.7147e-01],
        [9.9952e-01, 4.8469e-04],
        [3.1055e-06, 1.0000e+00],
        [2.1794e-07, 1.0000e+00],
        [6.2386e-08, 1.0000e+00],
        [9.9945e-01, 5.4849e-04],
        [3.9680e-07, 1.0000e+00],
        [9.1047e-05, 9.9991e-01],
        [1.5593e-06, 1.0000e+00],
        [4.5460e-07, 1.0000e+00],
        [4.7522e-07, 1.0000e+00],
        [1.6814e-06, 1.0000e+00],
        [3.0549e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1405e-08, 1.0000e+00],
        [2.0468e-06, 1.0000e+00],
        [5.5395e-10, 1.0000e+00],
        [3.0275e-02, 9.6972e-01],
        [4.9285e-06, 1.0000e+00],
        [9.9991e-01, 9.2472e-05],
        [3.1836e-08, 1.0000e+00],
        [9.9976e-01, 2.4151e-04],
        [2.2508e-06, 1.0000e+00],
        [9.9992e-01, 8.4688e-05],
        [1.5571e-06, 1.0000e+00],
        [9.9920e-01, 7.9893e-04],
        [1.8570e-07, 1.0000e+00],
        [3.1323e-05, 9.9997e-01],
        [2.8712e-04, 9.9971e-01],
        [2.2755e-06, 1.0000e+00],
        [2.3801e-04, 9.9976e-01],
        [1.5667e-02, 9.8433e-01],
        [2.7408e-06, 1.0000e+00],
        [9.2789e-06, 9.9999e-01],
        [9.9969e-01, 3.0991e-04],
        [9.3802e-06, 9.9999e-01],
        [1.5333e-05, 9.9998e-01],
        [1.2903e-04, 9.9987e-01],
        [1.4908e-02, 9.8509e-01],
        [7.7230e-05, 9.9992e-01],
        [3.5094e-05, 9.9996e-01],
        [1.8164e-05, 9.9998e-01],
        [8.2071e-05, 9.9992e-01],
        [8.9596e-07, 1.0000e+00],
        [3.6039e-03, 9.9640e-01],
        [4.7350e-06, 1.0000e+00],
        [5.7383e-07, 1.0000e+00],
        [6.8151e-06, 9.9999e-01],
        [9.9929e-01, 7.1498e-04],
        [7.6548e-01, 2.3452e-01],
        [2.8897e-08, 1.0000e+00],
        [3.1809e-05, 9.9997e-01],
        [2.1852e-05, 9.9998e-01],
        [9.9981e-01, 1.9362e-04],
        [7.3524e-05, 9.9993e-01],
        [2.1070e-01, 7.8930e-01],
        [2.6158e-06, 1.0000e+00],
        [7.2800e-06, 9.9999e-01],
        [1.5379e-07, 1.0000e+00],
        [4.8149e-08, 1.0000e+00],
        [8.2733e-05, 9.9992e-01],
        [5.6870e-08, 1.0000e+00],
        [1.0289e-05, 9.9999e-01],
        [1.4577e-03, 9.9854e-01],
        [1.9678e-03, 9.9803e-01],
        [1.0959e-02, 9.8904e-01],
        [3.8490e-03, 9.9615e-01],
        [4.9652e-09, 1.0000e+00],
        [6.9865e-06, 9.9999e-01],
        [5.7883e-04, 9.9942e-01],
        [2.0423e-03, 9.9796e-01],
        [1.0582e-06, 1.0000e+00],
        [7.2078e-01, 2.7922e-01],
        [3.9751e-03, 9.9602e-01],
        [1.0855e-09, 1.0000e+00],
        [9.8587e-01, 1.4127e-02],
        [7.0405e-04, 9.9930e-01],
        [5.0644e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7087e-07, 1.0000e+00],
        [7.5434e-04, 9.9925e-01],
        [4.8522e-07, 1.0000e+00],
        [1.7363e-05, 9.9998e-01],
        [9.0786e-04, 9.9909e-01],
        [2.1872e-07, 1.0000e+00],
        [6.3882e-06, 9.9999e-01],
        [1.9670e-05, 9.9998e-01],
        [5.6702e-07, 1.0000e+00],
        [6.9885e-01, 3.0115e-01],
        [5.2411e-03, 9.9476e-01],
        [2.7140e-05, 9.9997e-01],
        [2.2114e-09, 1.0000e+00],
        [7.3023e-08, 1.0000e+00],
        [9.6470e-05, 9.9990e-01],
        [9.9881e-01, 1.1853e-03],
        [9.9848e-03, 9.9002e-01],
        [3.6577e-08, 1.0000e+00],
        [1.7070e-06, 1.0000e+00],
        [1.1376e-04, 9.9989e-01],
        [1.0263e-05, 9.9999e-01],
        [1.0713e-07, 1.0000e+00],
        [1.2115e-05, 9.9999e-01],
        [8.7341e-01, 1.2659e-01],
        [2.1935e-04, 9.9978e-01],
        [1.0157e-05, 9.9999e-01],
        [1.7977e-05, 9.9998e-01],
        [5.2349e-03, 9.9477e-01],
        [2.5985e-03, 9.9740e-01],
        [5.5107e-04, 9.9945e-01],
        [9.4157e-01, 5.8433e-02],
        [5.7725e-07, 1.0000e+00],
        [8.1137e-07, 1.0000e+00],
        [6.2704e-03, 9.9373e-01],
        [1.9139e-07, 1.0000e+00],
        [3.8528e-04, 9.9961e-01],
        [2.5315e-04, 9.9975e-01],
        [1.7537e-05, 9.9998e-01],
        [1.6392e-09, 1.0000e+00],
        [9.4782e-06, 9.9999e-01],
        [1.0622e-09, 1.0000e+00],
        [3.1090e-06, 1.0000e+00],
        [3.8588e-05, 9.9996e-01],
        [7.8380e-01, 2.1620e-01],
        [9.6271e-02, 9.0373e-01],
        [6.9724e-06, 9.9999e-01],
        [9.9965e-01, 3.5382e-04],
        [3.6897e-06, 1.0000e+00],
        [9.9988e-01, 1.2148e-04],
        [1.5955e-05, 9.9998e-01],
        [4.1597e-04, 9.9958e-01],
        [3.6165e-03, 9.9638e-01],
        [3.4837e-05, 9.9997e-01],
        [9.9024e-01, 9.7619e-03],
        [1.9478e-05, 9.9998e-01],
        [1.1362e-06, 1.0000e+00],
        [6.7799e-10, 1.0000e+00],
        [9.9921e-01, 7.8896e-04],
        [1.2760e-08, 1.0000e+00],
        [5.4118e-08, 1.0000e+00],
        [1.2172e-04, 9.9988e-01],
        [1.3926e-05, 9.9999e-01],
        [8.8278e-06, 9.9999e-01],
        [1.8290e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.6354e-05, 9.9995e-01],
        [1.0928e-08, 1.0000e+00],
        [2.0327e-09, 1.0000e+00],
        [6.8860e-05, 9.9993e-01],
        [7.5652e-06, 9.9999e-01],
        [1.2844e-06, 1.0000e+00],
        [9.7895e-01, 2.1055e-02],
        [2.6230e-05, 9.9997e-01],
        [9.9979e-01, 2.1049e-04],
        [1.1354e-04, 9.9989e-01],
        [7.6399e-05, 9.9992e-01],
        [1.8500e-06, 1.0000e+00],
        [2.0995e-08, 1.0000e+00],
        [9.9984e-01, 1.6018e-04],
        [1.5876e-09, 1.0000e+00],
        [4.1417e-01, 5.8583e-01],
        [1.5518e-05, 9.9998e-01],
        [1.6555e-09, 1.0000e+00],
        [1.0386e-01, 8.9614e-01],
        [6.6052e-09, 1.0000e+00],
        [3.7573e-07, 1.0000e+00],
        [4.9985e-05, 9.9995e-01],
        [1.6135e-09, 1.0000e+00],
        [1.3801e-04, 9.9986e-01],
        [3.8439e-04, 9.9962e-01],
        [4.4941e-08, 1.0000e+00],
        [2.8539e-02, 9.7146e-01],
        [9.5983e-06, 9.9999e-01],
        [9.9985e-01, 1.4951e-04],
        [1.1199e-08, 1.0000e+00],
        [1.9097e-01, 8.0903e-01],
        [3.9152e-05, 9.9996e-01],
        [1.5261e-06, 1.0000e+00],
        [6.4344e-08, 1.0000e+00],
        [2.3755e-08, 1.0000e+00],
        [1.9807e-06, 1.0000e+00],
        [9.9943e-01, 5.7043e-04],
        [9.7421e-01, 2.5791e-02],
        [9.9788e-01, 2.1183e-03],
        [1.3138e-04, 9.9987e-01],
        [9.4145e-07, 1.0000e+00],
        [7.8678e-07, 1.0000e+00],
        [9.9808e-01, 1.9216e-03],
        [9.9964e-01, 3.5936e-04],
        [2.9825e-04, 9.9970e-01],
        [8.9022e-09, 1.0000e+00],
        [9.7821e-02, 9.0218e-01],
        [1.0977e-08, 1.0000e+00],
        [1.8660e-06, 1.0000e+00],
        [1.0554e-03, 9.9894e-01],
        [9.9997e-01, 3.4451e-05],
        [5.9820e-09, 1.0000e+00],
        [3.4911e-07, 1.0000e+00],
        [1.3477e-05, 9.9999e-01],
        [4.6831e-07, 1.0000e+00],
        [1.5983e-03, 9.9840e-01],
        [9.9952e-01, 4.7957e-04],
        [1.2518e-03, 9.9875e-01],
        [9.9973e-01, 2.6862e-04],
        [9.9909e-01, 9.1118e-04],
        [2.5315e-04, 9.9975e-01],
        [4.1720e-05, 9.9996e-01],
        [9.9977e-01, 2.2904e-04],
        [4.1012e-01, 5.8988e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0326e-08, 1.0000e+00],
        [9.2745e-03, 9.9073e-01],
        [9.9928e-01, 7.1997e-04],
        [9.2770e-11, 1.0000e+00],
        [1.3375e-05, 9.9999e-01],
        [7.3229e-04, 9.9927e-01],
        [1.9151e-07, 1.0000e+00],
        [9.9754e-01, 2.4601e-03],
        [2.3091e-06, 1.0000e+00],
        [1.4363e-05, 9.9999e-01],
        [4.0890e-07, 1.0000e+00],
        [3.3522e-02, 9.6648e-01],
        [9.9976e-01, 2.4461e-04],
        [2.3021e-04, 9.9977e-01],
        [1.2503e-05, 9.9999e-01],
        [9.9952e-01, 4.8130e-04],
        [8.8428e-02, 9.1157e-01],
        [8.7112e-06, 9.9999e-01],
        [5.1977e-03, 9.9480e-01],
        [9.4638e-01, 5.3621e-02],
        [3.2961e-08, 1.0000e+00],
        [3.2221e-06, 1.0000e+00],
        [1.5647e-05, 9.9998e-01],
        [2.7123e-08, 1.0000e+00],
        [1.7013e-04, 9.9983e-01],
        [1.4849e-08, 1.0000e+00],
        [2.4415e-06, 1.0000e+00],
        [2.9295e-03, 9.9707e-01],
        [2.5304e-07, 1.0000e+00],
        [4.1747e-04, 9.9958e-01],
        [9.9984e-01, 1.5738e-04],
        [1.0224e-09, 1.0000e+00],
        [9.5905e-01, 4.0954e-02],
        [1.2252e-07, 1.0000e+00],
        [9.9963e-01, 3.6554e-04],
        [1.9146e-06, 1.0000e+00],
        [7.9162e-08, 1.0000e+00],
        [6.9079e-06, 9.9999e-01],
        [1.1180e-05, 9.9999e-01],
        [2.1721e-03, 9.9783e-01],
        [1.6864e-08, 1.0000e+00],
        [9.9405e-03, 9.9006e-01],
        [6.4662e-08, 1.0000e+00],
        [1.0625e-08, 1.0000e+00],
        [4.3812e-09, 1.0000e+00],
        [2.4001e-09, 1.0000e+00],
        [5.6828e-09, 1.0000e+00],
        [3.6019e-08, 1.0000e+00],
        [1.5110e-01, 8.4890e-01],
        [5.3778e-07, 1.0000e+00],
        [9.9866e-01, 1.3409e-03],
        [2.6977e-08, 1.0000e+00],
        [3.4830e-05, 9.9997e-01],
        [3.8909e-06, 1.0000e+00],
        [8.7168e-06, 9.9999e-01],
        [1.2902e-07, 1.0000e+00],
        [9.9956e-01, 4.3924e-04],
        [9.7983e-07, 1.0000e+00],
        [8.0379e-08, 1.0000e+00],
        [9.4482e-01, 5.5177e-02],
        [1.7162e-03, 9.9828e-01],
        [9.4607e-07, 1.0000e+00],
        [8.8489e-05, 9.9991e-01],
        [6.0064e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9920e-01, 8.0266e-04],
        [5.9466e-06, 9.9999e-01],
        [6.2500e-04, 9.9938e-01],
        [2.7124e-07, 1.0000e+00],
        [2.3502e-05, 9.9998e-01],
        [5.1976e-07, 1.0000e+00],
        [1.4144e-02, 9.8586e-01],
        [9.9699e-01, 3.0121e-03],
        [2.7124e-07, 1.0000e+00],
        [1.1359e-03, 9.9886e-01],
        [1.8064e-04, 9.9982e-01],
        [1.5654e-02, 9.8435e-01],
        [9.9743e-01, 2.5726e-03],
        [3.3186e-07, 1.0000e+00],
        [1.6307e-03, 9.9837e-01],
        [1.0987e-06, 1.0000e+00],
        [2.0112e-01, 7.9888e-01],
        [7.4430e-07, 1.0000e+00],
        [9.9947e-01, 5.3001e-04],
        [4.4003e-04, 9.9956e-01],
        [1.7455e-09, 1.0000e+00],
        [3.6166e-05, 9.9996e-01],
        [9.9980e-01, 1.9859e-04],
        [9.9905e-01, 9.4977e-04],
        [9.6882e-09, 1.0000e+00],
        [9.9998e-01, 1.5299e-05],
        [6.6714e-07, 1.0000e+00],
        [9.9207e-01, 7.9324e-03],
        [5.9878e-08, 1.0000e+00],
        [4.8473e-02, 9.5153e-01],
        [1.9441e-02, 9.8056e-01],
        [5.9356e-06, 9.9999e-01],
        [9.8240e-01, 1.7598e-02],
        [3.5198e-07, 1.0000e+00],
        [9.9997e-01, 2.6002e-05],
        [9.9835e-01, 1.6483e-03],
        [2.3648e-08, 1.0000e+00],
        [4.2524e-05, 9.9996e-01],
        [9.1744e-09, 1.0000e+00],
        [8.4178e-07, 1.0000e+00],
        [9.9987e-01, 1.3012e-04],
        [4.7990e-04, 9.9952e-01],
        [1.5944e-07, 1.0000e+00],
        [1.4574e-04, 9.9985e-01],
        [7.6949e-03, 9.9231e-01],
        [1.3242e-05, 9.9999e-01],
        [9.9962e-01, 3.8265e-04],
        [5.0397e-05, 9.9995e-01],
        [1.6847e-05, 9.9998e-01],
        [9.9762e-01, 2.3771e-03],
        [6.0895e-04, 9.9939e-01],
        [1.0651e-06, 1.0000e+00],
        [3.7274e-02, 9.6273e-01],
        [9.9985e-01, 1.5370e-04],
        [6.9633e-08, 1.0000e+00],
        [9.9975e-01, 2.5367e-04],
        [1.2843e-08, 1.0000e+00],
        [4.6936e-05, 9.9995e-01],
        [9.9895e-01, 1.0495e-03],
        [2.1347e-08, 1.0000e+00],
        [5.0561e-02, 9.4944e-01],
        [4.6333e-06, 1.0000e+00],
        [7.2904e-01, 2.7096e-01],
        [2.2924e-04, 9.9977e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6104e-08, 1.0000e+00],
        [6.8722e-08, 1.0000e+00],
        [9.9681e-01, 3.1909e-03],
        [8.2480e-06, 9.9999e-01],
        [9.9966e-01, 3.3527e-04],
        [5.3130e-05, 9.9995e-01],
        [1.7634e-07, 1.0000e+00],
        [5.9458e-08, 1.0000e+00],
        [7.4636e-09, 1.0000e+00],
        [1.8938e-07, 1.0000e+00],
        [2.1754e-04, 9.9978e-01],
        [4.3078e-04, 9.9957e-01],
        [9.8999e-01, 1.0012e-02],
        [2.9934e-03, 9.9701e-01],
        [5.7953e-06, 9.9999e-01],
        [7.9291e-09, 1.0000e+00],
        [9.9852e-01, 1.4816e-03],
        [1.0125e-03, 9.9899e-01],
        [3.6120e-08, 1.0000e+00],
        [5.0899e-11, 1.0000e+00],
        [6.7081e-08, 1.0000e+00],
        [2.0238e-09, 1.0000e+00],
        [4.0565e-07, 1.0000e+00],
        [3.6560e-06, 1.0000e+00],
        [6.1868e-06, 9.9999e-01],
        [1.7942e-08, 1.0000e+00],
        [9.9552e-01, 4.4760e-03],
        [9.7833e-06, 9.9999e-01],
        [3.8819e-07, 1.0000e+00],
        [5.8087e-01, 4.1913e-01],
        [1.4866e-03, 9.9851e-01],
        [9.9240e-01, 7.6028e-03],
        [8.8209e-01, 1.1791e-01],
        [4.3854e-06, 1.0000e+00],
        [1.9470e-08, 1.0000e+00],
        [9.9613e-01, 3.8747e-03],
        [1.7805e-02, 9.8220e-01],
        [5.5832e-06, 9.9999e-01],
        [1.8337e-06, 1.0000e+00],
        [7.2899e-05, 9.9993e-01],
        [9.9954e-01, 4.6496e-04],
        [5.8448e-01, 4.1552e-01],
        [2.0445e-09, 1.0000e+00],
        [4.5947e-04, 9.9954e-01],
        [9.9988e-01, 1.2150e-04],
        [1.3176e-06, 1.0000e+00],
        [2.2318e-06, 1.0000e+00],
        [1.7317e-04, 9.9983e-01],
        [2.6550e-07, 1.0000e+00],
        [9.9979e-01, 2.0612e-04],
        [1.2217e-06, 1.0000e+00],
        [9.9939e-01, 6.1187e-04],
        [5.4847e-08, 1.0000e+00],
        [4.9457e-05, 9.9995e-01],
        [2.4774e-06, 1.0000e+00],
        [1.0081e-06, 1.0000e+00],
        [3.0890e-07, 1.0000e+00],
        [2.3273e-06, 1.0000e+00],
        [1.0940e-05, 9.9999e-01],
        [4.5609e-05, 9.9995e-01],
        [6.0258e-05, 9.9994e-01],
        [9.9970e-01, 3.0105e-04],
        [9.9955e-01, 4.4639e-04],
        [1.6008e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2065e-02, 9.8794e-01],
        [1.9452e-04, 9.9981e-01],
        [4.7347e-07, 1.0000e+00],
        [2.4383e-04, 9.9976e-01],
        [6.3876e-07, 1.0000e+00],
        [2.8739e-05, 9.9997e-01],
        [1.8252e-05, 9.9998e-01],
        [8.2254e-07, 1.0000e+00],
        [2.8401e-08, 1.0000e+00],
        [8.0660e-05, 9.9992e-01],
        [8.4973e-07, 1.0000e+00],
        [9.9945e-01, 5.4840e-04],
        [9.9949e-01, 5.0761e-04],
        [3.4976e-03, 9.9650e-01],
        [8.5221e-06, 9.9999e-01],
        [2.8458e-05, 9.9997e-01],
        [2.3313e-04, 9.9977e-01],
        [7.7896e-06, 9.9999e-01],
        [1.1120e-09, 1.0000e+00],
        [1.1140e-08, 1.0000e+00],
        [1.9556e-04, 9.9980e-01],
        [2.6222e-05, 9.9997e-01],
        [9.9876e-01, 1.2395e-03],
        [1.9714e-06, 1.0000e+00],
        [5.2893e-04, 9.9947e-01],
        [1.4792e-07, 1.0000e+00],
        [9.9971e-01, 2.8604e-04],
        [1.6836e-07, 1.0000e+00],
        [9.9951e-01, 4.9022e-04],
        [1.7612e-04, 9.9982e-01],
        [2.4450e-05, 9.9998e-01],
        [1.8272e-05, 9.9998e-01],
        [3.6779e-05, 9.9996e-01],
        [3.0296e-08, 1.0000e+00],
        [6.8512e-08, 1.0000e+00],
        [2.3380e-05, 9.9998e-01],
        [1.9994e-05, 9.9998e-01],
        [1.2536e-04, 9.9987e-01],
        [9.7860e-01, 2.1401e-02],
        [1.0107e-06, 1.0000e+00],
        [1.0000e+00, 5.4735e-07],
        [9.9994e-01, 5.9500e-05],
        [8.8815e-09, 1.0000e+00],
        [7.1890e-01, 2.8110e-01],
        [2.0308e-08, 1.0000e+00],
        [3.7304e-05, 9.9996e-01],
        [1.2635e-04, 9.9987e-01],
        [2.6857e-06, 1.0000e+00],
        [9.1393e-02, 9.0861e-01],
        [1.8573e-01, 8.1427e-01],
        [1.1987e-04, 9.9988e-01],
        [5.8250e-08, 1.0000e+00],
        [3.2500e-05, 9.9997e-01],
        [9.9982e-01, 1.8484e-04],
        [1.1957e-08, 1.0000e+00],
        [2.5013e-07, 1.0000e+00],
        [7.5840e-07, 1.0000e+00],
        [5.0155e-01, 4.9845e-01],
        [2.2113e-08, 1.0000e+00],
        [3.6152e-05, 9.9996e-01],
        [4.8661e-06, 1.0000e+00],
        [3.4100e-06, 1.0000e+00],
        [5.0856e-06, 9.9999e-01],
        [4.1424e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.2601e-07, 1.0000e+00],
        [2.7348e-06, 1.0000e+00],
        [3.0351e-04, 9.9970e-01],
        [4.9412e-05, 9.9995e-01],
        [1.9328e-07, 1.0000e+00],
        [4.8914e-06, 1.0000e+00],
        [8.2083e-07, 1.0000e+00],
        [3.2399e-07, 1.0000e+00],
        [1.4030e-07, 1.0000e+00],
        [1.1620e-04, 9.9988e-01],
        [1.7648e-05, 9.9998e-01],
        [9.9949e-01, 5.1344e-04],
        [6.3999e-04, 9.9936e-01],
        [5.7636e-06, 9.9999e-01],
        [3.6234e-03, 9.9638e-01],
        [3.7267e-07, 1.0000e+00],
        [7.9100e-04, 9.9921e-01],
        [9.3615e-07, 1.0000e+00],
        [2.5316e-06, 1.0000e+00],
        [7.1788e-06, 9.9999e-01],
        [2.4281e-06, 1.0000e+00],
        [6.2835e-06, 9.9999e-01],
        [9.9971e-01, 2.9009e-04],
        [3.0932e-04, 9.9969e-01],
        [4.0578e-05, 9.9996e-01],
        [5.8501e-05, 9.9994e-01],
        [9.9942e-01, 5.7752e-04],
        [5.8008e-07, 1.0000e+00],
        [4.4368e-01, 5.5632e-01],
        [3.4047e-03, 9.9660e-01],
        [1.7170e-06, 1.0000e+00],
        [9.9964e-01, 3.6053e-04],
        [1.0124e-04, 9.9990e-01],
        [3.4341e-03, 9.9657e-01],
        [3.1165e-05, 9.9997e-01],
        [5.0893e-02, 9.4911e-01],
        [2.2539e-07, 1.0000e+00],
        [9.9991e-01, 9.0650e-05],
        [9.9829e-01, 1.7129e-03],
        [1.0972e-06, 1.0000e+00],
        [8.0678e-09, 1.0000e+00],
        [9.9995e-01, 4.6875e-05],
        [9.9412e-01, 5.8849e-03],
        [3.1424e-08, 1.0000e+00],
        [1.4387e-02, 9.8561e-01],
        [9.9969e-01, 3.0743e-04],
        [2.3027e-05, 9.9998e-01],
        [7.6285e-06, 9.9999e-01],
        [9.7714e-01, 2.2861e-02],
        [1.6915e-04, 9.9983e-01],
        [5.6839e-05, 9.9994e-01],
        [3.1803e-08, 1.0000e+00],
        [4.5612e-05, 9.9995e-01],
        [9.9983e-01, 1.6807e-04],
        [7.4731e-05, 9.9993e-01],
        [3.0412e-04, 9.9970e-01],
        [1.4129e-07, 1.0000e+00],
        [9.7728e-01, 2.2723e-02],
        [3.1457e-04, 9.9969e-01],
        [9.4107e-01, 5.8934e-02],
        [4.8502e-07, 1.0000e+00],
        [9.9547e-01, 4.5304e-03],
        [1.1634e-03, 9.9884e-01],
        [8.7546e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8729e-01, 1.2713e-02],
        [9.1061e-04, 9.9909e-01],
        [3.1581e-05, 9.9997e-01],
        [6.8737e-02, 9.3126e-01],
        [6.4822e-04, 9.9935e-01],
        [9.9979e-01, 2.1269e-04],
        [9.3294e-06, 9.9999e-01],
        [9.7498e-01, 2.5016e-02],
        [1.0402e-06, 1.0000e+00],
        [2.4683e-03, 9.9753e-01],
        [5.3310e-03, 9.9467e-01],
        [7.5675e-07, 1.0000e+00],
        [9.3979e-08, 1.0000e+00],
        [2.2590e-05, 9.9998e-01],
        [9.9967e-01, 3.3287e-04],
        [1.0640e-02, 9.8936e-01],
        [2.8541e-05, 9.9997e-01],
        [9.0167e-07, 1.0000e+00],
        [7.4408e-01, 2.5592e-01],
        [3.0057e-06, 1.0000e+00],
        [4.7744e-08, 1.0000e+00],
        [9.9988e-01, 1.1989e-04],
        [2.5197e-07, 1.0000e+00],
        [2.4389e-03, 9.9756e-01],
        [3.0847e-03, 9.9692e-01],
        [9.9721e-01, 2.7854e-03],
        [9.9936e-01, 6.3551e-04],
        [1.4410e-05, 9.9999e-01],
        [3.9967e-03, 9.9600e-01],
        [9.9425e-01, 5.7526e-03],
        [8.3571e-06, 9.9999e-01],
        [1.3673e-01, 8.6327e-01],
        [3.5930e-07, 1.0000e+00],
        [2.7766e-04, 9.9972e-01],
        [6.6012e-08, 1.0000e+00],
        [2.6062e-02, 9.7394e-01],
        [2.2416e-06, 1.0000e+00],
        [3.6928e-07, 1.0000e+00],
        [1.8092e-04, 9.9982e-01],
        [9.9949e-01, 5.1360e-04],
        [2.8690e-04, 9.9971e-01],
        [1.9914e-06, 1.0000e+00],
        [8.2778e-09, 1.0000e+00],
        [9.8150e-08, 1.0000e+00],
        [9.9965e-01, 3.4963e-04],
        [9.9943e-01, 5.6862e-04],
        [3.1040e-04, 9.9969e-01],
        [9.6606e-07, 1.0000e+00],
        [9.9982e-01, 1.7756e-04],
        [5.5238e-03, 9.9448e-01],
        [1.6987e-06, 1.0000e+00],
        [9.9487e-01, 5.1261e-03],
        [4.6010e-04, 9.9954e-01],
        [1.9823e-08, 1.0000e+00],
        [3.8513e-05, 9.9996e-01],
        [4.4275e-07, 1.0000e+00],
        [9.9969e-01, 3.0882e-04],
        [4.5022e-04, 9.9955e-01],
        [3.2834e-09, 1.0000e+00],
        [9.9881e-01, 1.1862e-03],
        [2.2189e-07, 1.0000e+00],
        [4.4891e-06, 1.0000e+00],
        [4.3580e-09, 1.0000e+00],
        [1.1315e-04, 9.9989e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8023e-01, 1.9773e-02],
        [3.5471e-05, 9.9996e-01],
        [1.0353e-04, 9.9990e-01],
        [6.6398e-05, 9.9993e-01],
        [6.3502e-05, 9.9994e-01],
        [9.9902e-01, 9.7939e-04],
        [8.0211e-10, 1.0000e+00],
        [5.8736e-04, 9.9941e-01],
        [4.5478e-05, 9.9995e-01],
        [7.1663e-08, 1.0000e+00],
        [9.9996e-01, 3.6593e-05],
        [8.5334e-07, 1.0000e+00],
        [1.4219e-03, 9.9858e-01],
        [6.3070e-10, 1.0000e+00],
        [8.6355e-05, 9.9991e-01],
        [2.8025e-03, 9.9720e-01],
        [9.9985e-01, 1.4931e-04],
        [1.4003e-03, 9.9860e-01],
        [3.1738e-05, 9.9997e-01],
        [8.3065e-02, 9.1693e-01],
        [9.9997e-01, 3.1440e-05],
        [6.7023e-06, 9.9999e-01],
        [1.5306e-02, 9.8469e-01],
        [4.6016e-07, 1.0000e+00],
        [7.2357e-06, 9.9999e-01],
        [6.1421e-05, 9.9994e-01],
        [1.8311e-04, 9.9982e-01],
        [4.3312e-05, 9.9996e-01],
        [1.7452e-05, 9.9998e-01],
        [1.4797e-06, 1.0000e+00],
        [4.3715e-08, 1.0000e+00],
        [3.3730e-03, 9.9663e-01],
        [2.8196e-06, 1.0000e+00],
        [1.9503e-09, 1.0000e+00],
        [9.9809e-01, 1.9145e-03],
        [1.4818e-05, 9.9999e-01],
        [9.9966e-01, 3.3917e-04],
        [2.1271e-05, 9.9998e-01],
        [9.9999e-01, 1.0510e-05],
        [4.0037e-05, 9.9996e-01],
        [2.1054e-06, 1.0000e+00],
        [2.5809e-07, 1.0000e+00],
        [1.1441e-05, 9.9999e-01],
        [1.1139e-05, 9.9999e-01],
        [1.0881e-04, 9.9989e-01],
        [3.7257e-04, 9.9963e-01],
        [8.2558e-03, 9.9174e-01],
        [2.0948e-05, 9.9998e-01],
        [8.1796e-01, 1.8204e-01],
        [2.7270e-07, 1.0000e+00],
        [6.0695e-01, 3.9305e-01],
        [2.6332e-05, 9.9997e-01],
        [7.0438e-06, 9.9999e-01],
        [1.4916e-03, 9.9851e-01],
        [1.8567e-07, 1.0000e+00],
        [5.1354e-05, 9.9995e-01],
        [1.9938e-07, 1.0000e+00],
        [9.2685e-01, 7.3154e-02],
        [9.9968e-01, 3.1989e-04],
        [1.9984e-02, 9.8002e-01],
        [3.5164e-08, 1.0000e+00],
        [2.4755e-07, 1.0000e+00],
        [1.9938e-07, 1.0000e+00],
        [9.9983e-01, 1.6609e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9938e-01, 6.2058e-04],
        [8.6120e-05, 9.9991e-01],
        [3.0309e-07, 1.0000e+00],
        [1.0692e-08, 1.0000e+00],
        [4.3394e-06, 1.0000e+00],
        [3.5852e-06, 1.0000e+00],
        [9.9815e-01, 1.8463e-03],
        [2.5295e-08, 1.0000e+00],
        [7.9666e-05, 9.9992e-01],
        [9.9463e-01, 5.3691e-03],
        [9.9974e-01, 2.5554e-04],
        [8.5308e-01, 1.4692e-01],
        [2.7873e-01, 7.2127e-01],
        [9.9840e-01, 1.6008e-03],
        [9.9988e-01, 1.2126e-04],
        [8.3211e-04, 9.9917e-01],
        [2.9886e-05, 9.9997e-01],
        [4.8586e-04, 9.9951e-01],
        [7.7196e-01, 2.2804e-01],
        [2.4496e-02, 9.7550e-01],
        [4.1368e-04, 9.9959e-01],
        [5.5910e-05, 9.9994e-01],
        [9.9994e-01, 5.7682e-05],
        [9.9981e-01, 1.9500e-04],
        [2.9044e-02, 9.7096e-01],
        [9.9763e-01, 2.3695e-03],
        [9.5979e-05, 9.9990e-01],
        [9.9939e-01, 6.0581e-04],
        [1.4568e-04, 9.9985e-01],
        [2.8873e-02, 9.7113e-01],
        [1.0682e-03, 9.9893e-01],
        [9.7402e-01, 2.5978e-02],
        [9.9984e-01, 1.6079e-04],
        [9.9200e-01, 8.0001e-03],
        [9.9962e-01, 3.8435e-04],
        [1.2876e-06, 1.0000e+00],
        [9.9970e-01, 3.0117e-04],
        [9.9850e-02, 9.0015e-01],
        [8.4905e-04, 9.9915e-01],
        [3.7155e-04, 9.9963e-01],
        [2.4374e-04, 9.9976e-01],
        [4.4781e-06, 1.0000e+00],
        [9.9912e-01, 8.7629e-04],
        [3.3204e-05, 9.9997e-01],
        [4.7578e-06, 1.0000e+00],
        [9.9977e-01, 2.2942e-04],
        [9.9978e-01, 2.2168e-04],
        [1.5708e-07, 1.0000e+00],
        [9.9872e-01, 1.2829e-03],
        [9.4021e-05, 9.9991e-01],
        [4.9091e-04, 9.9951e-01],
        [1.1007e-02, 9.8899e-01],
        [1.0198e-14, 1.0000e+00],
        [1.6386e-07, 1.0000e+00],
        [9.9965e-01, 3.5272e-04],
        [8.7873e-07, 1.0000e+00],
        [5.3436e-06, 9.9999e-01],
        [1.4053e-02, 9.8595e-01],
        [6.3310e-09, 1.0000e+00],
        [5.5317e-07, 1.0000e+00],
        [1.8886e-06, 1.0000e+00],
        [1.4099e-04, 9.9986e-01],
        [9.9989e-01, 1.1028e-04],
        [1.9270e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.7637e-02, 9.7236e-01],
        [9.9254e-01, 7.4599e-03],
        [9.9967e-01, 3.3416e-04],
        [1.6837e-04, 9.9983e-01],
        [9.9978e-01, 2.1955e-04],
        [9.9989e-01, 1.1150e-04],
        [1.2446e-02, 9.8755e-01],
        [2.0551e-04, 9.9979e-01],
        [9.3624e-01, 6.3762e-02],
        [3.3356e-06, 1.0000e+00],
        [9.9987e-01, 1.3253e-04],
        [9.9963e-01, 3.6875e-04],
        [3.4554e-07, 1.0000e+00],
        [9.9038e-01, 9.6152e-03],
        [3.5447e-02, 9.6455e-01],
        [9.9797e-01, 2.0316e-03],
        [3.3895e-05, 9.9997e-01],
        [9.5665e-07, 1.0000e+00],
        [3.4554e-07, 1.0000e+00],
        [4.0927e-05, 9.9996e-01],
        [9.9558e-01, 4.4183e-03],
        [9.9979e-01, 2.1383e-04],
        [5.5015e-05, 9.9994e-01],
        [1.4580e-05, 9.9999e-01],
        [2.0023e-07, 1.0000e+00],
        [2.1196e-03, 9.9788e-01],
        [9.9986e-01, 1.4093e-04],
        [3.1328e-06, 1.0000e+00],
        [9.8577e-01, 1.4227e-02],
        [1.3848e-02, 9.8615e-01],
        [4.0078e-04, 9.9960e-01],
        [4.1079e-05, 9.9996e-01],
        [4.8189e-05, 9.9995e-01],
        [8.4860e-07, 1.0000e+00],
        [9.9989e-01, 1.1110e-04],
        [9.1600e-06, 9.9999e-01],
        [1.9579e-03, 9.9804e-01],
        [1.4901e-04, 9.9985e-01],
        [1.4789e-04, 9.9985e-01],
        [9.9988e-01, 1.1614e-04],
        [3.6209e-07, 1.0000e+00],
        [9.9949e-01, 5.1417e-04],
        [1.1811e-05, 9.9999e-01],
        [9.9964e-01, 3.6427e-04],
        [9.9904e-01, 9.5738e-04],
        [5.7891e-03, 9.9421e-01],
        [6.1526e-08, 1.0000e+00],
        [3.3746e-04, 9.9966e-01],
        [9.9957e-01, 4.3296e-04],
        [9.9982e-01, 1.8434e-04],
        [7.4620e-07, 1.0000e+00],
        [5.6464e-06, 9.9999e-01],
        [1.7872e-04, 9.9982e-01],
        [9.9988e-01, 1.1704e-04],
        [2.4847e-04, 9.9975e-01],
        [2.4788e-02, 9.7521e-01],
        [8.8279e-06, 9.9999e-01],
        [1.4640e-05, 9.9999e-01],
        [3.7724e-03, 9.9623e-01],
        [9.5248e-05, 9.9990e-01],
        [1.5974e-05, 9.9998e-01],
        [1.4680e-01, 8.5320e-01],
        [6.9801e-06, 9.9999e-01],
        [3.4581e-04, 9.9965e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0128e-05, 9.9998e-01],
        [9.9952e-01, 4.7898e-04],
        [9.9971e-01, 2.9233e-04],
        [2.1155e-06, 1.0000e+00],
        [5.0279e-03, 9.9497e-01],
        [4.1479e-07, 1.0000e+00],
        [8.8154e-02, 9.1185e-01],
        [2.3533e-04, 9.9976e-01],
        [3.2041e-05, 9.9997e-01],
        [9.9997e-01, 2.5701e-05],
        [8.5325e-01, 1.4675e-01],
        [1.1076e-07, 1.0000e+00],
        [1.3117e-08, 1.0000e+00],
        [3.0185e-04, 9.9970e-01],
        [9.9955e-01, 4.5303e-04],
        [9.9988e-01, 1.1639e-04],
        [6.4643e-07, 1.0000e+00],
        [5.5289e-01, 4.4711e-01],
        [5.8238e-01, 4.1762e-01],
        [4.0798e-04, 9.9959e-01],
        [9.9992e-01, 7.7294e-05],
        [3.6105e-03, 9.9639e-01],
        [3.2577e-09, 1.0000e+00],
        [5.1331e-05, 9.9995e-01],
        [4.1997e-05, 9.9996e-01],
        [9.9977e-01, 2.2838e-04],
        [9.9966e-01, 3.4052e-04],
        [9.9993e-01, 6.5807e-05],
        [9.9987e-01, 1.2767e-04],
        [1.7882e-04, 9.9982e-01],
        [3.8892e-01, 6.1108e-01],
        [2.6504e-05, 9.9997e-01],
        [9.9977e-01, 2.2778e-04],
        [7.9643e-09, 1.0000e+00],
        [9.9960e-01, 3.9590e-04],
        [4.4478e-08, 1.0000e+00],
        [9.9990e-01, 9.5030e-05],
        [2.6669e-02, 9.7333e-01],
        [2.1237e-07, 1.0000e+00],
        [5.8789e-06, 9.9999e-01],
        [1.3216e-03, 9.9868e-01],
        [4.4966e-03, 9.9550e-01],
        [7.7421e-01, 2.2579e-01],
        [1.8751e-07, 1.0000e+00],
        [2.3631e-02, 9.7637e-01],
        [9.9979e-01, 2.1148e-04],
        [9.9982e-01, 1.8155e-04],
        [1.7200e-07, 1.0000e+00],
        [2.3867e-06, 1.0000e+00],
        [9.9996e-01, 4.4524e-05],
        [5.3300e-06, 9.9999e-01],
        [1.3387e-05, 9.9999e-01],
        [7.8671e-03, 9.9213e-01],
        [8.9000e-01, 1.1000e-01],
        [3.9489e-04, 9.9961e-01],
        [6.2800e-04, 9.9937e-01],
        [2.0398e-05, 9.9998e-01],
        [2.8061e-06, 1.0000e+00],
        [5.6974e-02, 9.4303e-01],
        [7.6934e-06, 9.9999e-01],
        [9.0106e-05, 9.9991e-01],
        [1.4196e-05, 9.9999e-01],
        [9.9992e-01, 8.3641e-05],
        [9.9953e-01, 4.7002e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[8.8719e-06, 9.9999e-01],
        [3.7619e-03, 9.9624e-01],
        [3.2886e-05, 9.9997e-01],
        [9.9980e-01, 2.0072e-04],
        [9.9986e-01, 1.3899e-04],
        [2.6622e-03, 9.9734e-01],
        [8.9753e-07, 1.0000e+00],
        [2.8877e-05, 9.9997e-01],
        [9.9996e-01, 3.9530e-05],
        [9.9966e-01, 3.4463e-04],
        [4.5589e-04, 9.9954e-01],
        [9.9947e-01, 5.2545e-04],
        [4.5093e-03, 9.9549e-01],
        [2.1457e-04, 9.9979e-01],
        [3.1956e-03, 9.9680e-01],
        [1.9703e-04, 9.9980e-01],
        [2.5548e-04, 9.9974e-01],
        [9.9992e-01, 8.0137e-05],
        [9.9965e-01, 3.5134e-04],
        [2.8175e-07, 1.0000e+00],
        [9.9908e-01, 9.2444e-04],
        [4.6709e-05, 9.9995e-01],
        [9.9983e-01, 1.7156e-04],
        [2.2760e-03, 9.9772e-01],
        [1.6504e-05, 9.9998e-01],
        [2.1749e-02, 9.7825e-01],
        [5.1641e-08, 1.0000e+00],
        [1.0299e-07, 1.0000e+00],
        [2.7591e-02, 9.7241e-01],
        [3.0445e-06, 1.0000e+00],
        [9.9989e-01, 1.0901e-04],
        [1.1616e-03, 9.9884e-01],
        [3.6288e-05, 9.9996e-01],
        [2.4185e-04, 9.9976e-01],
        [3.5611e-05, 9.9996e-01],
        [9.5493e-03, 9.9045e-01],
        [2.3021e-01, 7.6979e-01],
        [3.3534e-06, 1.0000e+00],
        [3.7965e-04, 9.9962e-01],
        [9.9905e-01, 9.4924e-04],
        [1.1930e-01, 8.8070e-01],
        [9.9992e-01, 7.7479e-05],
        [6.4986e-05, 9.9994e-01],
        [1.6139e-08, 1.0000e+00],
        [9.9983e-01, 1.7281e-04],
        [1.0156e-07, 1.0000e+00],
        [9.9982e-01, 1.7848e-04],
        [4.9078e-07, 1.0000e+00],
        [6.1305e-04, 9.9939e-01],
        [7.9324e-07, 1.0000e+00],
        [1.1271e-05, 9.9999e-01],
        [1.6272e-08, 1.0000e+00],
        [9.9958e-01, 4.1660e-04],
        [2.3403e-09, 1.0000e+00],
        [9.9935e-01, 6.5294e-04],
        [2.3076e-06, 1.0000e+00],
        [3.0534e-06, 1.0000e+00],
        [4.5903e-05, 9.9995e-01],
        [9.0348e-05, 9.9991e-01],
        [7.1668e-03, 9.9283e-01],
        [9.9911e-01, 8.9408e-04],
        [9.3891e-01, 6.1093e-02],
        [2.1460e-06, 1.0000e+00],
        [5.2417e-01, 4.7583e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9909e-01, 9.1088e-04],
        [4.0888e-08, 1.0000e+00],
        [2.8851e-04, 9.9971e-01],
        [1.6371e-06, 1.0000e+00],
        [1.3982e-07, 1.0000e+00],
        [9.9990e-01, 9.7571e-05],
        [2.7808e-03, 9.9722e-01],
        [1.0758e-06, 1.0000e+00],
        [5.2903e-04, 9.9947e-01],
        [1.6470e-07, 1.0000e+00],
        [1.6263e-07, 1.0000e+00],
        [1.9594e-01, 8.0406e-01],
        [7.5970e-04, 9.9924e-01],
        [9.9891e-01, 1.0898e-03],
        [7.6293e-06, 9.9999e-01],
        [7.7079e-06, 9.9999e-01],
        [1.4357e-06, 1.0000e+00],
        [1.6837e-03, 9.9832e-01],
        [2.8459e-09, 1.0000e+00],
        [8.4952e-11, 1.0000e+00],
        [2.0515e-08, 1.0000e+00],
        [8.4651e-01, 1.5349e-01],
        [9.9981e-01, 1.9407e-04],
        [9.9985e-01, 1.5209e-04],
        [1.2451e-03, 9.9875e-01],
        [9.9371e-03, 9.9006e-01],
        [1.0857e-01, 8.9143e-01],
        [9.8655e-05, 9.9990e-01],
        [1.1485e-04, 9.9989e-01],
        [9.9970e-01, 3.0138e-04],
        [2.9255e-05, 9.9997e-01],
        [4.8742e-04, 9.9951e-01],
        [9.9854e-01, 1.4566e-03],
        [1.1363e-03, 9.9886e-01],
        [1.1082e-06, 1.0000e+00],
        [6.8505e-05, 9.9993e-01],
        [9.3092e-09, 1.0000e+00],
        [2.6887e-03, 9.9731e-01],
        [5.8597e-04, 9.9941e-01],
        [2.3671e-03, 9.9763e-01],
        [6.4011e-02, 9.3599e-01],
        [1.4289e-06, 1.0000e+00],
        [1.8493e-02, 9.8151e-01],
        [6.1435e-07, 1.0000e+00],
        [5.2236e-06, 9.9999e-01],
        [4.1818e-03, 9.9582e-01],
        [4.5044e-04, 9.9955e-01],
        [7.3343e-06, 9.9999e-01],
        [9.9167e-01, 8.3287e-03],
        [6.8827e-06, 9.9999e-01],
        [4.5337e-07, 1.0000e+00],
        [9.9950e-01, 5.0069e-04],
        [1.8680e-05, 9.9998e-01],
        [9.9974e-01, 2.5633e-04],
        [1.3545e-09, 1.0000e+00],
        [2.0444e-02, 9.7956e-01],
        [5.1357e-06, 9.9999e-01],
        [1.4011e-06, 1.0000e+00],
        [2.8489e-08, 1.0000e+00],
        [1.2518e-02, 9.8748e-01],
        [4.1935e-08, 1.0000e+00],
        [4.7993e-06, 1.0000e+00],
        [2.9617e-03, 9.9704e-01],
        [9.8795e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.2654e-08, 1.0000e+00],
        [3.0579e-05, 9.9997e-01],
        [1.3511e-08, 1.0000e+00],
        [9.9990e-01, 1.0331e-04],
        [3.0298e-06, 1.0000e+00],
        [4.9136e-05, 9.9995e-01],
        [2.7498e-07, 1.0000e+00],
        [6.4125e-01, 3.5875e-01],
        [3.8984e-06, 1.0000e+00],
        [6.7247e-06, 9.9999e-01],
        [4.2055e-07, 1.0000e+00],
        [1.4141e-07, 1.0000e+00],
        [4.4834e-05, 9.9996e-01],
        [9.5444e-02, 9.0456e-01],
        [9.7626e-01, 2.3744e-02],
        [2.0260e-07, 1.0000e+00],
        [1.4682e-06, 1.0000e+00],
        [2.1059e-06, 1.0000e+00],
        [9.9970e-01, 2.9559e-04],
        [4.5412e-01, 5.4588e-01],
        [2.3522e-05, 9.9998e-01],
        [9.9997e-01, 2.8448e-05],
        [1.8858e-02, 9.8114e-01],
        [9.9965e-01, 3.5400e-04],
        [8.6333e-08, 1.0000e+00],
        [1.2339e-04, 9.9988e-01],
        [2.9484e-06, 1.0000e+00],
        [9.9922e-01, 7.7706e-04],
        [1.1815e-08, 1.0000e+00],
        [9.9986e-01, 1.3993e-04],
        [8.8376e-01, 1.1624e-01],
        [5.8486e-06, 9.9999e-01],
        [2.5250e-05, 9.9997e-01],
        [1.5828e-09, 1.0000e+00],
        [1.5155e-06, 1.0000e+00],
        [1.4126e-04, 9.9986e-01],
        [9.9946e-01, 5.4081e-04],
        [4.1518e-01, 5.8482e-01],
        [1.6217e-03, 9.9838e-01],
        [4.5127e-08, 1.0000e+00],
        [8.9751e-07, 1.0000e+00],
        [2.4353e-05, 9.9998e-01],
        [4.2231e-02, 9.5777e-01],
        [3.9728e-06, 1.0000e+00],
        [4.5596e-05, 9.9995e-01],
        [1.9489e-05, 9.9998e-01],
        [1.7474e-02, 9.8253e-01],
        [6.3656e-02, 9.3634e-01],
        [1.5901e-04, 9.9984e-01],
        [5.4527e-06, 9.9999e-01],
        [9.9491e-01, 5.0865e-03],
        [9.1555e-03, 9.9084e-01],
        [1.2077e-04, 9.9988e-01],
        [6.5606e-02, 9.3439e-01],
        [9.9113e-01, 8.8718e-03],
        [9.3795e-02, 9.0620e-01],
        [9.9990e-01, 1.0105e-04],
        [9.9756e-01, 2.4356e-03],
        [4.2221e-05, 9.9996e-01],
        [9.9972e-01, 2.8099e-04],
        [1.0482e-04, 9.9990e-01],
        [9.1555e-03, 9.9084e-01],
        [2.7417e-08, 1.0000e+00],
        [1.4829e-04, 9.9985e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5283e-03, 9.9647e-01],
        [2.2352e-07, 1.0000e+00],
        [9.9795e-01, 2.0527e-03],
        [3.0066e-09, 1.0000e+00],
        [3.6061e-05, 9.9996e-01],
        [2.3307e-06, 1.0000e+00],
        [1.3401e-05, 9.9999e-01],
        [2.0344e-03, 9.9797e-01],
        [4.5157e-08, 1.0000e+00],
        [1.7192e-07, 1.0000e+00],
        [2.6207e-07, 1.0000e+00],
        [8.3637e-01, 1.6363e-01],
        [1.4889e-06, 1.0000e+00],
        [9.6651e-09, 1.0000e+00],
        [7.2931e-11, 1.0000e+00],
        [3.3051e-05, 9.9997e-01],
        [6.8545e-04, 9.9931e-01],
        [6.2801e-01, 3.7199e-01],
        [9.7262e-06, 9.9999e-01],
        [9.9982e-01, 1.7944e-04],
        [3.0073e-09, 1.0000e+00],
        [9.9750e-01, 2.4999e-03],
        [1.5665e-07, 1.0000e+00],
        [3.3321e-09, 1.0000e+00],
        [6.0338e-01, 3.9662e-01],
        [9.5099e-01, 4.9009e-02],
        [2.8800e-05, 9.9997e-01],
        [9.9954e-01, 4.6414e-04],
        [6.8354e-07, 1.0000e+00],
        [1.4397e-06, 1.0000e+00],
        [9.7593e-01, 2.4073e-02],
        [1.9792e-04, 9.9980e-01],
        [1.3785e-06, 1.0000e+00],
        [1.5071e-08, 1.0000e+00],
        [9.9958e-01, 4.2315e-04],
        [1.0045e-04, 9.9990e-01],
        [9.9954e-01, 4.5531e-04],
        [9.9914e-01, 8.5700e-04],
        [9.9914e-01, 8.5828e-04],
        [7.0365e-06, 9.9999e-01],
        [9.9792e-01, 2.0848e-03],
        [8.3563e-03, 9.9164e-01],
        [9.9981e-01, 1.8556e-04],
        [2.8040e-06, 1.0000e+00],
        [1.3574e-06, 1.0000e+00],
        [3.6795e-07, 1.0000e+00],
        [4.0143e-05, 9.9996e-01],
        [2.1472e-07, 1.0000e+00],
        [2.9895e-08, 1.0000e+00],
        [9.9996e-01, 3.6722e-05],
        [1.2260e-07, 1.0000e+00],
        [9.9969e-01, 3.1280e-04],
        [3.9238e-06, 1.0000e+00],
        [4.4251e-04, 9.9956e-01],
        [2.4459e-03, 9.9755e-01],
        [5.3434e-08, 1.0000e+00],
        [2.0060e-06, 1.0000e+00],
        [5.7716e-09, 1.0000e+00],
        [1.9614e-04, 9.9980e-01],
        [5.0665e-07, 1.0000e+00],
        [9.5931e-07, 1.0000e+00],
        [9.1661e-01, 8.3392e-02],
        [1.7460e-02, 9.8254e-01],
        [9.9958e-01, 4.1880e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4900e-03, 9.9851e-01],
        [1.8337e-08, 1.0000e+00],
        [1.4524e-07, 1.0000e+00],
        [7.0893e-05, 9.9993e-01],
        [9.9973e-01, 2.6978e-04],
        [2.3062e-08, 1.0000e+00],
        [1.7271e-05, 9.9998e-01],
        [9.6833e-06, 9.9999e-01],
        [3.1689e-07, 1.0000e+00],
        [8.1934e-05, 9.9992e-01],
        [3.4550e-05, 9.9997e-01],
        [9.8082e-01, 1.9184e-02],
        [1.5921e-05, 9.9998e-01],
        [6.7125e-01, 3.2875e-01],
        [1.0865e-06, 1.0000e+00],
        [6.3653e-10, 1.0000e+00],
        [2.0520e-07, 1.0000e+00],
        [5.3828e-07, 1.0000e+00],
        [9.9955e-01, 4.5474e-04],
        [1.6580e-04, 9.9983e-01],
        [2.6963e-06, 1.0000e+00],
        [1.7476e-05, 9.9998e-01],
        [3.7665e-07, 1.0000e+00],
        [1.1702e-05, 9.9999e-01],
        [1.1735e-04, 9.9988e-01],
        [2.6180e-08, 1.0000e+00],
        [3.4606e-08, 1.0000e+00],
        [1.2279e-04, 9.9988e-01],
        [8.9581e-08, 1.0000e+00],
        [5.2426e-05, 9.9995e-01],
        [9.9838e-01, 1.6245e-03],
        [4.7232e-06, 1.0000e+00],
        [9.9908e-01, 9.2325e-04],
        [5.7500e-09, 1.0000e+00],
        [1.7256e-08, 1.0000e+00],
        [3.5193e-04, 9.9965e-01],
        [9.9802e-01, 1.9807e-03],
        [1.4737e-03, 9.9853e-01],
        [4.2425e-06, 1.0000e+00],
        [6.3634e-01, 3.6366e-01],
        [1.6428e-06, 1.0000e+00],
        [4.7279e-05, 9.9995e-01],
        [6.3925e-01, 3.6075e-01],
        [1.7092e-07, 1.0000e+00],
        [4.4383e-07, 1.0000e+00],
        [6.0500e-05, 9.9994e-01],
        [8.4877e-07, 1.0000e+00],
        [1.9959e-03, 9.9800e-01],
        [2.6112e-05, 9.9997e-01],
        [6.2190e-06, 9.9999e-01],
        [8.1560e-05, 9.9992e-01],
        [9.9959e-01, 4.1214e-04],
        [9.9979e-01, 2.1098e-04],
        [2.5191e-05, 9.9997e-01],
        [2.6759e-08, 1.0000e+00],
        [5.7559e-06, 9.9999e-01],
        [1.6379e-04, 9.9984e-01],
        [1.3285e-06, 1.0000e+00],
        [4.0226e-09, 1.0000e+00],
        [1.2166e-02, 9.8783e-01],
        [3.4099e-04, 9.9966e-01],
        [3.0432e-01, 6.9568e-01],
        [9.9993e-01, 6.7683e-05],
        [9.9947e-01, 5.3433e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9925e-01, 7.5411e-04],
        [4.4183e-07, 1.0000e+00],
        [1.2419e-05, 9.9999e-01],
        [9.1139e-09, 1.0000e+00],
        [5.9430e-04, 9.9941e-01],
        [2.4703e-03, 9.9753e-01],
        [3.8384e-04, 9.9962e-01],
        [1.4714e-09, 1.0000e+00],
        [3.0761e-05, 9.9997e-01],
        [9.7097e-01, 2.9028e-02],
        [9.9628e-01, 3.7173e-03],
        [2.2455e-05, 9.9998e-01],
        [9.9989e-01, 1.0566e-04],
        [3.0984e-06, 1.0000e+00],
        [8.6299e-04, 9.9914e-01],
        [2.0674e-08, 1.0000e+00],
        [9.9993e-01, 7.1941e-05],
        [1.3177e-02, 9.8682e-01],
        [6.3816e-01, 3.6184e-01],
        [9.7150e-01, 2.8500e-02],
        [8.8861e-07, 1.0000e+00],
        [2.8258e-10, 1.0000e+00],
        [8.6474e-09, 1.0000e+00],
        [9.9270e-01, 7.2958e-03],
        [1.7815e-04, 9.9982e-01],
        [4.1805e-06, 1.0000e+00],
        [1.9420e-09, 1.0000e+00],
        [6.1445e-03, 9.9386e-01],
        [8.1807e-07, 1.0000e+00],
        [4.0168e-03, 9.9598e-01],
        [8.5452e-01, 1.4548e-01],
        [2.4367e-06, 1.0000e+00],
        [1.1176e-04, 9.9989e-01],
        [4.3776e-06, 1.0000e+00],
        [2.8796e-05, 9.9997e-01],
        [2.4516e-01, 7.5484e-01],
        [5.2841e-10, 1.0000e+00],
        [1.8693e-04, 9.9981e-01],
        [1.5872e-05, 9.9998e-01],
        [1.3446e-07, 1.0000e+00],
        [2.3246e-07, 1.0000e+00],
        [9.9992e-01, 7.5863e-05],
        [1.7971e-07, 1.0000e+00],
        [9.9962e-01, 3.7505e-04],
        [9.9966e-01, 3.3583e-04],
        [6.9483e-04, 9.9931e-01],
        [7.8592e-06, 9.9999e-01],
        [4.9088e-05, 9.9995e-01],
        [8.6649e-06, 9.9999e-01],
        [9.9973e-01, 2.6685e-04],
        [9.9957e-01, 4.3041e-04],
        [4.3288e-08, 1.0000e+00],
        [1.8606e-08, 1.0000e+00],
        [1.4140e-05, 9.9999e-01],
        [1.1840e-06, 1.0000e+00],
        [9.8970e-01, 1.0295e-02],
        [3.6616e-07, 1.0000e+00],
        [6.6682e-06, 9.9999e-01],
        [2.9401e-07, 1.0000e+00],
        [5.0161e-08, 1.0000e+00],
        [9.9972e-01, 2.7990e-04],
        [2.2401e-04, 9.9978e-01],
        [4.8950e-03, 9.9511e-01],
        [6.1113e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0317e-03, 9.9797e-01],
        [1.4294e-08, 1.0000e+00],
        [9.9988e-01, 1.2136e-04],
        [4.6872e-05, 9.9995e-01],
        [6.7071e-05, 9.9993e-01],
        [8.5709e-05, 9.9991e-01],
        [5.7368e-04, 9.9943e-01],
        [3.3124e-07, 1.0000e+00],
        [9.9746e-01, 2.5361e-03],
        [5.7153e-09, 1.0000e+00],
        [6.7431e-08, 1.0000e+00],
        [1.7468e-04, 9.9983e-01],
        [2.5059e-07, 1.0000e+00],
        [2.1156e-09, 1.0000e+00],
        [9.9268e-01, 7.3157e-03],
        [9.6530e-06, 9.9999e-01],
        [9.2356e-01, 7.6441e-02],
        [9.9895e-01, 1.0545e-03],
        [9.8257e-04, 9.9902e-01],
        [1.7266e-04, 9.9983e-01],
        [2.0904e-07, 1.0000e+00],
        [7.7433e-07, 1.0000e+00],
        [1.0804e-07, 1.0000e+00],
        [6.0272e-10, 1.0000e+00],
        [9.7529e-06, 9.9999e-01],
        [9.9982e-01, 1.8401e-04],
        [1.9299e-06, 1.0000e+00],
        [2.2172e-03, 9.9778e-01],
        [1.4028e-08, 1.0000e+00],
        [9.9922e-01, 7.7859e-04],
        [2.3995e-07, 1.0000e+00],
        [1.4038e-07, 1.0000e+00],
        [1.8513e-06, 1.0000e+00],
        [2.2137e-05, 9.9998e-01],
        [9.9955e-01, 4.5253e-04],
        [9.9980e-01, 1.9648e-04],
        [1.4406e-05, 9.9999e-01],
        [9.9985e-01, 1.5130e-04],
        [1.7862e-06, 1.0000e+00],
        [4.2897e-02, 9.5710e-01],
        [7.0710e-08, 1.0000e+00],
        [6.7328e-04, 9.9933e-01],
        [1.5248e-05, 9.9998e-01],
        [1.4877e-04, 9.9985e-01],
        [7.0205e-01, 2.9795e-01],
        [2.0851e-01, 7.9149e-01],
        [1.8189e-05, 9.9998e-01],
        [2.8244e-09, 1.0000e+00],
        [2.5007e-05, 9.9997e-01],
        [2.1548e-08, 1.0000e+00],
        [9.9940e-01, 5.9580e-04],
        [9.9999e-01, 1.2377e-05],
        [3.0399e-09, 1.0000e+00],
        [1.0716e-01, 8.9284e-01],
        [9.9959e-01, 4.1396e-04],
        [3.6009e-06, 1.0000e+00],
        [2.2401e-04, 9.9978e-01],
        [1.8807e-01, 8.1193e-01],
        [8.4662e-07, 1.0000e+00],
        [5.5409e-02, 9.4459e-01],
        [2.9164e-09, 1.0000e+00],
        [9.9988e-01, 1.2003e-04],
        [4.4667e-08, 1.0000e+00],
        [8.8508e-01, 1.1492e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2307e-09, 1.0000e+00],
        [3.4862e-04, 9.9965e-01],
        [9.9988e-01, 1.1728e-04],
        [7.5416e-05, 9.9992e-01],
        [3.4015e-02, 9.6598e-01],
        [9.9682e-01, 3.1784e-03],
        [9.9980e-01, 1.9628e-04],
        [1.0228e-03, 9.9898e-01],
        [2.8303e-07, 1.0000e+00],
        [9.9908e-01, 9.2181e-04],
        [9.9024e-01, 9.7588e-03],
        [4.4373e-06, 1.0000e+00],
        [4.5635e-02, 9.5437e-01],
        [2.8624e-07, 1.0000e+00],
        [9.8340e-09, 1.0000e+00],
        [9.8734e-01, 1.2664e-02],
        [5.2568e-01, 4.7432e-01],
        [9.9901e-01, 9.8911e-04],
        [3.6184e-05, 9.9996e-01],
        [9.9953e-01, 4.7350e-04],
        [3.1745e-01, 6.8255e-01],
        [9.7469e-01, 2.5313e-02],
        [3.9491e-06, 1.0000e+00],
        [1.4262e-04, 9.9986e-01],
        [8.9539e-06, 9.9999e-01],
        [9.0938e-08, 1.0000e+00],
        [4.6195e-07, 1.0000e+00],
        [9.9950e-01, 5.0464e-04],
        [9.9937e-01, 6.3270e-04],
        [1.2290e-06, 1.0000e+00],
        [1.4075e-03, 9.9859e-01],
        [1.7835e-07, 1.0000e+00],
        [8.8953e-04, 9.9911e-01],
        [2.0707e-07, 1.0000e+00],
        [7.7071e-07, 1.0000e+00],
        [1.8571e-06, 1.0000e+00],
        [3.5589e-01, 6.4411e-01],
        [9.5778e-01, 4.2223e-02],
        [9.9827e-01, 1.7315e-03],
        [1.2600e-07, 1.0000e+00],
        [9.9972e-01, 2.8319e-04],
        [9.9965e-01, 3.4615e-04],
        [1.0429e-06, 1.0000e+00],
        [9.0147e-01, 9.8528e-02],
        [4.0351e-08, 1.0000e+00],
        [9.9981e-01, 1.8788e-04],
        [9.9982e-01, 1.7649e-04],
        [1.8385e-06, 1.0000e+00],
        [1.3725e-01, 8.6275e-01],
        [9.9994e-01, 5.5219e-05],
        [1.0326e-08, 1.0000e+00],
        [3.6970e-04, 9.9963e-01],
        [7.4837e-10, 1.0000e+00],
        [8.5539e-05, 9.9991e-01],
        [9.9961e-01, 3.9370e-04],
        [3.4061e-07, 1.0000e+00],
        [7.1893e-03, 9.9281e-01],
        [1.4816e-06, 1.0000e+00],
        [1.1700e-05, 9.9999e-01],
        [7.8811e-08, 1.0000e+00],
        [9.9999e-01, 7.1881e-06],
        [1.3459e-09, 1.0000e+00],
        [8.7865e-07, 1.0000e+00],
        [8.6218e-01, 1.3782e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7381e-06, 1.0000e+00],
        [3.8879e-04, 9.9961e-01],
        [8.4200e-01, 1.5800e-01],
        [9.9909e-01, 9.1279e-04],
        [9.9984e-01, 1.6118e-04],
        [1.3497e-05, 9.9999e-01],
        [3.1437e-09, 1.0000e+00],
        [7.4953e-03, 9.9250e-01],
        [2.9301e-04, 9.9971e-01],
        [1.1413e-06, 1.0000e+00],
        [6.7374e-01, 3.2626e-01],
        [3.9805e-04, 9.9960e-01],
        [9.9061e-01, 9.3854e-03],
        [9.9996e-01, 4.2810e-05],
        [9.9904e-01, 9.5711e-04],
        [1.5957e-09, 1.0000e+00],
        [7.9554e-05, 9.9992e-01],
        [2.7199e-05, 9.9997e-01],
        [1.0734e-03, 9.9893e-01],
        [3.1173e-02, 9.6883e-01],
        [3.1263e-05, 9.9997e-01],
        [7.4417e-07, 1.0000e+00],
        [1.8046e-08, 1.0000e+00],
        [7.8038e-01, 2.1962e-01],
        [2.7376e-01, 7.2624e-01],
        [5.7570e-05, 9.9994e-01],
        [3.7013e-05, 9.9996e-01],
        [9.2572e-06, 9.9999e-01],
        [1.2277e-03, 9.9877e-01],
        [4.4058e-08, 1.0000e+00],
        [9.9953e-01, 4.7236e-04],
        [3.6532e-05, 9.9996e-01],
        [6.3066e-08, 1.0000e+00],
        [9.1997e-07, 1.0000e+00],
        [1.4614e-01, 8.5386e-01],
        [2.7528e-07, 1.0000e+00],
        [1.9442e-07, 1.0000e+00],
        [4.3672e-03, 9.9563e-01],
        [1.1406e-08, 1.0000e+00],
        [3.5395e-08, 1.0000e+00],
        [1.1628e-04, 9.9988e-01],
        [9.9896e-01, 1.0417e-03],
        [2.6875e-07, 1.0000e+00],
        [2.7823e-05, 9.9997e-01],
        [9.9914e-01, 8.5911e-04],
        [2.2168e-07, 1.0000e+00],
        [1.5269e-05, 9.9998e-01],
        [4.2124e-07, 1.0000e+00],
        [9.9977e-01, 2.2639e-04],
        [1.9282e-08, 1.0000e+00],
        [2.3328e-04, 9.9977e-01],
        [9.9893e-01, 1.0685e-03],
        [1.4016e-06, 1.0000e+00],
        [3.1909e-04, 9.9968e-01],
        [9.9902e-01, 9.8205e-04],
        [1.9754e-08, 1.0000e+00],
        [1.1436e-04, 9.9989e-01],
        [4.2910e-06, 1.0000e+00],
        [6.3699e-05, 9.9994e-01],
        [5.0383e-08, 1.0000e+00],
        [2.3405e-08, 1.0000e+00],
        [1.5732e-05, 9.9998e-01],
        [2.9114e-07, 1.0000e+00],
        [2.3304e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.6387e-07, 1.0000e+00],
        [9.7168e-06, 9.9999e-01],
        [1.3813e-09, 1.0000e+00],
        [6.3682e-05, 9.9994e-01],
        [1.7963e-08, 1.0000e+00],
        [1.7573e-04, 9.9982e-01],
        [9.9967e-01, 3.2750e-04],
        [4.1154e-05, 9.9996e-01],
        [1.5471e-06, 1.0000e+00],
        [2.3925e-08, 1.0000e+00],
        [1.4072e-06, 1.0000e+00],
        [5.2128e-06, 9.9999e-01],
        [1.4629e-04, 9.9985e-01],
        [1.2117e-07, 1.0000e+00],
        [1.2479e-09, 1.0000e+00],
        [9.9180e-01, 8.2014e-03],
        [1.6975e-08, 1.0000e+00],
        [1.5540e-05, 9.9998e-01],
        [9.9986e-01, 1.3793e-04],
        [8.1378e-05, 9.9992e-01],
        [9.9970e-01, 2.9995e-04],
        [8.5916e-11, 1.0000e+00],
        [9.9964e-01, 3.5625e-04],
        [8.8663e-05, 9.9991e-01],
        [9.8496e-01, 1.5042e-02],
        [9.9983e-01, 1.7150e-04],
        [5.7025e-05, 9.9994e-01],
        [1.9392e-09, 1.0000e+00],
        [1.3288e-08, 1.0000e+00],
        [2.7087e-06, 1.0000e+00],
        [1.3024e-04, 9.9987e-01],
        [3.6979e-05, 9.9996e-01],
        [3.2027e-02, 9.6797e-01],
        [3.0930e-07, 1.0000e+00],
        [2.8312e-06, 1.0000e+00],
        [1.0700e-05, 9.9999e-01],
        [1.5409e-04, 9.9985e-01],
        [2.6387e-07, 1.0000e+00],
        [1.3605e-07, 1.0000e+00],
        [9.9647e-01, 3.5275e-03],
        [1.4527e-03, 9.9855e-01],
        [1.8708e-03, 9.9813e-01],
        [9.8451e-01, 1.5495e-02],
        [9.9970e-01, 3.0158e-04],
        [6.9282e-04, 9.9931e-01],
        [9.9995e-01, 5.1474e-05],
        [5.0263e-01, 4.9737e-01],
        [4.4812e-03, 9.9552e-01],
        [9.9617e-01, 3.8285e-03],
        [5.2668e-04, 9.9947e-01],
        [9.9991e-01, 8.9850e-05],
        [4.9705e-06, 9.9999e-01],
        [9.8686e-01, 1.3143e-02],
        [1.2400e-11, 1.0000e+00],
        [4.2399e-05, 9.9996e-01],
        [1.6738e-03, 9.9833e-01],
        [4.8859e-05, 9.9995e-01],
        [9.9862e-01, 1.3847e-03],
        [8.9858e-09, 1.0000e+00],
        [1.3551e-05, 9.9999e-01],
        [1.7879e-07, 1.0000e+00],
        [2.2782e-07, 1.0000e+00],
        [2.5330e-04, 9.9975e-01],
        [1.3902e-03, 9.9861e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5510e-06, 1.0000e+00],
        [2.4897e-05, 9.9998e-01],
        [3.5782e-05, 9.9996e-01],
        [1.1114e-05, 9.9999e-01],
        [2.5174e-03, 9.9748e-01],
        [1.4806e-07, 1.0000e+00],
        [2.3500e-06, 1.0000e+00],
        [1.8860e-05, 9.9998e-01],
        [3.4612e-06, 1.0000e+00],
        [9.0942e-07, 1.0000e+00],
        [9.2621e-09, 1.0000e+00],
        [3.2121e-06, 1.0000e+00],
        [2.0803e-08, 1.0000e+00],
        [3.3014e-08, 1.0000e+00],
        [2.9968e-11, 1.0000e+00],
        [9.9935e-01, 6.4863e-04],
        [3.1042e-05, 9.9997e-01],
        [1.3966e-08, 1.0000e+00],
        [4.3878e-06, 1.0000e+00],
        [1.6222e-06, 1.0000e+00],
        [1.4117e-03, 9.9859e-01],
        [8.7733e-04, 9.9912e-01],
        [9.9937e-01, 6.3099e-04],
        [1.0428e-09, 1.0000e+00],
        [2.3314e-04, 9.9977e-01],
        [7.0973e-09, 1.0000e+00],
        [5.1999e-05, 9.9995e-01],
        [2.6570e-08, 1.0000e+00],
        [3.9892e-05, 9.9996e-01],
        [1.0649e-10, 1.0000e+00],
        [6.8651e-05, 9.9993e-01],
        [5.2495e-06, 9.9999e-01],
        [5.9707e-05, 9.9994e-01],
        [1.1455e-04, 9.9989e-01],
        [7.1400e-08, 1.0000e+00],
        [3.7957e-01, 6.2043e-01],
        [1.6560e-05, 9.9998e-01],
        [8.3441e-01, 1.6559e-01],
        [9.9839e-01, 1.6134e-03],
        [5.3694e-03, 9.9463e-01],
        [9.9994e-01, 6.0792e-05],
        [7.1004e-09, 1.0000e+00],
        [1.3605e-02, 9.8639e-01],
        [1.8114e-06, 1.0000e+00],
        [9.9956e-01, 4.4489e-04],
        [1.7746e-01, 8.2254e-01],
        [3.0541e-05, 9.9997e-01],
        [9.8921e-01, 1.0791e-02],
        [2.3455e-03, 9.9765e-01],
        [8.2272e-06, 9.9999e-01],
        [5.9531e-04, 9.9940e-01],
        [2.1732e-05, 9.9998e-01],
        [1.2167e-10, 1.0000e+00],
        [2.4457e-07, 1.0000e+00],
        [2.5453e-06, 1.0000e+00],
        [9.9661e-01, 3.3853e-03],
        [1.4847e-04, 9.9985e-01],
        [8.7533e-01, 1.2467e-01],
        [5.6551e-10, 1.0000e+00],
        [9.9956e-01, 4.4135e-04],
        [2.1077e-06, 1.0000e+00],
        [9.9977e-01, 2.3030e-04],
        [1.9146e-03, 9.9809e-01],
        [2.4153e-02, 9.7585e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9005e-01, 9.9477e-03],
        [3.8872e-08, 1.0000e+00],
        [3.3236e-09, 1.0000e+00],
        [1.4453e-06, 1.0000e+00],
        [1.9724e-02, 9.8028e-01],
        [1.8774e-06, 1.0000e+00],
        [9.9953e-01, 4.6589e-04],
        [1.5419e-05, 9.9998e-01],
        [1.5152e-07, 1.0000e+00],
        [2.9073e-02, 9.7093e-01],
        [2.4855e-06, 1.0000e+00],
        [1.0469e-05, 9.9999e-01],
        [2.8011e-09, 1.0000e+00],
        [5.9997e-01, 4.0003e-01],
        [1.2714e-04, 9.9987e-01],
        [1.2779e-06, 1.0000e+00],
        [1.9392e-04, 9.9981e-01],
        [2.3090e-05, 9.9998e-01],
        [9.8651e-01, 1.3495e-02],
        [9.9958e-01, 4.1827e-04],
        [1.0952e-05, 9.9999e-01],
        [4.1350e-08, 1.0000e+00],
        [2.5173e-04, 9.9975e-01],
        [1.1494e-04, 9.9989e-01],
        [2.0158e-04, 9.9980e-01],
        [3.8041e-05, 9.9996e-01],
        [1.7418e-07, 1.0000e+00],
        [3.2931e-10, 1.0000e+00],
        [3.1583e-06, 1.0000e+00],
        [3.0335e-09, 1.0000e+00],
        [9.0889e-06, 9.9999e-01],
        [4.7712e-07, 1.0000e+00],
        [2.0861e-11, 1.0000e+00],
        [9.6331e-04, 9.9904e-01],
        [8.5832e-06, 9.9999e-01],
        [3.1360e-05, 9.9997e-01],
        [9.0047e-01, 9.9534e-02],
        [7.1368e-08, 1.0000e+00],
        [9.8065e-01, 1.9353e-02],
        [9.6852e-04, 9.9903e-01],
        [9.4478e-01, 5.5221e-02],
        [2.1104e-06, 1.0000e+00],
        [1.0012e-05, 9.9999e-01],
        [7.7698e-06, 9.9999e-01],
        [3.5880e-07, 1.0000e+00],
        [1.7802e-04, 9.9982e-01],
        [2.3572e-07, 1.0000e+00],
        [6.0461e-08, 1.0000e+00],
        [1.0232e-03, 9.9898e-01],
        [2.5105e-05, 9.9997e-01],
        [6.1887e-07, 1.0000e+00],
        [2.8540e-05, 9.9997e-01],
        [9.9848e-01, 1.5184e-03],
        [2.1662e-05, 9.9998e-01],
        [9.9895e-01, 1.0462e-03],
        [6.3175e-09, 1.0000e+00],
        [9.9745e-01, 2.5459e-03],
        [4.3166e-07, 1.0000e+00],
        [1.0441e-08, 1.0000e+00],
        [1.8676e-05, 9.9998e-01],
        [8.4552e-01, 1.5448e-01],
        [1.7774e-05, 9.9998e-01],
        [9.9968e-01, 3.1936e-04],
        [1.2429e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9966e-01, 3.4401e-04],
        [9.7650e-01, 2.3496e-02],
        [1.1682e-03, 9.9883e-01],
        [3.0758e-03, 9.9692e-01],
        [3.8294e-05, 9.9996e-01],
        [9.5804e-10, 1.0000e+00],
        [1.8618e-05, 9.9998e-01],
        [7.5554e-06, 9.9999e-01],
        [8.5871e-05, 9.9991e-01],
        [8.0631e-06, 9.9999e-01],
        [7.1605e-01, 2.8395e-01],
        [1.0212e-06, 1.0000e+00],
        [9.9989e-01, 1.0661e-04],
        [3.7603e-01, 6.2397e-01],
        [9.9987e-01, 1.2505e-04],
        [9.5061e-01, 4.9389e-02],
        [7.6759e-01, 2.3241e-01],
        [2.6669e-05, 9.9997e-01],
        [9.9980e-01, 1.9505e-04],
        [9.9995e-01, 4.7537e-05],
        [2.6192e-07, 1.0000e+00],
        [3.6300e-07, 1.0000e+00],
        [9.9402e-01, 5.9832e-03],
        [7.8217e-01, 2.1783e-01],
        [8.7648e-01, 1.2352e-01],
        [9.9893e-01, 1.0687e-03],
        [6.3189e-06, 9.9999e-01],
        [8.2444e-05, 9.9992e-01],
        [2.3769e-05, 9.9998e-01],
        [1.2544e-05, 9.9999e-01],
        [8.0373e-04, 9.9920e-01],
        [3.6982e-08, 1.0000e+00],
        [4.2683e-05, 9.9996e-01],
        [1.1710e-06, 1.0000e+00],
        [2.1347e-08, 1.0000e+00],
        [6.8961e-06, 9.9999e-01],
        [4.9326e-09, 1.0000e+00],
        [2.8133e-06, 1.0000e+00],
        [6.2043e-07, 1.0000e+00],
        [1.5257e-04, 9.9985e-01],
        [1.2128e-05, 9.9999e-01],
        [8.3351e-08, 1.0000e+00],
        [1.3685e-06, 1.0000e+00],
        [5.8774e-01, 4.1226e-01],
        [5.4616e-05, 9.9995e-01],
        [7.5681e-08, 1.0000e+00],
        [2.4393e-07, 1.0000e+00],
        [3.3675e-05, 9.9997e-01],
        [5.2773e-10, 1.0000e+00],
        [8.9668e-07, 1.0000e+00],
        [4.4815e-06, 1.0000e+00],
        [2.2342e-05, 9.9998e-01],
        [9.9783e-01, 2.1700e-03],
        [9.9889e-01, 1.1054e-03],
        [9.9971e-01, 2.9371e-04],
        [9.9962e-01, 3.8465e-04],
        [2.1994e-07, 1.0000e+00],
        [9.8940e-01, 1.0596e-02],
        [4.1070e-06, 1.0000e+00],
        [3.0646e-07, 1.0000e+00],
        [9.9951e-01, 4.9020e-04],
        [9.9990e-01, 1.0342e-04],
        [4.7430e-07, 1.0000e+00],
        [4.5313e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9275e-01, 7.2514e-03],
        [3.2183e-08, 1.0000e+00],
        [7.1162e-06, 9.9999e-01],
        [3.0670e-04, 9.9969e-01],
        [1.1937e-01, 8.8063e-01],
        [3.0512e-07, 1.0000e+00],
        [3.4956e-06, 1.0000e+00],
        [9.9963e-01, 3.6948e-04],
        [9.9983e-01, 1.7167e-04],
        [9.9997e-01, 2.6492e-05],
        [2.9138e-05, 9.9997e-01],
        [1.7637e-06, 1.0000e+00],
        [3.9998e-07, 1.0000e+00],
        [9.7911e-01, 2.0890e-02],
        [7.4198e-03, 9.9258e-01],
        [9.2312e-07, 1.0000e+00],
        [9.9932e-01, 6.7693e-04],
        [8.2607e-06, 9.9999e-01],
        [6.1199e-06, 9.9999e-01],
        [9.8826e-01, 1.1743e-02],
        [1.9105e-06, 1.0000e+00],
        [2.0127e-03, 9.9799e-01],
        [9.6023e-01, 3.9773e-02],
        [3.8615e-08, 1.0000e+00],
        [1.8393e-03, 9.9816e-01],
        [9.9997e-01, 2.9766e-05],
        [8.7874e-06, 9.9999e-01],
        [9.9979e-01, 2.1482e-04],
        [2.3073e-06, 1.0000e+00],
        [1.0681e-08, 1.0000e+00],
        [9.9983e-01, 1.7456e-04],
        [7.8710e-09, 1.0000e+00],
        [1.5339e-03, 9.9847e-01],
        [2.7375e-06, 1.0000e+00],
        [1.9126e-04, 9.9981e-01],
        [1.3912e-01, 8.6088e-01],
        [7.0680e-06, 9.9999e-01],
        [5.1433e-06, 9.9999e-01],
        [9.9982e-01, 1.8052e-04],
        [1.8468e-06, 1.0000e+00],
        [4.3629e-08, 1.0000e+00],
        [1.5618e-06, 1.0000e+00],
        [5.6125e-04, 9.9944e-01],
        [4.5541e-07, 1.0000e+00],
        [3.1054e-07, 1.0000e+00],
        [9.3683e-01, 6.3174e-02],
        [7.0419e-06, 9.9999e-01],
        [9.9824e-01, 1.7561e-03],
        [6.1218e-03, 9.9388e-01],
        [9.4043e-01, 5.9573e-02],
        [1.9597e-06, 1.0000e+00],
        [1.7545e-04, 9.9982e-01],
        [4.4391e-06, 1.0000e+00],
        [9.9228e-01, 7.7165e-03],
        [1.7162e-04, 9.9983e-01],
        [9.9843e-01, 1.5661e-03],
        [5.4126e-06, 9.9999e-01],
        [1.4274e-05, 9.9999e-01],
        [9.9898e-01, 1.0166e-03],
        [3.6410e-08, 1.0000e+00],
        [6.9488e-09, 1.0000e+00],
        [1.4668e-05, 9.9999e-01],
        [5.0234e-10, 1.0000e+00],
        [9.5431e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9980e-01, 1.9505e-04],
        [2.2812e-06, 1.0000e+00],
        [5.0567e-11, 1.0000e+00],
        [1.7389e-07, 1.0000e+00],
        [2.5659e-03, 9.9743e-01],
        [4.8915e-07, 1.0000e+00],
        [9.9904e-01, 9.5769e-04],
        [1.8276e-09, 1.0000e+00],
        [9.9974e-01, 2.5892e-04],
        [2.9196e-04, 9.9971e-01],
        [1.5037e-06, 1.0000e+00],
        [2.3401e-09, 1.0000e+00],
        [9.9924e-01, 7.6354e-04],
        [1.0515e-08, 1.0000e+00],
        [9.9924e-01, 7.5833e-04],
        [3.1130e-08, 1.0000e+00],
        [3.2586e-03, 9.9674e-01],
        [4.1969e-05, 9.9996e-01],
        [7.3111e-05, 9.9993e-01],
        [1.6480e-06, 1.0000e+00],
        [9.9735e-01, 2.6508e-03],
        [9.9961e-01, 3.9166e-04],
        [1.6416e-05, 9.9998e-01],
        [5.7208e-06, 9.9999e-01],
        [1.3427e-05, 9.9999e-01],
        [9.6808e-01, 3.1924e-02],
        [1.5851e-05, 9.9998e-01],
        [1.3334e-03, 9.9867e-01],
        [7.5633e-07, 1.0000e+00],
        [9.9992e-01, 7.9068e-05],
        [3.5981e-06, 1.0000e+00],
        [6.3241e-05, 9.9994e-01],
        [9.9644e-01, 3.5569e-03],
        [2.5909e-04, 9.9974e-01],
        [1.0852e-04, 9.9989e-01],
        [9.9707e-01, 2.9276e-03],
        [3.1479e-07, 1.0000e+00],
        [1.6633e-03, 9.9834e-01],
        [5.0438e-06, 9.9999e-01],
        [9.9978e-01, 2.1862e-04],
        [4.3454e-08, 1.0000e+00],
        [9.9850e-01, 1.4963e-03],
        [9.9989e-01, 1.1493e-04],
        [7.9297e-06, 9.9999e-01],
        [3.6579e-01, 6.3421e-01],
        [2.4712e-07, 1.0000e+00],
        [3.9576e-07, 1.0000e+00],
        [6.2203e-04, 9.9938e-01],
        [2.5874e-09, 1.0000e+00],
        [7.8538e-05, 9.9992e-01],
        [9.5489e-01, 4.5107e-02],
        [8.1078e-04, 9.9919e-01],
        [3.7394e-06, 1.0000e+00],
        [3.5174e-04, 9.9965e-01],
        [8.4416e-01, 1.5584e-01],
        [3.6890e-08, 1.0000e+00],
        [3.5927e-05, 9.9996e-01],
        [1.0780e-07, 1.0000e+00],
        [2.4566e-06, 1.0000e+00],
        [2.5637e-06, 1.0000e+00],
        [1.0733e-01, 8.9267e-01],
        [6.9978e-07, 1.0000e+00],
        [2.4129e-04, 9.9976e-01],
        [7.4624e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.6030e-11, 1.0000e+00],
        [9.9989e-01, 1.1242e-04],
        [1.0010e-05, 9.9999e-01],
        [1.5004e-06, 1.0000e+00],
        [9.9883e-01, 1.1714e-03],
        [4.4109e-06, 1.0000e+00],
        [1.5292e-07, 1.0000e+00],
        [1.6774e-08, 1.0000e+00],
        [9.9903e-01, 9.7123e-04],
        [6.2666e-09, 1.0000e+00],
        [2.1801e-06, 1.0000e+00],
        [2.1389e-04, 9.9979e-01],
        [4.1580e-04, 9.9958e-01],
        [1.4714e-07, 1.0000e+00],
        [1.0038e-03, 9.9900e-01],
        [3.4589e-04, 9.9965e-01],
        [9.9995e-01, 5.0743e-05],
        [3.6089e-04, 9.9964e-01],
        [1.6250e-06, 1.0000e+00],
        [6.1192e-05, 9.9994e-01],
        [9.9853e-01, 1.4733e-03],
        [8.0078e-09, 1.0000e+00],
        [4.7705e-03, 9.9523e-01],
        [1.4868e-08, 1.0000e+00],
        [7.4492e-08, 1.0000e+00],
        [5.7032e-10, 1.0000e+00],
        [4.7091e-08, 1.0000e+00],
        [1.0826e-09, 1.0000e+00],
        [1.5384e-03, 9.9846e-01],
        [3.4828e-04, 9.9965e-01],
        [2.9214e-12, 1.0000e+00],
        [9.9828e-01, 1.7174e-03],
        [1.6944e-06, 1.0000e+00],
        [6.7178e-05, 9.9993e-01],
        [9.9964e-01, 3.6401e-04],
        [3.0553e-02, 9.6945e-01],
        [4.3521e-07, 1.0000e+00],
        [1.4011e-02, 9.8599e-01],
        [5.7073e-06, 9.9999e-01],
        [9.9983e-01, 1.7241e-04],
        [4.7019e-05, 9.9995e-01],
        [9.9983e-01, 1.7383e-04],
        [4.7716e-05, 9.9995e-01],
        [9.6268e-01, 3.7322e-02],
        [2.6537e-01, 7.3463e-01],
        [5.6893e-06, 9.9999e-01],
        [9.9855e-01, 1.4468e-03],
        [1.8864e-08, 1.0000e+00],
        [2.7066e-02, 9.7293e-01],
        [2.0115e-09, 1.0000e+00],
        [4.6900e-09, 1.0000e+00],
        [1.5967e-07, 1.0000e+00],
        [1.8991e-06, 1.0000e+00],
        [9.9721e-01, 2.7946e-03],
        [9.9911e-01, 8.8844e-04],
        [6.6222e-07, 1.0000e+00],
        [1.6729e-04, 9.9983e-01],
        [2.2040e-06, 1.0000e+00],
        [5.5008e-07, 1.0000e+00],
        [9.8809e-07, 1.0000e+00],
        [1.9301e-08, 1.0000e+00],
        [9.9977e-01, 2.3387e-04],
        [8.3038e-08, 1.0000e+00],
        [3.9127e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.1080e-10, 1.0000e+00],
        [1.7250e-08, 1.0000e+00],
        [4.1479e-06, 1.0000e+00],
        [3.9289e-04, 9.9961e-01],
        [1.2677e-08, 1.0000e+00],
        [1.6741e-02, 9.8326e-01],
        [2.2641e-04, 9.9977e-01],
        [6.3349e-10, 1.0000e+00],
        [1.2550e-04, 9.9987e-01],
        [9.9848e-01, 1.5157e-03],
        [4.1184e-08, 1.0000e+00],
        [5.0005e-11, 1.0000e+00],
        [9.9870e-01, 1.3032e-03],
        [5.9055e-01, 4.0945e-01],
        [9.9897e-01, 1.0312e-03],
        [9.3116e-04, 9.9907e-01],
        [9.9960e-01, 4.0442e-04],
        [9.9955e-01, 4.4632e-04],
        [1.4081e-04, 9.9986e-01],
        [9.9985e-01, 1.5416e-04],
        [2.2889e-09, 1.0000e+00],
        [9.9973e-01, 2.7269e-04],
        [2.7726e-08, 1.0000e+00],
        [4.5921e-10, 1.0000e+00],
        [9.2192e-05, 9.9991e-01],
        [1.8664e-07, 1.0000e+00],
        [8.5083e-04, 9.9915e-01],
        [8.9147e-07, 1.0000e+00],
        [7.1758e-07, 1.0000e+00],
        [8.0798e-01, 1.9202e-01],
        [5.4781e-05, 9.9995e-01],
        [9.9925e-01, 7.5193e-04],
        [9.9988e-01, 1.1865e-04],
        [2.1611e-03, 9.9784e-01],
        [2.0806e-06, 1.0000e+00],
        [4.5362e-06, 1.0000e+00],
        [7.2759e-07, 1.0000e+00],
        [6.6386e-10, 1.0000e+00],
        [9.9988e-01, 1.1793e-04],
        [7.4891e-06, 9.9999e-01],
        [4.8664e-03, 9.9513e-01],
        [1.8574e-04, 9.9981e-01],
        [2.1243e-04, 9.9979e-01],
        [2.7011e-05, 9.9997e-01],
        [9.7211e-01, 2.7885e-02],
        [9.9193e-02, 9.0081e-01],
        [3.7925e-03, 9.9621e-01],
        [3.9701e-04, 9.9960e-01],
        [7.8049e-07, 1.0000e+00],
        [1.5457e-03, 9.9845e-01],
        [5.6882e-09, 1.0000e+00],
        [1.4260e-07, 1.0000e+00],
        [9.9954e-01, 4.5855e-04],
        [1.7165e-05, 9.9998e-01],
        [3.4927e-03, 9.9651e-01],
        [4.7659e-08, 1.0000e+00],
        [2.9496e-04, 9.9971e-01],
        [9.9796e-01, 2.0404e-03],
        [8.4095e-08, 1.0000e+00],
        [2.7357e-03, 9.9726e-01],
        [9.5191e-05, 9.9990e-01],
        [1.0127e-02, 9.8987e-01],
        [9.9989e-01, 1.1061e-04],
        [9.3927e-05, 9.9991e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.8483e-05, 9.9997e-01],
        [1.0000e+00, 1.2339e-06],
        [1.9910e-08, 1.0000e+00],
        [9.7903e-01, 2.0972e-02],
        [3.5746e-05, 9.9996e-01],
        [9.9966e-01, 3.4235e-04],
        [9.9995e-01, 5.0122e-05],
        [9.9985e-01, 1.5193e-04],
        [5.7822e-06, 9.9999e-01],
        [3.3616e-04, 9.9966e-01],
        [9.7432e-08, 1.0000e+00],
        [5.3110e-06, 9.9999e-01],
        [3.4286e-06, 1.0000e+00],
        [2.3301e-04, 9.9977e-01],
        [9.9994e-01, 5.7935e-05],
        [1.8722e-03, 9.9813e-01],
        [9.8721e-01, 1.2788e-02],
        [2.3531e-05, 9.9998e-01],
        [4.2847e-03, 9.9572e-01],
        [9.9958e-01, 4.1946e-04],
        [1.4788e-04, 9.9985e-01],
        [2.3514e-06, 1.0000e+00],
        [2.3059e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
Epoch 5, loss: 0.3333
tensor([[1.8384e-06, 1.0000e+00],
        [9.9368e-08, 1.0000e+00],
        [1.4667e-08, 1.0000e+00],
        [9.9988e-01, 1.1644e-04],
        [4.5409e-03, 9.9546e-01],
        [2.4269e-08, 1.0000e+00],
        [2.9496e-09, 1.0000e+00],
        [2.0250e-02, 9.7975e-01],
        [8.6884e-03, 9.9131e-01],
        [3.4345e-02, 9.6565e-01],
        [5.8906e-07, 1.0000e+00],
        [9.8673e-03, 9.9013e-01],
        [7.6039e-04, 9.9924e-01],
        [4.5641e-07, 1.0000e+00],
        [4.5702e-08, 1.0000e+00],
        [3.0380e-04, 9.9970e-01],
        [6.8699e-05, 9.9993e-01],
        [1.5577e-03, 9.9844e-01],
        [9.9510e-07, 1.0000e+00],
        [8.6406e-01, 1.3594e-01],
        [6.2319e-06, 9.9999e-01],
        [9.9979e-01, 2.1462e-04],
        [1.0708e-04, 9.9989e-01],
        [2.1072e-03, 9.9789e-01],
        [1.0000e+00, 5.7493e-07],
        [1.0383e-02, 9.8962e-01],
        [6.9007e-08, 1.0000e+00],
        [9.9950e-01, 4.9898e-04],
        [2.8396e-01, 7.1604e-01],
        [5.8341e-03, 9.9417e-01],
        [2.1154e-04, 9.9979e-01],
        [7.4621e-05, 9.9993e-01],
        [5.9627e-04, 9.9940e-01],
        [1.5620e-03, 9.9844e-01],
        [4.5983e-03, 9.9540e-01],
        [4.1554e-03, 9.9584e-01],
        [4.4603e-03, 9.9554e-01],
        [3.9439e-05, 9.9996e-01],
        [2.4483e-04, 9.9976e-01],
        [9.9927e-01, 7.2799e-04],
        [9.4853e-09, 1.0000e+00],
        [9.5747e-08, 1.0000e+00],
        [2.8165e-09, 1.0000e+00],
        [3.1363e-05, 9.9997e-01],
        [1.5532e-04, 9.9984e-01],
        [1.6973e-05, 9.9998e-01],
        [1.4527e-08, 1.0000e+00],
        [9.9987e-01, 1.2639e-04],
        [8.8706e-01, 1.1294e-01],
        [8.9102e-01, 1.0898e-01],
        [1.2353e-07, 1.0000e+00],
        [2.5997e-05, 9.9997e-01],
        [1.1037e-06, 1.0000e+00],
        [1.2027e-04, 9.9988e-01],
        [4.9796e-06, 9.9999e-01],
        [2.7640e-07, 1.0000e+00],
        [3.2078e-06, 1.0000e+00],
        [9.9962e-01, 3.7577e-04],
        [9.9996e-01, 3.7616e-05],
        [1.1218e-01, 8.8782e-01],
        [9.4262e-01, 5.7378e-02],
        [6.5519e-05, 9.9993e-01],
        [1.6184e-01, 8.3816e-01],
        [9.8984e-01, 1.0161e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3643e-07, 1.0000e+00],
        [5.4618e-02, 9.4538e-01],
        [3.7719e-04, 9.9962e-01],
        [9.9993e-01, 7.3428e-05],
        [7.3855e-08, 1.0000e+00],
        [3.2048e-09, 1.0000e+00],
        [9.9998e-01, 2.3117e-05],
        [9.9963e-01, 3.6665e-04],
        [8.2185e-06, 9.9999e-01],
        [8.4466e-08, 1.0000e+00],
        [9.9857e-01, 1.4250e-03],
        [3.6709e-07, 1.0000e+00],
        [5.7945e-03, 9.9421e-01],
        [1.0013e-03, 9.9900e-01],
        [1.1300e-05, 9.9999e-01],
        [2.7409e-02, 9.7259e-01],
        [2.4665e-06, 1.0000e+00],
        [3.9430e-05, 9.9996e-01],
        [9.9989e-01, 1.0720e-04],
        [4.7996e-01, 5.2004e-01],
        [9.0461e-06, 9.9999e-01],
        [1.9599e-04, 9.9980e-01],
        [5.0262e-04, 9.9950e-01],
        [5.3688e-05, 9.9995e-01],
        [6.8268e-05, 9.9993e-01],
        [2.3061e-04, 9.9977e-01],
        [1.1723e-05, 9.9999e-01],
        [9.9992e-01, 8.3198e-05],
        [4.1077e-04, 9.9959e-01],
        [3.3584e-03, 9.9664e-01],
        [6.7434e-04, 9.9933e-01],
        [9.9967e-01, 3.2947e-04],
        [2.4306e-05, 9.9998e-01],
        [9.9942e-01, 5.7830e-04],
        [7.1639e-06, 9.9999e-01],
        [5.2904e-10, 1.0000e+00],
        [4.6467e-05, 9.9995e-01],
        [9.9989e-01, 1.1033e-04],
        [3.1980e-06, 1.0000e+00],
        [1.9229e-07, 1.0000e+00],
        [9.9997e-01, 2.6002e-05],
        [9.0685e-07, 1.0000e+00],
        [9.9942e-01, 5.8327e-04],
        [9.8851e-01, 1.1492e-02],
        [8.8457e-05, 9.9991e-01],
        [6.1447e-05, 9.9994e-01],
        [9.2359e-05, 9.9991e-01],
        [8.0125e-06, 9.9999e-01],
        [4.7513e-04, 9.9952e-01],
        [5.7465e-05, 9.9994e-01],
        [5.8505e-02, 9.4150e-01],
        [1.1883e-03, 9.9881e-01],
        [3.9007e-05, 9.9996e-01],
        [2.0338e-03, 9.9797e-01],
        [8.5884e-01, 1.4116e-01],
        [1.4560e-04, 9.9985e-01],
        [1.0760e-04, 9.9989e-01],
        [1.5437e-08, 1.0000e+00],
        [8.8908e-05, 9.9991e-01],
        [9.8491e-07, 1.0000e+00],
        [9.9694e-01, 3.0556e-03],
        [3.2841e-04, 9.9967e-01],
        [4.4795e-08, 1.0000e+00],
        [9.9185e-01, 8.1548e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4180e-01, 8.5820e-01],
        [9.9962e-01, 3.7891e-04],
        [9.9249e-01, 7.5126e-03],
        [9.9963e-01, 3.7482e-04],
        [2.9551e-05, 9.9997e-01],
        [1.7173e-06, 1.0000e+00],
        [1.6124e-03, 9.9839e-01],
        [5.1558e-08, 1.0000e+00],
        [9.9157e-05, 9.9990e-01],
        [1.2268e-03, 9.9877e-01],
        [1.0361e-04, 9.9990e-01],
        [9.9980e-01, 1.9567e-04],
        [9.9973e-01, 2.6517e-04],
        [2.9407e-06, 1.0000e+00],
        [9.9978e-01, 2.2274e-04],
        [3.9762e-04, 9.9960e-01],
        [3.7774e-05, 9.9996e-01],
        [1.6360e-08, 1.0000e+00],
        [9.9989e-01, 1.1303e-04],
        [2.1290e-06, 1.0000e+00],
        [9.8483e-01, 1.5172e-02],
        [1.8890e-08, 1.0000e+00],
        [9.5573e-01, 4.4265e-02],
        [7.4528e-06, 9.9999e-01],
        [9.5591e-03, 9.9044e-01],
        [9.2862e-08, 1.0000e+00],
        [7.0111e-01, 2.9889e-01],
        [3.6805e-01, 6.3195e-01],
        [3.2915e-04, 9.9967e-01],
        [2.3521e-04, 9.9976e-01],
        [1.7149e-04, 9.9983e-01],
        [3.6177e-02, 9.6382e-01],
        [7.3725e-07, 1.0000e+00],
        [3.2284e-04, 9.9968e-01],
        [9.9975e-01, 2.5041e-04],
        [9.9982e-01, 1.8368e-04],
        [2.3915e-04, 9.9976e-01],
        [9.9632e-01, 3.6792e-03],
        [3.3109e-04, 9.9967e-01],
        [1.4652e-05, 9.9999e-01],
        [9.5266e-07, 1.0000e+00],
        [2.7046e-07, 1.0000e+00],
        [2.3696e-06, 1.0000e+00],
        [9.9673e-01, 3.2704e-03],
        [2.7480e-03, 9.9725e-01],
        [1.2301e-04, 9.9988e-01],
        [7.1189e-03, 9.9288e-01],
        [3.1602e-05, 9.9997e-01],
        [9.4630e-05, 9.9991e-01],
        [9.9974e-01, 2.6001e-04],
        [3.7926e-06, 1.0000e+00],
        [4.4707e-02, 9.5529e-01],
        [5.1046e-06, 9.9999e-01],
        [7.1344e-04, 9.9929e-01],
        [8.1136e-08, 1.0000e+00],
        [4.2137e-07, 1.0000e+00],
        [2.3193e-03, 9.9768e-01],
        [5.6829e-06, 9.9999e-01],
        [1.5350e-07, 1.0000e+00],
        [2.2143e-03, 9.9779e-01],
        [9.9955e-01, 4.5231e-04],
        [9.9992e-01, 7.8372e-05],
        [8.0285e-04, 9.9920e-01],
        [1.3827e-02, 9.8617e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4101e-04, 9.9986e-01],
        [3.6873e-07, 1.0000e+00],
        [1.4230e-06, 1.0000e+00],
        [9.9892e-01, 1.0768e-03],
        [2.4050e-03, 9.9760e-01],
        [9.9987e-01, 1.3427e-04],
        [2.8330e-02, 9.7167e-01],
        [4.2559e-03, 9.9574e-01],
        [2.7572e-07, 1.0000e+00],
        [1.6715e-04, 9.9983e-01],
        [9.9162e-05, 9.9990e-01],
        [1.7556e-05, 9.9998e-01],
        [4.2038e-03, 9.9580e-01],
        [1.9994e-03, 9.9800e-01],
        [9.9719e-01, 2.8117e-03],
        [8.6961e-08, 1.0000e+00],
        [4.2592e-03, 9.9574e-01],
        [1.1294e-06, 1.0000e+00],
        [3.5627e-06, 1.0000e+00],
        [1.3690e-07, 1.0000e+00],
        [2.0358e-05, 9.9998e-01],
        [2.4585e-04, 9.9975e-01],
        [2.8737e-08, 1.0000e+00],
        [9.5616e-07, 1.0000e+00],
        [3.7271e-06, 1.0000e+00],
        [9.9988e-01, 1.2207e-04],
        [9.9970e-01, 2.9763e-04],
        [9.9993e-01, 7.3373e-05],
        [1.0000e+00, 1.2537e-06],
        [9.9968e-01, 3.1696e-04],
        [9.9991e-01, 8.6594e-05],
        [3.8583e-08, 1.0000e+00],
        [2.8724e-02, 9.7128e-01],
        [1.6947e-03, 9.9831e-01],
        [5.7534e-08, 1.0000e+00],
        [2.9023e-04, 9.9971e-01],
        [1.2076e-04, 9.9988e-01],
        [6.6698e-06, 9.9999e-01],
        [1.5604e-08, 1.0000e+00],
        [9.9926e-01, 7.4016e-04],
        [8.6707e-01, 1.3293e-01],
        [5.4519e-06, 9.9999e-01],
        [3.6530e-06, 1.0000e+00],
        [9.9918e-01, 8.2104e-04],
        [1.1224e-02, 9.8878e-01],
        [7.7137e-06, 9.9999e-01],
        [2.0799e-03, 9.9792e-01],
        [3.9812e-03, 9.9602e-01],
        [1.5542e-08, 1.0000e+00],
        [9.8148e-01, 1.8523e-02],
        [9.9974e-01, 2.5845e-04],
        [2.5814e-05, 9.9997e-01],
        [9.8775e-01, 1.2250e-02],
        [7.5533e-03, 9.9245e-01],
        [9.9906e-01, 9.3922e-04],
        [1.5237e-04, 9.9985e-01],
        [1.6828e-04, 9.9983e-01],
        [5.4798e-02, 9.4520e-01],
        [9.9968e-01, 3.2364e-04],
        [1.0852e-08, 1.0000e+00],
        [8.0342e-02, 9.1966e-01],
        [4.2713e-02, 9.5729e-01],
        [9.3247e-04, 9.9907e-01],
        [1.4402e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8802e-01, 1.1984e-02],
        [1.9419e-04, 9.9981e-01],
        [9.9991e-01, 9.4798e-05],
        [2.4487e-01, 7.5513e-01],
        [9.9997e-01, 3.2333e-05],
        [1.7339e-03, 9.9827e-01],
        [9.9981e-01, 1.9255e-04],
        [1.6286e-05, 9.9998e-01],
        [5.7460e-05, 9.9994e-01],
        [9.5450e-08, 1.0000e+00],
        [1.2978e-03, 9.9870e-01],
        [1.4447e-06, 1.0000e+00],
        [7.9284e-04, 9.9921e-01],
        [2.6340e-04, 9.9974e-01],
        [9.9968e-01, 3.1515e-04],
        [2.4855e-02, 9.7514e-01],
        [9.6898e-01, 3.1019e-02],
        [2.5063e-05, 9.9997e-01],
        [1.2300e-04, 9.9988e-01],
        [1.7032e-04, 9.9983e-01],
        [9.9910e-01, 9.0293e-04],
        [1.2048e-10, 1.0000e+00],
        [2.0126e-08, 1.0000e+00],
        [4.0921e-07, 1.0000e+00],
        [9.9382e-01, 6.1827e-03],
        [9.6667e-06, 9.9999e-01],
        [1.0002e-07, 1.0000e+00],
        [2.9698e-07, 1.0000e+00],
        [1.0621e-06, 1.0000e+00],
        [9.9958e-01, 4.2372e-04],
        [9.9901e-01, 9.9070e-04],
        [4.3460e-02, 9.5654e-01],
        [7.5292e-04, 9.9925e-01],
        [2.5130e-05, 9.9997e-01],
        [9.9825e-01, 1.7547e-03],
        [3.0653e-08, 1.0000e+00],
        [9.9878e-01, 1.2247e-03],
        [1.1089e-08, 1.0000e+00],
        [2.3358e-06, 1.0000e+00],
        [9.9931e-01, 6.8802e-04],
        [4.0938e-06, 1.0000e+00],
        [6.8796e-04, 9.9931e-01],
        [2.6765e-03, 9.9732e-01],
        [6.2018e-09, 1.0000e+00],
        [7.7041e-03, 9.9230e-01],
        [1.0520e-03, 9.9895e-01],
        [9.9868e-01, 1.3239e-03],
        [2.6601e-08, 1.0000e+00],
        [1.5118e-06, 1.0000e+00],
        [2.1962e-07, 1.0000e+00],
        [5.4582e-08, 1.0000e+00],
        [2.1938e-09, 1.0000e+00],
        [1.0850e-03, 9.9892e-01],
        [9.9989e-01, 1.1482e-04],
        [4.3119e-04, 9.9957e-01],
        [9.4614e-07, 1.0000e+00],
        [1.6711e-02, 9.8329e-01],
        [9.3759e-04, 9.9906e-01],
        [9.9359e-01, 6.4066e-03],
        [9.7463e-05, 9.9990e-01],
        [9.9424e-01, 5.7640e-03],
        [4.7127e-07, 1.0000e+00],
        [1.4757e-02, 9.8524e-01],
        [4.1218e-01, 5.8782e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8134e-07, 1.0000e+00],
        [1.0000e+00, 2.6450e-06],
        [9.9485e-01, 5.1518e-03],
        [1.4566e-05, 9.9999e-01],
        [3.6740e-05, 9.9996e-01],
        [4.4215e-02, 9.5578e-01],
        [5.4895e-07, 1.0000e+00],
        [9.9996e-01, 4.0582e-05],
        [9.8505e-01, 1.4953e-02],
        [9.9953e-01, 4.6768e-04],
        [8.7267e-04, 9.9913e-01],
        [4.8701e-04, 9.9951e-01],
        [1.1649e-05, 9.9999e-01],
        [1.0123e-04, 9.9990e-01],
        [2.4089e-05, 9.9998e-01],
        [4.0883e-05, 9.9996e-01],
        [3.3603e-05, 9.9997e-01],
        [7.8728e-07, 1.0000e+00],
        [1.0513e-01, 8.9487e-01],
        [3.9279e-10, 1.0000e+00],
        [2.1339e-01, 7.8661e-01],
        [6.5452e-02, 9.3455e-01],
        [3.4216e-08, 1.0000e+00],
        [8.3230e-01, 1.6770e-01],
        [9.4485e-01, 5.5150e-02],
        [8.1599e-04, 9.9918e-01],
        [5.7161e-06, 9.9999e-01],
        [6.6356e-08, 1.0000e+00],
        [2.1468e-05, 9.9998e-01],
        [4.6773e-07, 1.0000e+00],
        [9.9778e-01, 2.2204e-03],
        [2.3567e-01, 7.6433e-01],
        [5.4299e-08, 1.0000e+00],
        [1.9058e-03, 9.9809e-01],
        [3.3608e-09, 1.0000e+00],
        [5.6732e-02, 9.4327e-01],
        [2.6442e-04, 9.9974e-01],
        [1.7272e-04, 9.9983e-01],
        [1.5919e-05, 9.9998e-01],
        [2.4608e-06, 1.0000e+00],
        [8.8485e-06, 9.9999e-01],
        [1.5916e-03, 9.9841e-01],
        [1.2567e-05, 9.9999e-01],
        [9.9900e-01, 9.9910e-04],
        [2.6111e-05, 9.9997e-01],
        [1.8856e-03, 9.9811e-01],
        [9.9941e-01, 5.9343e-04],
        [2.9219e-03, 9.9708e-01],
        [5.3769e-04, 9.9946e-01],
        [9.9980e-01, 2.0360e-04],
        [1.2124e-09, 1.0000e+00],
        [2.4691e-05, 9.9998e-01],
        [7.7927e-03, 9.9221e-01],
        [2.0764e-02, 9.7924e-01],
        [5.6732e-02, 9.4327e-01],
        [9.4134e-03, 9.9059e-01],
        [1.5323e-08, 1.0000e+00],
        [9.9756e-01, 2.4444e-03],
        [9.9990e-01, 1.0384e-04],
        [2.0501e-06, 1.0000e+00],
        [2.5900e-02, 9.7410e-01],
        [9.9993e-01, 7.2744e-05],
        [9.9916e-01, 8.4335e-04],
        [3.9342e-04, 9.9961e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8551e-01, 1.4486e-02],
        [1.2525e-06, 1.0000e+00],
        [9.2237e-01, 7.7626e-02],
        [9.9512e-01, 4.8806e-03],
        [1.3550e-06, 1.0000e+00],
        [1.6532e-07, 1.0000e+00],
        [1.1878e-04, 9.9988e-01],
        [2.4541e-06, 1.0000e+00],
        [1.7482e-06, 1.0000e+00],
        [7.4765e-05, 9.9993e-01],
        [8.3759e-01, 1.6241e-01],
        [3.0280e-05, 9.9997e-01],
        [4.9666e-01, 5.0334e-01],
        [6.4167e-10, 1.0000e+00],
        [3.8876e-05, 9.9996e-01],
        [7.6539e-01, 2.3461e-01],
        [7.1024e-01, 2.8976e-01],
        [8.9159e-01, 1.0841e-01],
        [9.8908e-03, 9.9011e-01],
        [2.0968e-06, 1.0000e+00],
        [1.0413e-07, 1.0000e+00],
        [9.9176e-03, 9.9008e-01],
        [2.1546e-07, 1.0000e+00],
        [4.2054e-04, 9.9958e-01],
        [3.9648e-08, 1.0000e+00],
        [7.7644e-05, 9.9992e-01],
        [5.9848e-09, 1.0000e+00],
        [1.5636e-05, 9.9998e-01],
        [1.4444e-04, 9.9986e-01],
        [2.8738e-08, 1.0000e+00],
        [1.1339e-02, 9.8866e-01],
        [9.9961e-01, 3.9414e-04],
        [3.6398e-06, 1.0000e+00],
        [8.6944e-07, 1.0000e+00],
        [1.0049e-04, 9.9990e-01],
        [9.5394e-01, 4.6056e-02],
        [6.6039e-04, 9.9934e-01],
        [9.9915e-01, 8.4601e-04],
        [7.8952e-01, 2.1048e-01],
        [9.9965e-01, 3.4794e-04],
        [9.9864e-01, 1.3567e-03],
        [4.9268e-10, 1.0000e+00],
        [2.2383e-05, 9.9998e-01],
        [9.9972e-01, 2.8402e-04],
        [6.8302e-07, 1.0000e+00],
        [9.9972e-01, 2.8313e-04],
        [9.9961e-01, 3.8507e-04],
        [8.1169e-01, 1.8831e-01],
        [4.4197e-05, 9.9996e-01],
        [5.4916e-06, 9.9999e-01],
        [9.3376e-01, 6.6240e-02],
        [4.1865e-04, 9.9958e-01],
        [7.1955e-06, 9.9999e-01],
        [3.4248e-06, 1.0000e+00],
        [8.1530e-09, 1.0000e+00],
        [4.5867e-08, 1.0000e+00],
        [8.7259e-11, 1.0000e+00],
        [5.9854e-08, 1.0000e+00],
        [8.2644e-05, 9.9992e-01],
        [7.7168e-06, 9.9999e-01],
        [2.0852e-10, 1.0000e+00],
        [5.2930e-05, 9.9995e-01],
        [9.9455e-01, 5.4483e-03],
        [9.9977e-01, 2.3446e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9988e-01, 1.2327e-04],
        [9.9981e-01, 1.9344e-04],
        [9.9326e-01, 6.7413e-03],
        [1.0467e-04, 9.9990e-01],
        [7.5024e-05, 9.9993e-01],
        [5.4680e-04, 9.9945e-01],
        [2.1836e-06, 1.0000e+00],
        [1.8593e-02, 9.8141e-01],
        [3.2672e-06, 1.0000e+00],
        [9.9984e-01, 1.5854e-04],
        [9.9992e-01, 7.7739e-05],
        [5.5368e-08, 1.0000e+00],
        [6.5118e-05, 9.9993e-01],
        [1.0555e-04, 9.9989e-01],
        [1.0512e-06, 1.0000e+00],
        [9.6453e-05, 9.9990e-01],
        [5.0089e-06, 9.9999e-01],
        [1.6319e-12, 1.0000e+00],
        [9.1639e-09, 1.0000e+00],
        [4.2350e-04, 9.9958e-01],
        [9.9961e-01, 3.8663e-04],
        [1.2452e-03, 9.9875e-01],
        [6.8765e-09, 1.0000e+00],
        [1.7889e-03, 9.9821e-01],
        [5.3538e-05, 9.9995e-01],
        [1.0829e-07, 1.0000e+00],
        [3.0059e-02, 9.6994e-01],
        [9.9966e-01, 3.4098e-04],
        [1.1381e-05, 9.9999e-01],
        [7.4471e-03, 9.9255e-01],
        [6.6096e-06, 9.9999e-01],
        [9.6130e-06, 9.9999e-01],
        [8.3897e-03, 9.9161e-01],
        [4.0243e-01, 5.9757e-01],
        [3.5695e-01, 6.4305e-01],
        [1.1120e-07, 1.0000e+00],
        [7.2658e-05, 9.9993e-01],
        [5.9692e-08, 1.0000e+00],
        [9.9974e-01, 2.6486e-04],
        [4.5198e-09, 1.0000e+00],
        [1.7206e-05, 9.9998e-01],
        [1.2107e-07, 1.0000e+00],
        [1.5435e-03, 9.9846e-01],
        [9.5916e-05, 9.9990e-01],
        [3.5465e-02, 9.6454e-01],
        [6.2077e-06, 9.9999e-01],
        [7.6235e-05, 9.9992e-01],
        [7.8182e-06, 9.9999e-01],
        [3.1991e-07, 1.0000e+00],
        [9.9439e-01, 5.6054e-03],
        [2.4600e-06, 1.0000e+00],
        [5.7728e-03, 9.9423e-01],
        [1.7051e-07, 1.0000e+00],
        [1.5600e-08, 1.0000e+00],
        [6.6679e-05, 9.9993e-01],
        [7.0152e-01, 2.9848e-01],
        [9.9996e-01, 4.0954e-05],
        [1.6837e-08, 1.0000e+00],
        [2.7246e-07, 1.0000e+00],
        [9.9948e-01, 5.2275e-04],
        [5.1480e-05, 9.9995e-01],
        [1.0325e-09, 1.0000e+00],
        [5.8935e-08, 1.0000e+00],
        [1.6225e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.5009e-01, 1.4991e-01],
        [2.5262e-03, 9.9747e-01],
        [7.1218e-09, 1.0000e+00],
        [1.1979e-03, 9.9880e-01],
        [1.4153e-04, 9.9986e-01],
        [1.4362e-01, 8.5638e-01],
        [2.2744e-05, 9.9998e-01],
        [7.1638e-05, 9.9993e-01],
        [1.1741e-05, 9.9999e-01],
        [9.8552e-01, 1.4477e-02],
        [2.8900e-07, 1.0000e+00],
        [1.2823e-07, 1.0000e+00],
        [1.8661e-04, 9.9981e-01],
        [2.1238e-06, 1.0000e+00],
        [1.2488e-07, 1.0000e+00],
        [3.9798e-07, 1.0000e+00],
        [2.5988e-03, 9.9740e-01],
        [1.0498e-04, 9.9989e-01],
        [1.5611e-04, 9.9984e-01],
        [3.8959e-04, 9.9961e-01],
        [5.5384e-05, 9.9994e-01],
        [2.2333e-03, 9.9777e-01],
        [5.9594e-01, 4.0406e-01],
        [1.0393e-05, 9.9999e-01],
        [1.4522e-02, 9.8548e-01],
        [4.5841e-01, 5.4159e-01],
        [4.4394e-06, 1.0000e+00],
        [8.5890e-08, 1.0000e+00],
        [1.6587e-05, 9.9998e-01],
        [6.8423e-01, 3.1577e-01],
        [7.6198e-07, 1.0000e+00],
        [9.9964e-01, 3.6351e-04],
        [6.4530e-07, 1.0000e+00],
        [1.1650e-10, 1.0000e+00],
        [6.0740e-09, 1.0000e+00],
        [7.5078e-06, 9.9999e-01],
        [1.1524e-07, 1.0000e+00],
        [4.6444e-04, 9.9954e-01],
        [7.4014e-03, 9.9260e-01],
        [1.3993e-05, 9.9999e-01],
        [3.2203e-04, 9.9968e-01],
        [4.2724e-05, 9.9996e-01],
        [2.5893e-02, 9.7411e-01],
        [3.5233e-07, 1.0000e+00],
        [9.9997e-01, 2.6238e-05],
        [3.2746e-04, 9.9967e-01],
        [1.4333e-04, 9.9986e-01],
        [8.7431e-02, 9.1257e-01],
        [1.6450e-08, 1.0000e+00],
        [6.7781e-08, 1.0000e+00],
        [6.8551e-06, 9.9999e-01],
        [5.9294e-02, 9.4071e-01],
        [1.2043e-06, 1.0000e+00],
        [3.9961e-03, 9.9600e-01],
        [9.9936e-01, 6.3605e-04],
        [9.9983e-01, 1.7438e-04],
        [9.9866e-01, 1.3449e-03],
        [1.5249e-05, 9.9998e-01],
        [1.2471e-08, 1.0000e+00],
        [2.3188e-04, 9.9977e-01],
        [9.9993e-01, 7.0959e-05],
        [7.9943e-02, 9.2006e-01],
        [2.6497e-07, 1.0000e+00],
        [2.6081e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0550e-04, 9.9989e-01],
        [9.9981e-01, 1.9448e-04],
        [4.9649e-08, 1.0000e+00],
        [9.9958e-01, 4.1675e-04],
        [4.2940e-04, 9.9957e-01],
        [1.4019e-03, 9.9860e-01],
        [1.0130e-06, 1.0000e+00],
        [1.8880e-05, 9.9998e-01],
        [9.9859e-01, 1.4112e-03],
        [3.3506e-06, 1.0000e+00],
        [1.7046e-06, 1.0000e+00],
        [9.9972e-01, 2.8224e-04],
        [1.5394e-06, 1.0000e+00],
        [9.9996e-01, 4.1740e-05],
        [6.8280e-08, 1.0000e+00],
        [9.9975e-01, 2.4596e-04],
        [5.4742e-08, 1.0000e+00],
        [5.2615e-01, 4.7385e-01],
        [9.9059e-01, 9.4130e-03],
        [4.6023e-06, 1.0000e+00],
        [4.9689e-08, 1.0000e+00],
        [1.3122e-10, 1.0000e+00],
        [9.9983e-01, 1.6897e-04],
        [1.0552e-06, 1.0000e+00],
        [2.2503e-08, 1.0000e+00],
        [9.9953e-01, 4.6786e-04],
        [1.8352e-06, 1.0000e+00],
        [9.9986e-01, 1.3978e-04],
        [2.1467e-03, 9.9785e-01],
        [8.0818e-07, 1.0000e+00],
        [1.8652e-08, 1.0000e+00],
        [1.3748e-02, 9.8625e-01],
        [1.8890e-04, 9.9981e-01],
        [6.5302e-03, 9.9347e-01],
        [9.5099e-01, 4.9011e-02],
        [8.2685e-05, 9.9992e-01],
        [6.4896e-01, 3.5104e-01],
        [5.5481e-07, 1.0000e+00],
        [1.7537e-05, 9.9998e-01],
        [9.4496e-06, 9.9999e-01],
        [6.9823e-06, 9.9999e-01],
        [3.5334e-05, 9.9996e-01],
        [7.3174e-07, 1.0000e+00],
        [2.6508e-04, 9.9973e-01],
        [4.9808e-03, 9.9502e-01],
        [4.5910e-04, 9.9954e-01],
        [1.7664e-07, 1.0000e+00],
        [3.8529e-08, 1.0000e+00],
        [2.2503e-08, 1.0000e+00],
        [2.7250e-04, 9.9973e-01],
        [9.9959e-01, 4.0958e-04],
        [9.9970e-01, 3.0015e-04],
        [3.5120e-04, 9.9965e-01],
        [9.5883e-06, 9.9999e-01],
        [1.0532e-03, 9.9895e-01],
        [4.5103e-08, 1.0000e+00],
        [9.2368e-06, 9.9999e-01],
        [9.9972e-01, 2.8377e-04],
        [1.1556e-01, 8.8444e-01],
        [3.7143e-01, 6.2857e-01],
        [1.1872e-05, 9.9999e-01],
        [4.4538e-07, 1.0000e+00],
        [8.4402e-11, 1.0000e+00],
        [6.0635e-03, 9.9394e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2975e-06, 1.0000e+00],
        [1.8480e-06, 1.0000e+00],
        [3.0569e-07, 1.0000e+00],
        [8.4071e-03, 9.9159e-01],
        [1.1203e-08, 1.0000e+00],
        [1.0926e-06, 1.0000e+00],
        [2.4440e-06, 1.0000e+00],
        [4.5960e-06, 1.0000e+00],
        [7.7575e-04, 9.9922e-01],
        [4.6046e-04, 9.9954e-01],
        [6.0246e-03, 9.9398e-01],
        [9.1139e-01, 8.8607e-02],
        [3.6628e-05, 9.9996e-01],
        [3.0461e-05, 9.9997e-01],
        [9.9494e-01, 5.0592e-03],
        [1.7293e-08, 1.0000e+00],
        [3.7452e-06, 1.0000e+00],
        [9.9182e-11, 1.0000e+00],
        [5.8425e-07, 1.0000e+00],
        [4.7801e-10, 1.0000e+00],
        [9.9977e-01, 2.2862e-04],
        [9.9962e-01, 3.8005e-04],
        [1.5043e-07, 1.0000e+00],
        [3.1305e-08, 1.0000e+00],
        [3.5876e-08, 1.0000e+00],
        [1.3286e-05, 9.9999e-01],
        [1.2350e-07, 1.0000e+00],
        [2.3782e-08, 1.0000e+00],
        [3.4678e-05, 9.9997e-01],
        [8.8239e-05, 9.9991e-01],
        [2.5710e-06, 1.0000e+00],
        [7.4286e-06, 9.9999e-01],
        [8.7734e-09, 1.0000e+00],
        [9.9962e-01, 3.8459e-04],
        [4.5762e-06, 1.0000e+00],
        [2.4051e-05, 9.9998e-01],
        [9.5804e-01, 4.1964e-02],
        [7.1482e-07, 1.0000e+00],
        [1.8893e-05, 9.9998e-01],
        [9.9959e-01, 4.1444e-04],
        [9.5981e-12, 1.0000e+00],
        [2.2031e-06, 1.0000e+00],
        [5.4573e-07, 1.0000e+00],
        [3.2664e-07, 1.0000e+00],
        [3.2232e-07, 1.0000e+00],
        [5.9728e-04, 9.9940e-01],
        [6.6735e-04, 9.9933e-01],
        [9.9985e-01, 1.5100e-04],
        [5.8015e-09, 1.0000e+00],
        [1.5495e-07, 1.0000e+00],
        [5.5489e-04, 9.9945e-01],
        [1.1681e-03, 9.9883e-01],
        [1.6226e-04, 9.9984e-01],
        [2.6169e-07, 1.0000e+00],
        [2.2639e-08, 1.0000e+00],
        [1.4711e-06, 1.0000e+00],
        [1.9473e-08, 1.0000e+00],
        [2.9626e-09, 1.0000e+00],
        [3.3798e-08, 1.0000e+00],
        [9.9996e-01, 4.0762e-05],
        [1.4401e-05, 9.9999e-01],
        [1.0000e+00, 5.0214e-07],
        [3.8127e-06, 1.0000e+00],
        [9.9978e-01, 2.2163e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0975e-06, 1.0000e+00],
        [4.8853e-08, 1.0000e+00],
        [6.2079e-09, 1.0000e+00],
        [9.9867e-01, 1.3325e-03],
        [9.9712e-01, 2.8795e-03],
        [1.0473e-06, 1.0000e+00],
        [8.7812e-05, 9.9991e-01],
        [7.0478e-08, 1.0000e+00],
        [1.3666e-09, 1.0000e+00],
        [4.0222e-07, 1.0000e+00],
        [9.9935e-01, 6.5228e-04],
        [1.4975e-09, 1.0000e+00],
        [1.5488e-08, 1.0000e+00],
        [9.9991e-01, 9.4274e-05],
        [7.2536e-06, 9.9999e-01],
        [3.9486e-01, 6.0514e-01],
        [9.9877e-01, 1.2307e-03],
        [5.2675e-07, 1.0000e+00],
        [3.5176e-10, 1.0000e+00],
        [9.9954e-01, 4.5863e-04],
        [9.9407e-01, 5.9264e-03],
        [7.0363e-06, 9.9999e-01],
        [1.9772e-06, 1.0000e+00],
        [1.9270e-05, 9.9998e-01],
        [7.2328e-07, 1.0000e+00],
        [5.1829e-10, 1.0000e+00],
        [2.7083e-09, 1.0000e+00],
        [2.2115e-03, 9.9779e-01],
        [9.9296e-01, 7.0351e-03],
        [9.9773e-01, 2.2678e-03],
        [9.9994e-01, 5.6189e-05],
        [2.7066e-05, 9.9997e-01],
        [6.5948e-06, 9.9999e-01],
        [4.1522e-04, 9.9958e-01],
        [6.0730e-08, 1.0000e+00],
        [1.5942e-06, 1.0000e+00],
        [2.8047e-07, 1.0000e+00],
        [1.6390e-06, 1.0000e+00],
        [9.9330e-01, 6.6961e-03],
        [4.1593e-06, 1.0000e+00],
        [9.9903e-01, 9.6542e-04],
        [4.3733e-11, 1.0000e+00],
        [1.6087e-03, 9.9839e-01],
        [1.3483e-02, 9.8652e-01],
        [5.3848e-06, 9.9999e-01],
        [8.5691e-09, 1.0000e+00],
        [6.1063e-04, 9.9939e-01],
        [1.6365e-07, 1.0000e+00],
        [1.1377e-05, 9.9999e-01],
        [7.7259e-06, 9.9999e-01],
        [4.4045e-01, 5.5955e-01],
        [7.9039e-06, 9.9999e-01],
        [3.7967e-07, 1.0000e+00],
        [1.5267e-08, 1.0000e+00],
        [9.7591e-01, 2.4092e-02],
        [1.1371e-07, 1.0000e+00],
        [2.4105e-05, 9.9998e-01],
        [9.9880e-01, 1.2003e-03],
        [1.8155e-04, 9.9982e-01],
        [9.9900e-01, 9.9621e-04],
        [8.7330e-01, 1.2670e-01],
        [3.9206e-06, 1.0000e+00],
        [2.8572e-03, 9.9714e-01],
        [3.4568e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9217e-01, 7.8327e-03],
        [9.9972e-01, 2.7597e-04],
        [6.2997e-08, 1.0000e+00],
        [2.1418e-02, 9.7858e-01],
        [4.3682e-11, 1.0000e+00],
        [6.4056e-07, 1.0000e+00],
        [1.3931e-07, 1.0000e+00],
        [1.3089e-08, 1.0000e+00],
        [3.0741e-07, 1.0000e+00],
        [3.0453e-09, 1.0000e+00],
        [9.4416e-01, 5.5837e-02],
        [2.4495e-10, 1.0000e+00],
        [9.5873e-06, 9.9999e-01],
        [4.6410e-08, 1.0000e+00],
        [2.8749e-07, 1.0000e+00],
        [1.4135e-07, 1.0000e+00],
        [2.0227e-04, 9.9980e-01],
        [4.0067e-09, 1.0000e+00],
        [5.0131e-07, 1.0000e+00],
        [9.9981e-01, 1.9199e-04],
        [9.9985e-01, 1.4564e-04],
        [1.9946e-08, 1.0000e+00],
        [1.2248e-06, 1.0000e+00],
        [8.1336e-08, 1.0000e+00],
        [9.9985e-01, 1.5214e-04],
        [9.9894e-01, 1.0591e-03],
        [8.0915e-07, 1.0000e+00],
        [9.3403e-10, 1.0000e+00],
        [2.3573e-07, 1.0000e+00],
        [8.3010e-04, 9.9917e-01],
        [1.6037e-06, 1.0000e+00],
        [8.9720e-07, 1.0000e+00],
        [9.8903e-01, 1.0966e-02],
        [8.4958e-04, 9.9915e-01],
        [1.9034e-06, 1.0000e+00],
        [3.5434e-06, 1.0000e+00],
        [9.9987e-01, 1.2642e-04],
        [9.8488e-07, 1.0000e+00],
        [9.9994e-01, 5.6399e-05],
        [5.4871e-06, 9.9999e-01],
        [5.0360e-06, 9.9999e-01],
        [9.9968e-01, 3.2191e-04],
        [9.9955e-01, 4.4594e-04],
        [1.8179e-08, 1.0000e+00],
        [9.9923e-01, 7.7089e-04],
        [2.2623e-07, 1.0000e+00],
        [2.1676e-03, 9.9783e-01],
        [2.4398e-05, 9.9998e-01],
        [4.8388e-05, 9.9995e-01],
        [9.9857e-01, 1.4316e-03],
        [8.1394e-10, 1.0000e+00],
        [6.7876e-09, 1.0000e+00],
        [2.9599e-05, 9.9997e-01],
        [2.3912e-06, 1.0000e+00],
        [5.7493e-08, 1.0000e+00],
        [2.3097e-05, 9.9998e-01],
        [2.1552e-07, 1.0000e+00],
        [1.5379e-04, 9.9985e-01],
        [4.5406e-04, 9.9955e-01],
        [9.9940e-01, 6.0476e-04],
        [1.2887e-04, 9.9987e-01],
        [9.9408e-01, 5.9211e-03],
        [2.0015e-04, 9.9980e-01],
        [3.4859e-03, 9.9651e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1441e-03, 9.9886e-01],
        [3.0301e-06, 1.0000e+00],
        [2.5356e-06, 1.0000e+00],
        [1.8431e-07, 1.0000e+00],
        [2.0904e-06, 1.0000e+00],
        [9.1337e-07, 1.0000e+00],
        [9.8320e-01, 1.6803e-02],
        [1.4912e-10, 1.0000e+00],
        [2.6638e-06, 1.0000e+00],
        [5.4060e-05, 9.9995e-01],
        [2.2423e-07, 1.0000e+00],
        [4.2337e-05, 9.9996e-01],
        [5.5747e-06, 9.9999e-01],
        [5.6884e-05, 9.9994e-01],
        [1.1564e-10, 1.0000e+00],
        [1.9906e-05, 9.9998e-01],
        [1.2206e-03, 9.9878e-01],
        [1.2665e-09, 1.0000e+00],
        [1.1354e-09, 1.0000e+00],
        [1.7951e-06, 1.0000e+00],
        [7.2037e-08, 1.0000e+00],
        [1.0937e-02, 9.8906e-01],
        [5.7206e-04, 9.9943e-01],
        [9.7529e-01, 2.4705e-02],
        [5.7935e-08, 1.0000e+00],
        [6.1723e-08, 1.0000e+00],
        [7.5459e-06, 9.9999e-01],
        [1.5401e-08, 1.0000e+00],
        [3.1260e-06, 1.0000e+00],
        [6.8032e-01, 3.1968e-01],
        [8.6642e-01, 1.3358e-01],
        [2.7833e-03, 9.9722e-01],
        [2.0211e-10, 1.0000e+00],
        [9.9995e-01, 5.3757e-05],
        [6.4329e-08, 1.0000e+00],
        [9.9975e-01, 2.5014e-04],
        [4.0182e-05, 9.9996e-01],
        [9.7879e-01, 2.1207e-02],
        [7.3709e-09, 1.0000e+00],
        [4.9616e-05, 9.9995e-01],
        [4.1662e-08, 1.0000e+00],
        [2.2578e-07, 1.0000e+00],
        [3.1066e-07, 1.0000e+00],
        [6.5148e-04, 9.9935e-01],
        [8.3098e-05, 9.9992e-01],
        [2.3878e-07, 1.0000e+00],
        [3.3686e-08, 1.0000e+00],
        [9.7917e-01, 2.0828e-02],
        [6.2591e-03, 9.9374e-01],
        [2.0002e-08, 1.0000e+00],
        [9.6340e-06, 9.9999e-01],
        [1.8121e-08, 1.0000e+00],
        [1.0737e-06, 1.0000e+00],
        [9.9878e-01, 1.2247e-03],
        [2.7195e-02, 9.7280e-01],
        [1.5337e-05, 9.9998e-01],
        [1.3676e-07, 1.0000e+00],
        [1.4384e-05, 9.9999e-01],
        [4.8968e-06, 1.0000e+00],
        [1.5800e-09, 1.0000e+00],
        [9.2581e-05, 9.9991e-01],
        [1.5439e-05, 9.9998e-01],
        [9.9999e-01, 1.3326e-05],
        [3.3350e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.4690e-07, 1.0000e+00],
        [3.4144e-08, 1.0000e+00],
        [1.7360e-02, 9.8264e-01],
        [1.3701e-04, 9.9986e-01],
        [1.9476e-05, 9.9998e-01],
        [1.9195e-05, 9.9998e-01],
        [9.9973e-01, 2.6643e-04],
        [4.3298e-10, 1.0000e+00],
        [1.1305e-07, 1.0000e+00],
        [7.0172e-07, 1.0000e+00],
        [1.7276e-01, 8.2724e-01],
        [1.1991e-07, 1.0000e+00],
        [4.2302e-05, 9.9996e-01],
        [1.7353e-06, 1.0000e+00],
        [9.3302e-01, 6.6983e-02],
        [2.1931e-04, 9.9978e-01],
        [9.8874e-01, 1.1258e-02],
        [2.9540e-08, 1.0000e+00],
        [8.9009e-08, 1.0000e+00],
        [6.3583e-01, 3.6417e-01],
        [8.1660e-06, 9.9999e-01],
        [5.3354e-06, 9.9999e-01],
        [2.0026e-08, 1.0000e+00],
        [1.7775e-04, 9.9982e-01],
        [7.2972e-07, 1.0000e+00],
        [1.5580e-07, 1.0000e+00],
        [9.9010e-01, 9.9014e-03],
        [2.9003e-06, 1.0000e+00],
        [8.0330e-06, 9.9999e-01],
        [1.4978e-04, 9.9985e-01],
        [2.4531e-09, 1.0000e+00],
        [4.4747e-06, 1.0000e+00],
        [5.3722e-05, 9.9995e-01],
        [7.6739e-09, 1.0000e+00],
        [9.9940e-01, 5.9511e-04],
        [7.9709e-01, 2.0291e-01],
        [5.9787e-07, 1.0000e+00],
        [4.1611e-08, 1.0000e+00],
        [2.5660e-08, 1.0000e+00],
        [2.3967e-08, 1.0000e+00],
        [9.9988e-01, 1.2181e-04],
        [9.9941e-01, 5.8737e-04],
        [9.9974e-01, 2.6364e-04],
        [2.0383e-05, 9.9998e-01],
        [7.7520e-05, 9.9992e-01],
        [9.9992e-01, 8.4982e-05],
        [1.2714e-04, 9.9987e-01],
        [2.4406e-05, 9.9998e-01],
        [3.6128e-06, 1.0000e+00],
        [1.0201e-06, 1.0000e+00],
        [2.1834e-08, 1.0000e+00],
        [4.4037e-06, 1.0000e+00],
        [9.9970e-01, 2.9605e-04],
        [8.2075e-01, 1.7925e-01],
        [6.8100e-05, 9.9993e-01],
        [5.3354e-05, 9.9995e-01],
        [2.8436e-06, 1.0000e+00],
        [3.5027e-07, 1.0000e+00],
        [5.1864e-03, 9.9481e-01],
        [1.9584e-04, 9.9980e-01],
        [4.3700e-09, 1.0000e+00],
        [2.8820e-04, 9.9971e-01],
        [9.9937e-01, 6.3190e-04],
        [4.0146e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3371e-08, 1.0000e+00],
        [7.0883e-09, 1.0000e+00],
        [3.4889e-09, 1.0000e+00],
        [4.4314e-10, 1.0000e+00],
        [9.5060e-01, 4.9396e-02],
        [4.9170e-08, 1.0000e+00],
        [6.1457e-10, 1.0000e+00],
        [5.5644e-01, 4.4356e-01],
        [9.9888e-01, 1.1227e-03],
        [3.1933e-05, 9.9997e-01],
        [3.5588e-07, 1.0000e+00],
        [6.4200e-09, 1.0000e+00],
        [1.6149e-04, 9.9984e-01],
        [1.3450e-07, 1.0000e+00],
        [2.3777e-05, 9.9998e-01],
        [2.2977e-02, 9.7702e-01],
        [9.9983e-01, 1.6811e-04],
        [1.7487e-03, 9.9825e-01],
        [2.4178e-08, 1.0000e+00],
        [9.2531e-01, 7.4692e-02],
        [5.0421e-05, 9.9995e-01],
        [1.5753e-03, 9.9842e-01],
        [2.8125e-06, 1.0000e+00],
        [1.2279e-07, 1.0000e+00],
        [1.2389e-03, 9.9876e-01],
        [9.9613e-01, 3.8704e-03],
        [6.9021e-09, 1.0000e+00],
        [8.3818e-07, 1.0000e+00],
        [3.0982e-07, 1.0000e+00],
        [4.3289e-03, 9.9567e-01],
        [1.3433e-02, 9.8657e-01],
        [8.8572e-05, 9.9991e-01],
        [6.8457e-01, 3.1543e-01],
        [1.6728e-05, 9.9998e-01],
        [1.6132e-04, 9.9984e-01],
        [2.5332e-08, 1.0000e+00],
        [6.7123e-07, 1.0000e+00],
        [4.5892e-09, 1.0000e+00],
        [2.8591e-08, 1.0000e+00],
        [6.6610e-08, 1.0000e+00],
        [8.2091e-08, 1.0000e+00],
        [9.9217e-04, 9.9901e-01],
        [1.3660e-07, 1.0000e+00],
        [2.7094e-07, 1.0000e+00],
        [2.5675e-04, 9.9974e-01],
        [9.9947e-01, 5.3185e-04],
        [4.2091e-05, 9.9996e-01],
        [6.7610e-07, 1.0000e+00],
        [6.8023e-01, 3.1977e-01],
        [9.1507e-10, 1.0000e+00],
        [9.9985e-01, 1.4940e-04],
        [2.5248e-04, 9.9975e-01],
        [9.9979e-01, 2.0782e-04],
        [1.5542e-08, 1.0000e+00],
        [9.6083e-05, 9.9990e-01],
        [9.9950e-01, 5.0478e-04],
        [5.0279e-07, 1.0000e+00],
        [2.5139e-01, 7.4861e-01],
        [1.3097e-06, 1.0000e+00],
        [6.9170e-08, 1.0000e+00],
        [4.1794e-09, 1.0000e+00],
        [8.7660e-04, 9.9912e-01],
        [3.2849e-08, 1.0000e+00],
        [9.9955e-01, 4.5495e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9964e-01, 3.6377e-04],
        [9.9286e-01, 7.1363e-03],
        [8.8705e-01, 1.1295e-01],
        [3.4136e-07, 1.0000e+00],
        [9.9633e-01, 3.6721e-03],
        [9.9962e-01, 3.7769e-04],
        [9.9994e-01, 6.1893e-05],
        [2.1974e-07, 1.0000e+00],
        [2.9082e-09, 1.0000e+00],
        [5.7981e-08, 1.0000e+00],
        [8.1505e-01, 1.8495e-01],
        [1.9681e-09, 1.0000e+00],
        [3.6227e-02, 9.6377e-01],
        [9.9873e-01, 1.2663e-03],
        [1.1307e-07, 1.0000e+00],
        [9.9979e-01, 2.0900e-04],
        [1.1925e-06, 1.0000e+00],
        [2.5146e-05, 9.9997e-01],
        [9.9982e-01, 1.8246e-04],
        [5.6484e-06, 9.9999e-01],
        [5.1415e-06, 9.9999e-01],
        [9.7595e-01, 2.4050e-02],
        [1.9463e-06, 1.0000e+00],
        [1.3249e-03, 9.9868e-01],
        [3.0131e-01, 6.9869e-01],
        [9.9909e-01, 9.1425e-04],
        [2.9297e-09, 1.0000e+00],
        [2.4261e-03, 9.9757e-01],
        [7.6927e-08, 1.0000e+00],
        [9.6824e-01, 3.1761e-02],
        [4.9958e-03, 9.9500e-01],
        [9.9753e-01, 2.4701e-03],
        [9.9959e-01, 4.0544e-04],
        [2.2818e-08, 1.0000e+00],
        [1.1180e-06, 1.0000e+00],
        [5.5031e-07, 1.0000e+00],
        [9.9942e-01, 5.7960e-04],
        [1.3057e-09, 1.0000e+00],
        [9.9989e-01, 1.1242e-04],
        [1.6731e-08, 1.0000e+00],
        [3.7695e-10, 1.0000e+00],
        [3.0273e-10, 1.0000e+00],
        [1.7315e-06, 1.0000e+00],
        [9.9927e-01, 7.3493e-04],
        [9.4319e-07, 1.0000e+00],
        [1.3218e-02, 9.8678e-01],
        [1.9901e-06, 1.0000e+00],
        [5.1372e-05, 9.9995e-01],
        [7.0903e-05, 9.9993e-01],
        [8.7326e-01, 1.2674e-01],
        [5.9308e-01, 4.0692e-01],
        [9.9995e-01, 4.9901e-05],
        [9.7987e-06, 9.9999e-01],
        [2.7664e-05, 9.9997e-01],
        [9.9972e-01, 2.8261e-04],
        [2.6532e-09, 1.0000e+00],
        [9.9973e-01, 2.6849e-04],
        [8.2956e-04, 9.9917e-01],
        [9.9932e-01, 6.8196e-04],
        [9.4083e-01, 5.9172e-02],
        [1.3962e-06, 1.0000e+00],
        [8.8494e-01, 1.1506e-01],
        [2.9835e-06, 1.0000e+00],
        [4.6822e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1508e-04, 9.9988e-01],
        [3.8008e-07, 1.0000e+00],
        [7.5479e-08, 1.0000e+00],
        [9.9966e-01, 3.4026e-04],
        [8.9430e-08, 1.0000e+00],
        [1.3115e-05, 9.9999e-01],
        [9.9990e-01, 1.0490e-04],
        [9.9921e-01, 7.8585e-04],
        [3.4160e-04, 9.9966e-01],
        [2.8346e-02, 9.7165e-01],
        [3.0589e-02, 9.6941e-01],
        [2.1434e-08, 1.0000e+00],
        [9.9857e-01, 1.4349e-03],
        [9.0720e-01, 9.2796e-02],
        [5.0667e-04, 9.9949e-01],
        [3.9382e-07, 1.0000e+00],
        [1.6915e-07, 1.0000e+00],
        [2.3504e-04, 9.9976e-01],
        [1.1836e-05, 9.9999e-01],
        [3.2433e-05, 9.9997e-01],
        [1.1114e-05, 9.9999e-01],
        [9.4102e-03, 9.9059e-01],
        [4.0155e-09, 1.0000e+00],
        [3.3199e-03, 9.9668e-01],
        [2.5222e-09, 1.0000e+00],
        [1.2939e-01, 8.7061e-01],
        [9.9991e-01, 9.0344e-05],
        [1.6594e-06, 1.0000e+00],
        [9.9986e-01, 1.3830e-04],
        [9.3290e-01, 6.7101e-02],
        [7.9429e-04, 9.9921e-01],
        [9.9836e-01, 1.6355e-03],
        [8.3179e-09, 1.0000e+00],
        [2.3613e-07, 1.0000e+00],
        [2.4338e-06, 1.0000e+00],
        [9.9931e-01, 6.8639e-04],
        [3.3834e-06, 1.0000e+00],
        [1.0744e-08, 1.0000e+00],
        [9.9979e-01, 2.0506e-04],
        [2.5631e-05, 9.9997e-01],
        [9.9982e-01, 1.8243e-04],
        [9.9977e-01, 2.2744e-04],
        [8.1194e-05, 9.9992e-01],
        [2.0549e-08, 1.0000e+00],
        [9.9989e-01, 1.1140e-04],
        [9.9804e-01, 1.9609e-03],
        [9.9029e-01, 9.7109e-03],
        [6.1124e-05, 9.9994e-01],
        [9.9545e-01, 4.5497e-03],
        [9.9982e-01, 1.8054e-04],
        [9.5787e-07, 1.0000e+00],
        [1.1393e-04, 9.9989e-01],
        [8.9423e-08, 1.0000e+00],
        [5.3248e-05, 9.9995e-01],
        [1.5198e-05, 9.9998e-01],
        [7.0489e-03, 9.9295e-01],
        [9.9993e-01, 6.6693e-05],
        [9.9986e-01, 1.3720e-04],
        [1.4722e-05, 9.9999e-01],
        [4.5200e-04, 9.9955e-01],
        [4.9782e-06, 9.9999e-01],
        [9.9973e-01, 2.6661e-04],
        [1.9598e-07, 1.0000e+00],
        [1.1243e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8981e-01, 1.0188e-02],
        [1.3777e-07, 1.0000e+00],
        [5.1665e-03, 9.9483e-01],
        [1.1445e-01, 8.8555e-01],
        [9.6611e-01, 3.3893e-02],
        [9.9983e-01, 1.7175e-04],
        [8.8881e-01, 1.1119e-01],
        [7.6879e-02, 9.2312e-01],
        [4.5067e-04, 9.9955e-01],
        [1.5731e-06, 1.0000e+00],
        [4.7611e-01, 5.2389e-01],
        [6.9006e-06, 9.9999e-01],
        [7.5347e-07, 1.0000e+00],
        [6.5863e-03, 9.9341e-01],
        [9.9777e-01, 2.2252e-03],
        [1.1894e-03, 9.9881e-01],
        [3.8623e-04, 9.9961e-01],
        [1.2940e-05, 9.9999e-01],
        [9.9985e-01, 1.5483e-04],
        [2.7000e-04, 9.9973e-01],
        [1.8780e-02, 9.8122e-01],
        [2.9523e-04, 9.9970e-01],
        [9.9974e-01, 2.5715e-04],
        [1.9657e-03, 9.9803e-01],
        [7.2649e-05, 9.9993e-01],
        [7.0490e-03, 9.9295e-01],
        [9.9961e-01, 3.9264e-04],
        [2.3526e-03, 9.9765e-01],
        [6.0652e-05, 9.9994e-01],
        [2.4417e-07, 1.0000e+00],
        [9.9963e-01, 3.7306e-04],
        [1.3113e-08, 1.0000e+00],
        [5.3142e-02, 9.4686e-01],
        [1.4168e-07, 1.0000e+00],
        [1.6718e-07, 1.0000e+00],
        [1.6718e-04, 9.9983e-01],
        [2.1662e-06, 1.0000e+00],
        [9.9731e-01, 2.6866e-03],
        [3.0075e-03, 9.9699e-01],
        [1.8921e-07, 1.0000e+00],
        [1.0464e-05, 9.9999e-01],
        [1.1587e-04, 9.9988e-01],
        [2.6389e-03, 9.9736e-01],
        [8.8533e-01, 1.1467e-01],
        [9.9955e-01, 4.5497e-04],
        [3.0177e-05, 9.9997e-01],
        [1.9319e-01, 8.0681e-01],
        [2.3733e-03, 9.9763e-01],
        [9.9897e-01, 1.0276e-03],
        [2.1784e-08, 1.0000e+00],
        [2.0462e-04, 9.9980e-01],
        [9.9435e-01, 5.6457e-03],
        [2.1812e-03, 9.9782e-01],
        [1.6136e-03, 9.9839e-01],
        [3.6321e-08, 1.0000e+00],
        [1.6290e-03, 9.9837e-01],
        [9.9989e-01, 1.1338e-04],
        [9.9971e-01, 2.9110e-04],
        [9.8061e-01, 1.9393e-02],
        [7.7604e-12, 1.0000e+00],
        [1.4512e-04, 9.9985e-01],
        [9.9898e-01, 1.0209e-03],
        [4.3192e-03, 9.9568e-01],
        [1.9664e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.6795e-06, 9.9999e-01],
        [1.8020e-03, 9.9820e-01],
        [3.8896e-07, 1.0000e+00],
        [5.7611e-04, 9.9942e-01],
        [1.2144e-02, 9.8786e-01],
        [5.7101e-06, 9.9999e-01],
        [9.9977e-01, 2.2891e-04],
        [8.0077e-05, 9.9992e-01],
        [9.9998e-01, 2.2507e-05],
        [2.5211e-04, 9.9975e-01],
        [1.4170e-07, 1.0000e+00],
        [5.5831e-06, 9.9999e-01],
        [9.9998e-01, 1.9266e-05],
        [9.9867e-01, 1.3283e-03],
        [5.8926e-06, 9.9999e-01],
        [9.9999e-01, 1.4552e-05],
        [9.9974e-01, 2.6004e-04],
        [9.9987e-01, 1.2956e-04],
        [1.5025e-07, 1.0000e+00],
        [9.1502e-01, 8.4976e-02],
        [6.8552e-04, 9.9931e-01],
        [3.2494e-05, 9.9997e-01],
        [9.9910e-01, 9.0187e-04],
        [2.8786e-09, 1.0000e+00],
        [1.0396e-02, 9.8960e-01],
        [5.9328e-07, 1.0000e+00],
        [1.1533e-06, 1.0000e+00],
        [1.6789e-01, 8.3211e-01],
        [8.1471e-07, 1.0000e+00],
        [1.0000e+00, 8.8221e-07],
        [9.9998e-01, 2.3736e-05],
        [1.4762e-06, 1.0000e+00],
        [7.0072e-05, 9.9993e-01],
        [4.7930e-05, 9.9995e-01],
        [9.9972e-01, 2.8169e-04],
        [1.4001e-03, 9.9860e-01],
        [2.5459e-06, 1.0000e+00],
        [1.5033e-04, 9.9985e-01],
        [1.8647e-04, 9.9981e-01],
        [4.0939e-04, 9.9959e-01],
        [2.6651e-05, 9.9997e-01],
        [3.4524e-06, 1.0000e+00],
        [7.8092e-07, 1.0000e+00],
        [8.8423e-08, 1.0000e+00],
        [1.6022e-06, 1.0000e+00],
        [9.6523e-08, 1.0000e+00],
        [7.7474e-04, 9.9923e-01],
        [4.4435e-07, 1.0000e+00],
        [2.5406e-05, 9.9997e-01],
        [9.9974e-01, 2.6004e-04],
        [9.9977e-01, 2.2979e-04],
        [9.9977e-01, 2.3325e-04],
        [1.5985e-04, 9.9984e-01],
        [1.7271e-07, 1.0000e+00],
        [9.9967e-01, 3.3059e-04],
        [1.7757e-03, 9.9822e-01],
        [1.0082e-01, 8.9918e-01],
        [4.0184e-01, 5.9816e-01],
        [4.0874e-03, 9.9591e-01],
        [9.9891e-01, 1.0889e-03],
        [1.1745e-06, 1.0000e+00],
        [1.4113e-04, 9.9986e-01],
        [5.6737e-08, 1.0000e+00],
        [1.8160e-04, 9.9982e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.4069e-06, 1.0000e+00],
        [7.6465e-05, 9.9992e-01],
        [3.5428e-04, 9.9965e-01],
        [1.5739e-02, 9.8426e-01],
        [3.5264e-06, 1.0000e+00],
        [9.9953e-01, 4.6569e-04],
        [3.4462e-10, 1.0000e+00],
        [1.9970e-05, 9.9998e-01],
        [9.9997e-01, 2.8245e-05],
        [1.0584e-03, 9.9894e-01],
        [9.9992e-01, 8.4388e-05],
        [1.0998e-04, 9.9989e-01],
        [1.0000e+00, 2.1404e-06],
        [5.6505e-05, 9.9994e-01],
        [1.8444e-03, 9.9816e-01],
        [8.0754e-04, 9.9919e-01],
        [9.9968e-01, 3.1844e-04],
        [8.4873e-06, 9.9999e-01],
        [9.9961e-01, 3.9290e-04],
        [2.3723e-03, 9.9763e-01],
        [3.8730e-07, 1.0000e+00],
        [9.9981e-01, 1.8840e-04],
        [4.6290e-07, 1.0000e+00],
        [1.0908e-04, 9.9989e-01],
        [4.3152e-03, 9.9568e-01],
        [9.8512e-01, 1.4877e-02],
        [9.9980e-01, 2.0030e-04],
        [2.8264e-01, 7.1736e-01],
        [1.2389e-04, 9.9988e-01],
        [9.9850e-01, 1.5042e-03],
        [9.9981e-01, 1.9252e-04],
        [2.5265e-08, 1.0000e+00],
        [2.5130e-09, 1.0000e+00],
        [8.7679e-09, 1.0000e+00],
        [9.9942e-01, 5.8393e-04],
        [9.9990e-01, 9.8950e-05],
        [3.2276e-05, 9.9997e-01],
        [1.6228e-03, 9.9838e-01],
        [9.9989e-01, 1.1449e-04],
        [2.2664e-07, 1.0000e+00],
        [3.3497e-06, 1.0000e+00],
        [9.8721e-01, 1.2787e-02],
        [9.9939e-01, 6.0726e-04],
        [6.4967e-06, 9.9999e-01],
        [1.4227e-07, 1.0000e+00],
        [2.2645e-04, 9.9977e-01],
        [7.4249e-05, 9.9993e-01],
        [4.7074e-03, 9.9529e-01],
        [3.1209e-01, 6.8791e-01],
        [2.0555e-07, 1.0000e+00],
        [1.1770e-02, 9.8823e-01],
        [9.9940e-01, 5.9942e-04],
        [3.0237e-04, 9.9970e-01],
        [9.9971e-01, 2.8538e-04],
        [3.6618e-04, 9.9963e-01],
        [9.9972e-01, 2.7922e-04],
        [2.4163e-02, 9.7584e-01],
        [1.7219e-05, 9.9998e-01],
        [9.9987e-01, 1.2726e-04],
        [4.6385e-06, 1.0000e+00],
        [2.1554e-06, 1.0000e+00],
        [9.9981e-01, 1.8558e-04],
        [4.4060e-05, 9.9996e-01],
        [1.3649e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2092e-02, 9.6791e-01],
        [4.6902e-05, 9.9995e-01],
        [9.9973e-01, 2.7454e-04],
        [9.2831e-01, 7.1690e-02],
        [3.5289e-05, 9.9996e-01],
        [3.4396e-06, 1.0000e+00],
        [1.3021e-07, 1.0000e+00],
        [4.7751e-03, 9.9522e-01],
        [9.9993e-01, 6.8867e-05],
        [9.0380e-06, 9.9999e-01],
        [2.6704e-03, 9.9733e-01],
        [5.0395e-02, 9.4960e-01],
        [3.9461e-05, 9.9996e-01],
        [3.5697e-03, 9.9643e-01],
        [1.8865e-01, 8.1135e-01],
        [9.9968e-01, 3.2041e-04],
        [1.7837e-06, 1.0000e+00],
        [6.0852e-04, 9.9939e-01],
        [2.5007e-02, 9.7499e-01],
        [9.9966e-01, 3.4411e-04],
        [6.3978e-01, 3.6022e-01],
        [2.7298e-04, 9.9973e-01],
        [9.9934e-01, 6.5943e-04],
        [2.1688e-03, 9.9783e-01],
        [3.2263e-05, 9.9997e-01],
        [9.9987e-01, 1.2715e-04],
        [1.9168e-06, 1.0000e+00],
        [9.9978e-01, 2.2274e-04],
        [9.9960e-01, 4.0398e-04],
        [1.5638e-04, 9.9984e-01],
        [6.1271e-08, 1.0000e+00],
        [9.9945e-01, 5.5005e-04],
        [9.9353e-01, 6.4737e-03],
        [2.9286e-06, 1.0000e+00],
        [9.9955e-01, 4.5465e-04],
        [1.4639e-04, 9.9985e-01],
        [1.8422e-02, 9.8158e-01],
        [4.1297e-03, 9.9587e-01],
        [9.6411e-07, 1.0000e+00],
        [3.3575e-05, 9.9997e-01],
        [9.5850e-04, 9.9904e-01],
        [2.1103e-05, 9.9998e-01],
        [9.8832e-01, 1.1681e-02],
        [7.2892e-01, 2.7108e-01],
        [1.7592e-05, 9.9998e-01],
        [3.1068e-04, 9.9969e-01],
        [7.8956e-06, 9.9999e-01],
        [3.4591e-08, 1.0000e+00],
        [1.5824e-05, 9.9998e-01],
        [9.9987e-01, 1.3277e-04],
        [9.9974e-01, 2.6238e-04],
        [2.6721e-09, 1.0000e+00],
        [9.9973e-01, 2.7053e-04],
        [1.4873e-07, 1.0000e+00],
        [2.4395e-07, 1.0000e+00],
        [2.7790e-07, 1.0000e+00],
        [2.3707e-04, 9.9976e-01],
        [5.9924e-06, 9.9999e-01],
        [1.5848e-01, 8.4152e-01],
        [2.4162e-05, 9.9998e-01],
        [9.9864e-01, 1.3639e-03],
        [8.2632e-06, 9.9999e-01],
        [1.3672e-07, 1.0000e+00],
        [9.2368e-04, 9.9908e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8593e-06, 9.9999e-01],
        [2.2858e-05, 9.9998e-01],
        [9.9920e-01, 8.0246e-04],
        [9.9986e-01, 1.3651e-04],
        [9.9770e-01, 2.3017e-03],
        [4.2862e-06, 1.0000e+00],
        [9.9964e-01, 3.5536e-04],
        [5.7318e-05, 9.9994e-01],
        [1.2073e-04, 9.9988e-01],
        [2.1308e-07, 1.0000e+00],
        [9.8239e-02, 9.0176e-01],
        [3.2784e-05, 9.9997e-01],
        [9.9778e-01, 2.2226e-03],
        [9.9987e-01, 1.3112e-04],
        [1.1626e-02, 9.8837e-01],
        [4.8421e-04, 9.9952e-01],
        [3.0298e-05, 9.9997e-01],
        [2.4903e-04, 9.9975e-01],
        [3.6496e-05, 9.9996e-01],
        [6.1689e-05, 9.9994e-01],
        [1.9453e-03, 9.9805e-01],
        [1.6836e-04, 9.9983e-01],
        [1.0032e-03, 9.9900e-01],
        [1.4371e-04, 9.9986e-01],
        [9.9978e-01, 2.2191e-04],
        [1.7057e-07, 1.0000e+00],
        [1.8346e-03, 9.9817e-01],
        [9.2541e-01, 7.4592e-02],
        [9.9983e-01, 1.6518e-04],
        [9.9949e-01, 5.1118e-04],
        [1.8618e-05, 9.9998e-01],
        [9.5974e-01, 4.0260e-02],
        [8.2628e-01, 1.7372e-01],
        [4.9325e-06, 1.0000e+00],
        [8.6564e-08, 1.0000e+00],
        [1.2421e-05, 9.9999e-01],
        [1.6362e-03, 9.9836e-01],
        [6.8328e-05, 9.9993e-01],
        [1.5242e-08, 1.0000e+00],
        [9.9995e-01, 5.1887e-05],
        [2.1598e-02, 9.7840e-01],
        [3.5480e-08, 1.0000e+00],
        [9.9973e-01, 2.7273e-04],
        [9.9953e-01, 4.6902e-04],
        [3.3691e-06, 1.0000e+00],
        [2.6401e-04, 9.9974e-01],
        [2.5591e-05, 9.9997e-01],
        [6.5950e-08, 1.0000e+00],
        [1.2140e-03, 9.9879e-01],
        [9.9993e-01, 7.3971e-05],
        [2.2257e-05, 9.9998e-01],
        [1.5536e-04, 9.9984e-01],
        [1.6616e-01, 8.3384e-01],
        [2.5847e-02, 9.7415e-01],
        [1.7377e-07, 1.0000e+00],
        [1.5310e-04, 9.9985e-01],
        [2.2269e-04, 9.9978e-01],
        [9.8894e-01, 1.1061e-02],
        [5.3247e-03, 9.9468e-01],
        [3.6750e-02, 9.6325e-01],
        [1.0000e+00, 4.5116e-06],
        [2.1066e-04, 9.9979e-01],
        [2.0577e-03, 9.9794e-01],
        [3.4842e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9961e-01, 3.8980e-04],
        [9.9988e-01, 1.2312e-04],
        [9.9982e-01, 1.7526e-04],
        [2.7147e-06, 1.0000e+00],
        [8.6960e-01, 1.3040e-01],
        [9.9943e-01, 5.7369e-04],
        [3.9227e-06, 1.0000e+00],
        [9.9977e-01, 2.3444e-04],
        [3.7215e-06, 1.0000e+00],
        [9.9029e-01, 9.7081e-03],
        [8.4924e-06, 9.9999e-01],
        [4.7659e-08, 1.0000e+00],
        [2.0748e-03, 9.9793e-01],
        [1.2461e-04, 9.9988e-01],
        [2.2793e-09, 1.0000e+00],
        [3.5187e-03, 9.9648e-01],
        [9.9964e-01, 3.5539e-04],
        [3.4192e-07, 1.0000e+00],
        [2.6211e-07, 1.0000e+00],
        [2.4127e-02, 9.7587e-01],
        [2.2747e-05, 9.9998e-01],
        [9.9953e-01, 4.6912e-04],
        [3.7720e-06, 1.0000e+00],
        [1.2426e-06, 1.0000e+00],
        [5.1305e-01, 4.8695e-01],
        [3.5908e-03, 9.9641e-01],
        [9.0101e-02, 9.0990e-01],
        [5.2729e-07, 1.0000e+00],
        [2.0578e-04, 9.9979e-01],
        [4.7466e-05, 9.9995e-01],
        [9.9594e-01, 4.0596e-03],
        [2.5089e-06, 1.0000e+00],
        [7.2380e-06, 9.9999e-01],
        [4.8174e-08, 1.0000e+00],
        [9.9858e-01, 1.4242e-03],
        [2.2744e-04, 9.9977e-01],
        [1.7860e-05, 9.9998e-01],
        [2.3900e-08, 1.0000e+00],
        [1.3787e-05, 9.9999e-01],
        [1.1331e-06, 1.0000e+00],
        [8.9953e-05, 9.9991e-01],
        [9.9965e-01, 3.4525e-04],
        [1.2193e-06, 1.0000e+00],
        [7.2711e-07, 1.0000e+00],
        [2.3078e-06, 1.0000e+00],
        [6.5745e-02, 9.3425e-01],
        [2.5032e-03, 9.9750e-01],
        [9.6631e-07, 1.0000e+00],
        [9.9997e-01, 2.8488e-05],
        [1.8749e-07, 1.0000e+00],
        [1.5047e-03, 9.9850e-01],
        [4.4503e-05, 9.9996e-01],
        [9.9968e-01, 3.1980e-04],
        [8.5865e-06, 9.9999e-01],
        [4.1631e-03, 9.9584e-01],
        [1.1618e-08, 1.0000e+00],
        [9.9990e-01, 1.0367e-04],
        [4.4909e-06, 1.0000e+00],
        [9.9966e-01, 3.4450e-04],
        [2.3024e-06, 1.0000e+00],
        [1.1929e-03, 9.9881e-01],
        [1.0831e-03, 9.9892e-01],
        [1.7447e-05, 9.9998e-01],
        [9.8235e-01, 1.7648e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5346e-08, 1.0000e+00],
        [9.9983e-01, 1.7235e-04],
        [3.9749e-08, 1.0000e+00],
        [3.9700e-03, 9.9603e-01],
        [9.9782e-01, 2.1831e-03],
        [5.4599e-04, 9.9945e-01],
        [9.4674e-01, 5.3260e-02],
        [9.9992e-01, 8.0158e-05],
        [1.9189e-06, 1.0000e+00],
        [5.4749e-07, 1.0000e+00],
        [5.6645e-08, 1.0000e+00],
        [1.6100e-06, 1.0000e+00],
        [1.2855e-07, 1.0000e+00],
        [6.1524e-08, 1.0000e+00],
        [7.4774e-08, 1.0000e+00],
        [1.9431e-04, 9.9981e-01],
        [2.3552e-03, 9.9764e-01],
        [2.7933e-06, 1.0000e+00],
        [1.3678e-07, 1.0000e+00],
        [2.6272e-08, 1.0000e+00],
        [4.1312e-07, 1.0000e+00],
        [9.9981e-01, 1.9161e-04],
        [9.4639e-01, 5.3612e-02],
        [9.9715e-01, 2.8543e-03],
        [2.0660e-09, 1.0000e+00],
        [7.0608e-01, 2.9392e-01],
        [9.9557e-01, 4.4348e-03],
        [1.5796e-05, 9.9998e-01],
        [9.9789e-01, 2.1112e-03],
        [9.4917e-06, 9.9999e-01],
        [1.1133e-09, 1.0000e+00],
        [9.9980e-01, 2.0315e-04],
        [3.3798e-05, 9.9997e-01],
        [2.1285e-05, 9.9998e-01],
        [1.7559e-03, 9.9824e-01],
        [6.2569e-09, 1.0000e+00],
        [9.9988e-01, 1.2275e-04],
        [7.4472e-02, 9.2553e-01],
        [3.1690e-07, 1.0000e+00],
        [9.8151e-02, 9.0185e-01],
        [1.1169e-10, 1.0000e+00],
        [1.9397e-04, 9.9981e-01],
        [7.7809e-05, 9.9992e-01],
        [3.9683e-06, 1.0000e+00],
        [3.9774e-05, 9.9996e-01],
        [1.7504e-03, 9.9825e-01],
        [1.7951e-08, 1.0000e+00],
        [5.4681e-06, 9.9999e-01],
        [1.3769e-08, 1.0000e+00],
        [3.3667e-09, 1.0000e+00],
        [9.3237e-01, 6.7626e-02],
        [6.2100e-05, 9.9994e-01],
        [9.9998e-01, 1.7582e-05],
        [1.6097e-02, 9.8390e-01],
        [1.9440e-02, 9.8056e-01],
        [1.4778e-06, 1.0000e+00],
        [3.5675e-03, 9.9643e-01],
        [1.7958e-05, 9.9998e-01],
        [9.9918e-01, 8.2420e-04],
        [7.5187e-07, 1.0000e+00],
        [3.9431e-07, 1.0000e+00],
        [9.9961e-01, 3.8955e-04],
        [4.0495e-03, 9.9595e-01],
        [5.9558e-05, 9.9994e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.4504e-05, 9.9991e-01],
        [9.3233e-05, 9.9991e-01],
        [7.7435e-11, 1.0000e+00],
        [3.5650e-05, 9.9996e-01],
        [1.9139e-06, 1.0000e+00],
        [1.6925e-08, 1.0000e+00],
        [6.6626e-07, 1.0000e+00],
        [5.0937e-07, 1.0000e+00],
        [9.9997e-01, 2.8039e-05],
        [8.1794e-08, 1.0000e+00],
        [9.9987e-01, 1.3380e-04],
        [2.1542e-07, 1.0000e+00],
        [9.3419e-01, 6.5806e-02],
        [9.9089e-01, 9.1131e-03],
        [1.5357e-06, 1.0000e+00],
        [3.3693e-04, 9.9966e-01],
        [9.5376e-03, 9.9046e-01],
        [9.9998e-01, 1.6798e-05],
        [3.0225e-05, 9.9997e-01],
        [5.2685e-04, 9.9947e-01],
        [2.4634e-08, 1.0000e+00],
        [9.9900e-01, 1.0042e-03],
        [2.4284e-04, 9.9976e-01],
        [8.0296e-06, 9.9999e-01],
        [5.9949e-05, 9.9994e-01],
        [1.4250e-05, 9.9999e-01],
        [9.9989e-01, 1.1342e-04],
        [5.2410e-05, 9.9995e-01],
        [1.3747e-05, 9.9999e-01],
        [3.6635e-06, 1.0000e+00],
        [9.9984e-01, 1.6298e-04],
        [2.3860e-07, 1.0000e+00],
        [9.9983e-01, 1.6675e-04],
        [7.3154e-01, 2.6846e-01],
        [9.9987e-01, 1.3148e-04],
        [6.5328e-07, 1.0000e+00],
        [9.9985e-01, 1.4973e-04],
        [7.7527e-05, 9.9992e-01],
        [9.9550e-01, 4.5003e-03],
        [2.7465e-05, 9.9997e-01],
        [2.9827e-07, 1.0000e+00],
        [3.0263e-04, 9.9970e-01],
        [8.0706e-05, 9.9992e-01],
        [1.3192e-03, 9.9868e-01],
        [1.0414e-03, 9.9896e-01],
        [9.9677e-01, 3.2341e-03],
        [9.9978e-01, 2.1758e-04],
        [5.9296e-05, 9.9994e-01],
        [4.5388e-04, 9.9955e-01],
        [1.6155e-03, 9.9838e-01],
        [9.9847e-01, 1.5302e-03],
        [9.9769e-01, 2.3068e-03],
        [2.8767e-03, 9.9712e-01],
        [1.3653e-08, 1.0000e+00],
        [4.2609e-05, 9.9996e-01],
        [2.8802e-02, 9.7120e-01],
        [2.5183e-05, 9.9997e-01],
        [3.5489e-07, 1.0000e+00],
        [9.6231e-01, 3.7695e-02],
        [1.1468e-08, 1.0000e+00],
        [1.3653e-07, 1.0000e+00],
        [3.6353e-03, 9.9636e-01],
        [9.9950e-01, 5.0449e-04],
        [4.4275e-05, 9.9996e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.1027e-08, 1.0000e+00],
        [6.3038e-10, 1.0000e+00],
        [6.0771e-07, 1.0000e+00],
        [8.5467e-06, 9.9999e-01],
        [9.9858e-01, 1.4227e-03],
        [9.9977e-01, 2.2751e-04],
        [8.5826e-07, 1.0000e+00],
        [1.4194e-06, 1.0000e+00],
        [3.2538e-06, 1.0000e+00],
        [2.7408e-05, 9.9997e-01],
        [9.9948e-01, 5.1865e-04],
        [9.9945e-01, 5.4724e-04],
        [6.5774e-07, 1.0000e+00],
        [6.0164e-06, 9.9999e-01],
        [5.6193e-04, 9.9944e-01],
        [9.9793e-01, 2.0724e-03],
        [2.1927e-09, 1.0000e+00],
        [2.5047e-04, 9.9975e-01],
        [4.3050e-06, 1.0000e+00],
        [2.8100e-04, 9.9972e-01],
        [8.7466e-09, 1.0000e+00],
        [4.8335e-02, 9.5167e-01],
        [7.5795e-04, 9.9924e-01],
        [7.2123e-07, 1.0000e+00],
        [9.0109e-06, 9.9999e-01],
        [3.5018e-12, 1.0000e+00],
        [6.1433e-06, 9.9999e-01],
        [9.1470e-07, 1.0000e+00],
        [6.6746e-09, 1.0000e+00],
        [1.4274e-06, 1.0000e+00],
        [7.1540e-07, 1.0000e+00],
        [9.9975e-01, 2.4963e-04],
        [5.7084e-01, 4.2916e-01],
        [4.4291e-06, 1.0000e+00],
        [6.4004e-05, 9.9994e-01],
        [6.9533e-05, 9.9993e-01],
        [8.6304e-05, 9.9991e-01],
        [8.2511e-05, 9.9992e-01],
        [9.9948e-01, 5.2216e-04],
        [9.9957e-01, 4.2599e-04],
        [9.3271e-11, 1.0000e+00],
        [9.5306e-08, 1.0000e+00],
        [9.9967e-01, 3.3317e-04],
        [4.4474e-05, 9.9996e-01],
        [1.8889e-01, 8.1111e-01],
        [9.8104e-01, 1.8956e-02],
        [2.5972e-03, 9.9740e-01],
        [5.0502e-03, 9.9495e-01],
        [1.8184e-03, 9.9818e-01],
        [5.8055e-04, 9.9942e-01],
        [2.3196e-04, 9.9977e-01],
        [1.0635e-08, 1.0000e+00],
        [1.5616e-10, 1.0000e+00],
        [9.9966e-01, 3.4030e-04],
        [3.7766e-05, 9.9996e-01],
        [8.6475e-10, 1.0000e+00],
        [8.4773e-04, 9.9915e-01],
        [1.9029e-07, 1.0000e+00],
        [6.2831e-07, 1.0000e+00],
        [3.4984e-07, 1.0000e+00],
        [9.9968e-01, 3.1885e-04],
        [9.9919e-01, 8.0850e-04],
        [9.9977e-01, 2.2916e-04],
        [9.9979e-01, 2.0821e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3023e-08, 1.0000e+00],
        [9.9910e-01, 8.9561e-04],
        [3.3518e-07, 1.0000e+00],
        [5.3053e-09, 1.0000e+00],
        [9.1118e-08, 1.0000e+00],
        [1.8402e-04, 9.9982e-01],
        [1.7048e-06, 1.0000e+00],
        [2.6642e-04, 9.9973e-01],
        [4.1334e-05, 9.9996e-01],
        [1.4296e-04, 9.9986e-01],
        [9.9864e-01, 1.3567e-03],
        [1.7011e-08, 1.0000e+00],
        [9.8215e-09, 1.0000e+00],
        [2.8813e-02, 9.7119e-01],
        [1.3364e-14, 1.0000e+00],
        [7.9567e-06, 9.9999e-01],
        [3.6810e-04, 9.9963e-01],
        [2.5986e-11, 1.0000e+00],
        [6.2399e-08, 1.0000e+00],
        [9.9394e-01, 6.0554e-03],
        [9.9953e-01, 4.7325e-04],
        [4.9571e-08, 1.0000e+00],
        [9.6684e-10, 1.0000e+00],
        [2.7470e-05, 9.9997e-01],
        [9.9926e-03, 9.9001e-01],
        [2.8908e-06, 1.0000e+00],
        [1.8554e-06, 1.0000e+00],
        [9.9706e-01, 2.9440e-03],
        [5.1707e-08, 1.0000e+00],
        [1.3131e-01, 8.6869e-01],
        [9.9979e-01, 2.1127e-04],
        [8.3598e-05, 9.9992e-01],
        [9.5254e-01, 4.7459e-02],
        [9.9959e-01, 4.0825e-04],
        [7.1995e-07, 1.0000e+00],
        [5.5880e-09, 1.0000e+00],
        [3.5271e-04, 9.9965e-01],
        [2.9459e-06, 1.0000e+00],
        [1.9614e-05, 9.9998e-01],
        [2.8784e-07, 1.0000e+00],
        [3.7193e-06, 1.0000e+00],
        [3.1025e-05, 9.9997e-01],
        [3.4428e-07, 1.0000e+00],
        [4.8221e-04, 9.9952e-01],
        [1.7616e-05, 9.9998e-01],
        [2.2440e-06, 1.0000e+00],
        [1.5042e-01, 8.4958e-01],
        [2.1153e-08, 1.0000e+00],
        [9.9328e-01, 6.7196e-03],
        [9.9981e-01, 1.9432e-04],
        [9.9977e-01, 2.3253e-04],
        [1.8040e-07, 1.0000e+00],
        [9.3661e-05, 9.9991e-01],
        [9.5055e-05, 9.9990e-01],
        [7.8982e-08, 1.0000e+00],
        [9.9977e-01, 2.2992e-04],
        [3.2958e-05, 9.9997e-01],
        [2.6882e-07, 1.0000e+00],
        [1.9864e-07, 1.0000e+00],
        [2.9974e-10, 1.0000e+00],
        [2.8403e-03, 9.9716e-01],
        [4.0950e-01, 5.9050e-01],
        [9.9994e-01, 5.9126e-05],
        [2.1012e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8774e-12, 1.0000e+00],
        [9.9965e-01, 3.4907e-04],
        [1.9833e-07, 1.0000e+00],
        [9.9705e-01, 2.9486e-03],
        [1.2781e-06, 1.0000e+00],
        [4.5276e-09, 1.0000e+00],
        [8.9240e-06, 9.9999e-01],
        [1.3480e-09, 1.0000e+00],
        [7.6025e-11, 1.0000e+00],
        [1.5215e-06, 1.0000e+00],
        [7.7626e-04, 9.9922e-01],
        [7.0164e-04, 9.9930e-01],
        [1.6795e-02, 9.8321e-01],
        [2.0669e-03, 9.9793e-01],
        [1.0912e-04, 9.9989e-01],
        [2.2131e-08, 1.0000e+00],
        [9.9947e-01, 5.3152e-04],
        [2.9012e-06, 1.0000e+00],
        [4.5923e-07, 1.0000e+00],
        [9.9905e-01, 9.4706e-04],
        [9.9826e-01, 1.7412e-03],
        [9.9999e-01, 1.1970e-05],
        [9.9970e-01, 2.9896e-04],
        [2.6910e-07, 1.0000e+00],
        [6.8941e-04, 9.9931e-01],
        [7.5252e-03, 9.9247e-01],
        [9.5751e-01, 4.2489e-02],
        [3.4724e-06, 1.0000e+00],
        [3.3340e-06, 1.0000e+00],
        [9.2659e-08, 1.0000e+00],
        [3.9249e-05, 9.9996e-01],
        [1.2633e-07, 1.0000e+00],
        [5.3231e-09, 1.0000e+00],
        [9.9993e-01, 6.9973e-05],
        [1.1661e-07, 1.0000e+00],
        [2.4771e-07, 1.0000e+00],
        [7.9011e-10, 1.0000e+00],
        [4.4922e-01, 5.5078e-01],
        [5.8363e-05, 9.9994e-01],
        [2.1216e-03, 9.9788e-01],
        [5.3551e-02, 9.4645e-01],
        [9.9955e-01, 4.4988e-04],
        [7.7643e-07, 1.0000e+00],
        [6.6322e-05, 9.9993e-01],
        [9.7213e-06, 9.9999e-01],
        [4.3564e-09, 1.0000e+00],
        [9.9916e-01, 8.3961e-04],
        [2.6012e-06, 1.0000e+00],
        [9.9193e-08, 1.0000e+00],
        [2.7610e-04, 9.9972e-01],
        [4.3184e-05, 9.9996e-01],
        [3.1373e-04, 9.9969e-01],
        [1.1966e-10, 1.0000e+00],
        [1.6623e-03, 9.9834e-01],
        [3.3711e-09, 1.0000e+00],
        [9.9847e-01, 1.5349e-03],
        [8.1529e-06, 9.9999e-01],
        [1.8109e-06, 1.0000e+00],
        [3.5156e-09, 1.0000e+00],
        [7.2250e-06, 9.9999e-01],
        [9.6805e-10, 1.0000e+00],
        [9.9862e-01, 1.3784e-03],
        [1.6756e-07, 1.0000e+00],
        [2.5876e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9733e-01, 2.6653e-03],
        [1.3144e-06, 1.0000e+00],
        [1.6652e-08, 1.0000e+00],
        [2.6618e-10, 1.0000e+00],
        [1.0543e-04, 9.9989e-01],
        [1.1725e-06, 1.0000e+00],
        [5.0727e-01, 4.9273e-01],
        [4.6067e-09, 1.0000e+00],
        [1.8238e-01, 8.1762e-01],
        [1.3222e-02, 9.8678e-01],
        [7.6084e-04, 9.9924e-01],
        [2.5755e-06, 1.0000e+00],
        [6.5556e-07, 1.0000e+00],
        [6.9789e-06, 9.9999e-01],
        [8.7942e-02, 9.1206e-01],
        [9.9993e-01, 7.2246e-05],
        [6.3471e-06, 9.9999e-01],
        [2.3332e-05, 9.9998e-01],
        [8.9786e-06, 9.9999e-01],
        [1.2571e-05, 9.9999e-01],
        [4.5887e-05, 9.9995e-01],
        [9.9133e-01, 8.6675e-03],
        [6.1005e-06, 9.9999e-01],
        [1.1303e-11, 1.0000e+00],
        [2.4142e-06, 1.0000e+00],
        [3.4323e-10, 1.0000e+00],
        [8.4390e-01, 1.5610e-01],
        [9.9952e-01, 4.8011e-04],
        [5.3103e-08, 1.0000e+00],
        [3.2121e-07, 1.0000e+00],
        [2.6252e-04, 9.9974e-01],
        [6.8355e-06, 9.9999e-01],
        [3.5721e-07, 1.0000e+00],
        [2.7098e-02, 9.7290e-01],
        [2.7269e-04, 9.9973e-01],
        [9.9990e-01, 9.8037e-05],
        [1.5645e-06, 1.0000e+00],
        [1.9998e-05, 9.9998e-01],
        [4.4556e-10, 1.0000e+00],
        [3.7243e-05, 9.9996e-01],
        [2.0685e-04, 9.9979e-01],
        [1.1099e-04, 9.9989e-01],
        [3.6718e-02, 9.6328e-01],
        [3.1189e-09, 1.0000e+00],
        [5.9871e-07, 1.0000e+00],
        [1.2564e-07, 1.0000e+00],
        [9.6647e-01, 3.3531e-02],
        [8.1846e-01, 1.8154e-01],
        [8.5032e-01, 1.4968e-01],
        [7.9015e-04, 9.9921e-01],
        [2.2608e-08, 1.0000e+00],
        [3.3476e-05, 9.9997e-01],
        [2.1475e-04, 9.9979e-01],
        [9.9531e-01, 4.6864e-03],
        [1.0874e-06, 1.0000e+00],
        [9.9855e-01, 1.4494e-03],
        [1.6738e-06, 1.0000e+00],
        [4.7097e-07, 1.0000e+00],
        [9.1691e-05, 9.9991e-01],
        [9.0544e-01, 9.4556e-02],
        [9.9355e-01, 6.4452e-03],
        [2.3790e-05, 9.9998e-01],
        [6.8902e-09, 1.0000e+00],
        [9.9951e-01, 4.8546e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[5.8278e-10, 1.0000e+00],
        [9.9639e-01, 3.6061e-03],
        [1.0238e-05, 9.9999e-01],
        [9.9161e-01, 8.3948e-03],
        [9.3929e-01, 6.0706e-02],
        [9.9981e-01, 1.8507e-04],
        [8.5719e-06, 9.9999e-01],
        [6.8998e-01, 3.1002e-01],
        [9.9982e-01, 1.7905e-04],
        [1.0059e-08, 1.0000e+00],
        [8.4449e-09, 1.0000e+00],
        [1.5804e-06, 1.0000e+00],
        [2.2230e-02, 9.7777e-01],
        [7.0570e-01, 2.9430e-01],
        [9.8635e-01, 1.3645e-02],
        [7.8572e-05, 9.9992e-01],
        [9.9827e-01, 1.7285e-03],
        [2.6794e-01, 7.3206e-01],
        [7.0328e-01, 2.9672e-01],
        [5.7181e-07, 1.0000e+00],
        [2.7844e-08, 1.0000e+00],
        [4.8099e-07, 1.0000e+00],
        [2.3962e-08, 1.0000e+00],
        [2.9051e-05, 9.9997e-01],
        [9.4089e-01, 5.9111e-02],
        [9.9574e-01, 4.2570e-03],
        [3.2461e-04, 9.9968e-01],
        [1.1558e-01, 8.8442e-01],
        [9.6401e-01, 3.5993e-02],
        [4.4249e-05, 9.9996e-01],
        [9.0299e-01, 9.7014e-02],
        [1.2178e-05, 9.9999e-01],
        [3.1284e-09, 1.0000e+00],
        [9.9980e-01, 2.0207e-04],
        [9.9941e-01, 5.8777e-04],
        [2.8061e-02, 9.7194e-01],
        [1.3656e-09, 1.0000e+00],
        [5.6128e-06, 9.9999e-01],
        [1.1130e-07, 1.0000e+00],
        [1.0958e-03, 9.9890e-01],
        [1.9708e-04, 9.9980e-01],
        [3.8003e-09, 1.0000e+00],
        [4.2340e-07, 1.0000e+00],
        [9.9819e-01, 1.8090e-03],
        [6.0226e-06, 9.9999e-01],
        [1.6735e-04, 9.9983e-01],
        [2.9061e-06, 1.0000e+00],
        [9.9951e-01, 4.9476e-04],
        [1.3160e-05, 9.9999e-01],
        [4.9018e-08, 1.0000e+00],
        [3.5695e-05, 9.9996e-01],
        [1.4557e-03, 9.9854e-01],
        [7.0331e-06, 9.9999e-01],
        [9.4923e-07, 1.0000e+00],
        [7.8540e-05, 9.9992e-01],
        [9.9363e-01, 6.3708e-03],
        [7.1809e-09, 1.0000e+00],
        [5.9886e-06, 9.9999e-01],
        [2.8719e-03, 9.9713e-01],
        [2.4140e-04, 9.9976e-01],
        [9.9671e-01, 3.2931e-03],
        [4.4003e-07, 1.0000e+00],
        [1.9684e-06, 1.0000e+00],
        [9.3412e-01, 6.5885e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0995e-04, 9.9979e-01],
        [2.2730e-03, 9.9773e-01],
        [3.8645e-03, 9.9614e-01],
        [9.7261e-01, 2.7388e-02],
        [1.3998e-04, 9.9986e-01],
        [2.8903e-04, 9.9971e-01],
        [1.9378e-08, 1.0000e+00],
        [9.9991e-01, 8.6381e-05],
        [4.4089e-06, 1.0000e+00],
        [6.2339e-07, 1.0000e+00],
        [1.6107e-03, 9.9839e-01],
        [1.4989e-07, 1.0000e+00],
        [2.5656e-04, 9.9974e-01],
        [9.9975e-01, 2.4580e-04],
        [9.9983e-01, 1.7464e-04],
        [8.7431e-07, 1.0000e+00],
        [6.5589e-06, 9.9999e-01],
        [1.4726e-08, 1.0000e+00],
        [1.0354e-07, 1.0000e+00],
        [1.9267e-06, 1.0000e+00],
        [3.9256e-07, 1.0000e+00],
        [1.9534e-10, 1.0000e+00],
        [4.5752e-07, 1.0000e+00],
        [9.9976e-01, 2.4098e-04],
        [7.5240e-06, 9.9999e-01],
        [6.6396e-06, 9.9999e-01],
        [8.3649e-01, 1.6351e-01],
        [9.9949e-01, 5.0970e-04],
        [9.9650e-03, 9.9004e-01],
        [3.0078e-05, 9.9997e-01],
        [2.0802e-10, 1.0000e+00],
        [7.2513e-01, 2.7487e-01],
        [3.1731e-03, 9.9683e-01],
        [4.6512e-06, 1.0000e+00],
        [7.9664e-07, 1.0000e+00],
        [6.6320e-09, 1.0000e+00],
        [1.9928e-06, 1.0000e+00],
        [1.4271e-09, 1.0000e+00],
        [1.2907e-07, 1.0000e+00],
        [3.3711e-05, 9.9997e-01],
        [1.7820e-06, 1.0000e+00],
        [2.9160e-06, 1.0000e+00],
        [9.9994e-01, 6.2497e-05],
        [7.2275e-01, 2.7725e-01],
        [2.1904e-06, 1.0000e+00],
        [1.3734e-04, 9.9986e-01],
        [9.9965e-01, 3.5470e-04],
        [4.4434e-05, 9.9996e-01],
        [4.2973e-05, 9.9996e-01],
        [8.9157e-01, 1.0843e-01],
        [5.4305e-06, 9.9999e-01],
        [9.8691e-01, 1.3085e-02],
        [9.9999e-01, 6.4152e-06],
        [7.6598e-10, 1.0000e+00],
        [1.7406e-03, 9.9826e-01],
        [2.5772e-07, 1.0000e+00],
        [4.9317e-06, 1.0000e+00],
        [2.0024e-05, 9.9998e-01],
        [9.3610e-04, 9.9906e-01],
        [1.3776e-05, 9.9999e-01],
        [9.9900e-01, 1.0038e-03],
        [9.9987e-01, 1.3033e-04],
        [2.3346e-07, 1.0000e+00],
        [4.5557e-04, 9.9954e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2093e-06, 1.0000e+00],
        [9.7281e-03, 9.9027e-01],
        [9.6759e-01, 3.2413e-02],
        [4.8026e-07, 1.0000e+00],
        [3.5015e-05, 9.9996e-01],
        [5.6610e-03, 9.9434e-01],
        [9.9985e-01, 1.4777e-04],
        [4.6980e-10, 1.0000e+00],
        [4.0714e-08, 1.0000e+00],
        [6.3394e-07, 1.0000e+00],
        [4.9083e-10, 1.0000e+00],
        [9.7099e-01, 2.9008e-02],
        [6.3802e-10, 1.0000e+00],
        [3.8476e-03, 9.9615e-01],
        [1.0717e-08, 1.0000e+00],
        [7.3627e-01, 2.6373e-01],
        [2.7726e-04, 9.9972e-01],
        [9.9993e-01, 7.2945e-05],
        [1.8393e-09, 1.0000e+00],
        [8.3361e-03, 9.9166e-01],
        [3.7633e-03, 9.9624e-01],
        [9.3531e-01, 6.4692e-02],
        [3.4853e-01, 6.5147e-01],
        [2.0621e-08, 1.0000e+00],
        [1.5872e-05, 9.9998e-01],
        [9.9959e-01, 4.0973e-04],
        [2.8393e-05, 9.9997e-01],
        [1.2360e-01, 8.7640e-01],
        [2.8303e-06, 1.0000e+00],
        [2.8583e-10, 1.0000e+00],
        [1.6717e-06, 1.0000e+00],
        [3.6337e-06, 1.0000e+00],
        [1.5859e-08, 1.0000e+00],
        [5.6820e-07, 1.0000e+00],
        [3.5212e-07, 1.0000e+00],
        [1.1091e-06, 1.0000e+00],
        [5.5310e-06, 9.9999e-01],
        [5.7394e-08, 1.0000e+00],
        [9.9981e-01, 1.8900e-04],
        [4.5152e-03, 9.9548e-01],
        [9.9984e-01, 1.6466e-04],
        [5.4101e-04, 9.9946e-01],
        [9.9710e-01, 2.9049e-03],
        [1.8255e-06, 1.0000e+00],
        [2.3999e-07, 1.0000e+00],
        [7.0191e-05, 9.9993e-01],
        [2.1588e-05, 9.9998e-01],
        [8.1838e-04, 9.9918e-01],
        [9.9973e-01, 2.7041e-04],
        [9.9978e-01, 2.1827e-04],
        [4.9112e-05, 9.9995e-01],
        [2.7720e-10, 1.0000e+00],
        [9.3770e-01, 6.2295e-02],
        [1.4519e-05, 9.9999e-01],
        [5.1112e-08, 1.0000e+00],
        [1.2890e-05, 9.9999e-01],
        [1.1698e-05, 9.9999e-01],
        [5.1716e-01, 4.8284e-01],
        [3.1143e-01, 6.8857e-01],
        [9.9926e-01, 7.3960e-04],
        [9.3641e-08, 1.0000e+00],
        [2.0045e-04, 9.9980e-01],
        [1.4286e-06, 1.0000e+00],
        [3.2333e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9994e-01, 6.4196e-05],
        [7.6619e-07, 1.0000e+00],
        [1.2285e-08, 1.0000e+00],
        [2.5113e-06, 1.0000e+00],
        [8.2917e-05, 9.9992e-01],
        [8.7869e-02, 9.1213e-01],
        [9.4175e-05, 9.9991e-01],
        [3.0352e-07, 1.0000e+00],
        [2.1475e-06, 1.0000e+00],
        [1.7746e-08, 1.0000e+00],
        [9.9088e-01, 9.1158e-03],
        [8.8672e-07, 1.0000e+00],
        [1.6286e-08, 1.0000e+00],
        [4.0170e-08, 1.0000e+00],
        [3.2278e-05, 9.9997e-01],
        [9.9984e-01, 1.5534e-04],
        [2.7367e-08, 1.0000e+00],
        [7.1719e-08, 1.0000e+00],
        [9.1740e-06, 9.9999e-01],
        [6.3868e-05, 9.9994e-01],
        [5.2023e-05, 9.9995e-01],
        [4.0507e-04, 9.9959e-01],
        [1.5237e-02, 9.8476e-01],
        [9.9827e-01, 1.7348e-03],
        [9.9953e-01, 4.6594e-04],
        [3.9911e-05, 9.9996e-01],
        [1.0799e-05, 9.9999e-01],
        [9.9826e-01, 1.7377e-03],
        [6.8769e-06, 9.9999e-01],
        [9.9923e-01, 7.7440e-04],
        [2.5090e-07, 1.0000e+00],
        [1.7939e-06, 1.0000e+00],
        [1.1048e-06, 1.0000e+00],
        [1.2815e-03, 9.9872e-01],
        [7.5440e-09, 1.0000e+00],
        [5.1031e-04, 9.9949e-01],
        [6.7284e-05, 9.9993e-01],
        [1.8953e-07, 1.0000e+00],
        [7.3528e-06, 9.9999e-01],
        [2.8345e-03, 9.9717e-01],
        [6.0552e-07, 1.0000e+00],
        [1.3071e-04, 9.9987e-01],
        [2.4117e-06, 1.0000e+00],
        [1.0000e+00, 4.4068e-06],
        [1.9095e-07, 1.0000e+00],
        [9.5735e-07, 1.0000e+00],
        [1.6359e-06, 1.0000e+00],
        [2.0620e-01, 7.9380e-01],
        [8.2323e-05, 9.9992e-01],
        [9.9982e-01, 1.8245e-04],
        [1.0692e-06, 1.0000e+00],
        [1.2448e-06, 1.0000e+00],
        [9.9947e-01, 5.2889e-04],
        [1.3689e-07, 1.0000e+00],
        [1.8688e-05, 9.9998e-01],
        [1.0811e-05, 9.9999e-01],
        [3.2036e-06, 1.0000e+00],
        [9.9940e-01, 6.0365e-04],
        [1.9400e-03, 9.9806e-01],
        [6.5099e-07, 1.0000e+00],
        [9.9984e-01, 1.6014e-04],
        [2.4494e-06, 1.0000e+00],
        [1.6478e-05, 9.9998e-01],
        [9.9994e-01, 5.9261e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9335e-01, 6.6474e-03],
        [3.3192e-01, 6.6808e-01],
        [9.9978e-01, 2.1519e-04],
        [2.0975e-05, 9.9998e-01],
        [3.3852e-05, 9.9997e-01],
        [1.2019e-08, 1.0000e+00],
        [3.1052e-07, 1.0000e+00],
        [2.3280e-02, 9.7672e-01],
        [9.0491e-06, 9.9999e-01],
        [1.3889e-05, 9.9999e-01],
        [6.4235e-07, 1.0000e+00],
        [4.9919e-06, 9.9999e-01],
        [6.0124e-09, 1.0000e+00],
        [9.1030e-01, 8.9700e-02],
        [6.2771e-04, 9.9937e-01],
        [4.8515e-03, 9.9515e-01],
        [9.9997e-01, 2.6591e-05],
        [3.3860e-06, 1.0000e+00],
        [1.2221e-03, 9.9878e-01],
        [9.9986e-01, 1.4070e-04],
        [4.5107e-03, 9.9549e-01],
        [1.0394e-03, 9.9896e-01],
        [2.7200e-03, 9.9728e-01],
        [9.8959e-01, 1.0412e-02],
        [3.0123e-06, 1.0000e+00],
        [1.4441e-04, 9.9986e-01],
        [9.9972e-01, 2.8311e-04],
        [1.3613e-08, 1.0000e+00],
        [3.7011e-04, 9.9963e-01],
        [1.6394e-06, 1.0000e+00],
        [8.6878e-01, 1.3122e-01],
        [1.4401e-03, 9.9856e-01],
        [1.0605e-05, 9.9999e-01],
        [8.7452e-07, 1.0000e+00],
        [1.1468e-08, 1.0000e+00],
        [9.9985e-01, 1.4644e-04],
        [1.2610e-05, 9.9999e-01],
        [3.8429e-07, 1.0000e+00],
        [1.0222e-04, 9.9990e-01],
        [9.3482e-12, 1.0000e+00],
        [1.9721e-06, 1.0000e+00],
        [9.9968e-01, 3.1837e-04],
        [1.8069e-05, 9.9998e-01],
        [8.9069e-04, 9.9911e-01],
        [4.8648e-04, 9.9951e-01],
        [9.9502e-03, 9.9005e-01],
        [4.0401e-09, 1.0000e+00],
        [2.2357e-08, 1.0000e+00],
        [7.4603e-09, 1.0000e+00],
        [9.9992e-01, 7.5353e-05],
        [1.6130e-06, 1.0000e+00],
        [3.6640e-07, 1.0000e+00],
        [2.0478e-05, 9.9998e-01],
        [9.9668e-01, 3.3175e-03],
        [9.9909e-01, 9.1058e-04],
        [9.1685e-08, 1.0000e+00],
        [9.9836e-01, 1.6404e-03],
        [9.9989e-01, 1.1100e-04],
        [9.9977e-01, 2.3449e-04],
        [4.0565e-08, 1.0000e+00],
        [2.1290e-04, 9.9979e-01],
        [1.2605e-07, 1.0000e+00],
        [4.6595e-04, 9.9953e-01],
        [1.7945e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7584e-06, 1.0000e+00],
        [2.6264e-04, 9.9974e-01],
        [7.6423e-05, 9.9992e-01],
        [1.3532e-04, 9.9986e-01],
        [1.2708e-07, 1.0000e+00],
        [7.4508e-04, 9.9925e-01],
        [6.3441e-06, 9.9999e-01],
        [1.2993e-05, 9.9999e-01],
        [8.6253e-07, 1.0000e+00],
        [2.0318e-01, 7.9682e-01],
        [9.8996e-01, 1.0037e-02],
        [6.6257e-01, 3.3743e-01],
        [6.7778e-08, 1.0000e+00],
        [9.9968e-01, 3.1744e-04],
        [9.9978e-01, 2.1721e-04],
        [4.4494e-09, 1.0000e+00],
        [2.4928e-03, 9.9751e-01],
        [1.2997e-04, 9.9987e-01],
        [1.9034e-07, 1.0000e+00],
        [3.5768e-05, 9.9996e-01],
        [7.3643e-07, 1.0000e+00],
        [1.0291e-05, 9.9999e-01],
        [6.4086e-04, 9.9936e-01],
        [3.2577e-07, 1.0000e+00],
        [9.8924e-01, 1.0757e-02],
        [1.7590e-06, 1.0000e+00],
        [9.9986e-01, 1.3880e-04],
        [7.9287e-07, 1.0000e+00],
        [1.2267e-05, 9.9999e-01],
        [6.0106e-08, 1.0000e+00],
        [8.3897e-04, 9.9916e-01],
        [7.9837e-06, 9.9999e-01],
        [4.5234e-04, 9.9955e-01],
        [3.8231e-09, 1.0000e+00],
        [9.9223e-06, 9.9999e-01],
        [9.9985e-01, 1.5082e-04],
        [3.0384e-04, 9.9970e-01],
        [7.6393e-03, 9.9236e-01],
        [4.8473e-04, 9.9952e-01],
        [1.7853e-02, 9.8215e-01],
        [1.3297e-07, 1.0000e+00],
        [6.2807e-07, 1.0000e+00],
        [9.9956e-01, 4.3740e-04],
        [9.9959e-01, 4.0511e-04],
        [3.7368e-02, 9.6263e-01],
        [5.1227e-08, 1.0000e+00],
        [6.9286e-09, 1.0000e+00],
        [2.4659e-03, 9.9753e-01],
        [7.2060e-08, 1.0000e+00],
        [9.9961e-01, 3.9185e-04],
        [7.0340e-11, 1.0000e+00],
        [9.9982e-01, 1.8129e-04],
        [4.5817e-06, 1.0000e+00],
        [1.2597e-06, 1.0000e+00],
        [4.4456e-07, 1.0000e+00],
        [1.5192e-07, 1.0000e+00],
        [6.3555e-07, 1.0000e+00],
        [9.9942e-01, 5.7839e-04],
        [9.2673e-08, 1.0000e+00],
        [9.9987e-01, 1.3183e-04],
        [9.9998e-01, 1.5532e-05],
        [4.7487e-05, 9.9995e-01],
        [9.9994e-01, 5.9994e-05],
        [9.9633e-01, 3.6673e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[2.8962e-09, 1.0000e+00],
        [3.0030e-04, 9.9970e-01],
        [9.9888e-01, 1.1188e-03],
        [7.8139e-10, 1.0000e+00],
        [5.4415e-08, 1.0000e+00],
        [2.5964e-05, 9.9997e-01],
        [7.7788e-02, 9.2221e-01],
        [9.9987e-01, 1.3323e-04],
        [2.6705e-03, 9.9733e-01],
        [4.8949e-04, 9.9951e-01],
        [7.6513e-09, 1.0000e+00],
        [5.8615e-05, 9.9994e-01],
        [9.9985e-01, 1.4856e-04],
        [5.6025e-06, 9.9999e-01],
        [2.0948e-06, 1.0000e+00],
        [1.4472e-05, 9.9999e-01],
        [1.3867e-08, 1.0000e+00],
        [1.4373e-05, 9.9999e-01],
        [1.6336e-02, 9.8366e-01],
        [2.1639e-06, 1.0000e+00],
        [2.4016e-04, 9.9976e-01],
        [6.7845e-01, 3.2155e-01],
        [9.9952e-01, 4.7945e-04],
        [1.1582e-06, 1.0000e+00],
        [7.5979e-11, 1.0000e+00],
        [1.5629e-05, 9.9998e-01],
        [2.1709e-08, 1.0000e+00],
        [6.4861e-04, 9.9935e-01],
        [2.3094e-09, 1.0000e+00],
        [9.9989e-01, 1.0845e-04],
        [6.1312e-07, 1.0000e+00],
        [1.9172e-03, 9.9808e-01],
        [2.2084e-06, 1.0000e+00],
        [2.2865e-07, 1.0000e+00],
        [1.2202e-08, 1.0000e+00],
        [1.0846e-08, 1.0000e+00],
        [5.2335e-04, 9.9948e-01],
        [6.3166e-02, 9.3683e-01],
        [9.9982e-01, 1.8039e-04],
        [1.5199e-07, 1.0000e+00],
        [4.3698e-05, 9.9996e-01],
        [8.3936e-03, 9.9161e-01],
        [5.6695e-05, 9.9994e-01],
        [9.9947e-01, 5.2578e-04],
        [3.1468e-08, 1.0000e+00],
        [9.1815e-01, 8.1852e-02],
        [3.1864e-06, 1.0000e+00],
        [2.5901e-06, 1.0000e+00],
        [1.6289e-01, 8.3711e-01],
        [3.6931e-04, 9.9963e-01],
        [4.0002e-04, 9.9960e-01],
        [5.1532e-03, 9.9485e-01],
        [6.0681e-03, 9.9393e-01],
        [3.5843e-05, 9.9996e-01],
        [1.6121e-02, 9.8388e-01],
        [4.5064e-04, 9.9955e-01],
        [1.0081e-05, 9.9999e-01],
        [9.2459e-02, 9.0754e-01],
        [9.9965e-01, 3.5178e-04],
        [5.5615e-04, 9.9944e-01],
        [5.6620e-06, 9.9999e-01],
        [1.1978e-02, 9.8802e-01],
        [2.3262e-07, 1.0000e+00],
        [5.4753e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.0729e-04, 9.9909e-01],
        [8.9522e-06, 9.9999e-01],
        [5.8750e-08, 1.0000e+00],
        [9.9980e-01, 1.9641e-04],
        [1.2119e-04, 9.9988e-01],
        [4.6663e-05, 9.9995e-01],
        [2.4069e-01, 7.5931e-01],
        [6.2749e-07, 1.0000e+00],
        [6.4831e-08, 1.0000e+00],
        [9.8051e-01, 1.9492e-02],
        [9.2833e-02, 9.0717e-01],
        [4.5543e-07, 1.0000e+00],
        [9.9987e-01, 1.3430e-04],
        [1.5640e-08, 1.0000e+00],
        [2.0505e-08, 1.0000e+00],
        [6.8945e-06, 9.9999e-01],
        [1.3887e-01, 8.6113e-01],
        [1.5757e-01, 8.4243e-01],
        [3.1030e-02, 9.6897e-01],
        [5.6383e-07, 1.0000e+00],
        [3.4422e-09, 1.0000e+00],
        [1.5254e-03, 9.9847e-01],
        [1.7147e-04, 9.9983e-01],
        [3.7448e-03, 9.9626e-01],
        [9.0215e-05, 9.9991e-01],
        [9.9956e-01, 4.4042e-04],
        [1.7418e-07, 1.0000e+00],
        [1.5777e-03, 9.9842e-01],
        [9.9970e-01, 3.0016e-04],
        [2.7134e-04, 9.9973e-01],
        [8.1051e-07, 1.0000e+00],
        [5.1196e-05, 9.9995e-01],
        [4.2277e-03, 9.9577e-01],
        [8.2871e-07, 1.0000e+00],
        [7.8006e-06, 9.9999e-01],
        [2.6991e-09, 1.0000e+00],
        [7.1098e-08, 1.0000e+00],
        [1.9599e-02, 9.8040e-01],
        [7.1259e-05, 9.9993e-01],
        [1.0287e-07, 1.0000e+00],
        [1.4504e-06, 1.0000e+00],
        [1.0896e-03, 9.9891e-01],
        [1.3604e-07, 1.0000e+00],
        [9.9247e-01, 7.5309e-03],
        [1.7492e-02, 9.8251e-01],
        [9.9997e-01, 2.9813e-05],
        [2.5695e-05, 9.9997e-01],
        [2.5476e-04, 9.9975e-01],
        [4.1677e-06, 1.0000e+00],
        [5.8309e-03, 9.9417e-01],
        [8.2599e-05, 9.9992e-01],
        [2.6893e-08, 1.0000e+00],
        [3.3417e-05, 9.9997e-01],
        [4.9793e-08, 1.0000e+00],
        [9.9963e-01, 3.6930e-04],
        [1.9221e-10, 1.0000e+00],
        [9.8623e-01, 1.3774e-02],
        [1.6025e-06, 1.0000e+00],
        [9.9101e-01, 8.9857e-03],
        [9.9960e-01, 4.0086e-04],
        [3.5871e-05, 9.9996e-01],
        [4.1995e-08, 1.0000e+00],
        [9.9999e-01, 6.8051e-06],
        [2.8632e-04, 9.9971e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.1323e-01, 1.8677e-01],
        [1.1828e-06, 1.0000e+00],
        [9.9992e-01, 8.3269e-05],
        [9.9483e-01, 5.1657e-03],
        [9.9976e-01, 2.4155e-04],
        [4.2645e-07, 1.0000e+00],
        [4.4103e-05, 9.9996e-01],
        [1.0281e-06, 1.0000e+00],
        [1.5965e-06, 1.0000e+00],
        [6.5713e-05, 9.9993e-01],
        [1.6526e-06, 1.0000e+00],
        [1.5976e-07, 1.0000e+00],
        [8.1759e-05, 9.9992e-01],
        [1.1119e-08, 1.0000e+00],
        [9.9985e-01, 1.4844e-04],
        [8.9670e-12, 1.0000e+00],
        [7.2952e-02, 9.2705e-01],
        [7.0235e-05, 9.9993e-01],
        [9.9998e-01, 1.5317e-05],
        [1.1086e-04, 9.9989e-01],
        [3.7092e-01, 6.2908e-01],
        [5.3004e-09, 1.0000e+00],
        [9.9991e-01, 9.0771e-05],
        [8.1303e-06, 9.9999e-01],
        [2.5062e-06, 1.0000e+00],
        [7.2452e-05, 9.9993e-01],
        [9.9235e-01, 7.6459e-03],
        [4.5806e-05, 9.9995e-01],
        [4.3491e-06, 1.0000e+00],
        [4.0191e-04, 9.9960e-01],
        [3.9606e-08, 1.0000e+00],
        [5.1188e-07, 1.0000e+00],
        [9.9930e-01, 6.9993e-04],
        [6.2683e-10, 1.0000e+00],
        [9.8090e-01, 1.9101e-02],
        [2.1050e-08, 1.0000e+00],
        [2.1955e-05, 9.9998e-01],
        [9.9698e-01, 3.0178e-03],
        [4.4781e-07, 1.0000e+00],
        [2.0474e-04, 9.9980e-01],
        [5.7157e-09, 1.0000e+00],
        [9.9985e-01, 1.4734e-04],
        [3.8643e-07, 1.0000e+00],
        [7.4345e-08, 1.0000e+00],
        [9.6138e-01, 3.8623e-02],
        [1.3293e-06, 1.0000e+00],
        [8.3347e-04, 9.9917e-01],
        [2.4534e-07, 1.0000e+00],
        [9.8428e-01, 1.5721e-02],
        [9.9647e-01, 3.5277e-03],
        [6.5645e-05, 9.9993e-01],
        [9.0934e-08, 1.0000e+00],
        [9.9993e-01, 7.2608e-05],
        [7.3779e-04, 9.9926e-01],
        [9.9965e-01, 3.5290e-04],
        [6.8598e-07, 1.0000e+00],
        [9.9998e-01, 1.7116e-05],
        [6.1415e-07, 1.0000e+00],
        [1.1620e-07, 1.0000e+00],
        [4.5150e-05, 9.9995e-01],
        [9.1957e-01, 8.0425e-02],
        [1.6439e-05, 9.9998e-01],
        [3.5291e-05, 9.9996e-01],
        [9.9924e-01, 7.6085e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2584e-04, 9.9987e-01],
        [3.2532e-05, 9.9997e-01],
        [9.9989e-01, 1.0763e-04],
        [9.9856e-01, 1.4415e-03],
        [4.4506e-07, 1.0000e+00],
        [6.7862e-04, 9.9932e-01],
        [1.2565e-05, 9.9999e-01],
        [3.5100e-03, 9.9649e-01],
        [2.1054e-08, 1.0000e+00],
        [9.9993e-01, 6.7945e-05],
        [1.9345e-07, 1.0000e+00],
        [3.4548e-01, 6.5452e-01],
        [2.0732e-05, 9.9998e-01],
        [3.6794e-04, 9.9963e-01],
        [9.9990e-01, 1.0185e-04],
        [9.9993e-01, 7.0458e-05],
        [9.9989e-01, 1.1184e-04],
        [4.1475e-03, 9.9585e-01],
        [9.9972e-01, 2.8133e-04],
        [9.9979e-01, 2.1019e-04],
        [2.5944e-08, 1.0000e+00],
        [7.7797e-07, 1.0000e+00],
        [9.9980e-01, 1.9861e-04],
        [9.9977e-01, 2.2907e-04],
        [7.6561e-07, 1.0000e+00],
        [1.6370e-05, 9.9998e-01],
        [6.1368e-03, 9.9386e-01],
        [1.4954e-04, 9.9985e-01],
        [1.3764e-06, 1.0000e+00],
        [1.1209e-05, 9.9999e-01],
        [4.0574e-08, 1.0000e+00],
        [1.0427e-02, 9.8957e-01],
        [9.9974e-01, 2.5845e-04],
        [3.2400e-10, 1.0000e+00],
        [8.4570e-07, 1.0000e+00],
        [6.6500e-01, 3.3500e-01],
        [3.5930e-06, 1.0000e+00],
        [5.3887e-07, 1.0000e+00],
        [2.5812e-07, 1.0000e+00],
        [2.7752e-09, 1.0000e+00],
        [1.1334e-08, 1.0000e+00],
        [1.4535e-04, 9.9985e-01],
        [9.5493e-06, 9.9999e-01],
        [3.6830e-05, 9.9996e-01],
        [2.7661e-03, 9.9723e-01],
        [2.3193e-04, 9.9977e-01],
        [9.9980e-01, 1.9711e-04],
        [6.4093e-06, 9.9999e-01],
        [9.9988e-01, 1.1498e-04],
        [9.9990e-01, 1.0021e-04],
        [2.2992e-04, 9.9977e-01],
        [3.9448e-05, 9.9996e-01],
        [7.2583e-10, 1.0000e+00],
        [7.7995e-06, 9.9999e-01],
        [2.4302e-05, 9.9998e-01],
        [2.1393e-07, 1.0000e+00],
        [9.5336e-01, 4.6641e-02],
        [4.9354e-07, 1.0000e+00],
        [1.8326e-04, 9.9982e-01],
        [4.7367e-04, 9.9953e-01],
        [1.1695e-09, 1.0000e+00],
        [5.9962e-10, 1.0000e+00],
        [2.0265e-09, 1.0000e+00],
        [1.1671e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.6994e-10, 1.0000e+00],
        [9.9959e-01, 4.1374e-04],
        [9.9997e-01, 2.6894e-05],
        [9.3367e-05, 9.9991e-01],
        [9.9970e-01, 2.9693e-04],
        [1.7127e-07, 1.0000e+00],
        [2.3529e-04, 9.9976e-01],
        [9.9977e-01, 2.2544e-04],
        [3.8480e-04, 9.9962e-01],
        [7.5585e-01, 2.4415e-01],
        [6.7358e-04, 9.9933e-01],
        [2.3333e-05, 9.9998e-01],
        [5.6767e-06, 9.9999e-01],
        [3.2984e-03, 9.9670e-01],
        [7.9494e-07, 1.0000e+00],
        [3.6509e-06, 1.0000e+00],
        [1.2535e-01, 8.7465e-01],
        [2.2129e-05, 9.9998e-01],
        [1.5270e-04, 9.9985e-01],
        [7.7410e-09, 1.0000e+00],
        [6.8058e-02, 9.3194e-01],
        [9.5539e-09, 1.0000e+00],
        [2.1979e-08, 1.0000e+00],
        [2.5834e-05, 9.9997e-01],
        [9.9931e-01, 6.8726e-04],
        [5.9689e-06, 9.9999e-01],
        [1.1564e-05, 9.9999e-01],
        [9.9952e-01, 4.7846e-04],
        [9.9994e-01, 5.6454e-05],
        [3.1682e-05, 9.9997e-01],
        [9.9989e-01, 1.1367e-04],
        [9.9953e-01, 4.6630e-04],
        [3.6584e-06, 1.0000e+00],
        [1.4530e-07, 1.0000e+00],
        [6.1867e-08, 1.0000e+00],
        [2.2120e-06, 1.0000e+00],
        [6.0830e-06, 9.9999e-01],
        [4.8805e-03, 9.9512e-01],
        [2.4333e-06, 1.0000e+00],
        [9.4845e-07, 1.0000e+00],
        [2.7778e-04, 9.9972e-01],
        [1.7296e-04, 9.9983e-01],
        [9.9992e-01, 7.5420e-05],
        [9.9220e-01, 7.8013e-03],
        [1.7475e-06, 1.0000e+00],
        [1.6639e-03, 9.9834e-01],
        [1.4037e-03, 9.9860e-01],
        [1.2331e-06, 1.0000e+00],
        [4.7348e-07, 1.0000e+00],
        [5.6735e-02, 9.4326e-01],
        [3.6334e-06, 1.0000e+00],
        [1.1947e-05, 9.9999e-01],
        [7.5499e-03, 9.9245e-01],
        [1.3850e-07, 1.0000e+00],
        [5.6322e-04, 9.9944e-01],
        [6.9658e-06, 9.9999e-01],
        [9.9998e-01, 1.8985e-05],
        [2.7617e-08, 1.0000e+00],
        [9.9983e-01, 1.6816e-04],
        [1.2671e-06, 1.0000e+00],
        [9.9972e-01, 2.8233e-04],
        [2.4859e-04, 9.9975e-01],
        [3.1530e-08, 1.0000e+00],
        [1.0189e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2474e-03, 9.9875e-01],
        [3.6429e-05, 9.9996e-01],
        [9.9283e-01, 7.1721e-03],
        [2.7488e-03, 9.9725e-01],
        [6.9557e-04, 9.9930e-01],
        [9.9990e-01, 1.0478e-04],
        [1.2094e-05, 9.9999e-01],
        [9.9987e-01, 1.3369e-04],
        [9.9996e-01, 3.6977e-05],
        [1.4829e-07, 1.0000e+00],
        [5.2768e-05, 9.9995e-01],
        [1.0571e-04, 9.9989e-01],
        [2.4577e-03, 9.9754e-01],
        [9.8965e-01, 1.0352e-02],
        [9.4810e-04, 9.9905e-01],
        [9.9994e-01, 5.9549e-05],
        [9.9987e-01, 1.2866e-04],
        [1.1055e-05, 9.9999e-01],
        [7.7335e-08, 1.0000e+00],
        [7.9748e-09, 1.0000e+00],
        [3.8398e-06, 1.0000e+00],
        [2.0014e-04, 9.9980e-01],
        [3.9472e-07, 1.0000e+00],
        [3.1125e-03, 9.9689e-01],
        [3.8378e-05, 9.9996e-01],
        [1.3492e-04, 9.9987e-01],
        [7.9870e-02, 9.2013e-01],
        [3.8806e-02, 9.6119e-01],
        [5.7354e-04, 9.9943e-01],
        [7.3995e-07, 1.0000e+00],
        [2.2625e-05, 9.9998e-01],
        [9.9973e-01, 2.6507e-04],
        [8.3423e-04, 9.9917e-01],
        [3.4221e-06, 1.0000e+00],
        [2.4260e-06, 1.0000e+00],
        [1.3214e-04, 9.9987e-01],
        [9.9982e-01, 1.7721e-04],
        [1.6216e-04, 9.9984e-01],
        [1.9027e-03, 9.9810e-01],
        [5.6279e-07, 1.0000e+00],
        [9.9674e-01, 3.2603e-03],
        [1.4315e-08, 1.0000e+00],
        [2.4198e-07, 1.0000e+00],
        [1.5390e-04, 9.9985e-01],
        [4.2159e-06, 1.0000e+00],
        [2.3135e-05, 9.9998e-01],
        [3.0955e-06, 1.0000e+00],
        [1.0810e-05, 9.9999e-01],
        [6.2290e-06, 9.9999e-01],
        [1.0879e-07, 1.0000e+00],
        [9.8644e-01, 1.3563e-02],
        [7.0527e-03, 9.9295e-01],
        [9.9934e-01, 6.5970e-04],
        [9.9297e-01, 7.0340e-03],
        [5.1567e-05, 9.9995e-01],
        [8.9707e-07, 1.0000e+00],
        [1.9361e-07, 1.0000e+00],
        [9.9993e-01, 6.8147e-05],
        [3.4181e-02, 9.6582e-01],
        [9.9969e-01, 3.1295e-04],
        [3.5577e-07, 1.0000e+00],
        [4.3426e-09, 1.0000e+00],
        [1.2974e-04, 9.9987e-01],
        [1.1260e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5278e-04, 9.9975e-01],
        [1.4740e-07, 1.0000e+00],
        [9.9983e-01, 1.6692e-04],
        [1.1155e-03, 9.9888e-01],
        [7.0753e-02, 9.2925e-01],
        [7.9063e-08, 1.0000e+00],
        [9.0156e-01, 9.8441e-02],
        [3.1142e-07, 1.0000e+00],
        [9.3315e-01, 6.6849e-02],
        [5.5078e-15, 1.0000e+00],
        [7.0622e-07, 1.0000e+00],
        [1.8648e-07, 1.0000e+00],
        [9.9891e-01, 1.0920e-03],
        [9.6202e-09, 1.0000e+00],
        [8.4891e-06, 9.9999e-01],
        [9.9988e-01, 1.2465e-04],
        [2.1145e-06, 1.0000e+00],
        [9.9946e-01, 5.4179e-04],
        [9.9976e-01, 2.4122e-04],
        [2.7407e-04, 9.9973e-01],
        [1.0253e-08, 1.0000e+00],
        [9.4650e-04, 9.9905e-01],
        [9.9950e-01, 4.9808e-04],
        [1.8115e-05, 9.9998e-01],
        [5.8025e-07, 1.0000e+00],
        [7.3664e-04, 9.9926e-01],
        [6.0753e-08, 1.0000e+00],
        [6.7550e-08, 1.0000e+00],
        [3.5060e-05, 9.9996e-01],
        [3.2109e-05, 9.9997e-01],
        [8.6641e-01, 1.3359e-01],
        [7.8171e-06, 9.9999e-01],
        [5.8356e-05, 9.9994e-01],
        [1.3767e-08, 1.0000e+00],
        [9.9990e-01, 9.5765e-05],
        [8.2335e-08, 1.0000e+00],
        [1.1450e-03, 9.9885e-01],
        [1.6071e-03, 9.9839e-01],
        [9.9229e-01, 7.7129e-03],
        [6.4312e-06, 9.9999e-01],
        [6.6076e-04, 9.9934e-01],
        [9.9976e-01, 2.3553e-04],
        [3.4550e-07, 1.0000e+00],
        [3.2966e-03, 9.9670e-01],
        [2.5984e-04, 9.9974e-01],
        [9.9952e-01, 4.7925e-04],
        [1.3315e-05, 9.9999e-01],
        [9.9154e-01, 8.4585e-03],
        [2.4591e-03, 9.9754e-01],
        [8.9214e-06, 9.9999e-01],
        [1.4227e-05, 9.9999e-01],
        [6.9547e-01, 3.0453e-01],
        [3.5084e-09, 1.0000e+00],
        [3.3249e-01, 6.6751e-01],
        [1.1357e-05, 9.9999e-01],
        [2.7549e-04, 9.9972e-01],
        [7.5561e-04, 9.9924e-01],
        [9.9992e-01, 7.9160e-05],
        [3.5557e-07, 1.0000e+00],
        [6.2230e-07, 1.0000e+00],
        [8.7927e-06, 9.9999e-01],
        [5.7120e-02, 9.4288e-01],
        [1.9223e-06, 1.0000e+00],
        [1.6674e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4019e-06, 1.0000e+00],
        [2.1450e-06, 1.0000e+00],
        [4.8890e-08, 1.0000e+00],
        [4.2621e-06, 1.0000e+00],
        [3.2814e-04, 9.9967e-01],
        [2.3218e-05, 9.9998e-01],
        [9.9923e-01, 7.7299e-04],
        [1.7298e-04, 9.9983e-01],
        [1.7976e-05, 9.9998e-01],
        [1.1918e-05, 9.9999e-01],
        [9.9987e-01, 1.3318e-04],
        [1.3810e-02, 9.8619e-01],
        [2.7429e-05, 9.9997e-01],
        [7.1932e-09, 1.0000e+00],
        [2.1034e-05, 9.9998e-01],
        [2.1308e-06, 1.0000e+00],
        [9.9958e-01, 4.2135e-04],
        [8.3712e-08, 1.0000e+00],
        [9.9993e-01, 7.2699e-05],
        [6.7070e-07, 1.0000e+00],
        [9.9983e-01, 1.6907e-04],
        [5.7720e-05, 9.9994e-01],
        [6.6421e-09, 1.0000e+00],
        [1.5082e-05, 9.9998e-01],
        [9.9978e-01, 2.2393e-04],
        [6.6635e-07, 1.0000e+00],
        [1.8532e-06, 1.0000e+00],
        [9.9987e-01, 1.3454e-04],
        [8.0926e-07, 1.0000e+00],
        [1.6703e-03, 9.9833e-01],
        [9.9981e-01, 1.8918e-04],
        [1.1658e-03, 9.9883e-01],
        [9.5658e-01, 4.3415e-02],
        [1.9222e-04, 9.9981e-01],
        [9.9974e-01, 2.5591e-04],
        [1.1919e-07, 1.0000e+00],
        [3.9766e-09, 1.0000e+00],
        [1.1179e-06, 1.0000e+00],
        [1.1951e-02, 9.8805e-01],
        [4.2293e-03, 9.9577e-01],
        [1.8681e-04, 9.9981e-01],
        [8.3967e-01, 1.6033e-01],
        [3.1362e-01, 6.8638e-01],
        [1.5638e-07, 1.0000e+00],
        [1.9574e-05, 9.9998e-01],
        [9.9933e-01, 6.6742e-04],
        [1.7402e-08, 1.0000e+00],
        [1.9070e-08, 1.0000e+00],
        [1.8443e-06, 1.0000e+00],
        [1.0441e-07, 1.0000e+00],
        [8.1944e-06, 9.9999e-01],
        [3.6085e-03, 9.9639e-01],
        [9.3890e-06, 9.9999e-01],
        [1.1099e-03, 9.9889e-01],
        [2.2794e-07, 1.0000e+00],
        [9.9951e-01, 4.8807e-04],
        [1.4856e-02, 9.8514e-01],
        [4.1739e-07, 1.0000e+00],
        [9.9979e-01, 2.1044e-04],
        [9.9990e-01, 9.8918e-05],
        [2.8886e-04, 9.9971e-01],
        [4.6616e-05, 9.9995e-01],
        [4.8002e-07, 1.0000e+00],
        [1.0000e+00, 1.4376e-07]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7528e-02, 9.6247e-01],
        [4.7680e-04, 9.9952e-01],
        [2.3639e-06, 1.0000e+00],
        [3.2775e-04, 9.9967e-01],
        [9.0347e-02, 9.0965e-01],
        [4.2414e-07, 1.0000e+00],
        [1.4655e-03, 9.9853e-01],
        [1.6763e-04, 9.9983e-01],
        [9.9605e-01, 3.9525e-03],
        [1.8590e-09, 1.0000e+00],
        [6.5029e-02, 9.3497e-01],
        [1.4758e-05, 9.9999e-01],
        [3.7251e-04, 9.9963e-01],
        [5.4670e-05, 9.9995e-01],
        [9.8260e-01, 1.7396e-02],
        [1.5941e-05, 9.9998e-01],
        [1.3618e-01, 8.6382e-01],
        [3.1672e-09, 1.0000e+00],
        [2.8428e-06, 1.0000e+00],
        [9.9999e-01, 6.2963e-06],
        [2.2848e-05, 9.9998e-01],
        [5.3714e-04, 9.9946e-01],
        [9.9996e-01, 4.4019e-05],
        [6.3619e-05, 9.9994e-01],
        [1.6203e-03, 9.9838e-01],
        [1.4556e-06, 1.0000e+00],
        [6.6662e-05, 9.9993e-01],
        [3.2178e-06, 1.0000e+00],
        [1.6848e-06, 1.0000e+00],
        [2.8501e-01, 7.1499e-01],
        [3.7057e-03, 9.9629e-01],
        [1.1375e-08, 1.0000e+00],
        [1.1705e-07, 1.0000e+00],
        [9.9839e-01, 1.6112e-03],
        [9.9784e-01, 2.1648e-03],
        [9.9992e-01, 8.0632e-05],
        [1.3459e-09, 1.0000e+00],
        [7.6904e-01, 2.3096e-01],
        [9.9691e-01, 3.0910e-03],
        [8.8540e-07, 1.0000e+00],
        [9.9977e-01, 2.2520e-04],
        [9.9901e-01, 9.8976e-04],
        [9.9965e-01, 3.5007e-04],
        [2.0161e-04, 9.9980e-01],
        [7.4457e-08, 1.0000e+00],
        [3.5837e-06, 1.0000e+00],
        [4.8090e-05, 9.9995e-01],
        [5.6678e-06, 9.9999e-01],
        [5.3766e-01, 4.6234e-01],
        [2.4777e-04, 9.9975e-01],
        [9.9975e-01, 2.5181e-04],
        [2.4142e-06, 1.0000e+00],
        [2.2065e-07, 1.0000e+00],
        [8.9112e-01, 1.0888e-01],
        [1.1828e-07, 1.0000e+00],
        [2.8086e-01, 7.1914e-01],
        [9.9984e-01, 1.6301e-04],
        [9.9989e-01, 1.0861e-04],
        [1.3409e-04, 9.9987e-01],
        [3.1108e-05, 9.9997e-01],
        [2.5622e-01, 7.4378e-01],
        [9.9978e-01, 2.1512e-04],
        [9.5161e-05, 9.9990e-01],
        [3.0431e-01, 6.9569e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1275e-06, 1.0000e+00],
        [3.9923e-03, 9.9601e-01],
        [1.2559e-03, 9.9874e-01],
        [6.2168e-05, 9.9994e-01],
        [9.9989e-01, 1.1087e-04],
        [9.9970e-01, 3.0419e-04],
        [1.3350e-03, 9.9867e-01],
        [6.6676e-02, 9.3332e-01],
        [6.3283e-04, 9.9937e-01],
        [4.5217e-09, 1.0000e+00],
        [1.7248e-08, 1.0000e+00],
        [9.9906e-01, 9.3600e-04],
        [3.5207e-07, 1.0000e+00],
        [8.4597e-06, 9.9999e-01],
        [4.4119e-10, 1.0000e+00],
        [3.5552e-01, 6.4448e-01],
        [3.1102e-08, 1.0000e+00],
        [2.9720e-05, 9.9997e-01],
        [9.9886e-01, 1.1402e-03],
        [9.9987e-01, 1.3312e-04],
        [4.4538e-04, 9.9955e-01],
        [9.9631e-01, 3.6903e-03],
        [5.9668e-01, 4.0332e-01],
        [1.1255e-04, 9.9989e-01],
        [6.6164e-06, 9.9999e-01],
        [9.9298e-01, 7.0165e-03],
        [2.6277e-08, 1.0000e+00],
        [5.3929e-07, 1.0000e+00],
        [2.3995e-06, 1.0000e+00],
        [6.7527e-08, 1.0000e+00],
        [1.6614e-05, 9.9998e-01],
        [1.9070e-08, 1.0000e+00],
        [1.3430e-03, 9.9866e-01],
        [9.9970e-01, 3.0107e-04],
        [3.1020e-07, 1.0000e+00],
        [9.9579e-01, 4.2130e-03],
        [1.9301e-04, 9.9981e-01],
        [6.3142e-08, 1.0000e+00],
        [5.0158e-06, 9.9999e-01],
        [2.6743e-03, 9.9733e-01],
        [2.7841e-04, 9.9972e-01],
        [6.6420e-08, 1.0000e+00],
        [3.0053e-03, 9.9699e-01],
        [9.9971e-01, 2.9456e-04],
        [7.7019e-01, 2.2981e-01],
        [3.3415e-03, 9.9666e-01],
        [9.9961e-01, 3.9386e-04],
        [5.0109e-05, 9.9995e-01],
        [9.9986e-01, 1.3636e-04],
        [1.9317e-06, 1.0000e+00],
        [9.9962e-01, 3.7696e-04],
        [1.0811e-07, 1.0000e+00],
        [1.7123e-07, 1.0000e+00],
        [6.1688e-05, 9.9994e-01],
        [2.0618e-10, 1.0000e+00],
        [3.8093e-07, 1.0000e+00],
        [2.4063e-06, 1.0000e+00],
        [5.9074e-10, 1.0000e+00],
        [9.9971e-01, 2.9496e-04],
        [5.3542e-06, 9.9999e-01],
        [3.8662e-06, 1.0000e+00],
        [2.1108e-07, 1.0000e+00],
        [1.9833e-09, 1.0000e+00],
        [1.7019e-04, 9.9983e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3335e-06, 1.0000e+00],
        [2.7494e-05, 9.9997e-01],
        [1.0431e-07, 1.0000e+00],
        [7.0256e-02, 9.2974e-01],
        [7.5962e-09, 1.0000e+00],
        [1.3095e-07, 1.0000e+00],
        [5.9287e-02, 9.4071e-01],
        [5.9107e-07, 1.0000e+00],
        [3.4898e-06, 1.0000e+00],
        [1.6986e-07, 1.0000e+00],
        [1.1069e-07, 1.0000e+00],
        [2.2698e-08, 1.0000e+00],
        [1.0134e-06, 1.0000e+00],
        [9.9213e-01, 7.8652e-03],
        [2.0832e-08, 1.0000e+00],
        [9.9972e-01, 2.7738e-04],
        [3.2461e-01, 6.7539e-01],
        [8.6592e-07, 1.0000e+00],
        [6.4588e-06, 9.9999e-01],
        [2.1615e-11, 1.0000e+00],
        [9.2936e-06, 9.9999e-01],
        [1.4321e-04, 9.9986e-01],
        [1.2327e-09, 1.0000e+00],
        [5.1763e-05, 9.9995e-01],
        [2.1203e-11, 1.0000e+00],
        [1.3521e-02, 9.8648e-01],
        [9.9650e-01, 3.4979e-03],
        [8.9698e-05, 9.9991e-01],
        [9.9480e-01, 5.2009e-03],
        [8.1121e-11, 1.0000e+00],
        [5.0504e-04, 9.9949e-01],
        [1.1813e-07, 1.0000e+00],
        [2.2494e-06, 1.0000e+00],
        [5.3948e-05, 9.9995e-01],
        [2.8172e-05, 9.9997e-01],
        [9.9988e-01, 1.1521e-04],
        [6.4330e-05, 9.9994e-01],
        [1.7774e-06, 1.0000e+00],
        [9.9992e-01, 7.7451e-05],
        [6.4717e-03, 9.9353e-01],
        [3.4604e-01, 6.5396e-01],
        [1.5563e-06, 1.0000e+00],
        [1.9268e-05, 9.9998e-01],
        [1.1345e-03, 9.9887e-01],
        [4.0338e-09, 1.0000e+00],
        [9.0930e-08, 1.0000e+00],
        [1.0275e-05, 9.9999e-01],
        [9.9949e-01, 5.0734e-04],
        [1.0233e-10, 1.0000e+00],
        [2.2380e-07, 1.0000e+00],
        [1.2433e-06, 1.0000e+00],
        [1.7484e-05, 9.9998e-01],
        [9.8832e-01, 1.1678e-02],
        [5.9695e-09, 1.0000e+00],
        [1.6676e-06, 1.0000e+00],
        [9.9985e-01, 1.4690e-04],
        [2.0350e-01, 7.9650e-01],
        [3.7571e-09, 1.0000e+00],
        [7.9714e-07, 1.0000e+00],
        [2.2472e-05, 9.9998e-01],
        [4.2529e-05, 9.9996e-01],
        [1.3833e-03, 9.9862e-01],
        [8.1101e-01, 1.8899e-01],
        [3.1831e-04, 9.9968e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.4504e-02, 9.2550e-01],
        [2.0166e-04, 9.9980e-01],
        [1.7565e-09, 1.0000e+00],
        [1.6536e-03, 9.9835e-01],
        [9.9963e-01, 3.7184e-04],
        [4.5991e-07, 1.0000e+00],
        [9.1651e-09, 1.0000e+00],
        [1.0003e-03, 9.9900e-01],
        [4.5802e-06, 1.0000e+00],
        [3.6860e-06, 1.0000e+00],
        [8.9089e-08, 1.0000e+00],
        [1.4989e-08, 1.0000e+00],
        [9.9985e-01, 1.5398e-04],
        [4.6078e-03, 9.9539e-01],
        [3.7679e-04, 9.9962e-01],
        [5.2205e-05, 9.9995e-01],
        [2.1481e-03, 9.9785e-01],
        [5.9439e-05, 9.9994e-01],
        [4.1366e-04, 9.9959e-01],
        [3.8631e-08, 1.0000e+00],
        [1.8632e-07, 1.0000e+00],
        [9.6802e-01, 3.1985e-02],
        [9.9987e-01, 1.3019e-04],
        [8.0429e-07, 1.0000e+00],
        [3.5104e-10, 1.0000e+00],
        [5.9844e-05, 9.9994e-01],
        [9.9846e-01, 1.5409e-03],
        [1.3088e-09, 1.0000e+00],
        [6.0081e-08, 1.0000e+00],
        [9.2971e-01, 7.0285e-02],
        [9.9646e-01, 3.5377e-03],
        [9.9537e-01, 4.6344e-03],
        [5.0715e-04, 9.9949e-01],
        [1.3041e-02, 9.8696e-01],
        [1.5817e-06, 1.0000e+00],
        [9.8108e-01, 1.8919e-02],
        [9.9825e-01, 1.7455e-03],
        [7.7105e-01, 2.2895e-01],
        [1.9720e-03, 9.9803e-01],
        [2.4650e-07, 1.0000e+00],
        [4.0862e-06, 1.0000e+00],
        [9.9535e-01, 4.6536e-03],
        [9.9992e-01, 8.0970e-05],
        [1.9517e-05, 9.9998e-01],
        [2.5485e-07, 1.0000e+00],
        [9.9962e-01, 3.7571e-04],
        [1.3128e-07, 1.0000e+00],
        [9.9983e-01, 1.6660e-04],
        [8.7446e-08, 1.0000e+00],
        [8.9443e-07, 1.0000e+00],
        [9.4251e-01, 5.7486e-02],
        [8.4201e-08, 1.0000e+00],
        [9.7782e-01, 2.2180e-02],
        [9.9989e-01, 1.1451e-04],
        [2.1720e-06, 1.0000e+00],
        [5.4060e-04, 9.9946e-01],
        [5.6247e-01, 4.3753e-01],
        [4.7505e-09, 1.0000e+00],
        [3.1357e-04, 9.9969e-01],
        [1.2857e-05, 9.9999e-01],
        [9.9972e-01, 2.8264e-04],
        [9.7663e-09, 1.0000e+00],
        [1.9768e-06, 1.0000e+00],
        [2.3839e-02, 9.7616e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.7922e-04, 9.9922e-01],
        [9.9994e-01, 6.1887e-05],
        [9.9994e-01, 5.8620e-05],
        [5.8163e-05, 9.9994e-01],
        [3.1364e-06, 1.0000e+00],
        [2.1969e-06, 1.0000e+00],
        [6.9640e-04, 9.9930e-01],
        [9.9865e-01, 1.3526e-03],
        [2.1349e-10, 1.0000e+00],
        [3.0869e-04, 9.9969e-01],
        [1.2138e-05, 9.9999e-01],
        [2.1463e-04, 9.9979e-01],
        [9.5927e-01, 4.0730e-02],
        [1.8951e-08, 1.0000e+00],
        [6.7356e-05, 9.9993e-01],
        [1.3197e-07, 1.0000e+00],
        [1.3596e-06, 1.0000e+00],
        [2.8942e-05, 9.9997e-01],
        [7.5717e-05, 9.9992e-01],
        [9.9881e-01, 1.1928e-03],
        [1.1083e-08, 1.0000e+00],
        [3.1879e-07, 1.0000e+00],
        [4.4332e-05, 9.9996e-01],
        [8.2754e-04, 9.9917e-01],
        [9.9989e-01, 1.0731e-04],
        [2.3070e-08, 1.0000e+00],
        [9.9969e-01, 3.1318e-04],
        [9.9956e-01, 4.3619e-04],
        [2.6211e-07, 1.0000e+00],
        [1.7475e-08, 1.0000e+00],
        [3.0587e-06, 1.0000e+00],
        [9.9998e-01, 2.1750e-05],
        [4.2785e-08, 1.0000e+00],
        [7.6295e-08, 1.0000e+00],
        [1.0503e-06, 1.0000e+00],
        [2.1558e-08, 1.0000e+00],
        [9.9917e-01, 8.2554e-04],
        [2.9046e-06, 1.0000e+00],
        [9.9399e-01, 6.0143e-03],
        [2.8780e-09, 1.0000e+00],
        [3.1071e-02, 9.6893e-01],
        [1.1652e-06, 1.0000e+00],
        [8.3064e-07, 1.0000e+00],
        [4.1915e-07, 1.0000e+00],
        [1.0939e-03, 9.9891e-01],
        [9.9938e-01, 6.1751e-04],
        [8.6386e-04, 9.9914e-01],
        [3.7829e-06, 1.0000e+00],
        [3.5118e-04, 9.9965e-01],
        [4.1889e-03, 9.9581e-01],
        [1.7857e-04, 9.9982e-01],
        [2.7991e-06, 1.0000e+00],
        [3.5305e-07, 1.0000e+00],
        [5.9268e-08, 1.0000e+00],
        [9.9976e-01, 2.3747e-04],
        [3.9251e-04, 9.9961e-01],
        [3.5126e-05, 9.9996e-01],
        [3.0639e-05, 9.9997e-01],
        [5.6209e-07, 1.0000e+00],
        [9.8388e-06, 9.9999e-01],
        [4.0427e-06, 1.0000e+00],
        [7.1733e-05, 9.9993e-01],
        [1.8533e-04, 9.9981e-01],
        [1.1123e-04, 9.9989e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2441e-08, 1.0000e+00],
        [6.8562e-03, 9.9314e-01],
        [1.1924e-03, 9.9881e-01],
        [4.0956e-09, 1.0000e+00],
        [1.8870e-09, 1.0000e+00],
        [9.9980e-01, 1.9852e-04],
        [3.5984e-08, 1.0000e+00],
        [1.1537e-06, 1.0000e+00],
        [9.9256e-01, 7.4359e-03],
        [4.1787e-02, 9.5821e-01],
        [6.4832e-06, 9.9999e-01],
        [5.2859e-07, 1.0000e+00],
        [3.1525e-04, 9.9968e-01],
        [1.0107e-05, 9.9999e-01],
        [7.2765e-04, 9.9927e-01],
        [1.7828e-06, 1.0000e+00],
        [2.4135e-06, 1.0000e+00],
        [7.6353e-07, 1.0000e+00],
        [1.9160e-05, 9.9998e-01],
        [9.6749e-09, 1.0000e+00],
        [2.6506e-06, 1.0000e+00],
        [1.0805e-08, 1.0000e+00],
        [7.0345e-06, 9.9999e-01],
        [9.9964e-01, 3.5596e-04],
        [1.0624e-04, 9.9989e-01],
        [8.8242e-04, 9.9912e-01],
        [9.4626e-08, 1.0000e+00],
        [2.9168e-08, 1.0000e+00],
        [7.0578e-06, 9.9999e-01],
        [2.5114e-03, 9.9749e-01],
        [3.1502e-06, 1.0000e+00],
        [9.7744e-01, 2.2562e-02],
        [1.0037e-09, 1.0000e+00],
        [4.8014e-05, 9.9995e-01],
        [1.7590e-02, 9.8241e-01],
        [1.6179e-04, 9.9984e-01],
        [3.4511e-02, 9.6549e-01],
        [1.7043e-03, 9.9830e-01],
        [9.9883e-01, 1.1697e-03],
        [5.4676e-05, 9.9995e-01],
        [1.3652e-02, 9.8635e-01],
        [2.1212e-06, 1.0000e+00],
        [1.8913e-09, 1.0000e+00],
        [5.6971e-10, 1.0000e+00],
        [1.6287e-05, 9.9998e-01],
        [9.9962e-01, 3.7792e-04],
        [1.3606e-06, 1.0000e+00],
        [7.1423e-07, 1.0000e+00],
        [1.5103e-07, 1.0000e+00],
        [9.9976e-01, 2.4102e-04],
        [2.0061e-02, 9.7994e-01],
        [9.9954e-01, 4.6242e-04],
        [2.6329e-04, 9.9974e-01],
        [7.4415e-08, 1.0000e+00],
        [1.1446e-07, 1.0000e+00],
        [3.1643e-09, 1.0000e+00],
        [2.2321e-08, 1.0000e+00],
        [9.9974e-01, 2.5973e-04],
        [8.9350e-08, 1.0000e+00],
        [1.7810e-06, 1.0000e+00],
        [9.9968e-01, 3.2131e-04],
        [2.1154e-03, 9.9788e-01],
        [1.5398e-09, 1.0000e+00],
        [5.5692e-04, 9.9944e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6836e-01, 8.3164e-01],
        [5.3616e-01, 4.6384e-01],
        [9.6440e-01, 3.5601e-02],
        [9.9915e-01, 8.5269e-04],
        [1.3927e-03, 9.9861e-01],
        [9.9988e-01, 1.1654e-04],
        [3.9433e-01, 6.0567e-01],
        [1.8477e-05, 9.9998e-01],
        [8.5677e-01, 1.4323e-01],
        [9.4543e-07, 1.0000e+00],
        [9.6288e-01, 3.7117e-02],
        [1.6150e-04, 9.9984e-01],
        [1.6301e-06, 1.0000e+00],
        [3.0261e-05, 9.9997e-01],
        [1.0579e-09, 1.0000e+00],
        [7.5493e-09, 1.0000e+00],
        [2.8271e-08, 1.0000e+00],
        [2.3643e-01, 7.6357e-01],
        [1.6647e-06, 1.0000e+00],
        [3.5334e-07, 1.0000e+00],
        [1.8974e-01, 8.1026e-01],
        [9.9937e-01, 6.3288e-04],
        [1.0313e-05, 9.9999e-01],
        [9.9999e-01, 1.0390e-05],
        [1.7268e-06, 1.0000e+00],
        [7.4656e-11, 1.0000e+00],
        [1.7394e-09, 1.0000e+00],
        [3.4537e-04, 9.9965e-01],
        [1.7222e-04, 9.9983e-01],
        [5.5084e-07, 1.0000e+00],
        [1.3445e-08, 1.0000e+00],
        [9.9969e-01, 3.1186e-04],
        [1.4798e-07, 1.0000e+00],
        [2.5811e-05, 9.9997e-01],
        [4.0984e-07, 1.0000e+00],
        [9.7948e-06, 9.9999e-01],
        [8.2236e-06, 9.9999e-01],
        [9.5380e-06, 9.9999e-01],
        [1.0309e-05, 9.9999e-01],
        [2.7102e-07, 1.0000e+00],
        [7.2365e-06, 9.9999e-01],
        [9.9931e-01, 6.8720e-04],
        [6.4842e-08, 1.0000e+00],
        [3.0211e-07, 1.0000e+00],
        [4.4971e-08, 1.0000e+00],
        [4.2865e-06, 1.0000e+00],
        [5.3935e-07, 1.0000e+00],
        [1.5993e-04, 9.9984e-01],
        [9.9903e-01, 9.6775e-04],
        [8.2401e-01, 1.7599e-01],
        [1.1672e-08, 1.0000e+00],
        [1.1715e-07, 1.0000e+00],
        [2.5941e-08, 1.0000e+00],
        [4.3744e-06, 1.0000e+00],
        [2.5552e-03, 9.9744e-01],
        [9.9957e-01, 4.3321e-04],
        [5.3136e-06, 9.9999e-01],
        [9.9935e-01, 6.5136e-04],
        [2.9149e-09, 1.0000e+00],
        [1.6209e-03, 9.9838e-01],
        [4.2977e-08, 1.0000e+00],
        [9.9999e-01, 9.5082e-06],
        [9.9881e-01, 1.1871e-03],
        [9.9965e-01, 3.5334e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2406e-03, 9.9876e-01],
        [3.2464e-03, 9.9675e-01],
        [9.9988e-01, 1.1731e-04],
        [9.9738e-01, 2.6186e-03],
        [9.9984e-01, 1.5896e-04],
        [1.1097e-03, 9.9889e-01],
        [2.8828e-11, 1.0000e+00],
        [7.9520e-07, 1.0000e+00],
        [2.7905e-06, 1.0000e+00],
        [9.9919e-01, 8.1039e-04],
        [1.2993e-06, 1.0000e+00],
        [6.0260e-08, 1.0000e+00],
        [3.6464e-05, 9.9996e-01],
        [1.0754e-07, 1.0000e+00],
        [1.9252e-03, 9.9807e-01],
        [9.9974e-01, 2.6083e-04],
        [9.9915e-01, 8.4756e-04],
        [3.5181e-02, 9.6482e-01],
        [2.6489e-04, 9.9974e-01],
        [1.4025e-04, 9.9986e-01],
        [4.2350e-05, 9.9996e-01],
        [9.9962e-01, 3.8359e-04],
        [2.3026e-11, 1.0000e+00],
        [7.8768e-05, 9.9992e-01],
        [5.8758e-05, 9.9994e-01],
        [9.9952e-01, 4.8479e-04],
        [2.5363e-10, 1.0000e+00],
        [6.2128e-07, 1.0000e+00],
        [1.0445e-04, 9.9990e-01],
        [2.4359e-07, 1.0000e+00],
        [7.5864e-05, 9.9992e-01],
        [7.1703e-03, 9.9283e-01],
        [1.5811e-05, 9.9998e-01],
        [2.8411e-07, 1.0000e+00],
        [9.9913e-01, 8.6762e-04],
        [2.5382e-02, 9.7462e-01],
        [4.1315e-05, 9.9996e-01],
        [9.9946e-01, 5.3893e-04],
        [9.9996e-01, 3.6906e-05],
        [9.9981e-01, 1.8909e-04],
        [9.9924e-01, 7.6367e-04],
        [1.1022e-11, 1.0000e+00],
        [5.7572e-10, 1.0000e+00],
        [9.8635e-01, 1.3650e-02],
        [7.4200e-04, 9.9926e-01],
        [4.3930e-07, 1.0000e+00],
        [1.0000e+00, 1.3338e-06],
        [1.9370e-08, 1.0000e+00],
        [1.6645e-07, 1.0000e+00],
        [8.2831e-06, 9.9999e-01],
        [5.7219e-05, 9.9994e-01],
        [2.3493e-04, 9.9977e-01],
        [7.0141e-08, 1.0000e+00],
        [9.9950e-01, 4.9506e-04],
        [1.7703e-02, 9.8230e-01],
        [2.3295e-04, 9.9977e-01],
        [6.5524e-06, 9.9999e-01],
        [1.7363e-04, 9.9983e-01],
        [9.9954e-01, 4.5555e-04],
        [3.6368e-08, 1.0000e+00],
        [1.5751e-08, 1.0000e+00],
        [1.4209e-06, 1.0000e+00],
        [1.8528e-07, 1.0000e+00],
        [2.2527e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3891e-08, 1.0000e+00],
        [4.1466e-09, 1.0000e+00],
        [1.9667e-08, 1.0000e+00],
        [1.0802e-06, 1.0000e+00],
        [2.0450e-04, 9.9980e-01],
        [7.9920e-04, 9.9920e-01],
        [5.9247e-08, 1.0000e+00],
        [9.9986e-01, 1.3939e-04],
        [1.0741e-07, 1.0000e+00],
        [9.9997e-01, 3.3481e-05],
        [9.9977e-01, 2.3018e-04],
        [1.4727e-03, 9.9853e-01],
        [3.7679e-02, 9.6232e-01],
        [2.4634e-02, 9.7537e-01],
        [1.0140e-04, 9.9990e-01],
        [9.3118e-01, 6.8819e-02],
        [9.9975e-01, 2.4672e-04],
        [1.9000e-04, 9.9981e-01],
        [2.6302e-07, 1.0000e+00],
        [2.4081e-04, 9.9976e-01],
        [2.7339e-07, 1.0000e+00],
        [9.9986e-01, 1.4005e-04],
        [1.2710e-08, 1.0000e+00],
        [6.8835e-06, 9.9999e-01],
        [9.9414e-01, 5.8594e-03],
        [4.5642e-03, 9.9544e-01],
        [2.0803e-06, 1.0000e+00],
        [2.9458e-09, 1.0000e+00],
        [9.9956e-01, 4.3986e-04],
        [9.0231e-09, 1.0000e+00],
        [8.2861e-06, 9.9999e-01],
        [4.3821e-07, 1.0000e+00],
        [3.1156e-06, 1.0000e+00],
        [1.2985e-04, 9.9987e-01],
        [1.1297e-04, 9.9989e-01],
        [1.9538e-08, 1.0000e+00],
        [8.0064e-03, 9.9199e-01],
        [1.1549e-07, 1.0000e+00],
        [9.9680e-01, 3.1976e-03],
        [1.0000e+00, 1.5938e-06],
        [9.8013e-03, 9.9020e-01],
        [4.0874e-08, 1.0000e+00],
        [5.3261e-05, 9.9995e-01],
        [5.5901e-05, 9.9994e-01],
        [1.4526e-03, 9.9855e-01],
        [1.0322e-06, 1.0000e+00],
        [8.6301e-01, 1.3699e-01],
        [1.9259e-10, 1.0000e+00],
        [6.1731e-07, 1.0000e+00],
        [9.1557e-07, 1.0000e+00],
        [9.9975e-01, 2.5202e-04],
        [4.1341e-04, 9.9959e-01],
        [1.2414e-06, 1.0000e+00],
        [2.7142e-07, 1.0000e+00],
        [9.7708e-07, 1.0000e+00],
        [9.7681e-01, 2.3189e-02],
        [1.2028e-04, 9.9988e-01],
        [4.7325e-07, 1.0000e+00],
        [1.4216e-03, 9.9858e-01],
        [1.7444e-06, 1.0000e+00],
        [2.9982e-04, 9.9970e-01],
        [1.0131e-09, 1.0000e+00],
        [2.7034e-07, 1.0000e+00],
        [2.3634e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.9488e-06, 1.0000e+00],
        [1.4040e-03, 9.9860e-01],
        [9.9973e-01, 2.7397e-04],
        [1.5758e-09, 1.0000e+00],
        [9.9977e-01, 2.2676e-04],
        [9.6481e-01, 3.5188e-02],
        [9.9969e-01, 3.0518e-04],
        [9.9969e-01, 3.0892e-04],
        [3.7701e-06, 1.0000e+00],
        [1.0678e-05, 9.9999e-01],
        [9.9976e-01, 2.4040e-04],
        [2.3496e-06, 1.0000e+00],
        [9.9978e-01, 2.2025e-04],
        [9.9745e-01, 2.5466e-03],
        [2.7468e-05, 9.9997e-01],
        [5.0633e-07, 1.0000e+00],
        [5.0701e-03, 9.9493e-01],
        [4.3409e-05, 9.9996e-01],
        [9.9997e-01, 2.7344e-05],
        [1.2766e-04, 9.9987e-01],
        [9.9929e-01, 7.0861e-04],
        [1.1586e-07, 1.0000e+00],
        [1.3408e-09, 1.0000e+00],
        [3.3769e-09, 1.0000e+00],
        [2.7899e-07, 1.0000e+00],
        [9.9964e-01, 3.5951e-04],
        [9.7440e-01, 2.5595e-02],
        [1.3847e-08, 1.0000e+00],
        [1.0135e-09, 1.0000e+00],
        [9.9976e-01, 2.3645e-04],
        [6.8307e-05, 9.9993e-01],
        [2.2976e-09, 1.0000e+00],
        [7.9762e-04, 9.9920e-01],
        [9.9996e-01, 3.8407e-05],
        [1.5833e-02, 9.8417e-01],
        [4.8267e-06, 1.0000e+00],
        [9.2771e-04, 9.9907e-01],
        [2.6122e-04, 9.9974e-01],
        [1.3496e-09, 1.0000e+00],
        [1.8207e-06, 1.0000e+00],
        [4.9253e-07, 1.0000e+00],
        [2.9849e-06, 1.0000e+00],
        [1.3439e-02, 9.8656e-01],
        [2.8704e-07, 1.0000e+00],
        [9.9991e-01, 9.0464e-05],
        [1.9699e-01, 8.0301e-01],
        [9.9992e-01, 7.9513e-05],
        [2.8876e-04, 9.9971e-01],
        [4.0840e-07, 1.0000e+00],
        [2.3730e-06, 1.0000e+00],
        [8.3684e-05, 9.9992e-01],
        [4.4010e-02, 9.5599e-01],
        [4.0125e-06, 1.0000e+00],
        [1.8310e-09, 1.0000e+00],
        [5.7399e-08, 1.0000e+00],
        [1.3457e-04, 9.9987e-01],
        [5.0113e-05, 9.9995e-01],
        [9.9981e-01, 1.8558e-04],
        [9.2591e-06, 9.9999e-01],
        [1.9754e-04, 9.9980e-01],
        [7.9762e-04, 9.9920e-01],
        [9.5550e-01, 4.4497e-02],
        [1.4953e-09, 1.0000e+00],
        [1.4881e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.9112e-05, 9.9993e-01],
        [9.9959e-01, 4.1402e-04],
        [5.7349e-08, 1.0000e+00],
        [9.9945e-01, 5.5067e-04],
        [8.1800e-10, 1.0000e+00],
        [2.1451e-02, 9.7855e-01],
        [4.2176e-04, 9.9958e-01],
        [2.2590e-05, 9.9998e-01],
        [4.7509e-06, 1.0000e+00],
        [1.7925e-09, 1.0000e+00],
        [7.2449e-04, 9.9928e-01],
        [3.0344e-06, 1.0000e+00],
        [4.4665e-07, 1.0000e+00],
        [2.3582e-06, 1.0000e+00],
        [8.0706e-07, 1.0000e+00],
        [2.3291e-05, 9.9998e-01],
        [6.5220e-07, 1.0000e+00],
        [3.3250e-06, 1.0000e+00],
        [1.2332e-08, 1.0000e+00],
        [1.0776e-03, 9.9892e-01],
        [1.3330e-05, 9.9999e-01],
        [9.5515e-01, 4.4853e-02],
        [4.1032e-05, 9.9996e-01],
        [1.2114e-06, 1.0000e+00],
        [8.5504e-09, 1.0000e+00],
        [9.4456e-08, 1.0000e+00],
        [2.8945e-06, 1.0000e+00],
        [4.6797e-08, 1.0000e+00],
        [9.9859e-01, 1.4112e-03],
        [1.2748e-08, 1.0000e+00],
        [2.0135e-08, 1.0000e+00],
        [9.9842e-01, 1.5823e-03],
        [9.9991e-01, 9.2274e-05],
        [3.7460e-06, 1.0000e+00],
        [2.3138e-08, 1.0000e+00],
        [9.9184e-01, 8.1599e-03],
        [9.6405e-01, 3.5950e-02],
        [6.2490e-05, 9.9994e-01],
        [5.6246e-04, 9.9944e-01],
        [8.0331e-02, 9.1967e-01],
        [1.3364e-03, 9.9866e-01],
        [8.3240e-06, 9.9999e-01],
        [9.9711e-01, 2.8889e-03],
        [7.0022e-08, 1.0000e+00],
        [9.4324e-03, 9.9057e-01],
        [6.7745e-07, 1.0000e+00],
        [2.1222e-02, 9.7878e-01],
        [1.4428e-06, 1.0000e+00],
        [1.3433e-03, 9.9866e-01],
        [2.6115e-06, 1.0000e+00],
        [4.6229e-07, 1.0000e+00],
        [9.9961e-01, 3.8498e-04],
        [9.9989e-01, 1.0920e-04],
        [3.4027e-05, 9.9997e-01],
        [2.6215e-03, 9.9738e-01],
        [9.9981e-01, 1.9076e-04],
        [9.9968e-01, 3.1703e-04],
        [4.9233e-05, 9.9995e-01],
        [1.0131e-02, 9.8987e-01],
        [4.6704e-04, 9.9953e-01],
        [4.1416e-09, 1.0000e+00],
        [4.9350e-07, 1.0000e+00],
        [2.3137e-02, 9.7686e-01],
        [3.5670e-04, 9.9964e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.1386e-05, 9.9992e-01],
        [8.2018e-08, 1.0000e+00],
        [3.7747e-04, 9.9962e-01],
        [9.9362e-01, 6.3823e-03],
        [7.3450e-06, 9.9999e-01],
        [9.9987e-01, 1.3270e-04],
        [4.1486e-04, 9.9959e-01],
        [9.9706e-01, 2.9409e-03],
        [2.0390e-04, 9.9980e-01],
        [9.9838e-01, 1.6162e-03],
        [1.1008e-03, 9.9890e-01],
        [1.1622e-08, 1.0000e+00],
        [9.9990e-01, 9.8792e-05],
        [3.5350e-05, 9.9996e-01],
        [9.9942e-01, 5.8430e-04],
        [9.9994e-01, 5.8762e-05],
        [1.3206e-02, 9.8679e-01],
        [9.9990e-01, 1.0191e-04],
        [9.9963e-01, 3.6938e-04],
        [7.2611e-06, 9.9999e-01],
        [4.3182e-06, 1.0000e+00],
        [5.7654e-01, 4.2346e-01],
        [5.3189e-03, 9.9468e-01],
        [1.6329e-08, 1.0000e+00],
        [3.0362e-06, 1.0000e+00],
        [6.4587e-02, 9.3541e-01],
        [5.5990e-07, 1.0000e+00],
        [1.0137e-02, 9.8986e-01],
        [9.9998e-01, 2.1334e-05],
        [3.8011e-07, 1.0000e+00],
        [9.9957e-01, 4.3481e-04],
        [2.1943e-05, 9.9998e-01],
        [4.2256e-03, 9.9577e-01],
        [9.8576e-01, 1.4238e-02],
        [1.3863e-05, 9.9999e-01],
        [9.9989e-01, 1.1109e-04],
        [3.1698e-04, 9.9968e-01],
        [1.6782e-04, 9.9983e-01],
        [5.3718e-10, 1.0000e+00],
        [9.9988e-01, 1.2378e-04],
        [9.9974e-01, 2.5880e-04],
        [1.2266e-05, 9.9999e-01],
        [9.9988e-01, 1.2430e-04],
        [2.1579e-09, 1.0000e+00],
        [1.7453e-05, 9.9998e-01],
        [2.1970e-04, 9.9978e-01],
        [9.9924e-01, 7.6381e-04],
        [9.7184e-10, 1.0000e+00],
        [6.0263e-12, 1.0000e+00],
        [2.0412e-09, 1.0000e+00],
        [4.4573e-10, 1.0000e+00],
        [1.8863e-09, 1.0000e+00],
        [8.1246e-04, 9.9919e-01],
        [2.0937e-07, 1.0000e+00],
        [1.5753e-06, 1.0000e+00],
        [8.4186e-05, 9.9992e-01],
        [2.1923e-08, 1.0000e+00],
        [1.0000e+00, 4.2860e-07],
        [1.0004e-04, 9.9990e-01],
        [3.8719e-07, 1.0000e+00],
        [2.6157e-04, 9.9974e-01],
        [3.7097e-06, 1.0000e+00],
        [9.9976e-01, 2.3919e-04],
        [1.8493e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.1974e-09, 1.0000e+00],
        [3.9718e-03, 9.9603e-01],
        [1.8018e-06, 1.0000e+00],
        [2.5483e-07, 1.0000e+00],
        [8.2812e-01, 1.7188e-01],
        [9.9985e-01, 1.5430e-04],
        [4.5379e-06, 1.0000e+00],
        [2.3358e-05, 9.9998e-01],
        [1.6182e-03, 9.9838e-01],
        [9.9824e-01, 1.7627e-03],
        [9.3864e-03, 9.9061e-01],
        [1.3511e-08, 1.0000e+00],
        [9.9969e-01, 3.0812e-04],
        [5.6007e-02, 9.4399e-01],
        [9.9978e-01, 2.1599e-04],
        [5.8726e-05, 9.9994e-01],
        [1.3321e-04, 9.9987e-01],
        [1.2806e-07, 1.0000e+00],
        [4.0752e-09, 1.0000e+00],
        [9.9986e-01, 1.4452e-04],
        [1.8530e-01, 8.1470e-01],
        [9.9820e-01, 1.7958e-03],
        [3.1693e-06, 1.0000e+00],
        [9.9967e-01, 3.2713e-04],
        [4.6914e-04, 9.9953e-01],
        [9.9866e-01, 1.3400e-03],
        [2.1959e-07, 1.0000e+00],
        [4.2660e-07, 1.0000e+00],
        [1.2988e-06, 1.0000e+00],
        [1.4284e-02, 9.8572e-01],
        [1.6626e-03, 9.9834e-01],
        [1.7271e-04, 9.9983e-01],
        [6.8898e-01, 3.1102e-01],
        [4.3520e-03, 9.9565e-01],
        [3.0279e-01, 6.9721e-01],
        [3.1088e-06, 1.0000e+00],
        [2.3880e-05, 9.9998e-01],
        [9.9986e-01, 1.3522e-04],
        [3.6930e-02, 9.6307e-01],
        [6.7770e-07, 1.0000e+00],
        [9.9954e-01, 4.6104e-04],
        [1.9648e-03, 9.9804e-01],
        [8.4182e-04, 9.9916e-01],
        [9.9633e-01, 3.6692e-03],
        [9.9995e-01, 5.4208e-05],
        [1.1231e-09, 1.0000e+00],
        [4.0314e-08, 1.0000e+00],
        [9.2664e-09, 1.0000e+00],
        [1.1020e-07, 1.0000e+00],
        [8.7627e-04, 9.9912e-01],
        [1.2746e-03, 9.9873e-01],
        [9.9986e-01, 1.3739e-04],
        [6.6270e-06, 9.9999e-01],
        [8.6288e-08, 1.0000e+00],
        [1.1217e-08, 1.0000e+00],
        [2.0905e-08, 1.0000e+00],
        [1.7696e-05, 9.9998e-01],
        [3.7177e-06, 1.0000e+00],
        [9.9577e-01, 4.2308e-03],
        [1.0333e-06, 1.0000e+00],
        [4.1579e-04, 9.9958e-01],
        [9.9971e-01, 2.9360e-04],
        [7.8023e-03, 9.9220e-01],
        [6.4518e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.8176e-04, 9.9972e-01],
        [1.2582e-01, 8.7418e-01],
        [1.1482e-05, 9.9999e-01],
        [5.0868e-03, 9.9491e-01],
        [8.0569e-01, 1.9431e-01],
        [9.0780e-01, 9.2200e-02],
        [9.5752e-01, 4.2480e-02],
        [7.5772e-05, 9.9992e-01],
        [3.6666e-08, 1.0000e+00],
        [3.3588e-04, 9.9966e-01],
        [9.9983e-01, 1.6688e-04],
        [3.9036e-08, 1.0000e+00],
        [1.0138e-07, 1.0000e+00],
        [8.1893e-09, 1.0000e+00],
        [4.8479e-02, 9.5152e-01],
        [1.8449e-05, 9.9998e-01],
        [9.5649e-01, 4.3515e-02],
        [9.9987e-01, 1.2587e-04],
        [9.9579e-01, 4.2073e-03],
        [9.9983e-01, 1.6866e-04],
        [1.1095e-08, 1.0000e+00],
        [1.0359e-06, 1.0000e+00],
        [7.2011e-08, 1.0000e+00],
        [1.3755e-02, 9.8625e-01],
        [2.0792e-04, 9.9979e-01],
        [4.7905e-05, 9.9995e-01],
        [4.0812e-05, 9.9996e-01],
        [3.8510e-04, 9.9961e-01],
        [9.3835e-05, 9.9991e-01],
        [9.9556e-01, 4.4355e-03],
        [5.9649e-05, 9.9994e-01],
        [6.7341e-05, 9.9993e-01],
        [1.9559e-09, 1.0000e+00],
        [2.8634e-09, 1.0000e+00],
        [1.8191e-03, 9.9818e-01],
        [8.7074e-04, 9.9913e-01],
        [1.4505e-07, 1.0000e+00],
        [1.4962e-02, 9.8504e-01],
        [3.7142e-12, 1.0000e+00],
        [1.2019e-03, 9.9880e-01],
        [9.9964e-01, 3.5786e-04],
        [2.3934e-08, 1.0000e+00],
        [1.8178e-07, 1.0000e+00],
        [1.8958e-05, 9.9998e-01],
        [1.2764e-04, 9.9987e-01],
        [3.2955e-05, 9.9997e-01],
        [6.4881e-05, 9.9994e-01],
        [5.8006e-06, 9.9999e-01],
        [8.9126e-06, 9.9999e-01],
        [9.9999e-01, 5.7516e-06],
        [9.9982e-01, 1.8172e-04],
        [7.4984e-08, 1.0000e+00],
        [4.5130e-04, 9.9955e-01],
        [7.0496e-05, 9.9993e-01],
        [2.6696e-07, 1.0000e+00],
        [3.7340e-05, 9.9996e-01],
        [9.9475e-01, 5.2513e-03],
        [7.5022e-05, 9.9993e-01],
        [7.0241e-08, 1.0000e+00],
        [4.3358e-08, 1.0000e+00],
        [1.7543e-03, 9.9825e-01],
        [1.5645e-05, 9.9998e-01],
        [4.8611e-05, 9.9995e-01],
        [9.4402e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6411e-02, 9.8359e-01],
        [2.8868e-04, 9.9971e-01],
        [9.9981e-01, 1.9449e-04],
        [8.1496e-07, 1.0000e+00],
        [2.9314e-06, 1.0000e+00],
        [5.9121e-06, 9.9999e-01],
        [9.9997e-01, 2.8816e-05],
        [9.9997e-01, 3.4982e-05],
        [7.3423e-07, 1.0000e+00],
        [9.6682e-01, 3.3178e-02],
        [3.7009e-04, 9.9963e-01],
        [1.0000e+00, 4.5144e-07],
        [9.9989e-01, 1.1159e-04],
        [1.0412e-04, 9.9990e-01],
        [9.9980e-01, 2.0254e-04],
        [7.6207e-05, 9.9992e-01],
        [3.8569e-05, 9.9996e-01],
        [9.8073e-01, 1.9267e-02],
        [9.9981e-01, 1.9368e-04],
        [9.5711e-07, 1.0000e+00],
        [2.1674e-03, 9.9783e-01],
        [3.1372e-07, 1.0000e+00],
        [2.6712e-03, 9.9733e-01],
        [2.8169e-09, 1.0000e+00],
        [4.6496e-08, 1.0000e+00],
        [5.0811e-05, 9.9995e-01],
        [9.9999e-01, 1.0292e-05],
        [8.9858e-01, 1.0142e-01],
        [5.9363e-05, 9.9994e-01],
        [2.9017e-02, 9.7098e-01],
        [9.9966e-01, 3.3987e-04],
        [2.4882e-07, 1.0000e+00],
        [9.9981e-01, 1.9016e-04],
        [3.0474e-08, 1.0000e+00],
        [5.4451e-02, 9.4555e-01],
        [1.8308e-05, 9.9998e-01],
        [9.9043e-01, 9.5717e-03],
        [1.7311e-01, 8.2689e-01],
        [5.9499e-06, 9.9999e-01],
        [9.9980e-01, 1.9884e-04],
        [9.9970e-01, 3.0154e-04],
        [3.5248e-03, 9.9648e-01],
        [6.6491e-08, 1.0000e+00],
        [3.3977e-04, 9.9966e-01],
        [6.3409e-02, 9.3659e-01],
        [2.7385e-04, 9.9973e-01],
        [7.6552e-09, 1.0000e+00],
        [9.9995e-01, 5.3888e-05],
        [9.9969e-01, 3.0969e-04],
        [1.7160e-06, 1.0000e+00],
        [3.9135e-06, 1.0000e+00],
        [1.0296e-02, 9.8970e-01],
        [8.3850e-06, 9.9999e-01],
        [2.4763e-06, 1.0000e+00],
        [4.8017e-09, 1.0000e+00],
        [2.2674e-06, 1.0000e+00],
        [3.0112e-01, 6.9888e-01],
        [7.3253e-08, 1.0000e+00],
        [1.6678e-04, 9.9983e-01],
        [1.9427e-04, 9.9981e-01],
        [4.8377e-03, 9.9516e-01],
        [1.9668e-06, 1.0000e+00],
        [9.9976e-01, 2.3508e-04],
        [2.4755e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5918e-07, 1.0000e+00],
        [8.4155e-03, 9.9158e-01],
        [8.9680e-07, 1.0000e+00],
        [9.4790e-05, 9.9991e-01],
        [6.2071e-05, 9.9994e-01],
        [1.7463e-03, 9.9825e-01],
        [7.6436e-03, 9.9236e-01],
        [9.9983e-01, 1.6753e-04],
        [2.8507e-05, 9.9997e-01],
        [9.9815e-01, 1.8458e-03],
        [4.0721e-02, 9.5928e-01],
        [8.5212e-04, 9.9915e-01],
        [2.2539e-07, 1.0000e+00],
        [5.3130e-04, 9.9947e-01],
        [6.4527e-06, 9.9999e-01],
        [1.2072e-04, 9.9988e-01],
        [1.8397e-05, 9.9998e-01],
        [9.9988e-01, 1.2016e-04],
        [2.0218e-06, 1.0000e+00],
        [3.7457e-04, 9.9963e-01],
        [2.3495e-05, 9.9998e-01],
        [5.5466e-03, 9.9445e-01],
        [1.0700e-05, 9.9999e-01],
        [9.9961e-01, 3.8646e-04],
        [9.9987e-01, 1.3263e-04],
        [9.9999e-01, 9.1506e-06],
        [5.8446e-05, 9.9994e-01],
        [9.9975e-01, 2.5277e-04],
        [5.7406e-04, 9.9943e-01],
        [3.5165e-08, 1.0000e+00],
        [9.9997e-01, 3.3164e-05],
        [5.2616e-02, 9.4738e-01],
        [9.9996e-01, 4.3978e-05],
        [8.2122e-05, 9.9992e-01],
        [4.8851e-03, 9.9511e-01],
        [6.3887e-07, 1.0000e+00],
        [5.8455e-05, 9.9994e-01],
        [9.9983e-01, 1.6742e-04],
        [3.2475e-02, 9.6752e-01],
        [3.6872e-04, 9.9963e-01],
        [9.9987e-01, 1.2728e-04],
        [7.1981e-01, 2.8019e-01],
        [7.1981e-01, 2.8019e-01],
        [9.7278e-01, 2.7219e-02],
        [1.9767e-03, 9.9802e-01],
        [9.9957e-01, 4.3450e-04],
        [9.1045e-01, 8.9552e-02],
        [1.0616e-08, 1.0000e+00],
        [9.5626e-09, 1.0000e+00],
        [3.7626e-04, 9.9962e-01],
        [2.8938e-06, 1.0000e+00],
        [7.6303e-02, 9.2370e-01],
        [1.2913e-07, 1.0000e+00],
        [1.0291e-04, 9.9990e-01],
        [9.9570e-01, 4.2995e-03],
        [4.6355e-01, 5.3645e-01],
        [8.2811e-07, 1.0000e+00],
        [8.8549e-01, 1.1451e-01],
        [5.5198e-03, 9.9448e-01],
        [1.2427e-07, 1.0000e+00],
        [6.8787e-06, 9.9999e-01],
        [7.9825e-06, 9.9999e-01],
        [9.9986e-01, 1.3800e-04],
        [5.0643e-03, 9.9494e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.2080e-02, 9.2792e-01],
        [9.6736e-07, 1.0000e+00],
        [9.9971e-01, 2.9255e-04],
        [6.0823e-05, 9.9994e-01],
        [6.1215e-03, 9.9388e-01],
        [9.9993e-01, 6.7452e-05],
        [4.4172e-05, 9.9996e-01],
        [8.5547e-05, 9.9991e-01],
        [6.2232e-06, 9.9999e-01],
        [9.9980e-01, 2.0433e-04],
        [7.3526e-04, 9.9926e-01],
        [9.9977e-01, 2.3377e-04],
        [2.2292e-05, 9.9998e-01],
        [9.9987e-01, 1.3224e-04],
        [9.6784e-04, 9.9903e-01],
        [7.4768e-09, 1.0000e+00],
        [9.9926e-01, 7.3650e-04],
        [9.8945e-01, 1.0547e-02],
        [6.3011e-04, 9.9937e-01],
        [9.8555e-04, 9.9901e-01],
        [1.4642e-07, 1.0000e+00],
        [5.6783e-06, 9.9999e-01],
        [9.9984e-01, 1.5990e-04],
        [3.4889e-03, 9.9651e-01],
        [1.3330e-04, 9.9987e-01],
        [2.1395e-07, 1.0000e+00],
        [9.9982e-01, 1.7941e-04],
        [9.8387e-01, 1.6129e-02],
        [9.9989e-01, 1.1339e-04],
        [1.9463e-07, 1.0000e+00],
        [3.2012e-03, 9.9680e-01],
        [9.9982e-01, 1.8215e-04],
        [2.1789e-04, 9.9978e-01],
        [5.5222e-04, 9.9945e-01],
        [9.9997e-01, 3.3137e-05],
        [9.9987e-01, 1.2625e-04],
        [5.4871e-04, 9.9945e-01],
        [1.4180e-04, 9.9986e-01],
        [6.6000e-05, 9.9993e-01],
        [6.0870e-05, 9.9994e-01],
        [4.0967e-03, 9.9590e-01],
        [2.7683e-08, 1.0000e+00],
        [3.9624e-04, 9.9960e-01],
        [2.9750e-02, 9.7025e-01],
        [1.3778e-07, 1.0000e+00],
        [9.9975e-01, 2.5148e-04],
        [9.9906e-01, 9.3581e-04],
        [9.9994e-01, 6.2806e-05],
        [1.8144e-03, 9.9819e-01],
        [8.3830e-06, 9.9999e-01],
        [3.5019e-08, 1.0000e+00],
        [3.0452e-07, 1.0000e+00],
        [9.9981e-01, 1.9163e-04],
        [1.5765e-05, 9.9998e-01],
        [9.9930e-01, 7.0168e-04],
        [8.0235e-04, 9.9920e-01],
        [1.9170e-02, 9.8083e-01],
        [4.5121e-05, 9.9995e-01],
        [5.6051e-08, 1.0000e+00],
        [9.9942e-01, 5.8218e-04],
        [3.1187e-04, 9.9969e-01],
        [1.0420e-01, 8.9580e-01],
        [9.9950e-01, 5.0123e-04],
        [9.3520e-02, 9.0648e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9953e-01, 4.7346e-04],
        [1.1133e-08, 1.0000e+00],
        [7.5409e-05, 9.9992e-01],
        [4.5020e-04, 9.9955e-01],
        [5.0238e-07, 1.0000e+00],
        [1.5043e-06, 1.0000e+00],
        [5.6990e-05, 9.9994e-01],
        [1.8330e-05, 9.9998e-01],
        [2.8088e-01, 7.1912e-01],
        [8.6387e-01, 1.3613e-01],
        [3.5347e-07, 1.0000e+00],
        [1.0663e-04, 9.9989e-01],
        [4.6928e-05, 9.9995e-01],
        [9.9984e-01, 1.6416e-04],
        [1.9668e-06, 1.0000e+00],
        [3.6134e-05, 9.9996e-01],
        [9.9215e-07, 1.0000e+00],
        [1.0371e-04, 9.9990e-01],
        [6.6320e-07, 1.0000e+00],
        [9.9996e-01, 3.6319e-05],
        [9.7604e-01, 2.3958e-02],
        [6.8812e-04, 9.9931e-01],
        [6.5924e-07, 1.0000e+00],
        [9.9995e-01, 4.8155e-05],
        [1.2956e-05, 9.9999e-01],
        [3.6629e-03, 9.9634e-01],
        [6.9600e-05, 9.9993e-01],
        [3.8783e-05, 9.9996e-01],
        [7.1088e-05, 9.9993e-01],
        [2.0687e-05, 9.9998e-01],
        [7.5248e-01, 2.4752e-01],
        [1.7404e-04, 9.9983e-01],
        [1.8246e-05, 9.9998e-01],
        [9.9997e-01, 3.2956e-05],
        [7.6540e-07, 1.0000e+00],
        [9.9995e-01, 5.0615e-05],
        [1.4237e-03, 9.9858e-01],
        [4.1696e-07, 1.0000e+00],
        [1.0462e-04, 9.9990e-01],
        [7.4256e-09, 1.0000e+00],
        [2.0112e-06, 1.0000e+00],
        [3.0126e-06, 1.0000e+00],
        [2.6874e-04, 9.9973e-01],
        [8.3998e-06, 9.9999e-01],
        [8.7269e-01, 1.2731e-01],
        [2.9503e-05, 9.9997e-01],
        [7.8405e-05, 9.9992e-01],
        [1.1292e-01, 8.8708e-01],
        [9.9981e-01, 1.8629e-04],
        [2.3825e-07, 1.0000e+00],
        [9.9996e-01, 3.7248e-05],
        [9.6037e-01, 3.9632e-02],
        [9.9994e-01, 6.2415e-05],
        [5.6736e-04, 9.9943e-01],
        [8.0144e-10, 1.0000e+00],
        [1.9253e-07, 1.0000e+00],
        [6.9661e-01, 3.0339e-01],
        [5.6384e-07, 1.0000e+00],
        [2.9067e-04, 9.9971e-01],
        [9.9997e-01, 3.1593e-05],
        [9.9994e-01, 6.0803e-05],
        [9.6931e-04, 9.9903e-01],
        [1.0049e-03, 9.9900e-01],
        [2.0037e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0531e-07, 1.0000e+00],
        [6.4975e-03, 9.9350e-01],
        [9.9984e-01, 1.6029e-04],
        [7.9349e-06, 9.9999e-01],
        [1.7984e-07, 1.0000e+00],
        [1.7077e-03, 9.9829e-01],
        [3.1761e-04, 9.9968e-01],
        [1.7313e-05, 9.9998e-01],
        [9.9957e-01, 4.3264e-04],
        [9.9992e-01, 7.8272e-05],
        [4.7158e-06, 1.0000e+00],
        [9.9787e-01, 2.1262e-03],
        [3.8770e-08, 1.0000e+00],
        [9.7106e-01, 2.8942e-02],
        [2.1558e-04, 9.9978e-01],
        [9.9965e-01, 3.4831e-04],
        [4.2986e-04, 9.9957e-01],
        [5.8931e-04, 9.9941e-01],
        [1.3652e-07, 1.0000e+00],
        [9.9991e-01, 9.2915e-05],
        [1.8006e-04, 9.9982e-01],
        [1.7521e-06, 1.0000e+00],
        [9.9532e-01, 4.6804e-03],
        [9.9874e-01, 1.2560e-03],
        [1.0446e-06, 1.0000e+00],
        [1.2335e-07, 1.0000e+00],
        [5.6797e-09, 1.0000e+00],
        [2.4932e-03, 9.9751e-01],
        [2.1245e-05, 9.9998e-01],
        [1.5377e-04, 9.9985e-01],
        [7.0169e-08, 1.0000e+00],
        [4.0652e-07, 1.0000e+00],
        [7.0110e-04, 9.9930e-01],
        [1.5496e-03, 9.9845e-01],
        [8.1705e-05, 9.9992e-01],
        [4.3915e-03, 9.9561e-01],
        [6.4755e-10, 1.0000e+00],
        [2.1719e-05, 9.9998e-01],
        [9.9970e-01, 3.0143e-04],
        [1.1678e-01, 8.8322e-01],
        [4.2535e-05, 9.9996e-01],
        [2.8687e-06, 1.0000e+00],
        [1.6635e-03, 9.9834e-01],
        [9.2487e-08, 1.0000e+00],
        [2.0870e-05, 9.9998e-01],
        [7.6555e-04, 9.9923e-01],
        [4.2108e-09, 1.0000e+00],
        [2.4467e-06, 1.0000e+00],
        [1.6627e-07, 1.0000e+00],
        [9.9989e-01, 1.0536e-04],
        [2.6345e-06, 1.0000e+00],
        [9.9870e-01, 1.3047e-03],
        [9.9976e-01, 2.4319e-04],
        [3.4762e-03, 9.9652e-01],
        [9.9980e-01, 1.9677e-04],
        [6.2927e-07, 1.0000e+00],
        [3.5989e-06, 1.0000e+00],
        [9.8175e-01, 1.8245e-02],
        [9.9995e-01, 4.8455e-05],
        [9.9992e-01, 8.3578e-05],
        [4.8542e-06, 1.0000e+00],
        [9.9986e-01, 1.4391e-04],
        [5.6473e-06, 9.9999e-01],
        [1.1523e-04, 9.9988e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.3378e-08, 1.0000e+00],
        [9.7787e-01, 2.2126e-02],
        [1.0650e-05, 9.9999e-01],
        [7.3025e-05, 9.9993e-01],
        [9.5150e-01, 4.8500e-02],
        [9.2488e-01, 7.5117e-02],
        [2.8667e-04, 9.9971e-01],
        [1.9861e-05, 9.9998e-01],
        [1.9011e-07, 1.0000e+00],
        [2.2962e-05, 9.9998e-01],
        [9.9984e-01, 1.5571e-04],
        [9.9978e-01, 2.2043e-04],
        [9.9977e-01, 2.3184e-04],
        [8.0126e-01, 1.9874e-01],
        [7.2504e-06, 9.9999e-01],
        [9.9970e-01, 3.0167e-04],
        [3.7009e-07, 1.0000e+00],
        [7.3837e-05, 9.9993e-01],
        [2.7803e-07, 1.0000e+00],
        [9.9990e-01, 9.9032e-05],
        [4.8139e-04, 9.9952e-01],
        [3.8794e-08, 1.0000e+00],
        [5.1499e-04, 9.9949e-01],
        [1.0000e+00, 2.4195e-06],
        [1.8752e-07, 1.0000e+00],
        [1.5052e-05, 9.9998e-01],
        [9.9989e-01, 1.0777e-04],
        [5.5864e-06, 9.9999e-01],
        [5.6247e-05, 9.9994e-01],
        [2.0271e-02, 9.7973e-01],
        [1.8714e-04, 9.9981e-01],
        [8.0268e-04, 9.9920e-01],
        [3.5300e-07, 1.0000e+00],
        [1.7915e-05, 9.9998e-01],
        [3.3564e-05, 9.9997e-01],
        [1.7439e-06, 1.0000e+00],
        [6.2424e-07, 1.0000e+00],
        [8.2802e-06, 9.9999e-01],
        [2.6873e-05, 9.9997e-01],
        [9.9907e-01, 9.3050e-04],
        [3.4345e-08, 1.0000e+00],
        [2.1412e-07, 1.0000e+00],
        [4.1454e-06, 1.0000e+00],
        [1.9362e-05, 9.9998e-01],
        [5.4969e-10, 1.0000e+00],
        [9.9983e-01, 1.7463e-04],
        [1.5542e-10, 1.0000e+00],
        [5.1502e-09, 1.0000e+00],
        [2.4120e-03, 9.9759e-01],
        [9.9925e-01, 7.4927e-04],
        [1.4726e-07, 1.0000e+00],
        [1.3214e-04, 9.9987e-01],
        [2.8147e-06, 1.0000e+00],
        [1.0796e-02, 9.8920e-01],
        [1.6943e-03, 9.9831e-01],
        [9.9685e-01, 3.1530e-03],
        [4.1470e-01, 5.8530e-01],
        [9.9937e-01, 6.3090e-04],
        [4.7198e-05, 9.9995e-01],
        [4.9188e-02, 9.5081e-01],
        [9.9973e-01, 2.6817e-04],
        [9.7947e-04, 9.9902e-01],
        [3.4657e-07, 1.0000e+00],
        [1.5272e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2008e-04, 9.9978e-01],
        [1.0088e-06, 1.0000e+00],
        [1.7969e-05, 9.9998e-01],
        [6.9500e-05, 9.9993e-01],
        [6.5655e-07, 1.0000e+00],
        [9.9453e-01, 5.4717e-03],
        [9.9995e-01, 5.2634e-05],
        [7.0863e-09, 1.0000e+00],
        [9.9959e-01, 4.1046e-04],
        [1.8281e-06, 1.0000e+00],
        [9.7922e-05, 9.9990e-01],
        [7.0040e-04, 9.9930e-01],
        [3.7299e-07, 1.0000e+00],
        [9.9979e-01, 2.1254e-04],
        [3.8492e-04, 9.9962e-01],
        [3.1977e-09, 1.0000e+00],
        [4.2414e-10, 1.0000e+00],
        [2.9337e-09, 1.0000e+00],
        [1.1907e-07, 1.0000e+00],
        [3.7377e-08, 1.0000e+00],
        [2.2144e-05, 9.9998e-01],
        [1.3444e-08, 1.0000e+00],
        [8.7238e-04, 9.9913e-01],
        [9.9923e-01, 7.7440e-04],
        [9.9820e-01, 1.8032e-03],
        [1.0060e-03, 9.9899e-01],
        [1.6577e-10, 1.0000e+00],
        [9.9927e-01, 7.2789e-04],
        [3.8038e-03, 9.9620e-01],
        [1.0523e-06, 1.0000e+00],
        [9.9961e-01, 3.8732e-04],
        [4.6622e-07, 1.0000e+00],
        [7.4779e-05, 9.9993e-01],
        [1.2350e-07, 1.0000e+00],
        [5.9369e-02, 9.4063e-01],
        [5.6245e-02, 9.4376e-01],
        [9.9987e-01, 1.3490e-04],
        [3.3656e-09, 1.0000e+00],
        [7.2819e-07, 1.0000e+00],
        [8.1738e-07, 1.0000e+00],
        [9.8884e-01, 1.1162e-02],
        [9.9901e-01, 9.9449e-04],
        [9.9981e-01, 1.9102e-04],
        [9.9955e-01, 4.5397e-04],
        [1.6111e-07, 1.0000e+00],
        [7.7144e-06, 9.9999e-01],
        [3.0851e-05, 9.9997e-01],
        [1.8657e-05, 9.9998e-01],
        [9.9989e-01, 1.0921e-04],
        [1.7116e-07, 1.0000e+00],
        [9.9992e-01, 8.3667e-05],
        [1.7791e-05, 9.9998e-01],
        [2.4303e-07, 1.0000e+00],
        [6.7083e-06, 9.9999e-01],
        [6.3297e-07, 1.0000e+00],
        [4.8540e-08, 1.0000e+00],
        [1.6840e-04, 9.9983e-01],
        [8.1803e-07, 1.0000e+00],
        [9.9980e-01, 2.0354e-04],
        [9.8942e-01, 1.0579e-02],
        [8.8307e-08, 1.0000e+00],
        [6.5043e-07, 1.0000e+00],
        [1.9824e-09, 1.0000e+00],
        [4.5475e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.1438e-03, 9.9186e-01],
        [4.1303e-04, 9.9959e-01],
        [9.9389e-01, 6.1145e-03],
        [2.0652e-05, 9.9998e-01],
        [9.8268e-01, 1.7318e-02],
        [7.9010e-08, 1.0000e+00],
        [7.9508e-03, 9.9205e-01],
        [4.5570e-06, 1.0000e+00],
        [9.2796e-09, 1.0000e+00],
        [2.2399e-08, 1.0000e+00],
        [1.1339e-09, 1.0000e+00],
        [4.4196e-10, 1.0000e+00],
        [9.9990e-01, 9.5100e-05],
        [9.9991e-01, 9.3521e-05],
        [7.9343e-01, 2.0657e-01],
        [3.9367e-05, 9.9996e-01],
        [9.9978e-01, 2.2484e-04],
        [3.6277e-05, 9.9996e-01],
        [9.9983e-01, 1.7235e-04],
        [1.3770e-05, 9.9999e-01],
        [1.3600e-04, 9.9986e-01],
        [5.4944e-07, 1.0000e+00],
        [1.2448e-04, 9.9988e-01],
        [4.3937e-06, 1.0000e+00],
        [9.1186e-05, 9.9991e-01],
        [8.5154e-03, 9.9148e-01],
        [5.7451e-07, 1.0000e+00],
        [1.2039e-06, 1.0000e+00],
        [3.5900e-07, 1.0000e+00],
        [7.5745e-07, 1.0000e+00],
        [9.9969e-01, 3.1345e-04],
        [1.6064e-07, 1.0000e+00],
        [1.5732e-11, 1.0000e+00],
        [2.1354e-07, 1.0000e+00],
        [3.8890e-04, 9.9961e-01],
        [4.1687e-05, 9.9996e-01],
        [9.9763e-01, 2.3693e-03],
        [2.7288e-05, 9.9997e-01],
        [9.9992e-01, 7.9550e-05],
        [4.1700e-06, 1.0000e+00],
        [9.3758e-07, 1.0000e+00],
        [1.1499e-08, 1.0000e+00],
        [2.0110e-08, 1.0000e+00],
        [6.8577e-07, 1.0000e+00],
        [6.8482e-09, 1.0000e+00],
        [8.3827e-09, 1.0000e+00],
        [1.9188e-03, 9.9808e-01],
        [7.8198e-10, 1.0000e+00],
        [1.2925e-06, 1.0000e+00],
        [5.1092e-09, 1.0000e+00],
        [9.9969e-01, 3.0775e-04],
        [1.1878e-07, 1.0000e+00],
        [5.4520e-02, 9.4548e-01],
        [5.9174e-01, 4.0826e-01],
        [8.2280e-09, 1.0000e+00],
        [3.5913e-05, 9.9996e-01],
        [9.9983e-01, 1.6994e-04],
        [1.2985e-02, 9.8702e-01],
        [1.3546e-06, 1.0000e+00],
        [1.0871e-01, 8.9129e-01],
        [6.5319e-06, 9.9999e-01],
        [1.7259e-08, 1.0000e+00],
        [9.9961e-01, 3.8908e-04],
        [2.4844e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9984e-01, 1.5729e-04],
        [3.2973e-09, 1.0000e+00],
        [2.2255e-07, 1.0000e+00],
        [8.3247e-06, 9.9999e-01],
        [1.5441e-05, 9.9998e-01],
        [1.8097e-05, 9.9998e-01],
        [9.9927e-01, 7.3361e-04],
        [9.9695e-01, 3.0508e-03],
        [4.7039e-02, 9.5296e-01],
        [9.4230e-01, 5.7700e-02],
        [7.6475e-05, 9.9992e-01],
        [8.5939e-01, 1.4061e-01],
        [9.9658e-01, 3.4183e-03],
        [3.3222e-02, 9.6678e-01],
        [5.8432e-08, 1.0000e+00],
        [1.1781e-02, 9.8822e-01],
        [2.5634e-08, 1.0000e+00],
        [9.9968e-01, 3.1930e-04],
        [3.6534e-05, 9.9996e-01],
        [5.0160e-06, 9.9999e-01],
        [9.9971e-01, 2.9082e-04],
        [9.5367e-01, 4.6326e-02],
        [4.1690e-08, 1.0000e+00],
        [7.6271e-06, 9.9999e-01],
        [1.6468e-09, 1.0000e+00],
        [5.2428e-08, 1.0000e+00],
        [9.9762e-01, 2.3768e-03],
        [6.2578e-07, 1.0000e+00],
        [9.4801e-10, 1.0000e+00],
        [9.9740e-01, 2.5995e-03],
        [6.0488e-01, 3.9512e-01],
        [6.5651e-01, 3.4349e-01],
        [5.1245e-10, 1.0000e+00],
        [8.4739e-08, 1.0000e+00],
        [1.1984e-05, 9.9999e-01],
        [3.2287e-04, 9.9968e-01],
        [1.2290e-06, 1.0000e+00],
        [1.8326e-06, 1.0000e+00],
        [7.8180e-07, 1.0000e+00],
        [4.9253e-10, 1.0000e+00],
        [2.7212e-09, 1.0000e+00],
        [2.1229e-06, 1.0000e+00],
        [1.7881e-09, 1.0000e+00],
        [3.7522e-07, 1.0000e+00],
        [4.8584e-07, 1.0000e+00],
        [2.3345e-04, 9.9977e-01],
        [2.7005e-02, 9.7299e-01],
        [3.5806e-04, 9.9964e-01],
        [2.0657e-04, 9.9979e-01],
        [1.7571e-07, 1.0000e+00],
        [1.2294e-01, 8.7706e-01],
        [1.2533e-05, 9.9999e-01],
        [2.0059e-06, 1.0000e+00],
        [9.4787e-08, 1.0000e+00],
        [1.7864e-05, 9.9998e-01],
        [2.0808e-06, 1.0000e+00],
        [9.9982e-01, 1.8181e-04],
        [6.4130e-08, 1.0000e+00],
        [2.4533e-08, 1.0000e+00],
        [1.3268e-04, 9.9987e-01],
        [9.9998e-01, 1.7916e-05],
        [1.0840e-06, 1.0000e+00],
        [1.0697e-04, 9.9989e-01],
        [5.8516e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3047e-04, 9.9987e-01],
        [9.6943e-01, 3.0568e-02],
        [3.5721e-08, 1.0000e+00],
        [3.5271e-07, 1.0000e+00],
        [9.9452e-01, 5.4819e-03],
        [9.1915e-10, 1.0000e+00],
        [9.9908e-01, 9.1512e-04],
        [7.1384e-01, 2.8616e-01],
        [1.4659e-05, 9.9999e-01],
        [1.3482e-09, 1.0000e+00],
        [9.9144e-01, 8.5620e-03],
        [3.1912e-05, 9.9997e-01],
        [7.6032e-07, 1.0000e+00],
        [5.3885e-08, 1.0000e+00],
        [9.9961e-01, 3.8564e-04],
        [5.6842e-09, 1.0000e+00],
        [8.4847e-03, 9.9152e-01],
        [1.0147e-06, 1.0000e+00],
        [4.4983e-12, 1.0000e+00],
        [3.4438e-06, 1.0000e+00],
        [9.9168e-01, 8.3234e-03],
        [1.0247e-03, 9.9898e-01],
        [4.2135e-04, 9.9958e-01],
        [3.8272e-07, 1.0000e+00],
        [9.1496e-01, 8.5043e-02],
        [9.9998e-01, 1.7520e-05],
        [2.4541e-03, 9.9755e-01],
        [2.3223e-06, 1.0000e+00],
        [9.7672e-07, 1.0000e+00],
        [9.1133e-07, 1.0000e+00],
        [9.9939e-01, 6.0898e-04],
        [6.7315e-08, 1.0000e+00],
        [9.9823e-01, 1.7722e-03],
        [1.6245e-09, 1.0000e+00],
        [1.9222e-08, 1.0000e+00],
        [6.1791e-07, 1.0000e+00],
        [2.9398e-05, 9.9997e-01],
        [7.4112e-07, 1.0000e+00],
        [9.9948e-01, 5.1755e-04],
        [6.6041e-05, 9.9993e-01],
        [8.0665e-05, 9.9992e-01],
        [9.6069e-01, 3.9306e-02],
        [1.0000e+00, 1.5814e-08],
        [4.5005e-08, 1.0000e+00],
        [2.1691e-07, 1.0000e+00],
        [2.2095e-09, 1.0000e+00],
        [3.8796e-03, 9.9612e-01],
        [1.4123e-05, 9.9999e-01],
        [4.8327e-05, 9.9995e-01],
        [9.0194e-01, 9.8063e-02],
        [9.9980e-01, 1.9633e-04],
        [2.2350e-04, 9.9978e-01],
        [3.4690e-04, 9.9965e-01],
        [3.2221e-09, 1.0000e+00],
        [1.2260e-10, 1.0000e+00],
        [9.6257e-07, 1.0000e+00],
        [1.4089e-01, 8.5911e-01],
        [7.6355e-06, 9.9999e-01],
        [5.8524e-03, 9.9415e-01],
        [1.4128e-07, 1.0000e+00],
        [1.8907e-04, 9.9981e-01],
        [3.0925e-06, 1.0000e+00],
        [2.5335e-08, 1.0000e+00],
        [5.4125e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.3752e-01, 6.2480e-02],
        [1.4707e-05, 9.9999e-01],
        [9.9989e-01, 1.1486e-04],
        [5.5686e-06, 9.9999e-01],
        [2.6538e-08, 1.0000e+00],
        [2.3260e-05, 9.9998e-01],
        [9.7124e-01, 2.8765e-02],
        [1.0000e+00, 3.2582e-06],
        [9.8320e-01, 1.6803e-02],
        [8.8992e-01, 1.1008e-01],
        [1.7428e-05, 9.9998e-01],
        [4.1246e-05, 9.9996e-01],
        [8.2441e-06, 9.9999e-01],
        [2.7096e-04, 9.9973e-01],
        [9.9255e-01, 7.4525e-03],
        [1.3850e-04, 9.9986e-01],
        [2.8399e-04, 9.9972e-01],
        [4.4368e-04, 9.9956e-01],
        [1.7387e-05, 9.9998e-01],
        [3.2804e-09, 1.0000e+00],
        [9.9537e-01, 4.6331e-03],
        [9.8833e-06, 9.9999e-01],
        [8.1160e-03, 9.9188e-01],
        [3.1511e-03, 9.9685e-01],
        [2.3094e-04, 9.9977e-01],
        [1.4121e-03, 9.9859e-01],
        [9.9982e-01, 1.7927e-04],
        [1.2208e-03, 9.9878e-01],
        [9.9981e-01, 1.8934e-04],
        [4.3492e-03, 9.9565e-01],
        [1.1829e-09, 1.0000e+00],
        [4.1976e-05, 9.9996e-01],
        [3.0127e-04, 9.9970e-01],
        [8.8315e-10, 1.0000e+00],
        [6.6536e-06, 9.9999e-01],
        [2.8152e-07, 1.0000e+00],
        [9.9399e-04, 9.9901e-01],
        [1.4348e-06, 1.0000e+00],
        [7.2789e-12, 1.0000e+00],
        [1.4917e-07, 1.0000e+00],
        [9.9786e-01, 2.1357e-03],
        [5.7266e-06, 9.9999e-01],
        [1.5445e-06, 1.0000e+00],
        [1.9861e-07, 1.0000e+00],
        [5.1628e-07, 1.0000e+00],
        [1.3031e-05, 9.9999e-01],
        [7.3986e-04, 9.9926e-01],
        [9.9689e-01, 3.1099e-03],
        [9.9974e-01, 2.5780e-04],
        [9.1858e-09, 1.0000e+00],
        [6.4211e-04, 9.9936e-01],
        [1.0384e-06, 1.0000e+00],
        [1.6437e-10, 1.0000e+00],
        [9.9976e-01, 2.4117e-04],
        [1.3519e-06, 1.0000e+00],
        [9.9614e-01, 3.8566e-03],
        [2.0744e-05, 9.9998e-01],
        [8.0996e-06, 9.9999e-01],
        [3.0812e-08, 1.0000e+00],
        [5.7793e-04, 9.9942e-01],
        [5.0273e-09, 1.0000e+00],
        [3.3906e-05, 9.9997e-01],
        [1.2949e-04, 9.9987e-01],
        [2.2975e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9970e-01, 3.0004e-04],
        [1.8327e-06, 1.0000e+00],
        [4.4160e-06, 1.0000e+00],
        [1.2867e-04, 9.9987e-01],
        [9.8935e-01, 1.0653e-02],
        [6.7282e-09, 1.0000e+00],
        [1.6280e-04, 9.9984e-01],
        [9.9866e-01, 1.3367e-03],
        [5.9350e-03, 9.9406e-01],
        [3.6029e-03, 9.9640e-01],
        [1.4644e-03, 9.9854e-01],
        [9.9960e-01, 3.9898e-04],
        [9.9987e-01, 1.2604e-04],
        [3.1007e-09, 1.0000e+00],
        [9.9962e-01, 3.7579e-04],
        [4.5388e-09, 1.0000e+00],
        [3.6040e-05, 9.9996e-01],
        [9.9829e-01, 1.7128e-03],
        [1.6993e-04, 9.9983e-01],
        [6.3947e-04, 9.9936e-01],
        [9.9982e-01, 1.8347e-04],
        [2.9565e-03, 9.9704e-01],
        [1.8044e-08, 1.0000e+00],
        [9.9999e-01, 1.0051e-05],
        [3.2782e-05, 9.9997e-01],
        [9.3454e-01, 6.5459e-02],
        [6.4601e-09, 1.0000e+00],
        [1.5148e-05, 9.9998e-01],
        [9.9563e-01, 4.3667e-03],
        [3.1725e-05, 9.9997e-01],
        [2.4911e-03, 9.9751e-01],
        [5.0546e-02, 9.4945e-01],
        [3.2649e-05, 9.9997e-01],
        [1.3672e-06, 1.0000e+00],
        [7.1862e-04, 9.9928e-01],
        [4.1061e-07, 1.0000e+00],
        [1.1182e-08, 1.0000e+00],
        [1.3243e-07, 1.0000e+00],
        [9.9975e-01, 2.5186e-04],
        [3.2905e-07, 1.0000e+00],
        [4.4008e-03, 9.9560e-01],
        [6.7031e-06, 9.9999e-01],
        [5.0819e-10, 1.0000e+00],
        [9.6344e-01, 3.6561e-02],
        [9.7749e-01, 2.2514e-02],
        [1.1608e-05, 9.9999e-01],
        [9.9970e-01, 2.9895e-04],
        [2.8828e-03, 9.9712e-01],
        [9.9850e-01, 1.4962e-03],
        [1.1879e-04, 9.9988e-01],
        [1.1824e-06, 1.0000e+00],
        [8.6190e-07, 1.0000e+00],
        [1.1672e-04, 9.9988e-01],
        [9.9751e-01, 2.4881e-03],
        [6.8080e-06, 9.9999e-01],
        [3.4617e-05, 9.9997e-01],
        [2.1133e-02, 9.7887e-01],
        [1.0468e-06, 1.0000e+00],
        [3.0434e-07, 1.0000e+00],
        [9.9980e-01, 1.9787e-04],
        [3.0726e-08, 1.0000e+00],
        [9.9974e-01, 2.5659e-04],
        [9.9941e-01, 5.9152e-04],
        [1.4884e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9936e-01, 6.3546e-04],
        [4.7274e-01, 5.2726e-01],
        [2.0651e-05, 9.9998e-01],
        [1.3365e-04, 9.9987e-01],
        [1.0015e-04, 9.9990e-01],
        [5.9280e-03, 9.9407e-01],
        [4.0797e-08, 1.0000e+00],
        [9.9739e-01, 2.6118e-03],
        [9.9969e-01, 3.1041e-04],
        [9.9983e-01, 1.6940e-04],
        [9.9977e-01, 2.3213e-04],
        [2.2587e-07, 1.0000e+00],
        [1.5670e-04, 9.9984e-01],
        [7.4138e-04, 9.9926e-01],
        [4.8694e-08, 1.0000e+00],
        [2.5809e-08, 1.0000e+00],
        [3.1131e-03, 9.9689e-01],
        [1.0000e+00, 9.0009e-07],
        [3.8800e-05, 9.9996e-01],
        [9.9894e-01, 1.0616e-03],
        [3.0767e-08, 1.0000e+00],
        [8.3607e-05, 9.9992e-01],
        [1.6126e-07, 1.0000e+00],
        [7.3372e-06, 9.9999e-01],
        [6.6499e-05, 9.9993e-01],
        [1.1536e-06, 1.0000e+00],
        [8.1212e-03, 9.9188e-01],
        [6.5325e-06, 9.9999e-01],
        [1.2033e-08, 1.0000e+00],
        [5.6834e-06, 9.9999e-01],
        [4.0065e-06, 1.0000e+00],
        [9.8854e-01, 1.1461e-02],
        [6.0577e-04, 9.9939e-01],
        [9.9931e-01, 6.9395e-04],
        [1.5349e-05, 9.9998e-01],
        [9.9973e-01, 2.7088e-04],
        [2.7055e-08, 1.0000e+00],
        [3.7689e-01, 6.2311e-01],
        [2.7785e-02, 9.7222e-01],
        [4.4878e-08, 1.0000e+00],
        [6.8186e-07, 1.0000e+00],
        [2.0676e-08, 1.0000e+00],
        [5.8606e-05, 9.9994e-01],
        [7.3081e-01, 2.6919e-01],
        [9.3341e-01, 6.6588e-02],
        [5.8086e-01, 4.1914e-01],
        [7.7961e-03, 9.9220e-01],
        [6.4498e-05, 9.9994e-01],
        [9.3173e-03, 9.9068e-01],
        [7.2037e-05, 9.9993e-01],
        [1.4029e-03, 9.9860e-01],
        [9.9978e-01, 2.1561e-04],
        [5.1603e-08, 1.0000e+00],
        [9.9951e-01, 4.8910e-04],
        [1.4483e-06, 1.0000e+00],
        [9.9853e-01, 1.4731e-03],
        [2.6901e-07, 1.0000e+00],
        [5.8129e-07, 1.0000e+00],
        [3.7692e-05, 9.9996e-01],
        [8.9995e-01, 1.0005e-01],
        [1.2006e-06, 1.0000e+00],
        [9.9985e-01, 1.5495e-04],
        [9.9915e-01, 8.4860e-04],
        [1.7842e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2338e-05, 9.9999e-01],
        [9.9273e-01, 7.2651e-03],
        [1.7646e-03, 9.9824e-01],
        [3.5398e-06, 1.0000e+00],
        [9.9903e-01, 9.7470e-04],
        [6.1656e-06, 9.9999e-01],
        [9.9976e-01, 2.3738e-04],
        [1.1852e-06, 1.0000e+00],
        [1.9039e-07, 1.0000e+00],
        [3.3019e-03, 9.9670e-01],
        [9.8122e-01, 1.8778e-02],
        [9.9967e-01, 3.2961e-04],
        [7.7349e-04, 9.9923e-01],
        [1.0000e+00, 1.5901e-06],
        [1.6939e-01, 8.3061e-01],
        [9.9968e-01, 3.2289e-04],
        [3.4436e-07, 1.0000e+00],
        [7.3847e-07, 1.0000e+00],
        [6.8978e-06, 9.9999e-01],
        [7.5994e-06, 9.9999e-01],
        [5.9305e-06, 9.9999e-01],
        [9.9986e-01, 1.3746e-04],
        [1.0344e-06, 1.0000e+00],
        [9.9981e-01, 1.9499e-04],
        [1.4662e-02, 9.8534e-01],
        [1.9265e-09, 1.0000e+00],
        [5.1582e-02, 9.4842e-01],
        [2.1213e-04, 9.9979e-01],
        [7.8603e-07, 1.0000e+00],
        [8.4008e-08, 1.0000e+00],
        [1.2025e-03, 9.9880e-01],
        [3.3462e-06, 1.0000e+00],
        [9.9487e-01, 5.1303e-03],
        [9.9950e-01, 4.9968e-04],
        [3.6522e-05, 9.9996e-01],
        [2.1827e-05, 9.9998e-01],
        [1.7346e-03, 9.9827e-01],
        [9.9954e-01, 4.6054e-04],
        [2.5937e-05, 9.9997e-01],
        [9.9991e-01, 8.6453e-05],
        [8.0643e-02, 9.1936e-01],
        [1.1147e-06, 1.0000e+00],
        [9.9994e-01, 5.9250e-05],
        [1.4735e-02, 9.8527e-01],
        [8.9454e-04, 9.9911e-01],
        [9.9675e-01, 3.2519e-03],
        [2.5284e-10, 1.0000e+00],
        [9.3189e-09, 1.0000e+00],
        [5.2357e-08, 1.0000e+00],
        [1.6792e-09, 1.0000e+00],
        [5.5493e-04, 9.9945e-01],
        [1.7116e-08, 1.0000e+00],
        [9.9961e-01, 3.8703e-04],
        [4.2425e-05, 9.9996e-01],
        [1.0663e-04, 9.9989e-01],
        [3.4760e-11, 1.0000e+00],
        [4.0370e-08, 1.0000e+00],
        [3.2600e-05, 9.9997e-01],
        [4.9619e-06, 9.9999e-01],
        [2.6261e-04, 9.9974e-01],
        [3.9998e-04, 9.9960e-01],
        [6.8983e-03, 9.9310e-01],
        [2.3070e-06, 1.0000e+00],
        [2.6377e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9973e-01, 2.7307e-04],
        [3.4506e-03, 9.9655e-01],
        [1.6889e-06, 1.0000e+00],
        [6.5388e-08, 1.0000e+00],
        [9.7041e-06, 9.9999e-01],
        [9.9978e-01, 2.2178e-04],
        [8.2208e-04, 9.9918e-01],
        [4.1867e-02, 9.5813e-01],
        [3.4525e-06, 1.0000e+00],
        [1.9254e-02, 9.8075e-01],
        [1.5656e-04, 9.9984e-01],
        [7.4817e-02, 9.2518e-01],
        [9.9988e-01, 1.1723e-04],
        [4.0264e-09, 1.0000e+00],
        [2.3184e-07, 1.0000e+00],
        [9.5378e-05, 9.9990e-01],
        [9.9976e-01, 2.3651e-04],
        [9.9973e-01, 2.6512e-04],
        [9.9980e-01, 1.9522e-04],
        [1.8621e-05, 9.9998e-01],
        [9.9976e-01, 2.4484e-04],
        [1.0681e-01, 8.9319e-01],
        [7.1891e-02, 9.2811e-01],
        [3.0338e-08, 1.0000e+00],
        [2.8662e-08, 1.0000e+00],
        [9.8945e-01, 1.0547e-02],
        [1.7089e-05, 9.9998e-01],
        [1.4015e-05, 9.9999e-01],
        [5.2799e-05, 9.9995e-01],
        [1.0204e-06, 1.0000e+00],
        [6.3944e-06, 9.9999e-01],
        [9.9984e-01, 1.6413e-04],
        [2.4178e-02, 9.7582e-01],
        [5.9586e-04, 9.9940e-01],
        [4.3429e-04, 9.9957e-01],
        [4.7319e-01, 5.2681e-01],
        [2.5751e-05, 9.9997e-01],
        [1.1129e-06, 1.0000e+00],
        [5.8400e-09, 1.0000e+00],
        [2.4225e-04, 9.9976e-01],
        [3.9715e-06, 1.0000e+00],
        [1.3816e-05, 9.9999e-01],
        [9.1459e-06, 9.9999e-01],
        [4.6090e-05, 9.9995e-01],
        [9.9799e-01, 2.0076e-03],
        [2.9245e-06, 1.0000e+00],
        [2.2724e-06, 1.0000e+00],
        [2.0468e-07, 1.0000e+00],
        [1.1834e-05, 9.9999e-01],
        [7.0764e-06, 9.9999e-01],
        [1.4479e-06, 1.0000e+00],
        [4.4717e-07, 1.0000e+00],
        [8.2369e-05, 9.9992e-01],
        [8.2894e-05, 9.9992e-01],
        [3.1790e-03, 9.9682e-01],
        [9.9994e-01, 6.2434e-05],
        [3.0077e-05, 9.9997e-01],
        [4.3940e-07, 1.0000e+00],
        [3.3788e-03, 9.9662e-01],
        [3.8361e-03, 9.9616e-01],
        [1.2316e-08, 1.0000e+00],
        [8.6183e-06, 9.9999e-01],
        [3.4317e-09, 1.0000e+00],
        [6.1715e-10, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5389e-05, 9.9998e-01],
        [1.1361e-08, 1.0000e+00],
        [1.4128e-09, 1.0000e+00],
        [9.9809e-01, 1.9086e-03],
        [7.0964e-09, 1.0000e+00],
        [1.3101e-03, 9.9869e-01],
        [1.2613e-03, 9.9874e-01],
        [2.4836e-07, 1.0000e+00],
        [1.7141e-07, 1.0000e+00],
        [3.7857e-04, 9.9962e-01],
        [9.9523e-01, 4.7698e-03],
        [2.1331e-04, 9.9979e-01],
        [1.7482e-04, 9.9983e-01],
        [2.3677e-02, 9.7632e-01],
        [9.9332e-01, 6.6825e-03],
        [9.9459e-01, 5.4072e-03],
        [6.0201e-04, 9.9940e-01],
        [9.1150e-05, 9.9991e-01],
        [4.6448e-08, 1.0000e+00],
        [6.8793e-04, 9.9931e-01],
        [9.9960e-01, 4.0481e-04],
        [1.1927e-06, 1.0000e+00],
        [1.0324e-10, 1.0000e+00],
        [7.0376e-07, 1.0000e+00],
        [8.8087e-02, 9.1191e-01],
        [9.9989e-01, 1.1211e-04],
        [6.0908e-05, 9.9994e-01],
        [5.7397e-07, 1.0000e+00],
        [5.0883e-05, 9.9995e-01],
        [1.4242e-08, 1.0000e+00],
        [3.0107e-06, 1.0000e+00],
        [5.1180e-04, 9.9949e-01],
        [2.9892e-06, 1.0000e+00],
        [8.7737e-05, 9.9991e-01],
        [2.6877e-04, 9.9973e-01],
        [2.0278e-05, 9.9998e-01],
        [2.6780e-05, 9.9997e-01],
        [3.1610e-07, 1.0000e+00],
        [3.1747e-07, 1.0000e+00],
        [9.9987e-01, 1.2783e-04],
        [9.8026e-01, 1.9743e-02],
        [3.9900e-04, 9.9960e-01],
        [2.8349e-04, 9.9972e-01],
        [7.8897e-07, 1.0000e+00],
        [6.9012e-06, 9.9999e-01],
        [4.0788e-03, 9.9592e-01],
        [1.8516e-07, 1.0000e+00],
        [5.9886e-02, 9.4011e-01],
        [4.5526e-09, 1.0000e+00],
        [9.6508e-08, 1.0000e+00],
        [9.9995e-01, 5.4079e-05],
        [2.3525e-07, 1.0000e+00],
        [4.1621e-05, 9.9996e-01],
        [9.3185e-05, 9.9991e-01],
        [7.6133e-05, 9.9992e-01],
        [6.2627e-08, 1.0000e+00],
        [1.0991e-04, 9.9989e-01],
        [1.5629e-04, 9.9984e-01],
        [7.1097e-06, 9.9999e-01],
        [1.4384e-04, 9.9986e-01],
        [1.3463e-08, 1.0000e+00],
        [9.9993e-01, 6.7880e-05],
        [3.5820e-07, 1.0000e+00],
        [3.2424e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.3238e-04, 9.9947e-01],
        [1.5005e-07, 1.0000e+00],
        [3.3254e-05, 9.9997e-01],
        [6.6556e-09, 1.0000e+00],
        [1.4643e-08, 1.0000e+00],
        [1.0406e-06, 1.0000e+00],
        [1.3144e-06, 1.0000e+00],
        [1.1820e-04, 9.9988e-01],
        [6.3153e-06, 9.9999e-01],
        [9.8161e-06, 9.9999e-01],
        [3.0976e-04, 9.9969e-01],
        [1.7594e-08, 1.0000e+00],
        [3.9081e-08, 1.0000e+00],
        [9.9988e-01, 1.2463e-04],
        [1.4265e-07, 1.0000e+00],
        [2.6330e-01, 7.3670e-01],
        [9.0495e-07, 1.0000e+00],
        [6.4332e-08, 1.0000e+00],
        [2.6740e-07, 1.0000e+00],
        [9.6917e-01, 3.0832e-02],
        [1.8401e-07, 1.0000e+00],
        [1.6208e-09, 1.0000e+00],
        [6.9669e-06, 9.9999e-01],
        [8.6332e-08, 1.0000e+00],
        [9.9997e-01, 2.9124e-05],
        [2.1716e-06, 1.0000e+00],
        [6.1138e-06, 9.9999e-01],
        [1.4490e-09, 1.0000e+00],
        [3.0167e-07, 1.0000e+00],
        [2.4523e-09, 1.0000e+00],
        [9.3535e-04, 9.9906e-01],
        [9.9130e-01, 8.6970e-03],
        [1.1794e-06, 1.0000e+00],
        [7.7043e-03, 9.9230e-01],
        [4.0851e-08, 1.0000e+00],
        [1.8646e-07, 1.0000e+00],
        [9.9755e-01, 2.4548e-03],
        [3.3113e-05, 9.9997e-01],
        [3.3896e-05, 9.9997e-01],
        [4.9497e-03, 9.9505e-01],
        [9.9981e-01, 1.9117e-04],
        [1.6556e-05, 9.9998e-01],
        [1.1650e-06, 1.0000e+00],
        [4.9190e-02, 9.5081e-01],
        [3.2504e-05, 9.9997e-01],
        [9.9985e-01, 1.5242e-04],
        [1.1994e-08, 1.0000e+00],
        [2.7097e-05, 9.9997e-01],
        [9.9980e-01, 2.0127e-04],
        [1.5433e-10, 1.0000e+00],
        [1.4047e-06, 1.0000e+00],
        [1.0870e-05, 9.9999e-01],
        [1.4380e-06, 1.0000e+00],
        [8.5480e-05, 9.9991e-01],
        [1.3758e-09, 1.0000e+00],
        [2.6833e-03, 9.9732e-01],
        [2.8306e-08, 1.0000e+00],
        [1.4447e-05, 9.9999e-01],
        [1.8661e-07, 1.0000e+00],
        [5.7003e-01, 4.2997e-01],
        [5.9401e-04, 9.9941e-01],
        [5.5306e-06, 9.9999e-01],
        [2.8765e-08, 1.0000e+00],
        [1.3447e-04, 9.9987e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9917e-01, 8.2832e-04],
        [1.4990e-07, 1.0000e+00],
        [6.7753e-03, 9.9322e-01],
        [1.7834e-07, 1.0000e+00],
        [1.0226e-03, 9.9898e-01],
        [1.9649e-07, 1.0000e+00],
        [1.7031e-09, 1.0000e+00],
        [9.9971e-01, 2.9482e-04],
        [2.0188e-09, 1.0000e+00],
        [3.6820e-04, 9.9963e-01],
        [9.9189e-01, 8.1113e-03],
        [2.5481e-08, 1.0000e+00],
        [7.3312e-01, 2.6688e-01],
        [3.9529e-06, 1.0000e+00],
        [2.8878e-05, 9.9997e-01],
        [4.1580e-07, 1.0000e+00],
        [7.2140e-06, 9.9999e-01],
        [5.1042e-02, 9.4896e-01],
        [3.1130e-06, 1.0000e+00],
        [2.8113e-09, 1.0000e+00],
        [8.1483e-08, 1.0000e+00],
        [3.7190e-05, 9.9996e-01],
        [3.9862e-08, 1.0000e+00],
        [2.3378e-07, 1.0000e+00],
        [9.9982e-01, 1.7862e-04],
        [6.2996e-06, 9.9999e-01],
        [3.5815e-07, 1.0000e+00],
        [5.1567e-03, 9.9484e-01],
        [3.0910e-09, 1.0000e+00],
        [8.7164e-09, 1.0000e+00],
        [1.1877e-06, 1.0000e+00],
        [9.9902e-01, 9.7783e-04],
        [3.9843e-07, 1.0000e+00],
        [5.1557e-05, 9.9995e-01],
        [3.5704e-04, 9.9964e-01],
        [3.6629e-03, 9.9634e-01],
        [9.9974e-01, 2.6176e-04],
        [3.7880e-07, 1.0000e+00],
        [2.2444e-05, 9.9998e-01],
        [2.1752e-06, 1.0000e+00],
        [9.9987e-01, 1.3344e-04],
        [7.3384e-06, 9.9999e-01],
        [9.9980e-01, 1.9754e-04],
        [8.8377e-01, 1.1623e-01],
        [1.9011e-08, 1.0000e+00],
        [9.9960e-01, 3.9901e-04],
        [3.5759e-03, 9.9642e-01],
        [9.9987e-01, 1.2578e-04],
        [4.3948e-05, 9.9996e-01],
        [9.9977e-01, 2.3167e-04],
        [3.6684e-08, 1.0000e+00],
        [4.2225e-04, 9.9958e-01],
        [3.6786e-07, 1.0000e+00],
        [9.9984e-01, 1.6258e-04],
        [6.0208e-07, 1.0000e+00],
        [1.9165e-05, 9.9998e-01],
        [1.5459e-08, 1.0000e+00],
        [2.4057e-05, 9.9998e-01],
        [9.9846e-01, 1.5386e-03],
        [8.0026e-05, 9.9992e-01],
        [1.5157e-02, 9.8484e-01],
        [8.3536e-03, 9.9165e-01],
        [2.5066e-09, 1.0000e+00],
        [6.5141e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9343e-03, 9.9807e-01],
        [9.9724e-01, 2.7584e-03],
        [1.9019e-07, 1.0000e+00],
        [9.9522e-01, 4.7840e-03],
        [1.8491e-05, 9.9998e-01],
        [4.2472e-05, 9.9996e-01],
        [5.6970e-06, 9.9999e-01],
        [1.0830e-05, 9.9999e-01],
        [9.6600e-04, 9.9903e-01],
        [9.9974e-01, 2.6227e-04],
        [3.4225e-05, 9.9997e-01],
        [3.0528e-07, 1.0000e+00],
        [1.9272e-06, 1.0000e+00],
        [3.2303e-09, 1.0000e+00],
        [3.1396e-05, 9.9997e-01],
        [2.0416e-09, 1.0000e+00],
        [9.9765e-01, 2.3495e-03],
        [1.0414e-08, 1.0000e+00],
        [2.1219e-10, 1.0000e+00],
        [9.9987e-01, 1.2523e-04],
        [8.1281e-08, 1.0000e+00],
        [9.9946e-01, 5.4218e-04],
        [3.5128e-06, 1.0000e+00],
        [1.0404e-09, 1.0000e+00],
        [9.9992e-01, 8.3620e-05],
        [2.3784e-03, 9.9762e-01],
        [9.9828e-01, 1.7159e-03],
        [8.0032e-01, 1.9968e-01],
        [8.2877e-04, 9.9917e-01],
        [1.2651e-04, 9.9987e-01],
        [9.0876e-09, 1.0000e+00],
        [9.9990e-01, 1.0302e-04],
        [2.3864e-04, 9.9976e-01],
        [9.2953e-07, 1.0000e+00],
        [8.2472e-11, 1.0000e+00],
        [8.8169e-01, 1.1831e-01],
        [5.0744e-08, 1.0000e+00],
        [1.1919e-07, 1.0000e+00],
        [1.4693e-07, 1.0000e+00],
        [7.0379e-09, 1.0000e+00],
        [3.5405e-04, 9.9965e-01],
        [1.2673e-05, 9.9999e-01],
        [4.3737e-03, 9.9563e-01],
        [1.0151e-06, 1.0000e+00],
        [9.8068e-01, 1.9322e-02],
        [9.9987e-01, 1.3186e-04],
        [8.1309e-09, 1.0000e+00],
        [6.1275e-04, 9.9939e-01],
        [6.5486e-06, 9.9999e-01],
        [9.8079e-01, 1.9215e-02],
        [2.5234e-10, 1.0000e+00],
        [9.9122e-01, 8.7808e-03],
        [3.3935e-03, 9.9661e-01],
        [9.9984e-01, 1.5503e-04],
        [5.0965e-05, 9.9995e-01],
        [1.0318e-07, 1.0000e+00],
        [6.0417e-07, 1.0000e+00],
        [6.7417e-04, 9.9933e-01],
        [9.9579e-01, 4.2100e-03],
        [1.7707e-06, 1.0000e+00],
        [9.9567e-01, 4.3293e-03],
        [2.7095e-03, 9.9729e-01],
        [5.1472e-09, 1.0000e+00],
        [2.9268e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6435e-10, 1.0000e+00],
        [8.9203e-01, 1.0797e-01],
        [9.3190e-04, 9.9907e-01],
        [6.0233e-09, 1.0000e+00],
        [2.2619e-07, 1.0000e+00],
        [6.5392e-02, 9.3461e-01],
        [9.9872e-01, 1.2814e-03],
        [2.3471e-10, 1.0000e+00],
        [3.5712e-08, 1.0000e+00],
        [9.9978e-01, 2.2082e-04],
        [2.1251e-08, 1.0000e+00],
        [9.9982e-01, 1.8468e-04],
        [2.9639e-05, 9.9997e-01],
        [9.9985e-01, 1.5055e-04],
        [2.6359e-01, 7.3641e-01],
        [8.4794e-06, 9.9999e-01],
        [3.1542e-06, 1.0000e+00],
        [8.8635e-10, 1.0000e+00],
        [8.9109e-04, 9.9911e-01],
        [1.6865e-04, 9.9983e-01],
        [3.2708e-06, 1.0000e+00],
        [2.7816e-05, 9.9997e-01],
        [7.8966e-05, 9.9992e-01],
        [1.0921e-03, 9.9891e-01],
        [1.3927e-06, 1.0000e+00],
        [1.6675e-07, 1.0000e+00],
        [8.8490e-01, 1.1510e-01],
        [1.8891e-04, 9.9981e-01],
        [3.0247e-06, 1.0000e+00],
        [9.9998e-01, 1.5885e-05],
        [1.1174e-05, 9.9999e-01],
        [7.7350e-08, 1.0000e+00],
        [1.1321e-04, 9.9989e-01],
        [9.9974e-01, 2.6271e-04],
        [1.1237e-08, 1.0000e+00],
        [1.3362e-09, 1.0000e+00],
        [8.3234e-03, 9.9168e-01],
        [2.3443e-08, 1.0000e+00],
        [1.7520e-07, 1.0000e+00],
        [9.6095e-06, 9.9999e-01],
        [9.9991e-01, 8.7302e-05],
        [2.4826e-02, 9.7517e-01],
        [9.9825e-01, 1.7450e-03],
        [2.8937e-06, 1.0000e+00],
        [9.7170e-01, 2.8303e-02],
        [1.5407e-09, 1.0000e+00],
        [9.9998e-01, 1.8673e-05],
        [4.9220e-06, 1.0000e+00],
        [2.2619e-07, 1.0000e+00],
        [3.4352e-07, 1.0000e+00],
        [1.1199e-07, 1.0000e+00],
        [7.2097e-07, 1.0000e+00],
        [3.1282e-05, 9.9997e-01],
        [1.7920e-04, 9.9982e-01],
        [9.9991e-01, 9.4855e-05],
        [9.9579e-01, 4.2079e-03],
        [4.5596e-05, 9.9995e-01],
        [2.5263e-08, 1.0000e+00],
        [5.0524e-07, 1.0000e+00],
        [5.7409e-08, 1.0000e+00],
        [6.3576e-08, 1.0000e+00],
        [9.9035e-01, 9.6478e-03],
        [9.9977e-01, 2.3372e-04],
        [3.7392e-04, 9.9963e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0877e-08, 1.0000e+00],
        [6.9624e-05, 9.9993e-01],
        [2.4379e-06, 1.0000e+00],
        [2.8446e-08, 1.0000e+00],
        [2.4200e-05, 9.9998e-01],
        [7.0771e-06, 9.9999e-01],
        [2.0045e-05, 9.9998e-01],
        [5.7112e-02, 9.4289e-01],
        [5.9773e-08, 1.0000e+00],
        [6.3576e-07, 1.0000e+00],
        [2.4758e-03, 9.9752e-01],
        [7.2823e-01, 2.7177e-01],
        [1.4547e-04, 9.9985e-01],
        [5.6136e-06, 9.9999e-01],
        [1.5332e-07, 1.0000e+00],
        [2.9298e-06, 1.0000e+00],
        [9.3197e-05, 9.9991e-01],
        [2.2411e-11, 1.0000e+00],
        [9.9984e-01, 1.6496e-04],
        [1.3477e-07, 1.0000e+00],
        [1.0817e-07, 1.0000e+00],
        [3.2377e-07, 1.0000e+00],
        [1.8351e-08, 1.0000e+00],
        [4.2781e-08, 1.0000e+00],
        [6.4192e-07, 1.0000e+00],
        [2.1727e-08, 1.0000e+00],
        [1.9961e-09, 1.0000e+00],
        [9.9941e-01, 5.9433e-04],
        [1.4870e-07, 1.0000e+00],
        [1.2190e-05, 9.9999e-01],
        [3.9799e-07, 1.0000e+00],
        [9.9987e-01, 1.3352e-04],
        [9.9820e-01, 1.7991e-03],
        [7.5535e-01, 2.4465e-01],
        [9.9978e-01, 2.1663e-04],
        [1.1329e-02, 9.8867e-01],
        [5.0566e-07, 1.0000e+00],
        [2.8228e-07, 1.0000e+00],
        [1.3379e-04, 9.9987e-01],
        [9.3975e-04, 9.9906e-01],
        [9.9863e-01, 1.3677e-03],
        [1.0191e-06, 1.0000e+00],
        [2.1628e-09, 1.0000e+00],
        [2.8439e-03, 9.9716e-01],
        [3.6924e-07, 1.0000e+00],
        [1.3809e-04, 9.9986e-01],
        [1.7108e-10, 1.0000e+00],
        [9.9949e-01, 5.0973e-04],
        [2.5479e-09, 1.0000e+00],
        [2.4766e-07, 1.0000e+00],
        [2.1052e-08, 1.0000e+00],
        [9.9980e-01, 2.0253e-04],
        [9.7413e-01, 2.5874e-02],
        [9.9750e-01, 2.5000e-03],
        [9.9993e-08, 1.0000e+00],
        [9.9987e-01, 1.2675e-04],
        [9.9967e-01, 3.2942e-04],
        [2.1134e-06, 1.0000e+00],
        [1.7311e-06, 1.0000e+00],
        [1.5462e-09, 1.0000e+00],
        [2.5479e-09, 1.0000e+00],
        [3.2128e-07, 1.0000e+00],
        [9.9989e-01, 1.0521e-04],
        [1.5000e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.7198e-07, 1.0000e+00],
        [9.9781e-01, 2.1947e-03],
        [5.1063e-09, 1.0000e+00],
        [4.2591e-09, 1.0000e+00],
        [1.7478e-06, 1.0000e+00],
        [3.3238e-04, 9.9967e-01],
        [1.9341e-07, 1.0000e+00],
        [9.4474e-05, 9.9991e-01],
        [5.4241e-07, 1.0000e+00],
        [1.8442e-09, 1.0000e+00],
        [9.9981e-01, 1.8692e-04],
        [1.2340e-07, 1.0000e+00],
        [9.9987e-01, 1.2886e-04],
        [4.2473e-04, 9.9958e-01],
        [1.0579e-03, 9.9894e-01],
        [1.1507e-01, 8.8493e-01],
        [5.7940e-06, 9.9999e-01],
        [1.8563e-03, 9.9814e-01],
        [1.5427e-03, 9.9846e-01],
        [1.2977e-09, 1.0000e+00],
        [1.3236e-03, 9.9868e-01],
        [1.1923e-05, 9.9999e-01],
        [1.3255e-07, 1.0000e+00],
        [1.2762e-07, 1.0000e+00],
        [9.9974e-01, 2.5786e-04],
        [1.4717e-04, 9.9985e-01],
        [4.3011e-09, 1.0000e+00],
        [9.9993e-01, 7.0271e-05],
        [8.7042e-05, 9.9991e-01],
        [2.0265e-06, 1.0000e+00],
        [1.2268e-03, 9.9877e-01],
        [6.8287e-08, 1.0000e+00],
        [1.4812e-04, 9.9985e-01],
        [9.9984e-01, 1.5986e-04],
        [6.5597e-08, 1.0000e+00],
        [2.1073e-04, 9.9979e-01],
        [3.0142e-05, 9.9997e-01],
        [1.8888e-06, 1.0000e+00],
        [1.6373e-04, 9.9984e-01],
        [9.9944e-01, 5.5813e-04],
        [1.7570e-03, 9.9824e-01],
        [5.0348e-07, 1.0000e+00],
        [1.1683e-06, 1.0000e+00],
        [2.5228e-01, 7.4772e-01],
        [9.3072e-01, 6.9281e-02],
        [8.4314e-07, 1.0000e+00],
        [1.1583e-09, 1.0000e+00],
        [2.6719e-07, 1.0000e+00],
        [9.9988e-01, 1.1868e-04],
        [8.1217e-05, 9.9992e-01],
        [4.2854e-08, 1.0000e+00],
        [3.5055e-07, 1.0000e+00],
        [7.3731e-07, 1.0000e+00],
        [1.3134e-03, 9.9869e-01],
        [4.5431e-06, 1.0000e+00],
        [4.6906e-06, 1.0000e+00],
        [5.1102e-07, 1.0000e+00],
        [9.8509e-01, 1.4910e-02],
        [5.1797e-06, 9.9999e-01],
        [4.3442e-10, 1.0000e+00],
        [1.1770e-07, 1.0000e+00],
        [9.9970e-01, 3.0339e-04],
        [2.1496e-04, 9.9979e-01],
        [2.0358e-04, 9.9980e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9983e-01, 1.6789e-04],
        [4.8424e-06, 1.0000e+00],
        [1.1550e-04, 9.9988e-01],
        [9.9980e-01, 2.0188e-04],
        [1.0882e-07, 1.0000e+00],
        [2.5717e-01, 7.4283e-01],
        [1.3159e-04, 9.9987e-01],
        [9.9968e-01, 3.2465e-04],
        [9.9053e-01, 9.4712e-03],
        [2.3671e-05, 9.9998e-01],
        [2.7794e-11, 1.0000e+00],
        [2.6462e-04, 9.9974e-01],
        [2.8811e-04, 9.9971e-01],
        [9.0788e-08, 1.0000e+00],
        [2.8544e-06, 1.0000e+00],
        [3.6661e-09, 1.0000e+00],
        [1.7718e-05, 9.9998e-01],
        [3.8463e-07, 1.0000e+00],
        [9.9995e-01, 4.8278e-05],
        [5.2612e-06, 9.9999e-01],
        [7.3412e-11, 1.0000e+00],
        [9.9468e-01, 5.3195e-03],
        [3.0823e-02, 9.6918e-01],
        [8.3658e-06, 9.9999e-01],
        [9.9994e-01, 6.0534e-05],
        [1.2353e-05, 9.9999e-01],
        [1.0754e-05, 9.9999e-01],
        [9.9970e-01, 3.0441e-04],
        [6.9777e-06, 9.9999e-01],
        [4.0449e-07, 1.0000e+00],
        [9.4617e-10, 1.0000e+00],
        [2.4410e-02, 9.7559e-01],
        [2.8607e-07, 1.0000e+00],
        [6.1305e-07, 1.0000e+00],
        [8.2197e-06, 9.9999e-01],
        [2.0356e-04, 9.9980e-01],
        [2.8299e-05, 9.9997e-01],
        [1.6172e-05, 9.9998e-01],
        [9.9990e-01, 9.6067e-05],
        [2.0662e-07, 1.0000e+00],
        [7.9120e-08, 1.0000e+00],
        [2.4763e-05, 9.9998e-01],
        [9.9886e-01, 1.1359e-03],
        [5.5398e-07, 1.0000e+00],
        [9.9341e-03, 9.9007e-01],
        [8.4884e-08, 1.0000e+00],
        [7.3481e-04, 9.9927e-01],
        [1.3653e-04, 9.9986e-01],
        [9.9975e-01, 2.5005e-04],
        [4.5041e-06, 1.0000e+00],
        [4.6670e-07, 1.0000e+00],
        [3.4179e-05, 9.9997e-01],
        [1.1221e-08, 1.0000e+00],
        [3.1290e-06, 1.0000e+00],
        [2.6574e-07, 1.0000e+00],
        [9.5664e-11, 1.0000e+00],
        [9.9967e-01, 3.3029e-04],
        [9.9992e-01, 8.4392e-05],
        [8.2706e-09, 1.0000e+00],
        [1.4864e-07, 1.0000e+00],
        [7.2132e-05, 9.9993e-01],
        [5.6614e-03, 9.9434e-01],
        [9.9095e-01, 9.0480e-03],
        [9.9997e-01, 3.0106e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[3.4069e-07, 1.0000e+00],
        [9.9991e-01, 9.3676e-05],
        [9.9943e-01, 5.7420e-04],
        [9.9970e-01, 3.0441e-04],
        [9.9776e-07, 1.0000e+00],
        [3.1817e-08, 1.0000e+00],
        [9.9752e-01, 2.4847e-03],
        [3.4206e-09, 1.0000e+00],
        [3.4195e-04, 9.9966e-01],
        [3.9696e-07, 1.0000e+00],
        [9.9995e-01, 4.8932e-05],
        [1.3841e-02, 9.8616e-01],
        [3.7916e-06, 1.0000e+00],
        [1.6917e-04, 9.9983e-01],
        [5.6254e-08, 1.0000e+00],
        [4.1434e-05, 9.9996e-01],
        [2.3531e-07, 1.0000e+00],
        [9.9992e-01, 7.9394e-05],
        [4.5479e-03, 9.9545e-01],
        [9.9531e-01, 4.6947e-03],
        [3.1698e-10, 1.0000e+00],
        [1.4560e-07, 1.0000e+00],
        [9.9736e-01, 2.6368e-03],
        [8.8738e-10, 1.0000e+00],
        [1.0401e-07, 1.0000e+00],
        [2.4556e-08, 1.0000e+00],
        [4.0447e-07, 1.0000e+00],
        [6.3239e-05, 9.9994e-01],
        [9.9981e-01, 1.9088e-04],
        [5.6138e-06, 9.9999e-01],
        [1.6020e-03, 9.9840e-01],
        [9.8483e-01, 1.5170e-02],
        [7.1433e-10, 1.0000e+00],
        [4.5482e-08, 1.0000e+00],
        [2.2183e-03, 9.9778e-01],
        [9.9991e-01, 8.8066e-05],
        [2.8944e-05, 9.9997e-01],
        [6.4498e-01, 3.5502e-01],
        [2.1948e-09, 1.0000e+00],
        [4.7170e-04, 9.9953e-01],
        [1.1514e-07, 1.0000e+00],
        [2.1213e-07, 1.0000e+00],
        [2.4656e-02, 9.7534e-01],
        [6.6800e-06, 9.9999e-01],
        [9.9968e-01, 3.2460e-04],
        [9.9988e-01, 1.2264e-04],
        [7.8900e-01, 2.1100e-01],
        [9.2259e-06, 9.9999e-01],
        [1.8721e-08, 1.0000e+00],
        [9.9984e-01, 1.6293e-04],
        [2.6569e-06, 1.0000e+00],
        [3.0283e-09, 1.0000e+00],
        [9.6865e-01, 3.1353e-02],
        [3.4260e-04, 9.9966e-01],
        [2.9175e-06, 1.0000e+00],
        [2.4368e-08, 1.0000e+00],
        [4.4485e-05, 9.9996e-01],
        [4.5806e-07, 1.0000e+00],
        [1.1426e-03, 9.9886e-01],
        [8.1513e-09, 1.0000e+00],
        [9.9994e-01, 5.5830e-05],
        [1.6981e-03, 9.9830e-01],
        [4.3498e-05, 9.9996e-01],
        [1.1622e-03, 9.9884e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.7397e-01, 2.6026e-02],
        [3.1746e-05, 9.9997e-01],
        [9.9955e-01, 4.5014e-04],
        [1.4208e-07, 1.0000e+00],
        [9.9990e-01, 1.0389e-04],
        [7.3408e-01, 2.6592e-01],
        [7.6372e-03, 9.9236e-01],
        [9.9739e-01, 2.6059e-03],
        [1.3161e-07, 1.0000e+00],
        [9.9985e-01, 1.4941e-04],
        [3.2617e-06, 1.0000e+00],
        [1.1579e-05, 9.9999e-01],
        [2.3041e-06, 1.0000e+00],
        [3.0752e-10, 1.0000e+00],
        [6.1938e-04, 9.9938e-01],
        [3.1067e-03, 9.9689e-01],
        [5.3906e-05, 9.9995e-01],
        [7.4043e-06, 9.9999e-01],
        [4.7372e-06, 1.0000e+00],
        [4.7411e-07, 1.0000e+00],
        [5.4514e-07, 1.0000e+00],
        [2.4879e-08, 1.0000e+00],
        [2.5773e-04, 9.9974e-01],
        [6.8162e-06, 9.9999e-01],
        [2.4492e-04, 9.9976e-01],
        [2.1111e-08, 1.0000e+00],
        [9.4079e-07, 1.0000e+00],
        [3.2072e-05, 9.9997e-01],
        [4.6375e-05, 9.9995e-01],
        [3.9572e-07, 1.0000e+00],
        [6.2425e-05, 9.9994e-01],
        [9.9918e-01, 8.2315e-04],
        [4.8562e-07, 1.0000e+00],
        [4.6322e-07, 1.0000e+00],
        [5.3476e-08, 1.0000e+00],
        [7.3671e-07, 1.0000e+00],
        [2.0460e-10, 1.0000e+00],
        [2.4091e-03, 9.9759e-01],
        [1.1786e-02, 9.8821e-01],
        [9.9984e-01, 1.6089e-04],
        [2.1190e-04, 9.9979e-01],
        [9.9819e-01, 1.8137e-03],
        [3.5508e-04, 9.9964e-01],
        [8.9934e-01, 1.0066e-01],
        [8.1053e-09, 1.0000e+00],
        [1.0000e+00, 2.0719e-07],
        [9.9982e-01, 1.8111e-04],
        [9.9998e-01, 1.6591e-05],
        [1.2967e-06, 1.0000e+00],
        [8.2028e-09, 1.0000e+00],
        [1.6198e-09, 1.0000e+00],
        [1.1611e-08, 1.0000e+00],
        [5.8294e-06, 9.9999e-01],
        [3.1647e-04, 9.9968e-01],
        [2.4918e-06, 1.0000e+00],
        [1.1300e-02, 9.8870e-01],
        [3.6848e-04, 9.9963e-01],
        [2.2073e-05, 9.9998e-01],
        [9.9987e-01, 1.3177e-04],
        [5.4816e-04, 9.9945e-01],
        [9.9956e-01, 4.3758e-04],
        [5.2397e-05, 9.9995e-01],
        [1.1046e-05, 9.9999e-01],
        [1.0653e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.4992e-01, 5.0077e-02],
        [9.9993e-01, 6.8649e-05],
        [3.1345e-05, 9.9997e-01],
        [9.9980e-01, 1.9708e-04],
        [1.5803e-03, 9.9842e-01],
        [9.4070e-07, 1.0000e+00],
        [9.9850e-01, 1.4956e-03],
        [1.1631e-05, 9.9999e-01],
        [9.9997e-01, 3.3033e-05],
        [2.5667e-05, 9.9997e-01],
        [1.4081e-08, 1.0000e+00],
        [1.7316e-06, 1.0000e+00],
        [1.3166e-04, 9.9987e-01],
        [9.9961e-01, 3.8537e-04],
        [1.2099e-04, 9.9988e-01],
        [1.9550e-01, 8.0450e-01],
        [9.9972e-01, 2.8138e-04],
        [5.0912e-07, 1.0000e+00],
        [1.5004e-06, 1.0000e+00],
        [1.4717e-05, 9.9999e-01],
        [1.0448e-04, 9.9990e-01],
        [9.1130e-05, 9.9991e-01],
        [6.9094e-06, 9.9999e-01],
        [9.9933e-01, 6.6698e-04],
        [1.7638e-07, 1.0000e+00],
        [9.9996e-01, 4.0592e-05],
        [3.0371e-05, 9.9997e-01],
        [9.8443e-01, 1.5573e-02],
        [9.5023e-05, 9.9990e-01],
        [9.9968e-01, 3.1840e-04],
        [5.9726e-05, 9.9994e-01],
        [3.8452e-02, 9.6155e-01],
        [3.0559e-09, 1.0000e+00],
        [4.3134e-05, 9.9996e-01],
        [8.1084e-06, 9.9999e-01],
        [1.5182e-05, 9.9998e-01],
        [2.5672e-04, 9.9974e-01],
        [2.7618e-03, 9.9724e-01],
        [4.4123e-01, 5.5877e-01],
        [6.3604e-08, 1.0000e+00],
        [2.5304e-03, 9.9747e-01],
        [1.0384e-10, 1.0000e+00],
        [2.0262e-04, 9.9980e-01],
        [4.2341e-08, 1.0000e+00],
        [1.0920e-05, 9.9999e-01],
        [9.9983e-01, 1.6955e-04],
        [4.8804e-05, 9.9995e-01],
        [4.7326e-06, 1.0000e+00],
        [9.9952e-01, 4.8471e-04],
        [9.9990e-01, 9.9604e-05],
        [9.9976e-01, 2.3824e-04],
        [4.7207e-02, 9.5279e-01],
        [5.0023e-03, 9.9500e-01],
        [9.9996e-01, 4.3286e-05],
        [1.0153e-07, 1.0000e+00],
        [4.9834e-05, 9.9995e-01],
        [9.1639e-03, 9.9084e-01],
        [2.7995e-04, 9.9972e-01],
        [8.1667e-06, 9.9999e-01],
        [5.4707e-03, 9.9453e-01],
        [3.3093e-02, 9.6691e-01],
        [2.8321e-01, 7.1679e-01],
        [9.9828e-01, 1.7237e-03],
        [9.9972e-01, 2.7867e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8802e-01, 1.1984e-02],
        [1.3244e-07, 1.0000e+00],
        [6.7683e-03, 9.9323e-01],
        [8.3294e-07, 1.0000e+00],
        [6.5111e-07, 1.0000e+00],
        [2.2933e-04, 9.9977e-01],
        [3.0625e-05, 9.9997e-01],
        [8.7828e-08, 1.0000e+00],
        [7.0991e-06, 9.9999e-01],
        [6.9924e-03, 9.9301e-01],
        [8.7651e-06, 9.9999e-01],
        [6.0522e-08, 1.0000e+00],
        [8.4632e-05, 9.9992e-01],
        [9.9967e-01, 3.3303e-04],
        [1.0422e-06, 1.0000e+00],
        [1.6045e-07, 1.0000e+00],
        [8.5143e-07, 1.0000e+00],
        [2.3379e-05, 9.9998e-01],
        [9.9990e-01, 9.7051e-05],
        [6.7463e-05, 9.9993e-01],
        [4.9423e-02, 9.5058e-01],
        [4.0555e-05, 9.9996e-01],
        [1.2799e-03, 9.9872e-01],
        [9.9792e-01, 2.0768e-03],
        [6.1325e-06, 9.9999e-01],
        [2.0026e-06, 1.0000e+00],
        [8.1600e-03, 9.9184e-01],
        [1.2350e-04, 9.9988e-01],
        [9.9960e-01, 4.0017e-04],
        [5.6239e-05, 9.9994e-01],
        [2.1719e-05, 9.9998e-01],
        [9.9971e-01, 2.9028e-04],
        [9.9988e-01, 1.2484e-04],
        [3.2765e-03, 9.9672e-01],
        [2.9193e-06, 1.0000e+00],
        [9.6673e-01, 3.3273e-02],
        [9.9971e-01, 2.9118e-04],
        [2.1256e-08, 1.0000e+00],
        [9.0239e-09, 1.0000e+00],
        [9.8878e-01, 1.1222e-02],
        [1.3553e-06, 1.0000e+00],
        [5.3315e-03, 9.9467e-01],
        [1.2584e-03, 9.9874e-01],
        [7.9402e-07, 1.0000e+00],
        [2.3566e-07, 1.0000e+00],
        [1.4746e-05, 9.9999e-01],
        [4.7680e-07, 1.0000e+00],
        [1.4731e-06, 1.0000e+00],
        [2.3581e-04, 9.9976e-01],
        [9.8778e-01, 1.2216e-02],
        [1.2134e-07, 1.0000e+00],
        [9.9973e-01, 2.6998e-04],
        [7.9478e-06, 9.9999e-01],
        [9.9983e-01, 1.7053e-04],
        [5.3230e-09, 1.0000e+00],
        [5.0087e-06, 9.9999e-01],
        [1.7017e-01, 8.2983e-01],
        [9.9929e-01, 7.1228e-04],
        [1.0987e-06, 1.0000e+00],
        [9.9988e-01, 1.1920e-04],
        [6.7845e-06, 9.9999e-01],
        [9.4306e-10, 1.0000e+00],
        [9.9994e-01, 6.4986e-05],
        [3.8936e-05, 9.9996e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5490e-04, 9.9955e-01],
        [9.9997e-01, 2.7688e-05],
        [9.9478e-01, 5.2214e-03],
        [9.9985e-01, 1.5335e-04],
        [1.0215e-03, 9.9898e-01],
        [9.9980e-01, 2.0401e-04],
        [4.7697e-05, 9.9995e-01],
        [3.3033e-05, 9.9997e-01],
        [9.9459e-02, 9.0054e-01],
        [9.5552e-05, 9.9990e-01],
        [9.9960e-01, 4.0117e-04],
        [5.0112e-02, 9.4989e-01],
        [4.1164e-04, 9.9959e-01],
        [7.7907e-06, 9.9999e-01],
        [1.3117e-10, 1.0000e+00],
        [9.1874e-08, 1.0000e+00],
        [1.9293e-05, 9.9998e-01],
        [1.1589e-04, 9.9988e-01],
        [2.5788e-03, 9.9742e-01],
        [2.2955e-10, 1.0000e+00],
        [9.9887e-01, 1.1252e-03],
        [1.6114e-03, 9.9839e-01],
        [1.3607e-07, 1.0000e+00],
        [2.0484e-05, 9.9998e-01],
        [1.3499e-07, 1.0000e+00],
        [9.9985e-01, 1.5310e-04],
        [3.0292e-05, 9.9997e-01],
        [1.1365e-11, 1.0000e+00],
        [6.2990e-09, 1.0000e+00],
        [2.0195e-05, 9.9998e-01],
        [7.6701e-09, 1.0000e+00],
        [2.1322e-05, 9.9998e-01],
        [8.4235e-05, 9.9992e-01],
        [2.3507e-03, 9.9765e-01],
        [1.6252e-05, 9.9998e-01],
        [2.5082e-11, 1.0000e+00],
        [2.3701e-04, 9.9976e-01],
        [3.2376e-10, 1.0000e+00],
        [3.2419e-06, 1.0000e+00],
        [8.6621e-05, 9.9991e-01],
        [9.9992e-01, 7.8151e-05],
        [1.5014e-05, 9.9998e-01],
        [1.9334e-05, 9.9998e-01],
        [6.9167e-04, 9.9931e-01],
        [9.9884e-01, 1.1550e-03],
        [7.7037e-04, 9.9923e-01],
        [7.9003e-05, 9.9992e-01],
        [1.0434e-07, 1.0000e+00],
        [2.9779e-03, 9.9702e-01],
        [2.3614e-05, 9.9998e-01],
        [1.3153e-08, 1.0000e+00],
        [9.9614e-01, 3.8585e-03],
        [2.3015e-03, 9.9770e-01],
        [2.9899e-07, 1.0000e+00],
        [9.9984e-01, 1.5883e-04],
        [6.9600e-06, 9.9999e-01],
        [1.5384e-05, 9.9998e-01],
        [1.3419e-04, 9.9987e-01],
        [9.9967e-01, 3.2796e-04],
        [5.4928e-06, 9.9999e-01],
        [4.0846e-06, 1.0000e+00],
        [9.9987e-01, 1.2756e-04],
        [9.9990e-01, 1.0217e-04],
        [3.1116e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2177e-05, 9.9997e-01],
        [2.6068e-07, 1.0000e+00],
        [1.4049e-07, 1.0000e+00],
        [9.9994e-01, 5.9384e-05],
        [5.5089e-06, 9.9999e-01],
        [1.1533e-08, 1.0000e+00],
        [3.5433e-05, 9.9996e-01],
        [9.9983e-01, 1.7079e-04],
        [8.7200e-06, 9.9999e-01],
        [3.7667e-06, 1.0000e+00],
        [3.1427e-09, 1.0000e+00],
        [8.9125e-09, 1.0000e+00],
        [8.4026e-06, 9.9999e-01],
        [6.5749e-05, 9.9993e-01],
        [4.3583e-05, 9.9996e-01],
        [1.0999e-03, 9.9890e-01],
        [6.5154e-07, 1.0000e+00],
        [5.7007e-05, 9.9994e-01],
        [4.7097e-07, 1.0000e+00],
        [2.4759e-05, 9.9998e-01],
        [8.5289e-07, 1.0000e+00],
        [9.0739e-02, 9.0926e-01],
        [1.3081e-09, 1.0000e+00],
        [5.3545e-07, 1.0000e+00],
        [2.6520e-06, 1.0000e+00],
        [9.9989e-01, 1.0619e-04],
        [1.1693e-05, 9.9999e-01],
        [8.2213e-06, 9.9999e-01],
        [6.3566e-05, 9.9994e-01],
        [6.4253e-06, 9.9999e-01],
        [2.5972e-09, 1.0000e+00],
        [2.6958e-07, 1.0000e+00],
        [9.9975e-01, 2.4877e-04],
        [9.9808e-01, 1.9221e-03],
        [9.9950e-01, 4.9517e-04],
        [3.7268e-06, 1.0000e+00],
        [5.2146e-06, 9.9999e-01],
        [4.3752e-06, 1.0000e+00],
        [1.9179e-06, 1.0000e+00],
        [1.5171e-02, 9.8483e-01],
        [3.1147e-02, 9.6885e-01],
        [6.9120e-07, 1.0000e+00],
        [2.8501e-05, 9.9997e-01],
        [1.0687e-06, 1.0000e+00],
        [6.3761e-05, 9.9994e-01],
        [4.1182e-09, 1.0000e+00],
        [1.0573e-05, 9.9999e-01],
        [9.9873e-01, 1.2726e-03],
        [1.9699e-02, 9.8030e-01],
        [9.9985e-01, 1.4826e-04],
        [1.7077e-03, 9.9829e-01],
        [3.0490e-07, 1.0000e+00],
        [9.1054e-02, 9.0895e-01],
        [5.2860e-09, 1.0000e+00],
        [9.9973e-01, 2.6951e-04],
        [8.5045e-09, 1.0000e+00],
        [1.7043e-02, 9.8296e-01],
        [1.8245e-04, 9.9982e-01],
        [1.2008e-06, 1.0000e+00],
        [1.3179e-07, 1.0000e+00],
        [2.4032e-08, 1.0000e+00],
        [9.1350e-05, 9.9991e-01],
        [2.9886e-04, 9.9970e-01],
        [1.3570e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0716e-07, 1.0000e+00],
        [1.3088e-04, 9.9987e-01],
        [7.5148e-07, 1.0000e+00],
        [1.3272e-08, 1.0000e+00],
        [7.7975e-04, 9.9922e-01],
        [2.5378e-05, 9.9997e-01],
        [9.9987e-01, 1.2674e-04],
        [9.9990e-01, 1.0417e-04],
        [5.4277e-04, 9.9946e-01],
        [9.9990e-01, 9.6737e-05],
        [5.3199e-05, 9.9995e-01],
        [9.1963e-10, 1.0000e+00],
        [9.9998e-01, 1.7678e-05],
        [2.0332e-02, 9.7967e-01],
        [1.3509e-08, 1.0000e+00],
        [9.9984e-01, 1.6051e-04],
        [9.9959e-01, 4.1052e-04],
        [1.3658e-03, 9.9863e-01],
        [8.2719e-10, 1.0000e+00],
        [1.8205e-07, 1.0000e+00],
        [9.9897e-01, 1.0322e-03],
        [9.8673e-01, 1.3273e-02],
        [4.1029e-06, 1.0000e+00],
        [9.6691e-11, 1.0000e+00],
        [3.7476e-07, 1.0000e+00],
        [3.5288e-03, 9.9647e-01],
        [4.2452e-04, 9.9958e-01],
        [9.4348e-05, 9.9991e-01],
        [9.9975e-01, 2.5053e-04],
        [9.9956e-01, 4.3539e-04],
        [2.5297e-04, 9.9975e-01],
        [9.9992e-01, 8.2327e-05],
        [1.3189e-04, 9.9987e-01],
        [9.9972e-01, 2.7913e-04],
        [9.9971e-01, 2.9069e-04],
        [2.9328e-07, 1.0000e+00],
        [1.2090e-05, 9.9999e-01],
        [1.1603e-05, 9.9999e-01],
        [2.4119e-05, 9.9998e-01],
        [4.5110e-07, 1.0000e+00],
        [1.0676e-05, 9.9999e-01],
        [2.1813e-06, 1.0000e+00],
        [3.6765e-08, 1.0000e+00],
        [1.0344e-08, 1.0000e+00],
        [1.0035e-03, 9.9900e-01],
        [6.5432e-06, 9.9999e-01],
        [1.2314e-07, 1.0000e+00],
        [3.0044e-06, 1.0000e+00],
        [1.3455e-06, 1.0000e+00],
        [7.5701e-07, 1.0000e+00],
        [2.5294e-03, 9.9747e-01],
        [8.7384e-06, 9.9999e-01],
        [9.9576e-01, 4.2387e-03],
        [9.9972e-01, 2.8220e-04],
        [9.9972e-01, 2.8141e-04],
        [8.9288e-05, 9.9991e-01],
        [1.3395e-05, 9.9999e-01],
        [3.6219e-09, 1.0000e+00],
        [9.9971e-01, 2.8836e-04],
        [3.0691e-06, 1.0000e+00],
        [9.9990e-01, 9.6143e-05],
        [7.8267e-09, 1.0000e+00],
        [8.5547e-01, 1.4453e-01],
        [2.1874e-04, 9.9978e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5872e-05, 9.9995e-01],
        [1.2739e-04, 9.9987e-01],
        [1.0553e-05, 9.9999e-01],
        [2.3610e-06, 1.0000e+00],
        [1.2872e-03, 9.9871e-01],
        [9.9892e-01, 1.0773e-03],
        [1.7346e-06, 1.0000e+00],
        [4.5503e-06, 1.0000e+00],
        [8.6934e-06, 9.9999e-01],
        [4.8596e-06, 1.0000e+00],
        [9.9977e-01, 2.3425e-04],
        [5.9442e-07, 1.0000e+00],
        [2.4130e-07, 1.0000e+00],
        [7.2171e-09, 1.0000e+00],
        [9.9856e-01, 1.4367e-03],
        [3.3196e-07, 1.0000e+00],
        [6.2828e-10, 1.0000e+00],
        [9.9982e-01, 1.8327e-04],
        [3.7844e-06, 1.0000e+00],
        [9.9936e-01, 6.3626e-04],
        [2.8884e-03, 9.9711e-01],
        [9.0340e-06, 9.9999e-01],
        [2.5328e-03, 9.9747e-01],
        [9.9479e-01, 5.2146e-03],
        [9.9948e-01, 5.1524e-04],
        [7.2407e-06, 9.9999e-01],
        [1.7319e-04, 9.9983e-01],
        [1.3217e-08, 1.0000e+00],
        [1.0343e-04, 9.9990e-01],
        [1.0071e-03, 9.9899e-01],
        [9.9999e-01, 4.9687e-06],
        [1.1718e-05, 9.9999e-01],
        [2.4635e-08, 1.0000e+00],
        [3.9212e-04, 9.9961e-01],
        [1.8121e-05, 9.9998e-01],
        [3.9331e-07, 1.0000e+00],
        [3.3486e-08, 1.0000e+00],
        [6.6083e-01, 3.3917e-01],
        [2.5498e-05, 9.9997e-01],
        [1.4390e-05, 9.9999e-01],
        [1.8541e-03, 9.9815e-01],
        [9.9984e-01, 1.5606e-04],
        [9.9772e-01, 2.2828e-03],
        [6.6039e-09, 1.0000e+00],
        [3.6738e-06, 1.0000e+00],
        [9.9980e-01, 1.9593e-04],
        [7.7983e-05, 9.9992e-01],
        [2.0497e-07, 1.0000e+00],
        [1.0782e-01, 8.9218e-01],
        [3.0219e-08, 1.0000e+00],
        [5.2663e-01, 4.7337e-01],
        [5.5378e-08, 1.0000e+00],
        [9.9988e-01, 1.2102e-04],
        [3.0060e-06, 1.0000e+00],
        [6.4166e-07, 1.0000e+00],
        [8.3325e-09, 1.0000e+00],
        [1.9803e-07, 1.0000e+00],
        [9.2622e-08, 1.0000e+00],
        [8.5496e-06, 9.9999e-01],
        [2.1492e-06, 1.0000e+00],
        [6.6540e-06, 9.9999e-01],
        [3.1529e-04, 9.9968e-01],
        [3.9071e-04, 9.9961e-01],
        [7.2410e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.5776e-05, 9.9991e-01],
        [1.4873e-09, 1.0000e+00],
        [9.9979e-01, 2.1188e-04],
        [3.0035e-05, 9.9997e-01],
        [5.8514e-06, 9.9999e-01],
        [3.5566e-04, 9.9964e-01],
        [7.1388e-06, 9.9999e-01],
        [1.5357e-07, 1.0000e+00],
        [1.7647e-09, 1.0000e+00],
        [6.7523e-04, 9.9932e-01],
        [2.4953e-07, 1.0000e+00],
        [1.5157e-09, 1.0000e+00],
        [5.8025e-05, 9.9994e-01],
        [1.0636e-04, 9.9989e-01],
        [9.9988e-01, 1.1557e-04],
        [6.5827e-06, 9.9999e-01],
        [1.2464e-07, 1.0000e+00],
        [9.4542e-06, 9.9999e-01],
        [8.4857e-05, 9.9992e-01],
        [7.4687e-06, 9.9999e-01],
        [7.6423e-06, 9.9999e-01],
        [9.9511e-01, 4.8867e-03],
        [1.5249e-05, 9.9998e-01],
        [9.9981e-01, 1.8703e-04],
        [7.9581e-06, 9.9999e-01],
        [2.9571e-05, 9.9997e-01],
        [5.2523e-07, 1.0000e+00],
        [1.7055e-04, 9.9983e-01],
        [1.8035e-05, 9.9998e-01],
        [2.5517e-06, 1.0000e+00],
        [1.7210e-05, 9.9998e-01],
        [2.3422e-07, 1.0000e+00],
        [1.9699e-07, 1.0000e+00],
        [9.9977e-01, 2.2549e-04],
        [1.8020e-06, 1.0000e+00],
        [7.8894e-06, 9.9999e-01],
        [3.1923e-05, 9.9997e-01],
        [3.2436e-04, 9.9968e-01],
        [6.0124e-06, 9.9999e-01],
        [9.9959e-01, 4.0697e-04],
        [2.4841e-03, 9.9752e-01],
        [7.9097e-07, 1.0000e+00],
        [5.5856e-08, 1.0000e+00],
        [1.0136e-09, 1.0000e+00],
        [2.4505e-07, 1.0000e+00],
        [2.3897e-03, 9.9761e-01],
        [6.9815e-05, 9.9993e-01],
        [1.3900e-04, 9.9986e-01],
        [2.5244e-05, 9.9997e-01],
        [9.9974e-01, 2.6289e-04],
        [1.1136e-05, 9.9999e-01],
        [5.2764e-05, 9.9995e-01],
        [7.8587e-07, 1.0000e+00],
        [7.9781e-03, 9.9202e-01],
        [6.8188e-03, 9.9318e-01],
        [3.8665e-04, 9.9961e-01],
        [1.0113e-06, 1.0000e+00],
        [8.5230e-03, 9.9148e-01],
        [2.9603e-06, 1.0000e+00],
        [9.9985e-01, 1.4550e-04],
        [9.9985e-01, 1.5215e-04],
        [6.5703e-03, 9.9343e-01],
        [7.1364e-05, 9.9993e-01],
        [1.1609e-03, 9.9884e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0646e-08, 1.0000e+00],
        [2.9172e-04, 9.9971e-01],
        [9.9936e-01, 6.4223e-04],
        [9.9993e-01, 6.7444e-05],
        [9.9891e-01, 1.0909e-03],
        [2.7789e-05, 9.9997e-01],
        [1.4610e-05, 9.9999e-01],
        [2.5489e-08, 1.0000e+00],
        [1.7568e-06, 1.0000e+00],
        [1.7568e-06, 1.0000e+00],
        [2.8576e-09, 1.0000e+00],
        [4.1992e-04, 9.9958e-01],
        [1.4997e-03, 9.9850e-01],
        [1.8491e-02, 9.8151e-01],
        [6.5846e-05, 9.9993e-01],
        [9.9987e-01, 1.2709e-04],
        [5.2767e-06, 9.9999e-01],
        [1.7589e-04, 9.9982e-01],
        [9.9986e-01, 1.4446e-04],
        [8.8670e-08, 1.0000e+00],
        [2.4496e-06, 1.0000e+00],
        [8.0170e-06, 9.9999e-01],
        [9.9990e-01, 1.0426e-04],
        [3.8187e-05, 9.9996e-01],
        [3.4132e-03, 9.9659e-01],
        [9.9994e-01, 6.3474e-05],
        [9.5033e-06, 9.9999e-01],
        [1.9921e-04, 9.9980e-01],
        [2.1465e-09, 1.0000e+00],
        [1.1202e-03, 9.9888e-01],
        [9.0311e-08, 1.0000e+00],
        [9.2644e-08, 1.0000e+00],
        [7.8898e-01, 2.1102e-01],
        [8.8808e-04, 9.9911e-01],
        [3.3909e-03, 9.9661e-01],
        [9.9983e-01, 1.7173e-04],
        [1.2529e-07, 1.0000e+00],
        [3.6364e-06, 1.0000e+00],
        [4.6135e-01, 5.3865e-01],
        [1.7861e-04, 9.9982e-01],
        [1.7569e-04, 9.9982e-01],
        [1.0000e+00, 6.1035e-07],
        [2.9994e-01, 7.0006e-01],
        [1.0032e-08, 1.0000e+00],
        [9.9994e-01, 6.3894e-05],
        [9.9915e-01, 8.4556e-04],
        [2.7470e-03, 9.9725e-01],
        [9.9991e-01, 8.9974e-05],
        [9.9995e-01, 5.4624e-05],
        [7.6902e-07, 1.0000e+00],
        [1.1014e-01, 8.8986e-01],
        [5.4251e-10, 1.0000e+00],
        [9.9977e-01, 2.2585e-04],
        [9.9977e-01, 2.3425e-04],
        [2.1528e-06, 1.0000e+00],
        [2.9410e-08, 1.0000e+00],
        [9.9976e-01, 2.4439e-04],
        [4.5795e-10, 1.0000e+00],
        [1.1131e-05, 9.9999e-01],
        [9.9986e-01, 1.4226e-04],
        [1.6283e-04, 9.9984e-01],
        [4.6352e-06, 1.0000e+00],
        [2.2081e-09, 1.0000e+00],
        [1.0508e-02, 9.8949e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.4723e-01, 5.5277e-01],
        [8.3661e-07, 1.0000e+00],
        [3.1637e-03, 9.9684e-01],
        [1.7907e-04, 9.9982e-01],
        [2.6091e-08, 1.0000e+00],
        [8.4774e-07, 1.0000e+00],
        [4.3211e-04, 9.9957e-01],
        [2.0678e-08, 1.0000e+00],
        [7.0933e-05, 9.9993e-01],
        [9.9968e-01, 3.1979e-04],
        [9.9990e-01, 1.0075e-04],
        [9.9564e-01, 4.3595e-03],
        [4.9307e-07, 1.0000e+00],
        [9.9983e-01, 1.7345e-04],
        [1.5559e-03, 9.9844e-01],
        [4.0000e-05, 9.9996e-01],
        [5.1056e-06, 9.9999e-01],
        [9.9994e-01, 6.0106e-05],
        [1.0008e-05, 9.9999e-01],
        [5.0116e-08, 1.0000e+00],
        [9.9969e-01, 3.1204e-04],
        [1.0114e-01, 8.9886e-01],
        [1.3770e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5918e-09, 1.0000e+00],
        [9.9976e-01, 2.4127e-04],
        [7.4094e-08, 1.0000e+00],
        [9.9999e-01, 1.3106e-05],
        [3.2154e-09, 1.0000e+00],
        [4.7841e-05, 9.9995e-01],
        [8.9732e-07, 1.0000e+00],
        [8.4099e-07, 1.0000e+00],
        [1.6135e-06, 1.0000e+00],
        [9.9980e-01, 2.0393e-04],
        [2.1976e-04, 9.9978e-01],
        [1.4667e-07, 1.0000e+00],
        [5.5495e-06, 9.9999e-01],
        [8.3237e-06, 9.9999e-01],
        [7.7736e-04, 9.9922e-01],
        [7.2287e-04, 9.9928e-01],
        [4.7940e-06, 1.0000e+00],
        [2.1229e-08, 1.0000e+00],
        [9.9985e-01, 1.5034e-04],
        [5.9429e-06, 9.9999e-01],
        [2.2860e-03, 9.9771e-01],
        [9.9965e-01, 3.5471e-04],
        [3.0339e-03, 9.9697e-01],
        [2.2683e-08, 1.0000e+00],
        [3.0463e-03, 9.9695e-01],
        [1.0543e-01, 8.9457e-01],
        [9.9949e-01, 5.1140e-04],
        [9.9994e-01, 6.2069e-05],
        [1.4648e-08, 1.0000e+00],
        [4.7518e-06, 1.0000e+00],
        [1.9295e-08, 1.0000e+00],
        [3.2745e-06, 1.0000e+00],
        [9.9995e-01, 4.6089e-05],
        [8.0587e-06, 9.9999e-01],
        [1.2942e-03, 9.9871e-01],
        [5.1822e-05, 9.9995e-01],
        [1.0658e-04, 9.9989e-01],
        [1.5228e-05, 9.9998e-01],
        [5.9241e-05, 9.9994e-01],
        [9.9982e-01, 1.7743e-04],
        [7.3564e-05, 9.9993e-01],
        [4.3587e-05, 9.9996e-01],
        [3.9765e-06, 1.0000e+00],
        [1.1440e-09, 1.0000e+00],
        [5.5520e-04, 9.9944e-01],
        [9.7806e-01, 2.1944e-02],
        [9.9966e-01, 3.4285e-04],
        [3.1139e-04, 9.9969e-01],
        [4.2947e-07, 1.0000e+00],
        [1.7224e-08, 1.0000e+00],
        [9.5882e-01, 4.1176e-02],
        [6.3193e-06, 9.9999e-01],
        [2.1092e-06, 1.0000e+00],
        [9.9974e-01, 2.6034e-04],
        [3.1756e-05, 9.9997e-01],
        [4.6120e-06, 1.0000e+00],
        [1.0778e-03, 9.9892e-01],
        [8.5531e-07, 1.0000e+00],
        [9.9998e-01, 1.8357e-05],
        [6.5902e-04, 9.9934e-01],
        [9.6239e-08, 1.0000e+00],
        [8.8358e-08, 1.0000e+00],
        [9.9997e-01, 2.9817e-05],
        [5.8877e-04, 9.9941e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9926e-01, 7.3703e-04],
        [3.2465e-05, 9.9997e-01],
        [2.1303e-08, 1.0000e+00],
        [1.3423e-12, 1.0000e+00],
        [4.8606e-02, 9.5139e-01],
        [2.2567e-06, 1.0000e+00],
        [9.9730e-10, 1.0000e+00],
        [6.3254e-08, 1.0000e+00],
        [1.7432e-09, 1.0000e+00],
        [6.7848e-08, 1.0000e+00],
        [9.8352e-01, 1.6484e-02],
        [2.1471e-08, 1.0000e+00],
        [9.9984e-01, 1.5848e-04],
        [4.6755e-09, 1.0000e+00],
        [5.6566e-05, 9.9994e-01],
        [1.1953e-06, 1.0000e+00],
        [4.7355e-06, 1.0000e+00],
        [1.8589e-07, 1.0000e+00],
        [1.7314e-08, 1.0000e+00],
        [9.9961e-01, 3.9149e-04],
        [5.5359e-06, 9.9999e-01],
        [5.3584e-04, 9.9946e-01],
        [2.1775e-06, 1.0000e+00],
        [2.2360e-04, 9.9978e-01],
        [4.8702e-06, 1.0000e+00],
        [1.9711e-02, 9.8029e-01],
        [9.9999e-01, 6.7847e-06],
        [4.0871e-06, 1.0000e+00],
        [2.9971e-04, 9.9970e-01],
        [6.2833e-05, 9.9994e-01],
        [6.6321e-10, 1.0000e+00],
        [1.8603e-06, 1.0000e+00],
        [1.5425e-03, 9.9846e-01],
        [9.9954e-01, 4.6046e-04],
        [9.9964e-01, 3.5657e-04],
        [1.0897e-06, 1.0000e+00],
        [4.2235e-07, 1.0000e+00],
        [3.3330e-04, 9.9967e-01],
        [4.4895e-07, 1.0000e+00],
        [5.9797e-07, 1.0000e+00],
        [6.2298e-07, 1.0000e+00],
        [9.9957e-01, 4.3189e-04],
        [2.0886e-06, 1.0000e+00],
        [2.3576e-07, 1.0000e+00],
        [5.6618e-05, 9.9994e-01],
        [1.1933e-06, 1.0000e+00],
        [9.9984e-01, 1.6399e-04],
        [4.3644e-05, 9.9996e-01],
        [7.0414e-01, 2.9586e-01],
        [4.6782e-05, 9.9995e-01],
        [5.5465e-04, 9.9945e-01],
        [6.0335e-07, 1.0000e+00],
        [1.3318e-08, 1.0000e+00],
        [9.9988e-01, 1.2086e-04],
        [9.9845e-01, 1.5511e-03],
        [1.2887e-06, 1.0000e+00],
        [1.0000e+00, 1.2537e-08],
        [1.9723e-08, 1.0000e+00],
        [3.8181e-02, 9.6182e-01],
        [9.8851e-01, 1.1487e-02],
        [9.9990e-01, 1.0373e-04],
        [7.4344e-07, 1.0000e+00],
        [1.5160e-07, 1.0000e+00],
        [2.7637e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.6663e-07, 1.0000e+00],
        [8.0592e-08, 1.0000e+00],
        [2.6644e-07, 1.0000e+00],
        [1.1410e-05, 9.9999e-01],
        [1.0820e-05, 9.9999e-01],
        [1.2896e-03, 9.9871e-01],
        [5.4099e-01, 4.5901e-01],
        [9.9986e-01, 1.3748e-04],
        [9.9987e-01, 1.3099e-04],
        [1.0566e-05, 9.9999e-01],
        [1.5006e-04, 9.9985e-01],
        [5.7399e-05, 9.9994e-01],
        [1.7338e-03, 9.9827e-01],
        [2.3994e-05, 9.9998e-01],
        [7.0653e-01, 2.9347e-01],
        [3.7139e-05, 9.9996e-01],
        [1.7361e-09, 1.0000e+00],
        [8.2804e-07, 1.0000e+00],
        [9.9970e-01, 3.0051e-04],
        [9.9988e-01, 1.2454e-04],
        [9.9983e-01, 1.7456e-04],
        [7.2351e-10, 1.0000e+00],
        [1.3324e-07, 1.0000e+00],
        [1.6904e-08, 1.0000e+00],
        [1.9814e-03, 9.9802e-01],
        [9.9950e-01, 4.9903e-04],
        [9.9997e-01, 2.9003e-05],
        [1.1066e-02, 9.8893e-01],
        [3.3388e-05, 9.9997e-01],
        [9.9969e-01, 3.0954e-04],
        [2.7074e-08, 1.0000e+00],
        [5.5449e-08, 1.0000e+00],
        [5.0474e-09, 1.0000e+00],
        [6.0940e-09, 1.0000e+00],
        [6.9737e-08, 1.0000e+00],
        [1.0933e-11, 1.0000e+00],
        [9.9426e-01, 5.7408e-03],
        [9.8501e-01, 1.4991e-02],
        [4.3975e-05, 9.9996e-01],
        [1.1522e-03, 9.9885e-01],
        [1.0878e-06, 1.0000e+00],
        [5.5846e-10, 1.0000e+00],
        [5.0528e-05, 9.9995e-01],
        [9.9984e-01, 1.6295e-04],
        [4.7610e-06, 1.0000e+00],
        [3.1982e-05, 9.9997e-01],
        [3.1706e-07, 1.0000e+00],
        [1.1210e-07, 1.0000e+00],
        [2.4205e-05, 9.9998e-01],
        [5.0575e-06, 9.9999e-01],
        [5.8015e-08, 1.0000e+00],
        [5.1489e-05, 9.9995e-01],
        [9.9914e-01, 8.5787e-04],
        [1.8486e-02, 9.8151e-01],
        [9.9985e-01, 1.5434e-04],
        [9.7424e-08, 1.0000e+00],
        [6.0788e-07, 1.0000e+00],
        [9.9958e-01, 4.1841e-04],
        [5.6164e-07, 1.0000e+00],
        [3.7169e-09, 1.0000e+00],
        [6.5144e-06, 9.9999e-01],
        [9.9988e-01, 1.1943e-04],
        [8.5867e-01, 1.4133e-01],
        [7.1705e-04, 9.9928e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5196e-06, 1.0000e+00],
        [1.1815e-07, 1.0000e+00],
        [2.1053e-05, 9.9998e-01],
        [2.8275e-05, 9.9997e-01],
        [9.9946e-01, 5.4486e-04],
        [1.5151e-05, 9.9998e-01],
        [1.1622e-09, 1.0000e+00],
        [3.5743e-07, 1.0000e+00],
        [1.6402e-08, 1.0000e+00],
        [9.9723e-01, 2.7698e-03],
        [9.5061e-01, 4.9390e-02],
        [8.0621e-05, 9.9992e-01],
        [8.5530e-06, 9.9999e-01],
        [1.5153e-06, 1.0000e+00],
        [2.3835e-05, 9.9998e-01],
        [5.1232e-07, 1.0000e+00],
        [1.1826e-04, 9.9988e-01],
        [2.5347e-07, 1.0000e+00],
        [2.6716e-08, 1.0000e+00],
        [1.9014e-07, 1.0000e+00],
        [9.9989e-01, 1.0912e-04],
        [9.9993e-01, 6.8134e-05],
        [1.0782e-08, 1.0000e+00],
        [1.2305e-08, 1.0000e+00],
        [9.9988e-01, 1.2302e-04],
        [9.9989e-01, 1.0896e-04],
        [2.4081e-05, 9.9998e-01],
        [5.2629e-01, 4.7371e-01],
        [9.9978e-01, 2.2292e-04],
        [2.0015e-06, 1.0000e+00],
        [5.7240e-08, 1.0000e+00],
        [9.9930e-01, 6.9585e-04],
        [9.9998e-01, 1.5037e-05],
        [2.4178e-05, 9.9998e-01],
        [5.8652e-02, 9.4135e-01],
        [9.9971e-01, 2.8692e-04],
        [3.1312e-06, 1.0000e+00],
        [4.2594e-04, 9.9957e-01],
        [2.6961e-04, 9.9973e-01],
        [2.6457e-06, 1.0000e+00],
        [2.5148e-03, 9.9749e-01],
        [2.2162e-09, 1.0000e+00],
        [5.4891e-05, 9.9995e-01],
        [5.7177e-03, 9.9428e-01],
        [9.9967e-01, 3.3338e-04],
        [9.9744e-01, 2.5551e-03],
        [4.0733e-05, 9.9996e-01],
        [2.4817e-06, 1.0000e+00],
        [4.7186e-06, 1.0000e+00],
        [2.9420e-08, 1.0000e+00],
        [9.9975e-01, 2.4536e-04],
        [9.4493e-11, 1.0000e+00],
        [1.5798e-09, 1.0000e+00],
        [9.9784e-01, 2.1638e-03],
        [1.9067e-08, 1.0000e+00],
        [5.9469e-06, 9.9999e-01],
        [6.5525e-01, 3.4475e-01],
        [4.6697e-07, 1.0000e+00],
        [1.2820e-07, 1.0000e+00],
        [7.1466e-03, 9.9285e-01],
        [8.7849e-01, 1.2151e-01],
        [2.4820e-05, 9.9998e-01],
        [1.1234e-09, 1.0000e+00],
        [6.1617e-03, 9.9384e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9947e-01, 5.3322e-04],
        [8.8845e-07, 1.0000e+00],
        [9.8374e-01, 1.6263e-02],
        [4.4012e-05, 9.9996e-01],
        [2.6293e-04, 9.9974e-01],
        [5.2358e-06, 9.9999e-01],
        [7.2698e-01, 2.7302e-01],
        [5.5237e-08, 1.0000e+00],
        [9.9994e-01, 6.3247e-05],
        [1.4162e-07, 1.0000e+00],
        [9.5561e-01, 4.4390e-02],
        [8.5895e-08, 1.0000e+00],
        [2.3158e-07, 1.0000e+00],
        [1.7103e-03, 9.9829e-01],
        [1.3047e-08, 1.0000e+00],
        [9.9956e-01, 4.4354e-04],
        [9.9927e-01, 7.2580e-04],
        [2.3402e-02, 9.7660e-01],
        [1.5902e-08, 1.0000e+00],
        [4.2081e-02, 9.5792e-01],
        [3.4533e-05, 9.9997e-01],
        [9.9975e-01, 2.4658e-04],
        [1.8930e-09, 1.0000e+00],
        [2.1655e-06, 1.0000e+00],
        [3.1868e-06, 1.0000e+00],
        [9.9944e-01, 5.6444e-04],
        [5.5344e-10, 1.0000e+00],
        [9.9950e-01, 5.0345e-04],
        [1.1561e-05, 9.9999e-01],
        [9.9971e-01, 2.8578e-04],
        [1.4801e-03, 9.9852e-01],
        [9.9969e-01, 3.0981e-04],
        [9.7570e-04, 9.9902e-01],
        [1.1451e-05, 9.9999e-01],
        [5.5843e-06, 9.9999e-01],
        [9.9954e-01, 4.5840e-04],
        [4.5372e-08, 1.0000e+00],
        [8.1234e-05, 9.9992e-01],
        [2.3023e-07, 1.0000e+00],
        [9.9881e-01, 1.1851e-03],
        [1.9079e-04, 9.9981e-01],
        [1.5128e-05, 9.9998e-01],
        [2.2427e-07, 1.0000e+00],
        [9.9113e-10, 1.0000e+00],
        [1.0663e-05, 9.9999e-01],
        [1.4019e-08, 1.0000e+00],
        [9.9979e-01, 2.1105e-04],
        [1.0243e-04, 9.9990e-01],
        [7.8249e-01, 2.1751e-01],
        [1.2697e-06, 1.0000e+00],
        [1.7687e-10, 1.0000e+00],
        [6.5868e-07, 1.0000e+00],
        [1.2141e-07, 1.0000e+00],
        [1.4900e-07, 1.0000e+00],
        [9.9963e-01, 3.7382e-04],
        [9.9973e-01, 2.6993e-04],
        [5.2262e-07, 1.0000e+00],
        [1.4014e-03, 9.9860e-01],
        [2.8424e-06, 1.0000e+00],
        [7.8237e-02, 9.2176e-01],
        [1.5684e-07, 1.0000e+00],
        [6.5050e-07, 1.0000e+00],
        [5.2896e-05, 9.9995e-01],
        [1.9837e-03, 9.9802e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.8126e-08, 1.0000e+00],
        [7.8507e-05, 9.9992e-01],
        [9.9987e-01, 1.3008e-04],
        [1.0000e+00, 1.1106e-06],
        [3.6479e-05, 9.9996e-01],
        [1.1244e-04, 9.9989e-01],
        [1.6296e-10, 1.0000e+00],
        [4.1217e-06, 1.0000e+00],
        [9.8436e-01, 1.5639e-02],
        [8.7747e-10, 1.0000e+00],
        [9.9941e-01, 5.8715e-04],
        [1.5109e-09, 1.0000e+00],
        [8.0061e-04, 9.9920e-01],
        [3.7570e-03, 9.9624e-01],
        [9.9951e-01, 4.9143e-04],
        [1.9755e-08, 1.0000e+00],
        [2.3749e-04, 9.9976e-01],
        [1.7495e-06, 1.0000e+00],
        [2.4643e-05, 9.9998e-01],
        [4.0930e-09, 1.0000e+00],
        [2.4714e-04, 9.9975e-01],
        [1.3118e-09, 1.0000e+00],
        [9.6368e-01, 3.6317e-02],
        [4.6433e-05, 9.9995e-01],
        [7.1292e-07, 1.0000e+00],
        [3.6033e-06, 1.0000e+00],
        [1.7070e-06, 1.0000e+00],
        [1.9905e-04, 9.9980e-01],
        [9.9981e-01, 1.9360e-04],
        [1.1693e-06, 1.0000e+00],
        [9.9983e-01, 1.7148e-04],
        [9.9941e-01, 5.8668e-04],
        [8.2361e-05, 9.9992e-01],
        [8.8789e-06, 9.9999e-01],
        [4.0764e-05, 9.9996e-01],
        [1.1664e-03, 9.9883e-01],
        [2.6310e-07, 1.0000e+00],
        [7.2113e-06, 9.9999e-01],
        [9.9997e-01, 2.8489e-05],
        [9.9983e-01, 1.6614e-04],
        [3.2987e-05, 9.9997e-01],
        [6.4921e-06, 9.9999e-01],
        [1.0809e-07, 1.0000e+00],
        [5.0495e-07, 1.0000e+00],
        [6.0910e-05, 9.9994e-01],
        [3.1487e-06, 1.0000e+00],
        [7.2639e-06, 9.9999e-01],
        [2.3196e-06, 1.0000e+00],
        [3.8747e-04, 9.9961e-01],
        [1.5514e-03, 9.9845e-01],
        [7.5554e-08, 1.0000e+00],
        [1.4890e-04, 9.9985e-01],
        [1.5248e-08, 1.0000e+00],
        [7.3538e-09, 1.0000e+00],
        [3.9819e-08, 1.0000e+00],
        [1.1929e-04, 9.9988e-01],
        [1.5373e-06, 1.0000e+00],
        [5.5135e-07, 1.0000e+00],
        [3.4742e-06, 1.0000e+00],
        [1.7581e-03, 9.9824e-01],
        [7.7635e-06, 9.9999e-01],
        [1.0451e-03, 9.9895e-01],
        [4.1672e-07, 1.0000e+00],
        [4.8650e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6205e-04, 9.9984e-01],
        [1.5405e-05, 9.9998e-01],
        [3.6174e-02, 9.6383e-01],
        [5.9404e-08, 1.0000e+00],
        [1.5573e-08, 1.0000e+00],
        [1.9228e-03, 9.9808e-01],
        [9.0976e-01, 9.0235e-02],
        [2.3014e-05, 9.9998e-01],
        [1.5725e-04, 9.9984e-01],
        [3.8458e-05, 9.9996e-01],
        [3.7212e-08, 1.0000e+00],
        [1.2475e-04, 9.9988e-01],
        [9.9097e-01, 9.0264e-03],
        [7.6656e-03, 9.9233e-01],
        [7.3443e-07, 1.0000e+00],
        [9.3989e-10, 1.0000e+00],
        [1.1836e-05, 9.9999e-01],
        [2.5853e-04, 9.9974e-01],
        [2.1379e-09, 1.0000e+00],
        [2.4086e-05, 9.9998e-01],
        [1.5369e-05, 9.9998e-01],
        [7.0103e-08, 1.0000e+00],
        [1.9452e-07, 1.0000e+00],
        [5.1006e-06, 9.9999e-01],
        [2.2564e-05, 9.9998e-01],
        [2.0248e-05, 9.9998e-01],
        [1.8576e-08, 1.0000e+00],
        [1.6678e-06, 1.0000e+00],
        [7.6599e-08, 1.0000e+00],
        [5.1272e-09, 1.0000e+00],
        [3.7433e-08, 1.0000e+00],
        [8.0027e-06, 9.9999e-01],
        [4.4537e-06, 1.0000e+00],
        [9.1295e-08, 1.0000e+00],
        [1.7793e-01, 8.2207e-01],
        [1.1192e-03, 9.9888e-01],
        [5.0320e-08, 1.0000e+00],
        [1.0168e-03, 9.9898e-01],
        [2.8055e-06, 1.0000e+00],
        [9.9984e-01, 1.6382e-04],
        [2.3952e-06, 1.0000e+00],
        [9.9577e-01, 4.2283e-03],
        [8.2823e-06, 9.9999e-01],
        [8.2338e-04, 9.9918e-01],
        [9.4924e-04, 9.9905e-01],
        [5.0736e-06, 9.9999e-01],
        [1.5181e-04, 9.9985e-01],
        [4.7045e-11, 1.0000e+00],
        [9.9936e-01, 6.3949e-04],
        [9.9985e-01, 1.5498e-04],
        [6.7953e-09, 1.0000e+00],
        [6.2737e-04, 9.9937e-01],
        [3.2367e-07, 1.0000e+00],
        [9.9931e-01, 6.9199e-04],
        [5.0240e-08, 1.0000e+00],
        [9.9987e-01, 1.2649e-04],
        [4.2573e-08, 1.0000e+00],
        [7.2059e-06, 9.9999e-01],
        [9.9987e-01, 1.3076e-04],
        [1.6817e-07, 1.0000e+00],
        [2.9031e-05, 9.9997e-01],
        [2.1274e-06, 1.0000e+00],
        [1.1834e-06, 1.0000e+00],
        [1.3448e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2857e-06, 1.0000e+00],
        [5.7715e-06, 9.9999e-01],
        [1.2696e-04, 9.9987e-01],
        [1.2284e-04, 9.9988e-01],
        [2.4879e-04, 9.9975e-01],
        [1.3088e-07, 1.0000e+00],
        [2.1974e-07, 1.0000e+00],
        [1.0584e-05, 9.9999e-01],
        [2.6861e-05, 9.9997e-01],
        [1.6945e-04, 9.9983e-01],
        [9.9784e-01, 2.1616e-03],
        [9.9979e-01, 2.0650e-04],
        [2.7910e-05, 9.9997e-01],
        [2.7143e-03, 9.9729e-01],
        [4.2261e-09, 1.0000e+00],
        [9.9735e-01, 2.6484e-03],
        [2.0443e-08, 1.0000e+00],
        [9.9960e-01, 4.0350e-04],
        [2.8999e-09, 1.0000e+00],
        [5.3185e-08, 1.0000e+00],
        [9.9920e-01, 8.0286e-04],
        [3.6634e-08, 1.0000e+00],
        [2.8509e-04, 9.9971e-01],
        [9.4072e-01, 5.9280e-02],
        [6.5793e-02, 9.3421e-01],
        [9.9327e-08, 1.0000e+00],
        [3.2863e-05, 9.9997e-01],
        [2.2374e-03, 9.9776e-01],
        [3.7607e-09, 1.0000e+00],
        [4.2065e-04, 9.9958e-01],
        [9.9985e-01, 1.5488e-04],
        [1.9720e-07, 1.0000e+00],
        [1.6371e-10, 1.0000e+00],
        [7.4788e-10, 1.0000e+00],
        [3.6784e-10, 1.0000e+00],
        [2.3324e-06, 1.0000e+00],
        [2.3793e-04, 9.9976e-01],
        [2.2324e-06, 1.0000e+00],
        [3.2216e-06, 1.0000e+00],
        [6.3999e-07, 1.0000e+00],
        [1.7330e-06, 1.0000e+00],
        [9.9468e-01, 5.3234e-03],
        [1.2429e-06, 1.0000e+00],
        [1.2419e-07, 1.0000e+00],
        [1.4653e-05, 9.9999e-01],
        [3.9620e-05, 9.9996e-01],
        [4.3753e-04, 9.9956e-01],
        [1.6604e-08, 1.0000e+00],
        [2.7296e-06, 1.0000e+00],
        [2.3805e-04, 9.9976e-01],
        [1.2060e-03, 9.9879e-01],
        [9.9968e-01, 3.2256e-04],
        [2.4201e-08, 1.0000e+00],
        [6.0742e-05, 9.9994e-01],
        [5.6299e-01, 4.3701e-01],
        [1.1157e-06, 1.0000e+00],
        [1.6633e-11, 1.0000e+00],
        [8.9984e-01, 1.0016e-01],
        [4.0478e-08, 1.0000e+00],
        [1.5790e-05, 9.9998e-01],
        [3.9549e-04, 9.9960e-01],
        [1.2586e-04, 9.9987e-01],
        [9.0581e-04, 9.9909e-01],
        [7.7729e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.3694e-03, 9.9263e-01],
        [9.9974e-01, 2.5595e-04],
        [5.2113e-05, 9.9995e-01],
        [9.2660e-05, 9.9991e-01],
        [9.9959e-01, 4.1268e-04],
        [9.9983e-01, 1.7075e-04],
        [1.3573e-06, 1.0000e+00],
        [9.9943e-01, 5.7097e-04],
        [3.9453e-08, 1.0000e+00],
        [1.7949e-05, 9.9998e-01],
        [1.0521e-03, 9.9895e-01],
        [7.0436e-08, 1.0000e+00],
        [4.0165e-09, 1.0000e+00],
        [2.3783e-06, 1.0000e+00],
        [1.0000e+00, 7.2424e-07],
        [3.1204e-07, 1.0000e+00],
        [1.0939e-08, 1.0000e+00],
        [1.6609e-07, 1.0000e+00],
        [3.4037e-06, 1.0000e+00],
        [7.6852e-04, 9.9923e-01],
        [6.6461e-06, 9.9999e-01],
        [3.2249e-06, 1.0000e+00],
        [3.4545e-07, 1.0000e+00],
        [9.9791e-01, 2.0888e-03],
        [1.6221e-07, 1.0000e+00],
        [2.5423e-05, 9.9997e-01],
        [4.0742e-07, 1.0000e+00],
        [2.5541e-05, 9.9997e-01],
        [5.2305e-09, 1.0000e+00],
        [1.5918e-03, 9.9841e-01],
        [9.3039e-03, 9.9070e-01],
        [2.5423e-05, 9.9997e-01],
        [8.8439e-09, 1.0000e+00],
        [1.3710e-02, 9.8629e-01],
        [9.9980e-01, 1.9947e-04],
        [9.6864e-01, 3.1365e-02],
        [2.2425e-06, 1.0000e+00],
        [9.9976e-01, 2.4365e-04],
        [9.9289e-01, 7.1141e-03],
        [1.5008e-03, 9.9850e-01],
        [9.9969e-01, 3.1033e-04],
        [9.9471e-01, 5.2860e-03],
        [4.3541e-06, 1.0000e+00],
        [2.0150e-06, 1.0000e+00],
        [8.5784e-09, 1.0000e+00],
        [1.0126e-02, 9.8987e-01],
        [1.0350e-06, 1.0000e+00],
        [6.1787e-05, 9.9994e-01],
        [9.9997e-01, 3.0071e-05],
        [2.8733e-06, 1.0000e+00],
        [9.9959e-01, 4.1402e-04],
        [3.7800e-07, 1.0000e+00],
        [9.9962e-01, 3.7601e-04],
        [2.6530e-06, 1.0000e+00],
        [3.5245e-09, 1.0000e+00],
        [7.8853e-03, 9.9211e-01],
        [2.4248e-05, 9.9998e-01],
        [9.8678e-08, 1.0000e+00],
        [2.5862e-06, 1.0000e+00],
        [2.7130e-08, 1.0000e+00],
        [2.0131e-08, 1.0000e+00],
        [1.8130e-06, 1.0000e+00],
        [4.7140e-07, 1.0000e+00],
        [7.5814e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2678e-03, 9.9873e-01],
        [1.2473e-03, 9.9875e-01],
        [1.2857e-10, 1.0000e+00],
        [8.7313e-07, 1.0000e+00],
        [4.7132e-04, 9.9953e-01],
        [1.7525e-06, 1.0000e+00],
        [9.2589e-01, 7.4111e-02],
        [9.9983e-01, 1.7477e-04],
        [9.9599e-01, 4.0123e-03],
        [9.9888e-01, 1.1177e-03],
        [3.9598e-06, 1.0000e+00],
        [1.0308e-01, 8.9692e-01],
        [1.6575e-03, 9.9834e-01],
        [8.7299e-09, 1.0000e+00],
        [3.4168e-06, 1.0000e+00],
        [1.9251e-06, 1.0000e+00],
        [1.2765e-03, 9.9872e-01],
        [5.8952e-07, 1.0000e+00],
        [9.6618e-01, 3.3825e-02],
        [9.9971e-01, 2.9428e-04],
        [1.5599e-09, 1.0000e+00],
        [4.7256e-07, 1.0000e+00],
        [2.2771e-10, 1.0000e+00],
        [2.9870e-08, 1.0000e+00],
        [9.9981e-01, 1.8802e-04],
        [4.1508e-09, 1.0000e+00],
        [1.2367e-05, 9.9999e-01],
        [7.4604e-03, 9.9254e-01],
        [4.2867e-08, 1.0000e+00],
        [3.4996e-04, 9.9965e-01],
        [6.1757e-06, 9.9999e-01],
        [3.9423e-07, 1.0000e+00],
        [9.9991e-01, 8.7878e-05],
        [2.0197e-06, 1.0000e+00],
        [9.9823e-01, 1.7702e-03],
        [8.1851e-05, 9.9992e-01],
        [2.1469e-06, 1.0000e+00],
        [9.9995e-01, 4.6469e-05],
        [1.2906e-03, 9.9871e-01],
        [1.1196e-07, 1.0000e+00],
        [1.6312e-01, 8.3688e-01],
        [4.6333e-07, 1.0000e+00],
        [4.5774e-07, 1.0000e+00],
        [9.9973e-01, 2.7489e-04],
        [9.9992e-01, 7.9136e-05],
        [9.4349e-10, 1.0000e+00],
        [2.1657e-08, 1.0000e+00],
        [6.0413e-07, 1.0000e+00],
        [9.9994e-01, 6.2861e-05],
        [2.0332e-03, 9.9797e-01],
        [3.4996e-04, 9.9965e-01],
        [9.2295e-01, 7.7051e-02],
        [4.8763e-01, 5.1237e-01],
        [9.8579e-01, 1.4211e-02],
        [3.9433e-06, 1.0000e+00],
        [1.6399e-06, 1.0000e+00],
        [1.6993e-05, 9.9998e-01],
        [2.8381e-07, 1.0000e+00],
        [1.3658e-05, 9.9999e-01],
        [4.2356e-04, 9.9958e-01],
        [8.8417e-09, 1.0000e+00],
        [1.1702e-04, 9.9988e-01],
        [1.3658e-05, 9.9999e-01],
        [2.3887e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9965e-01, 3.5176e-04],
        [9.8894e-01, 1.1062e-02],
        [9.9979e-01, 2.0572e-04],
        [9.9512e-07, 1.0000e+00],
        [1.1486e-07, 1.0000e+00],
        [1.5714e-08, 1.0000e+00],
        [5.6188e-07, 1.0000e+00],
        [3.0691e-04, 9.9969e-01],
        [4.0424e-03, 9.9596e-01],
        [9.9882e-01, 1.1764e-03],
        [9.9992e-01, 7.8964e-05],
        [5.1290e-11, 1.0000e+00],
        [8.4820e-07, 1.0000e+00],
        [3.0160e-05, 9.9997e-01],
        [6.0540e-08, 1.0000e+00],
        [4.9521e-02, 9.5048e-01],
        [5.9490e-02, 9.4051e-01],
        [2.5285e-07, 1.0000e+00],
        [9.9990e-01, 1.0284e-04],
        [1.0731e-01, 8.9269e-01],
        [1.4228e-06, 1.0000e+00],
        [2.8786e-05, 9.9997e-01],
        [6.6961e-09, 1.0000e+00],
        [7.0739e-05, 9.9993e-01],
        [3.5423e-08, 1.0000e+00],
        [1.8285e-05, 9.9998e-01],
        [9.9986e-01, 1.4233e-04],
        [2.4585e-07, 1.0000e+00],
        [9.9979e-01, 2.0903e-04],
        [9.9993e-01, 6.6376e-05],
        [9.9984e-01, 1.6180e-04],
        [1.1596e-01, 8.8404e-01],
        [9.9993e-01, 7.1135e-05],
        [3.0282e-05, 9.9997e-01],
        [9.9984e-01, 1.6002e-04],
        [9.9991e-01, 9.2643e-05],
        [7.0923e-08, 1.0000e+00],
        [3.5207e-08, 1.0000e+00],
        [9.9968e-01, 3.2211e-04],
        [9.7416e-01, 2.5838e-02],
        [2.1656e-09, 1.0000e+00],
        [1.7444e-07, 1.0000e+00],
        [9.9979e-01, 2.0903e-04],
        [8.6535e-08, 1.0000e+00],
        [6.2717e-05, 9.9994e-01],
        [3.7014e-04, 9.9963e-01],
        [9.9706e-01, 2.9383e-03],
        [3.6727e-07, 1.0000e+00],
        [6.6339e-06, 9.9999e-01],
        [1.2878e-05, 9.9999e-01],
        [9.9306e-01, 6.9409e-03],
        [9.9751e-01, 2.4892e-03],
        [8.9901e-07, 1.0000e+00],
        [8.0509e-10, 1.0000e+00],
        [3.1060e-09, 1.0000e+00],
        [1.0113e-02, 9.8989e-01],
        [8.9942e-07, 1.0000e+00],
        [5.3130e-06, 9.9999e-01],
        [1.4983e-02, 9.8502e-01],
        [1.0674e-02, 9.8933e-01],
        [9.9991e-01, 9.0155e-05],
        [6.4350e-06, 9.9999e-01],
        [2.3076e-05, 9.9998e-01],
        [5.0634e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.6978e-05, 9.9995e-01],
        [1.3430e-08, 1.0000e+00],
        [4.8870e-05, 9.9995e-01],
        [1.3400e-09, 1.0000e+00],
        [1.6780e-07, 1.0000e+00],
        [3.0671e-05, 9.9997e-01],
        [3.6326e-05, 9.9996e-01],
        [9.9957e-01, 4.3031e-04],
        [6.5706e-11, 1.0000e+00],
        [9.9991e-01, 8.8777e-05],
        [3.2310e-07, 1.0000e+00],
        [3.0340e-08, 1.0000e+00],
        [4.8195e-06, 1.0000e+00],
        [2.3016e-03, 9.9770e-01],
        [9.9992e-01, 7.5819e-05],
        [1.1360e-05, 9.9999e-01],
        [9.9995e-01, 4.5322e-05],
        [9.9686e-01, 3.1422e-03],
        [4.9843e-02, 9.5016e-01],
        [6.2270e-07, 1.0000e+00],
        [9.9904e-01, 9.5858e-04],
        [1.0000e+00, 5.2524e-07],
        [2.1025e-07, 1.0000e+00],
        [7.8600e-07, 1.0000e+00],
        [1.0164e-09, 1.0000e+00],
        [9.9936e-01, 6.3607e-04],
        [1.5599e-05, 9.9998e-01],
        [1.3608e-07, 1.0000e+00],
        [4.5853e-03, 9.9541e-01],
        [5.7639e-07, 1.0000e+00],
        [2.1269e-04, 9.9979e-01],
        [1.3083e-05, 9.9999e-01],
        [7.0499e-10, 1.0000e+00],
        [3.6893e-01, 6.3107e-01],
        [9.9994e-01, 5.5295e-05],
        [1.2391e-06, 1.0000e+00],
        [9.1912e-05, 9.9991e-01],
        [7.2436e-07, 1.0000e+00],
        [2.9194e-03, 9.9708e-01],
        [8.5922e-08, 1.0000e+00],
        [3.9069e-08, 1.0000e+00],
        [2.4165e-02, 9.7584e-01],
        [2.9146e-03, 9.9709e-01],
        [1.8171e-08, 1.0000e+00],
        [5.0274e-06, 9.9999e-01],
        [8.9877e-08, 1.0000e+00],
        [4.1231e-08, 1.0000e+00],
        [9.9999e-01, 1.4321e-05],
        [3.7795e-06, 1.0000e+00],
        [1.6764e-06, 1.0000e+00],
        [7.4163e-05, 9.9993e-01],
        [8.0948e-09, 1.0000e+00],
        [4.4169e-05, 9.9996e-01],
        [9.4218e-07, 1.0000e+00],
        [4.6996e-08, 1.0000e+00],
        [9.9946e-01, 5.4189e-04],
        [5.4119e-02, 9.4588e-01],
        [9.9972e-01, 2.7696e-04],
        [9.9994e-01, 6.1721e-05],
        [3.4366e-06, 1.0000e+00],
        [9.8049e-09, 1.0000e+00],
        [9.9817e-01, 1.8314e-03],
        [5.7997e-02, 9.4200e-01],
        [9.9977e-01, 2.3072e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9951e-01, 4.8986e-04],
        [1.7662e-07, 1.0000e+00],
        [9.9988e-01, 1.2393e-04],
        [7.6574e-06, 9.9999e-01],
        [1.5471e-04, 9.9985e-01],
        [1.4682e-05, 9.9999e-01],
        [1.8749e-04, 9.9981e-01],
        [1.0945e-06, 1.0000e+00],
        [3.3102e-05, 9.9997e-01],
        [3.7845e-05, 9.9996e-01],
        [8.6167e-07, 1.0000e+00],
        [1.9852e-06, 1.0000e+00],
        [1.4032e-04, 9.9986e-01],
        [9.9982e-01, 1.8128e-04],
        [9.5209e-08, 1.0000e+00],
        [8.1378e-06, 9.9999e-01],
        [4.4509e-07, 1.0000e+00],
        [9.9991e-01, 8.5546e-05],
        [5.2799e-08, 1.0000e+00],
        [8.3720e-05, 9.9992e-01],
        [4.3650e-07, 1.0000e+00],
        [9.9298e-01, 7.0239e-03],
        [8.8877e-01, 1.1123e-01],
        [1.8251e-05, 9.9998e-01],
        [1.1218e-04, 9.9989e-01],
        [3.9163e-09, 1.0000e+00],
        [2.1667e-05, 9.9998e-01],
        [4.9199e-02, 9.5080e-01],
        [1.1859e-09, 1.0000e+00],
        [3.0860e-09, 1.0000e+00],
        [9.9971e-01, 2.8594e-04],
        [5.5505e-09, 1.0000e+00],
        [3.5317e-07, 1.0000e+00],
        [3.7357e-03, 9.9626e-01],
        [1.2851e-04, 9.9987e-01],
        [2.3655e-08, 1.0000e+00],
        [2.7802e-05, 9.9997e-01],
        [3.8117e-05, 9.9996e-01],
        [1.9823e-05, 9.9998e-01],
        [5.9500e-09, 1.0000e+00],
        [6.7174e-04, 9.9933e-01],
        [4.6483e-08, 1.0000e+00],
        [1.9200e-07, 1.0000e+00],
        [7.4102e-08, 1.0000e+00],
        [6.1463e-09, 1.0000e+00],
        [3.0940e-09, 1.0000e+00],
        [7.3650e-04, 9.9926e-01],
        [4.9335e-07, 1.0000e+00],
        [1.6641e-08, 1.0000e+00],
        [2.0490e-09, 1.0000e+00],
        [3.2885e-03, 9.9671e-01],
        [8.9241e-08, 1.0000e+00],
        [2.1227e-06, 1.0000e+00],
        [4.5782e-08, 1.0000e+00],
        [3.5722e-10, 1.0000e+00],
        [5.2764e-01, 4.7236e-01],
        [8.9070e-09, 1.0000e+00],
        [3.5242e-05, 9.9996e-01],
        [5.6744e-04, 9.9943e-01],
        [2.5641e-06, 1.0000e+00],
        [5.1626e-03, 9.9484e-01],
        [9.9986e-01, 1.4123e-04],
        [1.3624e-14, 1.0000e+00],
        [9.9976e-01, 2.3976e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4594e-02, 9.8541e-01],
        [3.1984e-05, 9.9997e-01],
        [5.8940e-03, 9.9411e-01],
        [1.1136e-02, 9.8886e-01],
        [9.9830e-01, 1.7010e-03],
        [9.9995e-01, 5.4449e-05],
        [1.9343e-03, 9.9807e-01],
        [1.3366e-01, 8.6634e-01],
        [3.2654e-06, 1.0000e+00],
        [2.0311e-06, 1.0000e+00],
        [5.5442e-05, 9.9994e-01],
        [9.6169e-08, 1.0000e+00],
        [1.2359e-04, 9.9988e-01],
        [8.5431e-05, 9.9991e-01],
        [1.0508e-06, 1.0000e+00],
        [3.7060e-05, 9.9996e-01],
        [1.2406e-05, 9.9999e-01],
        [8.7815e-05, 9.9991e-01],
        [1.2638e-04, 9.9987e-01],
        [5.2082e-02, 9.4792e-01],
        [2.9021e-05, 9.9997e-01],
        [9.9995e-01, 5.3814e-05],
        [9.9989e-01, 1.1076e-04],
        [1.7172e-04, 9.9983e-01],
        [5.5665e-02, 9.4434e-01],
        [4.2693e-08, 1.0000e+00],
        [8.9777e-04, 9.9910e-01],
        [9.5658e-01, 4.3419e-02],
        [3.2973e-06, 1.0000e+00],
        [2.3614e-08, 1.0000e+00],
        [2.1557e-03, 9.9784e-01],
        [1.7695e-09, 1.0000e+00],
        [8.5409e-06, 9.9999e-01],
        [3.7976e-03, 9.9620e-01],
        [1.7692e-03, 9.9823e-01],
        [7.0958e-08, 1.0000e+00],
        [1.8107e-05, 9.9998e-01],
        [1.5875e-08, 1.0000e+00],
        [9.9297e-01, 7.0324e-03],
        [4.2957e-06, 1.0000e+00],
        [1.3881e-07, 1.0000e+00],
        [1.1373e-01, 8.8627e-01],
        [2.3174e-03, 9.9768e-01],
        [7.5331e-04, 9.9925e-01],
        [8.5496e-04, 9.9915e-01],
        [7.7917e-07, 1.0000e+00],
        [9.9876e-01, 1.2447e-03],
        [2.7436e-05, 9.9997e-01],
        [4.5793e-05, 9.9995e-01],
        [9.9955e-01, 4.5380e-04],
        [3.2954e-07, 1.0000e+00],
        [8.2642e-08, 1.0000e+00],
        [8.1047e-01, 1.8953e-01],
        [1.8541e-03, 9.9815e-01],
        [7.4138e-07, 1.0000e+00],
        [2.5528e-08, 1.0000e+00],
        [7.8632e-05, 9.9992e-01],
        [1.3273e-06, 1.0000e+00],
        [1.2383e-03, 9.9876e-01],
        [4.1086e-09, 1.0000e+00],
        [6.8282e-07, 1.0000e+00],
        [1.3112e-07, 1.0000e+00],
        [5.0671e-05, 9.9995e-01],
        [3.0160e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.4138e-04, 9.9966e-01],
        [1.1010e-04, 9.9989e-01],
        [9.5368e-01, 4.6318e-02],
        [4.7112e-03, 9.9529e-01],
        [3.0159e-03, 9.9698e-01],
        [5.6611e-05, 9.9994e-01],
        [9.4937e-02, 9.0506e-01],
        [1.1702e-06, 1.0000e+00],
        [1.0752e-05, 9.9999e-01],
        [4.8162e-03, 9.9518e-01],
        [1.1258e-03, 9.9887e-01],
        [1.2153e-04, 9.9988e-01],
        [1.0098e-03, 9.9899e-01],
        [8.1675e-06, 9.9999e-01],
        [2.2599e-02, 9.7740e-01],
        [8.0411e-05, 9.9992e-01],
        [2.1888e-07, 1.0000e+00],
        [9.9993e-01, 6.5149e-05],
        [9.9982e-01, 1.8497e-04],
        [9.9991e-01, 9.1763e-05],
        [4.7578e-04, 9.9952e-01],
        [1.1345e-04, 9.9989e-01],
        [4.7170e-05, 9.9995e-01],
        [1.6877e-04, 9.9983e-01],
        [9.8565e-05, 9.9990e-01],
        [9.9989e-01, 1.0800e-04],
        [1.1631e-07, 1.0000e+00],
        [1.6327e-06, 1.0000e+00],
        [2.3823e-07, 1.0000e+00],
        [4.3520e-06, 1.0000e+00],
        [1.4467e-03, 9.9855e-01],
        [9.9962e-01, 3.8243e-04],
        [3.9091e-04, 9.9961e-01],
        [9.9989e-01, 1.1156e-04],
        [3.9487e-04, 9.9961e-01],
        [2.3790e-05, 9.9998e-01],
        [9.9992e-01, 8.1264e-05],
        [1.9812e-04, 9.9980e-01],
        [1.2866e-04, 9.9987e-01],
        [8.9184e-05, 9.9991e-01],
        [2.4984e-07, 1.0000e+00],
        [9.9951e-01, 4.8911e-04],
        [1.2695e-07, 1.0000e+00],
        [3.6671e-06, 1.0000e+00],
        [1.4453e-08, 1.0000e+00],
        [1.3738e-06, 1.0000e+00],
        [1.2824e-04, 9.9987e-01],
        [7.0947e-01, 2.9053e-01],
        [1.2566e-05, 9.9999e-01],
        [4.5280e-08, 1.0000e+00],
        [1.6280e-05, 9.9998e-01],
        [1.4145e-04, 9.9986e-01],
        [9.9997e-01, 3.3145e-05],
        [2.1903e-10, 1.0000e+00],
        [3.6989e-07, 1.0000e+00],
        [9.9513e-01, 4.8741e-03],
        [1.0207e-04, 9.9990e-01],
        [9.9685e-01, 3.1484e-03],
        [8.1158e-05, 9.9992e-01],
        [9.9999e-01, 1.1024e-05],
        [1.0328e-04, 9.9990e-01],
        [1.4029e-05, 9.9999e-01],
        [1.9428e-02, 9.8057e-01],
        [6.0588e-03, 9.9394e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9983e-01, 1.7028e-04],
        [8.1608e-07, 1.0000e+00],
        [9.4854e-07, 1.0000e+00],
        [2.0102e-07, 1.0000e+00],
        [7.9948e-04, 9.9920e-01],
        [9.9971e-01, 2.8991e-04],
        [5.6876e-05, 9.9994e-01],
        [2.9898e-07, 1.0000e+00],
        [9.9321e-01, 6.7881e-03],
        [4.1657e-06, 1.0000e+00],
        [3.4949e-08, 1.0000e+00],
        [1.9958e-05, 9.9998e-01],
        [9.6923e-01, 3.0767e-02],
        [2.2190e-09, 1.0000e+00],
        [3.0017e-04, 9.9970e-01],
        [9.9741e-01, 2.5936e-03],
        [1.8051e-05, 9.9998e-01],
        [9.4162e-06, 9.9999e-01],
        [4.2776e-03, 9.9572e-01],
        [4.4138e-08, 1.0000e+00],
        [7.1630e-02, 9.2837e-01],
        [2.1067e-04, 9.9979e-01],
        [3.1918e-03, 9.9681e-01],
        [1.7182e-06, 1.0000e+00],
        [1.0000e+00, 2.0791e-06],
        [9.9971e-01, 2.8824e-04],
        [1.0969e-04, 9.9989e-01],
        [6.8331e-07, 1.0000e+00],
        [2.0016e-04, 9.9980e-01],
        [2.0713e-06, 1.0000e+00],
        [1.7118e-03, 9.9829e-01],
        [3.6573e-08, 1.0000e+00],
        [3.0270e-01, 6.9730e-01],
        [3.5847e-05, 9.9996e-01],
        [1.7540e-03, 9.9825e-01],
        [9.9792e-01, 2.0833e-03],
        [4.0272e-03, 9.9597e-01],
        [4.4966e-09, 1.0000e+00],
        [1.8737e-02, 9.8126e-01],
        [4.8714e-03, 9.9513e-01],
        [1.5474e-04, 9.9985e-01],
        [1.0000e+00, 2.5361e-06],
        [4.1335e-05, 9.9996e-01],
        [9.9982e-01, 1.8192e-04],
        [6.7290e-04, 9.9933e-01],
        [9.9368e-01, 6.3213e-03],
        [1.3792e-05, 9.9999e-01],
        [9.9994e-01, 5.9550e-05],
        [2.8403e-07, 1.0000e+00],
        [4.3728e-08, 1.0000e+00],
        [2.0728e-04, 9.9979e-01],
        [1.0000e+00, 4.9131e-06],
        [1.9855e-02, 9.8014e-01],
        [9.9994e-01, 6.2541e-05],
        [5.8902e-02, 9.4110e-01],
        [1.3808e-04, 9.9986e-01],
        [8.9404e-07, 1.0000e+00],
        [2.7501e-01, 7.2499e-01],
        [1.5083e-05, 9.9998e-01],
        [1.6449e-06, 1.0000e+00],
        [5.2441e-07, 1.0000e+00],
        [5.5565e-04, 9.9944e-01],
        [9.1182e-01, 8.8185e-02],
        [3.7865e-10, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.1747e-02, 9.3825e-01],
        [2.2242e-02, 9.7776e-01],
        [2.8229e-05, 9.9997e-01],
        [1.5744e-08, 1.0000e+00],
        [3.1567e-03, 9.9684e-01],
        [5.2238e-09, 1.0000e+00],
        [9.9985e-01, 1.4833e-04],
        [9.9963e-01, 3.6546e-04],
        [2.3605e-04, 9.9976e-01],
        [9.8709e-01, 1.2912e-02],
        [8.4130e-08, 1.0000e+00],
        [5.3974e-04, 9.9946e-01],
        [3.3387e-07, 1.0000e+00],
        [4.3118e-05, 9.9996e-01],
        [9.9994e-01, 6.2380e-05],
        [3.0723e-01, 6.9277e-01],
        [4.6605e-04, 9.9953e-01],
        [6.3790e-03, 9.9362e-01],
        [5.1851e-04, 9.9948e-01],
        [1.2223e-07, 1.0000e+00],
        [9.9993e-01, 6.5730e-05],
        [9.7355e-01, 2.6453e-02],
        [1.2049e-01, 8.7951e-01],
        [4.4920e-08, 1.0000e+00],
        [9.1247e-04, 9.9909e-01],
        [9.9982e-01, 1.8220e-04],
        [2.3072e-08, 1.0000e+00],
        [9.9988e-01, 1.1968e-04],
        [9.9966e-01, 3.4334e-04],
        [9.9989e-01, 1.0526e-04],
        [9.6976e-05, 9.9990e-01],
        [9.9898e-01, 1.0224e-03],
        [7.5513e-03, 9.9245e-01],
        [9.0875e-06, 9.9999e-01],
        [1.0795e-05, 9.9999e-01],
        [1.5873e-07, 1.0000e+00],
        [9.9976e-01, 2.3967e-04],
        [9.9923e-01, 7.6953e-04],
        [9.7469e-01, 2.5307e-02],
        [1.2091e-09, 1.0000e+00],
        [9.9976e-01, 2.4241e-04],
        [1.9821e-06, 1.0000e+00],
        [1.3115e-04, 9.9987e-01],
        [4.5527e-06, 1.0000e+00],
        [4.5799e-06, 1.0000e+00],
        [1.3985e-08, 1.0000e+00],
        [1.9549e-05, 9.9998e-01],
        [1.4095e-05, 9.9999e-01],
        [1.4125e-05, 9.9999e-01],
        [2.7697e-06, 1.0000e+00],
        [1.0428e-06, 1.0000e+00],
        [2.1685e-10, 1.0000e+00],
        [5.5778e-01, 4.4222e-01],
        [6.1655e-07, 1.0000e+00],
        [9.9996e-01, 4.0536e-05],
        [9.9979e-01, 2.1414e-04],
        [6.4446e-05, 9.9994e-01],
        [4.7099e-03, 9.9529e-01],
        [9.9991e-01, 9.2131e-05],
        [1.3168e-06, 1.0000e+00],
        [3.5387e-05, 9.9996e-01],
        [1.2174e-05, 9.9999e-01],
        [9.9958e-01, 4.2022e-04],
        [3.3269e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9706e-01, 2.9355e-03],
        [3.2810e-04, 9.9967e-01],
        [1.2788e-06, 1.0000e+00],
        [9.9060e-01, 9.3994e-03],
        [1.7688e-08, 1.0000e+00],
        [5.9097e-06, 9.9999e-01],
        [9.9992e-01, 8.0230e-05],
        [9.9988e-01, 1.2382e-04],
        [9.9973e-01, 2.6881e-04],
        [2.7135e-04, 9.9973e-01],
        [4.6101e-03, 9.9539e-01],
        [9.9974e-01, 2.6270e-04],
        [1.6899e-05, 9.9998e-01],
        [9.3310e-01, 6.6898e-02],
        [8.5648e-05, 9.9991e-01],
        [9.7287e-07, 1.0000e+00],
        [4.3525e-10, 1.0000e+00],
        [2.2438e-04, 9.9978e-01],
        [9.9981e-01, 1.8750e-04],
        [1.0307e-06, 1.0000e+00],
        [9.9387e-01, 6.1260e-03],
        [2.6346e-07, 1.0000e+00],
        [2.6918e-07, 1.0000e+00],
        [6.7786e-04, 9.9932e-01],
        [2.9064e-05, 9.9997e-01],
        [7.5863e-06, 9.9999e-01],
        [3.7939e-09, 1.0000e+00],
        [6.1940e-09, 1.0000e+00],
        [1.1436e-04, 9.9989e-01],
        [4.1138e-07, 1.0000e+00],
        [1.1289e-08, 1.0000e+00],
        [1.0156e-05, 9.9999e-01],
        [1.3856e-09, 1.0000e+00],
        [4.6164e-05, 9.9995e-01],
        [9.7829e-01, 2.1708e-02],
        [2.3348e-01, 7.6652e-01],
        [3.9507e-07, 1.0000e+00],
        [3.8382e-06, 1.0000e+00],
        [9.1727e-08, 1.0000e+00],
        [9.9967e-01, 3.2568e-04],
        [9.8360e-01, 1.6399e-02],
        [5.3486e-09, 1.0000e+00],
        [2.5638e-07, 1.0000e+00],
        [3.2750e-03, 9.9673e-01],
        [6.1691e-05, 9.9994e-01],
        [2.4767e-04, 9.9975e-01],
        [1.6962e-06, 1.0000e+00],
        [3.5531e-05, 9.9996e-01],
        [4.7631e-06, 1.0000e+00],
        [9.9901e-01, 9.9492e-04],
        [9.9919e-01, 8.1076e-04],
        [5.1463e-06, 9.9999e-01],
        [1.7908e-07, 1.0000e+00],
        [9.9709e-01, 2.9081e-03],
        [1.3690e-09, 1.0000e+00],
        [2.8527e-05, 9.9997e-01],
        [6.2605e-07, 1.0000e+00],
        [9.9882e-01, 1.1751e-03],
        [9.7169e-04, 9.9903e-01],
        [1.9076e-06, 1.0000e+00],
        [1.8850e-10, 1.0000e+00],
        [1.7630e-06, 1.0000e+00],
        [3.9148e-04, 9.9961e-01],
        [6.1279e-01, 3.8721e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.3883e-09, 1.0000e+00],
        [2.3100e-11, 1.0000e+00],
        [5.5907e-07, 1.0000e+00],
        [5.7541e-04, 9.9942e-01],
        [2.2752e-08, 1.0000e+00],
        [1.2112e-08, 1.0000e+00],
        [7.2717e-07, 1.0000e+00],
        [4.4919e-07, 1.0000e+00],
        [2.4361e-06, 1.0000e+00],
        [3.5601e-04, 9.9964e-01],
        [9.9941e-01, 5.9012e-04],
        [1.6880e-05, 9.9998e-01],
        [2.5867e-03, 9.9741e-01],
        [4.4219e-07, 1.0000e+00],
        [2.1351e-02, 9.7865e-01],
        [2.5513e-07, 1.0000e+00],
        [1.5878e-09, 1.0000e+00],
        [3.4633e-08, 1.0000e+00],
        [2.3688e-06, 1.0000e+00],
        [6.8163e-08, 1.0000e+00],
        [5.3877e-10, 1.0000e+00],
        [9.9938e-01, 6.1576e-04],
        [1.1999e-09, 1.0000e+00],
        [1.2058e-04, 9.9988e-01],
        [4.3957e-02, 9.5604e-01],
        [9.9978e-01, 2.1877e-04],
        [5.9613e-05, 9.9994e-01],
        [9.8716e-09, 1.0000e+00],
        [1.3608e-04, 9.9986e-01],
        [7.8310e-06, 9.9999e-01],
        [2.4289e-09, 1.0000e+00],
        [9.9980e-01, 1.9940e-04],
        [9.9183e-01, 8.1650e-03],
        [9.9988e-01, 1.2218e-04],
        [5.0752e-08, 1.0000e+00],
        [7.1286e-07, 1.0000e+00],
        [9.2162e-01, 7.8377e-02],
        [9.9546e-01, 4.5364e-03],
        [4.4610e-05, 9.9996e-01],
        [5.5358e-11, 1.0000e+00],
        [1.1183e-09, 1.0000e+00],
        [2.8379e-05, 9.9997e-01],
        [1.1518e-03, 9.9885e-01],
        [2.1101e-09, 1.0000e+00],
        [3.4777e-03, 9.9652e-01],
        [9.9989e-01, 1.1248e-04],
        [9.5264e-01, 4.7360e-02],
        [1.0731e-06, 1.0000e+00],
        [9.9987e-01, 1.2594e-04],
        [9.1641e-01, 8.3591e-02],
        [1.4420e-08, 1.0000e+00],
        [9.1626e-06, 9.9999e-01],
        [1.0093e-01, 8.9907e-01],
        [2.6559e-04, 9.9973e-01],
        [4.2348e-08, 1.0000e+00],
        [9.9954e-01, 4.6295e-04],
        [7.4222e-02, 9.2578e-01],
        [1.4663e-07, 1.0000e+00],
        [7.4244e-05, 9.9993e-01],
        [3.7054e-09, 1.0000e+00],
        [7.2665e-06, 9.9999e-01],
        [9.9985e-01, 1.5460e-04],
        [9.9975e-01, 2.5244e-04],
        [6.5693e-04, 9.9934e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.4843e-09, 1.0000e+00],
        [3.5447e-02, 9.6455e-01],
        [2.2840e-02, 9.7716e-01],
        [3.0030e-06, 1.0000e+00],
        [1.7004e-05, 9.9998e-01],
        [1.8781e-09, 1.0000e+00],
        [3.4432e-08, 1.0000e+00],
        [9.9698e-01, 3.0170e-03],
        [6.3125e-02, 9.3688e-01],
        [4.3681e-06, 1.0000e+00],
        [4.8407e-04, 9.9952e-01],
        [5.9387e-05, 9.9994e-01],
        [2.1450e-09, 1.0000e+00],
        [2.4833e-08, 1.0000e+00],
        [3.2404e-13, 1.0000e+00],
        [2.3986e-09, 1.0000e+00],
        [1.0487e-02, 9.8951e-01],
        [9.9979e-01, 2.0961e-04],
        [2.5926e-04, 9.9974e-01],
        [9.9992e-01, 7.8301e-05],
        [1.0566e-04, 9.9989e-01],
        [8.6549e-08, 1.0000e+00],
        [2.2404e-07, 1.0000e+00],
        [1.3013e-05, 9.9999e-01],
        [6.1286e-09, 1.0000e+00],
        [8.7425e-05, 9.9991e-01],
        [1.7373e-05, 9.9998e-01],
        [9.9609e-01, 3.9083e-03],
        [1.3881e-08, 1.0000e+00],
        [3.2870e-06, 1.0000e+00],
        [1.3228e-07, 1.0000e+00],
        [2.3132e-03, 9.9769e-01],
        [2.2497e-09, 1.0000e+00],
        [6.5151e-11, 1.0000e+00],
        [3.0733e-10, 1.0000e+00],
        [9.3728e-01, 6.2717e-02],
        [9.9969e-01, 3.1252e-04],
        [4.0098e-09, 1.0000e+00],
        [1.7841e-02, 9.8216e-01],
        [1.5341e-08, 1.0000e+00],
        [3.9771e-06, 1.0000e+00],
        [1.5387e-01, 8.4613e-01],
        [2.8871e-04, 9.9971e-01],
        [9.9984e-01, 1.6362e-04],
        [1.1317e-07, 1.0000e+00],
        [9.9986e-01, 1.4477e-04],
        [9.9968e-01, 3.1528e-04],
        [9.4896e-01, 5.1045e-02],
        [2.1908e-07, 1.0000e+00],
        [3.5029e-06, 1.0000e+00],
        [9.9962e-01, 3.7927e-04],
        [9.5717e-01, 4.2831e-02],
        [1.8539e-07, 1.0000e+00],
        [9.9976e-01, 2.3808e-04],
        [9.7421e-01, 2.5787e-02],
        [5.5365e-04, 9.9945e-01],
        [9.8840e-08, 1.0000e+00],
        [1.5511e-06, 1.0000e+00],
        [5.6979e-08, 1.0000e+00],
        [1.7070e-06, 1.0000e+00],
        [5.1860e-07, 1.0000e+00],
        [2.0237e-07, 1.0000e+00],
        [9.9967e-01, 3.3050e-04],
        [1.3441e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.1006e-07, 1.0000e+00],
        [2.5885e-05, 9.9997e-01],
        [9.9955e-01, 4.5235e-04],
        [1.4587e-04, 9.9985e-01],
        [4.1574e-08, 1.0000e+00],
        [3.9144e-04, 9.9961e-01],
        [2.5024e-04, 9.9975e-01],
        [9.1118e-06, 9.9999e-01],
        [9.7522e-06, 9.9999e-01],
        [9.9943e-01, 5.7068e-04],
        [4.1031e-09, 1.0000e+00],
        [9.9965e-01, 3.5005e-04],
        [8.6043e-05, 9.9991e-01],
        [1.5285e-06, 1.0000e+00],
        [9.9981e-01, 1.8662e-04],
        [6.0908e-06, 9.9999e-01],
        [7.6942e-09, 1.0000e+00],
        [1.8769e-05, 9.9998e-01],
        [2.4705e-05, 9.9998e-01],
        [3.4825e-02, 9.6518e-01],
        [3.2306e-10, 1.0000e+00],
        [2.0942e-07, 1.0000e+00],
        [9.2046e-01, 7.9543e-02],
        [3.2759e-05, 9.9997e-01],
        [1.5598e-04, 9.9984e-01],
        [2.0108e-06, 1.0000e+00],
        [1.9873e-10, 1.0000e+00],
        [4.9587e-06, 9.9999e-01],
        [9.3333e-01, 6.6665e-02],
        [9.9966e-01, 3.3864e-04],
        [5.6349e-09, 1.0000e+00],
        [1.0792e-06, 1.0000e+00],
        [2.4130e-05, 9.9998e-01],
        [9.9996e-01, 3.7871e-05],
        [2.9715e-09, 1.0000e+00],
        [9.9954e-01, 4.5568e-04],
        [2.8333e-05, 9.9997e-01],
        [6.5655e-01, 3.4345e-01],
        [1.5285e-07, 1.0000e+00],
        [2.8826e-05, 9.9997e-01],
        [1.5623e-05, 9.9998e-01],
        [2.3609e-01, 7.6391e-01],
        [1.2423e-09, 1.0000e+00],
        [9.9961e-01, 3.9017e-04],
        [1.9087e-06, 1.0000e+00],
        [9.9956e-01, 4.3705e-04],
        [7.9883e-07, 1.0000e+00],
        [9.8597e-09, 1.0000e+00],
        [4.5720e-08, 1.0000e+00],
        [3.0237e-05, 9.9997e-01],
        [9.4310e-05, 9.9991e-01],
        [1.3392e-07, 1.0000e+00],
        [1.0417e-07, 1.0000e+00],
        [6.0924e-06, 9.9999e-01],
        [1.7421e-04, 9.9983e-01],
        [4.3189e-05, 9.9996e-01],
        [3.2834e-04, 9.9967e-01],
        [8.7374e-08, 1.0000e+00],
        [2.2464e-07, 1.0000e+00],
        [2.1721e-05, 9.9998e-01],
        [1.2517e-08, 1.0000e+00],
        [1.7480e-04, 9.9983e-01],
        [1.0879e-06, 1.0000e+00],
        [1.3025e-04, 9.9987e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5536e-08, 1.0000e+00],
        [8.7281e-07, 1.0000e+00],
        [1.8659e-03, 9.9813e-01],
        [7.3266e-07, 1.0000e+00],
        [4.1279e-07, 1.0000e+00],
        [4.3631e-09, 1.0000e+00],
        [9.0227e-05, 9.9991e-01],
        [1.5681e-06, 1.0000e+00],
        [4.8221e-06, 1.0000e+00],
        [9.9995e-01, 5.1737e-05],
        [1.7661e-09, 1.0000e+00],
        [1.8262e-05, 9.9998e-01],
        [9.9976e-01, 2.3922e-04],
        [9.9675e-01, 3.2546e-03],
        [3.1696e-01, 6.8304e-01],
        [5.0292e-09, 1.0000e+00],
        [4.6166e-08, 1.0000e+00],
        [8.8261e-09, 1.0000e+00],
        [5.0684e-09, 1.0000e+00],
        [1.4964e-06, 1.0000e+00],
        [8.5277e-09, 1.0000e+00],
        [7.4468e-08, 1.0000e+00],
        [6.9814e-06, 9.9999e-01],
        [1.4020e-05, 9.9999e-01],
        [6.3198e-06, 9.9999e-01],
        [9.9972e-01, 2.7505e-04],
        [1.0045e-05, 9.9999e-01],
        [1.3168e-04, 9.9987e-01],
        [3.1561e-05, 9.9997e-01],
        [2.7431e-10, 1.0000e+00],
        [1.9133e-06, 1.0000e+00],
        [2.7431e-10, 1.0000e+00],
        [9.9980e-01, 1.9542e-04],
        [9.9974e-01, 2.5923e-04],
        [1.0214e-06, 1.0000e+00],
        [5.6434e-03, 9.9436e-01],
        [7.6658e-05, 9.9992e-01],
        [3.1150e-06, 1.0000e+00],
        [5.2436e-07, 1.0000e+00],
        [9.9984e-01, 1.6170e-04],
        [9.9967e-01, 3.3066e-04],
        [3.3455e-06, 1.0000e+00],
        [1.4579e-06, 1.0000e+00],
        [1.5335e-06, 1.0000e+00],
        [5.7735e-01, 4.2265e-01],
        [6.5125e-06, 9.9999e-01],
        [1.3420e-04, 9.9987e-01],
        [1.7004e-06, 1.0000e+00],
        [1.5265e-09, 1.0000e+00],
        [5.4851e-08, 1.0000e+00],
        [1.9453e-06, 1.0000e+00],
        [1.7264e-05, 9.9998e-01],
        [9.9975e-01, 2.5309e-04],
        [9.9986e-01, 1.4241e-04],
        [1.0286e-05, 9.9999e-01],
        [9.9128e-01, 8.7161e-03],
        [2.0619e-10, 1.0000e+00],
        [1.1570e-07, 1.0000e+00],
        [2.8356e-08, 1.0000e+00],
        [9.9961e-01, 3.8863e-04],
        [1.0207e-01, 8.9793e-01],
        [2.9158e-05, 9.9997e-01],
        [9.9916e-01, 8.4024e-04],
        [3.8933e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8811e-10, 1.0000e+00],
        [3.9476e-09, 1.0000e+00],
        [3.1573e-03, 9.9684e-01],
        [1.8463e-07, 1.0000e+00],
        [7.9397e-09, 1.0000e+00],
        [7.8677e-05, 9.9992e-01],
        [1.9910e-06, 1.0000e+00],
        [3.1455e-03, 9.9685e-01],
        [4.8047e-08, 1.0000e+00],
        [9.9934e-01, 6.5531e-04],
        [1.1453e-04, 9.9989e-01],
        [9.9802e-01, 1.9804e-03],
        [9.9783e-01, 2.1656e-03],
        [1.3204e-05, 9.9999e-01],
        [3.7017e-05, 9.9996e-01],
        [9.9620e-01, 3.7953e-03],
        [9.9221e-01, 7.7872e-03],
        [9.0261e-07, 1.0000e+00],
        [3.6003e-07, 1.0000e+00],
        [9.9920e-01, 7.9583e-04],
        [4.9169e-04, 9.9951e-01],
        [3.5893e-07, 1.0000e+00],
        [8.1089e-06, 9.9999e-01],
        [1.5148e-06, 1.0000e+00],
        [3.8143e-09, 1.0000e+00],
        [9.9987e-01, 1.2832e-04],
        [9.9961e-01, 3.9368e-04],
        [2.5718e-06, 1.0000e+00],
        [1.1740e-04, 9.9988e-01],
        [9.9818e-01, 1.8231e-03],
        [2.8782e-06, 1.0000e+00],
        [1.0059e-07, 1.0000e+00],
        [9.9977e-01, 2.3368e-04],
        [5.4457e-07, 1.0000e+00],
        [1.7289e-02, 9.8271e-01],
        [4.7252e-06, 1.0000e+00],
        [1.1518e-04, 9.9988e-01],
        [8.9562e-08, 1.0000e+00],
        [2.6456e-05, 9.9997e-01],
        [9.9988e-01, 1.1680e-04],
        [1.0545e-07, 1.0000e+00],
        [1.1895e-08, 1.0000e+00],
        [6.9275e-06, 9.9999e-01],
        [2.4215e-04, 9.9976e-01],
        [9.9964e-01, 3.6195e-04],
        [1.2694e-04, 9.9987e-01],
        [1.7006e-05, 9.9998e-01],
        [9.5525e-01, 4.4749e-02],
        [5.3210e-06, 9.9999e-01],
        [9.4515e-01, 5.4847e-02],
        [1.9994e-05, 9.9998e-01],
        [1.6105e-06, 1.0000e+00],
        [4.2913e-09, 1.0000e+00],
        [9.4664e-06, 9.9999e-01],
        [3.8337e-02, 9.6166e-01],
        [1.2821e-05, 9.9999e-01],
        [8.3461e-09, 1.0000e+00],
        [8.3624e-08, 1.0000e+00],
        [7.4136e-06, 9.9999e-01],
        [3.8761e-10, 1.0000e+00],
        [2.6233e-04, 9.9974e-01],
        [1.3306e-04, 9.9987e-01],
        [9.0258e-11, 1.0000e+00],
        [3.8448e-04, 9.9962e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5790e-05, 9.9997e-01],
        [3.8099e-02, 9.6190e-01],
        [4.1406e-07, 1.0000e+00],
        [5.7896e-01, 4.2104e-01],
        [2.1602e-06, 1.0000e+00],
        [9.9976e-01, 2.4384e-04],
        [9.9864e-01, 1.3587e-03],
        [3.3638e-07, 1.0000e+00],
        [9.3345e-01, 6.6546e-02],
        [3.4845e-09, 1.0000e+00],
        [3.9746e-05, 9.9996e-01],
        [9.6053e-08, 1.0000e+00],
        [1.3823e-05, 9.9999e-01],
        [2.0822e-06, 1.0000e+00],
        [9.9959e-01, 4.1405e-04],
        [1.7005e-04, 9.9983e-01],
        [9.9985e-01, 1.5424e-04],
        [1.2906e-05, 9.9999e-01],
        [1.6119e-05, 9.9998e-01],
        [2.4295e-07, 1.0000e+00],
        [3.9739e-05, 9.9996e-01],
        [9.9406e-01, 5.9446e-03],
        [4.7948e-09, 1.0000e+00],
        [1.8950e-05, 9.9998e-01],
        [9.9335e-09, 1.0000e+00],
        [4.2088e-09, 1.0000e+00],
        [3.3851e-08, 1.0000e+00],
        [9.9995e-01, 5.4459e-05],
        [9.9992e-01, 8.0550e-05],
        [9.9997e-01, 2.8012e-05],
        [5.5290e-09, 1.0000e+00],
        [9.9936e-01, 6.3512e-04],
        [2.8464e-04, 9.9972e-01],
        [5.9663e-06, 9.9999e-01],
        [4.7471e-08, 1.0000e+00],
        [9.9568e-01, 4.3161e-03],
        [8.1128e-04, 9.9919e-01],
        [1.3973e-02, 9.8603e-01],
        [9.9451e-01, 5.4908e-03],
        [9.6561e-01, 3.4388e-02],
        [6.2084e-05, 9.9994e-01],
        [1.6901e-07, 1.0000e+00],
        [1.1005e-03, 9.9890e-01],
        [1.3030e-01, 8.6970e-01],
        [1.5257e-02, 9.8474e-01],
        [2.4147e-09, 1.0000e+00],
        [9.3761e-02, 9.0624e-01],
        [9.9807e-01, 1.9328e-03],
        [9.9252e-01, 7.4846e-03],
        [8.5438e-08, 1.0000e+00],
        [1.2579e-09, 1.0000e+00],
        [9.9886e-01, 1.1351e-03],
        [3.8634e-05, 9.9996e-01],
        [1.5947e-07, 1.0000e+00],
        [5.0078e-06, 9.9999e-01],
        [1.7811e-04, 9.9982e-01],
        [1.0956e-07, 1.0000e+00],
        [1.6643e-06, 1.0000e+00],
        [1.8659e-08, 1.0000e+00],
        [1.9309e-06, 1.0000e+00],
        [1.7949e-07, 1.0000e+00],
        [1.2519e-01, 8.7481e-01],
        [3.1738e-07, 1.0000e+00],
        [5.3441e-01, 4.6559e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.3891e-09, 1.0000e+00],
        [9.9946e-01, 5.4069e-04],
        [5.8142e-07, 1.0000e+00],
        [2.0406e-06, 1.0000e+00],
        [8.2814e-06, 9.9999e-01],
        [3.2180e-06, 1.0000e+00],
        [1.8387e-04, 9.9982e-01],
        [3.0631e-05, 9.9997e-01],
        [5.1085e-03, 9.9489e-01],
        [3.9418e-07, 1.0000e+00],
        [1.0664e-08, 1.0000e+00],
        [4.6399e-06, 1.0000e+00],
        [4.5867e-10, 1.0000e+00],
        [3.3658e-08, 1.0000e+00],
        [9.9699e-01, 3.0082e-03],
        [6.8514e-04, 9.9931e-01],
        [1.9536e-02, 9.8046e-01],
        [9.9650e-01, 3.5020e-03],
        [1.9158e-04, 9.9981e-01],
        [7.6823e-06, 9.9999e-01],
        [9.9941e-01, 5.9137e-04],
        [9.9803e-01, 1.9736e-03],
        [9.9868e-01, 1.3229e-03],
        [1.7055e-08, 1.0000e+00],
        [9.0697e-05, 9.9991e-01],
        [1.2256e-08, 1.0000e+00],
        [1.7092e-04, 9.9983e-01],
        [1.9050e-03, 9.9809e-01],
        [7.6809e-07, 1.0000e+00],
        [9.9371e-01, 6.2899e-03],
        [9.9974e-01, 2.6326e-04],
        [2.4089e-04, 9.9976e-01],
        [9.9931e-01, 6.8966e-04],
        [2.4224e-05, 9.9998e-01],
        [9.2797e-01, 7.2029e-02],
        [9.9952e-01, 4.7784e-04],
        [4.2927e-07, 1.0000e+00],
        [3.5618e-06, 1.0000e+00],
        [1.1685e-08, 1.0000e+00],
        [9.9963e-01, 3.7478e-04],
        [9.9301e-01, 6.9934e-03],
        [5.5846e-04, 9.9944e-01],
        [9.9970e-01, 3.0386e-04],
        [3.6824e-08, 1.0000e+00],
        [9.4191e-09, 1.0000e+00],
        [1.0749e-04, 9.9989e-01],
        [9.3972e-01, 6.0278e-02],
        [4.2661e-06, 1.0000e+00],
        [9.9976e-01, 2.3802e-04],
        [1.4094e-06, 1.0000e+00],
        [7.0271e-02, 9.2973e-01],
        [3.0421e-08, 1.0000e+00],
        [1.3933e-06, 1.0000e+00],
        [8.1678e-05, 9.9992e-01],
        [9.3647e-01, 6.3528e-02],
        [8.6192e-07, 1.0000e+00],
        [9.0872e-01, 9.1279e-02],
        [5.0729e-05, 9.9995e-01],
        [9.9979e-01, 2.1450e-04],
        [2.8454e-08, 1.0000e+00],
        [2.2261e-11, 1.0000e+00],
        [4.1140e-09, 1.0000e+00],
        [8.6378e-04, 9.9914e-01],
        [9.7749e-10, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2674e-05, 9.9999e-01],
        [1.4304e-04, 9.9986e-01],
        [5.2144e-05, 9.9995e-01],
        [9.9830e-01, 1.6987e-03],
        [2.7476e-09, 1.0000e+00],
        [3.1576e-05, 9.9997e-01],
        [1.0411e-08, 1.0000e+00],
        [9.8001e-07, 1.0000e+00],
        [6.8687e-08, 1.0000e+00],
        [5.3770e-01, 4.6230e-01],
        [1.6128e-07, 1.0000e+00],
        [3.6951e-09, 1.0000e+00],
        [9.9966e-01, 3.3745e-04],
        [9.9993e-01, 6.8024e-05],
        [1.2013e-07, 1.0000e+00],
        [9.1234e-07, 1.0000e+00],
        [9.9976e-01, 2.4283e-04],
        [9.9994e-01, 6.0179e-05],
        [8.8740e-01, 1.1260e-01],
        [1.9936e-08, 1.0000e+00],
        [1.2143e-07, 1.0000e+00],
        [2.3761e-09, 1.0000e+00],
        [9.2282e-06, 9.9999e-01],
        [2.3080e-07, 1.0000e+00],
        [2.6764e-05, 9.9997e-01],
        [2.0225e-09, 1.0000e+00],
        [4.9291e-06, 1.0000e+00],
        [2.2917e-06, 1.0000e+00],
        [9.9941e-01, 5.8829e-04],
        [9.9997e-01, 2.8974e-05],
        [3.3589e-11, 1.0000e+00],
        [9.9984e-01, 1.6105e-04],
        [9.9999e-01, 5.4808e-06],
        [2.1769e-07, 1.0000e+00],
        [8.7950e-10, 1.0000e+00],
        [1.3379e-06, 1.0000e+00],
        [9.9994e-01, 6.3484e-05],
        [6.3078e-05, 9.9994e-01],
        [1.0566e-03, 9.9894e-01],
        [9.9947e-01, 5.3379e-04],
        [9.9700e-01, 2.9974e-03],
        [2.1489e-05, 9.9998e-01],
        [1.2670e-01, 8.7330e-01],
        [8.2998e-08, 1.0000e+00],
        [1.9163e-02, 9.8084e-01],
        [2.9272e-06, 1.0000e+00],
        [6.9043e-01, 3.0957e-01],
        [1.8033e-04, 9.9982e-01],
        [1.1530e-10, 1.0000e+00],
        [3.2382e-06, 1.0000e+00],
        [5.8933e-07, 1.0000e+00],
        [1.7340e-07, 1.0000e+00],
        [1.7772e-06, 1.0000e+00],
        [9.9948e-01, 5.2351e-04],
        [1.8223e-07, 1.0000e+00],
        [8.7531e-08, 1.0000e+00],
        [9.9942e-01, 5.8418e-04],
        [9.9951e-01, 4.8604e-04],
        [1.3879e-02, 9.8612e-01],
        [2.9166e-04, 9.9971e-01],
        [2.6500e-05, 9.9997e-01],
        [7.3456e-06, 9.9999e-01],
        [2.4984e-06, 1.0000e+00],
        [9.9993e-01, 7.3036e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9937e-01, 6.3383e-04],
        [2.8898e-07, 1.0000e+00],
        [1.6769e-07, 1.0000e+00],
        [1.0539e-08, 1.0000e+00],
        [9.9987e-01, 1.3457e-04],
        [2.9989e-04, 9.9970e-01],
        [2.0768e-08, 1.0000e+00],
        [9.0445e-01, 9.5553e-02],
        [9.9962e-01, 3.8325e-04],
        [3.5884e-05, 9.9996e-01],
        [3.3838e-05, 9.9997e-01],
        [2.0877e-03, 9.9791e-01],
        [1.9922e-05, 9.9998e-01],
        [9.9944e-01, 5.6235e-04],
        [8.2521e-05, 9.9992e-01],
        [7.9614e-10, 1.0000e+00],
        [6.7147e-01, 3.2853e-01],
        [9.9008e-05, 9.9990e-01],
        [1.5018e-06, 1.0000e+00],
        [8.7121e-07, 1.0000e+00],
        [9.9994e-01, 6.1311e-05],
        [9.9985e-01, 1.5225e-04],
        [5.3707e-04, 9.9946e-01],
        [1.0352e-08, 1.0000e+00],
        [1.6102e-05, 9.9998e-01],
        [9.9998e-01, 1.5372e-05],
        [4.5753e-07, 1.0000e+00],
        [7.5362e-05, 9.9992e-01],
        [6.1066e-08, 1.0000e+00],
        [7.0795e-04, 9.9929e-01],
        [5.6894e-06, 9.9999e-01],
        [9.9959e-01, 4.0754e-04],
        [9.8235e-01, 1.7646e-02],
        [1.9210e-07, 1.0000e+00],
        [3.3801e-06, 1.0000e+00],
        [1.4215e-08, 1.0000e+00],
        [1.3424e-07, 1.0000e+00],
        [8.5290e-02, 9.1471e-01],
        [4.3767e-08, 1.0000e+00],
        [4.2984e-06, 1.0000e+00],
        [1.5515e-09, 1.0000e+00],
        [1.3581e-06, 1.0000e+00],
        [3.8925e-03, 9.9611e-01],
        [9.9829e-01, 1.7135e-03],
        [6.7779e-07, 1.0000e+00],
        [9.9952e-01, 4.7605e-04],
        [2.6133e-04, 9.9974e-01],
        [1.3570e-06, 1.0000e+00],
        [7.0820e-09, 1.0000e+00],
        [1.9451e-06, 1.0000e+00],
        [1.7983e-01, 8.2017e-01],
        [1.1172e-06, 1.0000e+00],
        [1.2442e-05, 9.9999e-01],
        [8.0297e-05, 9.9992e-01],
        [2.6204e-04, 9.9974e-01],
        [1.7493e-06, 1.0000e+00],
        [9.9980e-01, 2.0028e-04],
        [9.9904e-01, 9.5627e-04],
        [9.2543e-10, 1.0000e+00],
        [1.2853e-08, 1.0000e+00],
        [3.4447e-05, 9.9997e-01],
        [9.9752e-01, 2.4824e-03],
        [1.4654e-03, 9.9853e-01],
        [4.5300e-04, 9.9955e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.5319e-10, 1.0000e+00],
        [4.2317e-06, 1.0000e+00],
        [2.5056e-05, 9.9997e-01],
        [2.1446e-02, 9.7855e-01],
        [8.4500e-07, 1.0000e+00],
        [4.5504e-08, 1.0000e+00],
        [9.4001e-02, 9.0600e-01],
        [5.8129e-10, 1.0000e+00],
        [1.2836e-09, 1.0000e+00],
        [1.0288e-04, 9.9990e-01],
        [7.4605e-04, 9.9925e-01],
        [2.0195e-08, 1.0000e+00],
        [5.9390e-09, 1.0000e+00],
        [1.1495e-02, 9.8850e-01],
        [4.6695e-08, 1.0000e+00],
        [9.9984e-01, 1.6186e-04],
        [2.4101e-04, 9.9976e-01],
        [2.2189e-04, 9.9978e-01],
        [9.6312e-01, 3.6882e-02],
        [7.4423e-12, 1.0000e+00],
        [1.5698e-04, 9.9984e-01],
        [9.0435e-01, 9.5651e-02],
        [9.9998e-01, 2.2383e-05],
        [9.9987e-01, 1.3440e-04],
        [1.4614e-04, 9.9985e-01],
        [9.9939e-01, 6.1138e-04],
        [6.0932e-05, 9.9994e-01],
        [9.9942e-01, 5.7975e-04],
        [1.1687e-07, 1.0000e+00],
        [1.2343e-06, 1.0000e+00],
        [2.7506e-06, 1.0000e+00],
        [2.5636e-05, 9.9997e-01],
        [1.1021e-06, 1.0000e+00],
        [5.4704e-10, 1.0000e+00],
        [4.7819e-02, 9.5218e-01],
        [2.9008e-04, 9.9971e-01],
        [9.1859e-06, 9.9999e-01],
        [2.1104e-06, 1.0000e+00],
        [9.9694e-01, 3.0632e-03],
        [1.9231e-02, 9.8077e-01],
        [1.1481e-05, 9.9999e-01],
        [9.9994e-01, 5.5030e-05],
        [8.8911e-06, 9.9999e-01],
        [2.9357e-03, 9.9706e-01],
        [1.2149e-05, 9.9999e-01],
        [9.9917e-01, 8.2836e-04],
        [9.9983e-01, 1.7313e-04],
        [1.1077e-03, 9.9889e-01],
        [5.1963e-07, 1.0000e+00],
        [3.5699e-04, 9.9964e-01],
        [7.6712e-10, 1.0000e+00],
        [5.8221e-07, 1.0000e+00],
        [9.9857e-01, 1.4343e-03],
        [3.6565e-06, 1.0000e+00],
        [6.3660e-02, 9.3634e-01],
        [1.1554e-03, 9.9884e-01],
        [1.4132e-04, 9.9986e-01],
        [2.2334e-09, 1.0000e+00],
        [4.1166e-06, 1.0000e+00],
        [4.3204e-05, 9.9996e-01],
        [9.1058e-05, 9.9991e-01],
        [1.3485e-03, 9.9865e-01],
        [4.6447e-07, 1.0000e+00],
        [6.1151e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.9998e-08, 1.0000e+00],
        [2.4047e-06, 1.0000e+00],
        [1.5944e-08, 1.0000e+00],
        [9.7753e-01, 2.2472e-02],
        [8.3827e-01, 1.6173e-01],
        [5.6210e-05, 9.9994e-01],
        [7.2800e-04, 9.9927e-01],
        [5.1640e-07, 1.0000e+00],
        [9.9996e-01, 4.2505e-05],
        [4.1218e-09, 1.0000e+00],
        [3.1976e-06, 1.0000e+00],
        [8.6655e-07, 1.0000e+00],
        [9.9984e-01, 1.5870e-04],
        [4.2458e-03, 9.9575e-01],
        [7.6377e-10, 1.0000e+00],
        [2.0020e-08, 1.0000e+00],
        [1.3667e-09, 1.0000e+00],
        [5.7843e-03, 9.9422e-01],
        [1.2829e-05, 9.9999e-01],
        [9.9939e-01, 6.0906e-04],
        [2.7040e-03, 9.9730e-01],
        [3.2396e-07, 1.0000e+00],
        [1.0370e-03, 9.9896e-01],
        [9.7825e-01, 2.1751e-02],
        [9.9677e-01, 3.2331e-03],
        [9.8022e-01, 1.9782e-02],
        [2.3856e-07, 1.0000e+00],
        [1.0090e-05, 9.9999e-01],
        [9.9993e-01, 6.9906e-05],
        [2.8627e-07, 1.0000e+00],
        [4.8394e-10, 1.0000e+00],
        [9.9926e-01, 7.4310e-04],
        [6.9932e-04, 9.9930e-01],
        [1.3072e-03, 9.9869e-01],
        [8.8255e-06, 9.9999e-01],
        [3.7444e-04, 9.9963e-01],
        [9.9989e-01, 1.1239e-04],
        [1.3145e-04, 9.9987e-01],
        [1.3853e-06, 1.0000e+00],
        [7.7341e-07, 1.0000e+00],
        [9.6982e-03, 9.9030e-01],
        [1.9573e-10, 1.0000e+00],
        [3.5790e-05, 9.9996e-01],
        [3.9454e-07, 1.0000e+00],
        [3.5868e-05, 9.9996e-01],
        [3.4690e-06, 1.0000e+00],
        [2.2011e-06, 1.0000e+00],
        [8.2177e-05, 9.9992e-01],
        [9.9989e-01, 1.0653e-04],
        [2.4573e-04, 9.9975e-01],
        [4.6681e-06, 1.0000e+00],
        [6.6023e-04, 9.9934e-01],
        [9.8022e-01, 1.9782e-02],
        [2.2829e-06, 1.0000e+00],
        [7.1345e-05, 9.9993e-01],
        [7.0010e-06, 9.9999e-01],
        [6.1920e-10, 1.0000e+00],
        [9.9754e-01, 2.4630e-03],
        [9.8691e-01, 1.3094e-02],
        [2.9915e-04, 9.9970e-01],
        [1.4434e-07, 1.0000e+00],
        [9.9998e-01, 2.3014e-05],
        [8.6744e-04, 9.9913e-01],
        [2.4767e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1750e-08, 1.0000e+00],
        [1.5830e-08, 1.0000e+00],
        [9.8603e-04, 9.9901e-01],
        [3.3585e-06, 1.0000e+00],
        [5.1791e-05, 9.9995e-01],
        [5.4303e-08, 1.0000e+00],
        [8.3250e-07, 1.0000e+00],
        [9.9908e-01, 9.2148e-04],
        [8.1164e-01, 1.8836e-01],
        [2.4305e-06, 1.0000e+00],
        [9.9974e-01, 2.5992e-04],
        [4.8131e-09, 1.0000e+00],
        [9.8475e-06, 9.9999e-01],
        [9.9955e-01, 4.4834e-04],
        [6.1185e-06, 9.9999e-01],
        [3.9538e-09, 1.0000e+00],
        [1.1718e-05, 9.9999e-01],
        [1.0739e-03, 9.9893e-01],
        [9.9984e-01, 1.6302e-04],
        [1.2988e-03, 9.9870e-01],
        [1.7610e-05, 9.9998e-01],
        [2.5604e-05, 9.9997e-01],
        [3.1000e-05, 9.9997e-01],
        [6.6524e-04, 9.9933e-01],
        [1.2541e-04, 9.9987e-01],
        [9.9968e-01, 3.1909e-04],
        [2.4918e-06, 1.0000e+00],
        [9.9250e-01, 7.5031e-03],
        [9.9993e-01, 6.7373e-05],
        [8.4002e-07, 1.0000e+00],
        [1.6639e-06, 1.0000e+00],
        [2.0562e-03, 9.9794e-01],
        [1.6391e-07, 1.0000e+00],
        [1.5983e-08, 1.0000e+00],
        [3.4588e-06, 1.0000e+00],
        [6.6062e-04, 9.9934e-01],
        [9.9994e-01, 6.0019e-05],
        [4.7736e-04, 9.9952e-01],
        [1.0862e-07, 1.0000e+00],
        [9.9981e-01, 1.9306e-04],
        [3.9600e-03, 9.9604e-01],
        [1.5862e-07, 1.0000e+00],
        [1.2023e-08, 1.0000e+00],
        [6.9364e-09, 1.0000e+00],
        [1.2966e-05, 9.9999e-01],
        [1.6391e-07, 1.0000e+00],
        [3.0022e-04, 9.9970e-01],
        [3.9080e-07, 1.0000e+00],
        [9.9933e-01, 6.7288e-04],
        [9.9943e-01, 5.6730e-04],
        [8.6281e-01, 1.3719e-01],
        [3.0502e-07, 1.0000e+00],
        [9.9951e-01, 4.8883e-04],
        [1.2167e-05, 9.9999e-01],
        [6.2270e-07, 1.0000e+00],
        [1.1967e-05, 9.9999e-01],
        [1.0354e-05, 9.9999e-01],
        [1.4490e-06, 1.0000e+00],
        [1.1934e-06, 1.0000e+00],
        [3.8639e-05, 9.9996e-01],
        [6.3285e-09, 1.0000e+00],
        [9.8411e-01, 1.5886e-02],
        [8.7346e-05, 9.9991e-01],
        [9.9266e-01, 7.3386e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[4.8147e-10, 1.0000e+00],
        [2.8551e-04, 9.9971e-01],
        [1.8439e-04, 9.9982e-01],
        [3.5898e-05, 9.9996e-01],
        [9.9979e-01, 2.1170e-04],
        [4.9186e-06, 1.0000e+00],
        [1.6876e-03, 9.9831e-01],
        [4.4863e-06, 1.0000e+00],
        [1.4535e-06, 1.0000e+00],
        [2.0000e-06, 1.0000e+00],
        [9.9578e-01, 4.2152e-03],
        [4.7637e-04, 9.9952e-01],
        [5.8351e-08, 1.0000e+00],
        [1.4461e-09, 1.0000e+00],
        [9.9884e-01, 1.1569e-03],
        [9.9975e-01, 2.4500e-04],
        [9.1660e-09, 1.0000e+00],
        [9.9992e-01, 8.0398e-05],
        [9.9976e-01, 2.3804e-04],
        [2.0178e-01, 7.9822e-01],
        [5.1866e-04, 9.9948e-01],
        [2.0020e-06, 1.0000e+00],
        [9.9876e-01, 1.2362e-03],
        [9.9985e-01, 1.4653e-04],
        [7.1882e-01, 2.8118e-01],
        [1.4778e-03, 9.9852e-01],
        [9.9991e-01, 8.9353e-05],
        [3.0034e-06, 1.0000e+00],
        [1.9749e-06, 1.0000e+00],
        [1.2496e-05, 9.9999e-01],
        [2.0318e-06, 1.0000e+00],
        [9.9975e-01, 2.5464e-04],
        [5.6921e-07, 1.0000e+00],
        [1.7856e-07, 1.0000e+00],
        [7.4606e-01, 2.5394e-01],
        [4.1442e-05, 9.9996e-01],
        [2.3331e-07, 1.0000e+00],
        [9.9991e-01, 8.5543e-05],
        [4.8004e-04, 9.9952e-01],
        [6.4431e-09, 1.0000e+00],
        [3.0652e-08, 1.0000e+00],
        [7.2909e-06, 9.9999e-01],
        [9.9983e-01, 1.7032e-04],
        [1.9680e-07, 1.0000e+00],
        [1.1533e-01, 8.8467e-01],
        [4.1682e-02, 9.5832e-01],
        [2.7451e-02, 9.7255e-01],
        [6.5650e-04, 9.9934e-01],
        [3.9958e-05, 9.9996e-01],
        [9.9989e-01, 1.0690e-04],
        [3.0762e-09, 1.0000e+00],
        [9.9908e-01, 9.2211e-04],
        [2.1013e-07, 1.0000e+00],
        [9.1139e-07, 1.0000e+00],
        [3.4143e-08, 1.0000e+00],
        [1.7085e-06, 1.0000e+00],
        [9.9980e-01, 1.9640e-04],
        [5.0897e-06, 9.9999e-01],
        [3.1721e-08, 1.0000e+00],
        [1.4128e-09, 1.0000e+00],
        [9.9988e-01, 1.2192e-04],
        [9.9516e-01, 4.8422e-03],
        [5.2499e-07, 1.0000e+00],
        [3.3517e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.5830e-10, 1.0000e+00],
        [4.9042e-08, 1.0000e+00],
        [1.7008e-05, 9.9998e-01],
        [1.1521e-05, 9.9999e-01],
        [3.0806e-09, 1.0000e+00],
        [6.5519e-05, 9.9993e-01],
        [4.9178e-10, 1.0000e+00],
        [1.0000e+00, 4.8670e-06],
        [5.5991e-09, 1.0000e+00],
        [9.8949e-05, 9.9990e-01],
        [2.3392e-07, 1.0000e+00],
        [9.9983e-01, 1.6892e-04],
        [9.9984e-01, 1.5528e-04],
        [5.9839e-10, 1.0000e+00],
        [5.3951e-06, 9.9999e-01],
        [3.4876e-03, 9.9651e-01],
        [9.9967e-01, 3.3155e-04],
        [2.0701e-11, 1.0000e+00],
        [1.0291e-10, 1.0000e+00],
        [6.1708e-04, 9.9938e-01],
        [1.3295e-03, 9.9867e-01],
        [1.5400e-05, 9.9998e-01],
        [3.5740e-08, 1.0000e+00],
        [2.5491e-05, 9.9997e-01],
        [1.7369e-04, 9.9983e-01],
        [2.7255e-06, 1.0000e+00],
        [4.8564e-07, 1.0000e+00],
        [1.6126e-03, 9.9839e-01],
        [1.6918e-09, 1.0000e+00],
        [1.5570e-01, 8.4430e-01],
        [3.8636e-05, 9.9996e-01],
        [3.3966e-03, 9.9660e-01],
        [9.9987e-01, 1.2535e-04],
        [4.3983e-08, 1.0000e+00],
        [5.8593e-04, 9.9941e-01],
        [2.5497e-07, 1.0000e+00],
        [4.2735e-08, 1.0000e+00],
        [7.9334e-11, 1.0000e+00],
        [2.7308e-08, 1.0000e+00],
        [1.5381e-04, 9.9985e-01],
        [2.7472e-07, 1.0000e+00],
        [9.9986e-01, 1.4000e-04],
        [7.8846e-04, 9.9921e-01],
        [2.8548e-01, 7.1452e-01],
        [3.5007e-07, 1.0000e+00],
        [1.1166e-01, 8.8834e-01],
        [2.8419e-06, 1.0000e+00],
        [1.4713e-05, 9.9999e-01],
        [4.2957e-08, 1.0000e+00],
        [7.5342e-05, 9.9992e-01],
        [2.0843e-10, 1.0000e+00],
        [3.8098e-07, 1.0000e+00],
        [1.2671e-06, 1.0000e+00],
        [2.4175e-04, 9.9976e-01],
        [5.0598e-06, 9.9999e-01],
        [8.9062e-09, 1.0000e+00],
        [8.0205e-06, 9.9999e-01],
        [9.9971e-01, 2.9282e-04],
        [1.0982e-07, 1.0000e+00],
        [9.0856e-05, 9.9991e-01],
        [1.1292e-05, 9.9999e-01],
        [2.9578e-03, 9.9704e-01],
        [9.9989e-01, 1.1248e-04],
        [9.9893e-01, 1.0674e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[5.4820e-05, 9.9995e-01],
        [4.0860e-07, 1.0000e+00],
        [9.8770e-01, 1.2302e-02],
        [2.3779e-05, 9.9998e-01],
        [9.9999e-01, 8.9319e-06],
        [7.1937e-06, 9.9999e-01],
        [4.6982e-06, 1.0000e+00],
        [2.4159e-05, 9.9998e-01],
        [5.2540e-04, 9.9947e-01],
        [1.4583e-06, 1.0000e+00],
        [9.8026e-09, 1.0000e+00],
        [3.8773e-05, 9.9996e-01],
        [1.5863e-04, 9.9984e-01],
        [9.9975e-01, 2.5473e-04],
        [9.9990e-01, 9.9072e-05],
        [2.8000e-06, 1.0000e+00],
        [3.8532e-07, 1.0000e+00],
        [8.8062e-05, 9.9991e-01],
        [6.2714e-03, 9.9373e-01],
        [1.6200e-05, 9.9998e-01],
        [9.9973e-01, 2.6642e-04],
        [9.8384e-01, 1.6160e-02],
        [1.0357e-03, 9.9896e-01],
        [8.7614e-08, 1.0000e+00],
        [4.5640e-05, 9.9995e-01],
        [1.4364e-10, 1.0000e+00],
        [2.3584e-06, 1.0000e+00],
        [6.3540e-05, 9.9994e-01],
        [9.9813e-01, 1.8725e-03],
        [7.0075e-03, 9.9299e-01],
        [1.4515e-08, 1.0000e+00],
        [2.0954e-04, 9.9979e-01],
        [9.7208e-11, 1.0000e+00],
        [2.0158e-08, 1.0000e+00],
        [7.6670e-07, 1.0000e+00],
        [9.9987e-01, 1.3325e-04],
        [4.6883e-09, 1.0000e+00],
        [9.4777e-06, 9.9999e-01],
        [1.6268e-06, 1.0000e+00],
        [2.6850e-09, 1.0000e+00],
        [9.7191e-07, 1.0000e+00],
        [9.9018e-01, 9.8195e-03],
        [8.6853e-06, 9.9999e-01],
        [2.6571e-09, 1.0000e+00],
        [3.8312e-02, 9.6169e-01],
        [2.0767e-04, 9.9979e-01],
        [9.9964e-01, 3.5528e-04],
        [2.5819e-06, 1.0000e+00],
        [9.9935e-01, 6.4814e-04],
        [9.9998e-01, 1.7110e-05],
        [9.9773e-01, 2.2651e-03],
        [8.4362e-10, 1.0000e+00],
        [4.1699e-06, 1.0000e+00],
        [3.7046e-05, 9.9996e-01],
        [9.9985e-01, 1.5032e-04],
        [9.9988e-01, 1.1881e-04],
        [9.5899e-01, 4.1012e-02],
        [1.0968e-06, 1.0000e+00],
        [9.9986e-01, 1.3790e-04],
        [1.4544e-07, 1.0000e+00],
        [4.1572e-04, 9.9958e-01],
        [5.8549e-02, 9.4145e-01],
        [3.4875e-08, 1.0000e+00],
        [5.4742e-04, 9.9945e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3455e-05, 9.9999e-01],
        [1.9528e-03, 9.9805e-01],
        [1.2712e-04, 9.9987e-01],
        [1.6365e-05, 9.9998e-01],
        [1.9099e-02, 9.8090e-01],
        [1.5835e-06, 1.0000e+00],
        [4.3948e-05, 9.9996e-01],
        [9.9924e-01, 7.6208e-04],
        [1.7486e-04, 9.9983e-01],
        [4.8252e-06, 1.0000e+00],
        [2.6055e-02, 9.7394e-01],
        [1.3528e-08, 1.0000e+00],
        [5.3968e-10, 1.0000e+00],
        [3.4757e-10, 1.0000e+00],
        [7.8637e-08, 1.0000e+00],
        [5.2851e-04, 9.9947e-01],
        [3.5353e-06, 1.0000e+00],
        [1.4777e-07, 1.0000e+00],
        [1.0024e-04, 9.9990e-01],
        [6.7668e-08, 1.0000e+00],
        [2.9633e-05, 9.9997e-01],
        [1.0214e-06, 1.0000e+00],
        [2.0998e-05, 9.9998e-01],
        [2.2515e-02, 9.7748e-01],
        [1.1753e-04, 9.9988e-01],
        [3.1478e-06, 1.0000e+00],
        [9.6964e-01, 3.0359e-02],
        [9.9977e-01, 2.2750e-04],
        [1.7618e-01, 8.2382e-01],
        [1.2210e-07, 1.0000e+00],
        [2.1920e-09, 1.0000e+00],
        [2.6122e-03, 9.9739e-01],
        [6.0119e-06, 9.9999e-01],
        [9.1988e-04, 9.9908e-01],
        [9.2324e-04, 9.9908e-01],
        [9.9994e-01, 5.9116e-05],
        [9.8216e-05, 9.9990e-01],
        [1.4128e-04, 9.9986e-01],
        [1.7574e-05, 9.9998e-01],
        [1.8350e-02, 9.8165e-01],
        [6.9538e-05, 9.9993e-01],
        [6.5414e-05, 9.9993e-01],
        [9.9995e-01, 4.7567e-05],
        [6.0623e-06, 9.9999e-01],
        [8.3720e-07, 1.0000e+00],
        [5.0884e-04, 9.9949e-01],
        [1.3659e-04, 9.9986e-01],
        [1.7216e-06, 1.0000e+00],
        [9.9982e-01, 1.7917e-04],
        [9.9981e-01, 1.8967e-04],
        [4.6688e-05, 9.9995e-01],
        [9.9894e-01, 1.0574e-03],
        [2.0510e-06, 1.0000e+00],
        [4.6674e-02, 9.5333e-01],
        [1.8947e-06, 1.0000e+00],
        [5.7803e-05, 9.9994e-01],
        [1.4327e-05, 9.9999e-01],
        [2.1167e-09, 1.0000e+00],
        [3.3967e-06, 1.0000e+00],
        [9.3290e-09, 1.0000e+00],
        [2.2586e-06, 1.0000e+00],
        [1.1309e-04, 9.9989e-01],
        [9.9329e-01, 6.7054e-03],
        [1.4431e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0212e-03, 9.9798e-01],
        [9.9983e-01, 1.6880e-04],
        [8.9592e-03, 9.9104e-01],
        [8.6079e-05, 9.9991e-01],
        [1.0055e-03, 9.9899e-01],
        [1.8456e-04, 9.9982e-01],
        [1.6091e-04, 9.9984e-01],
        [9.9998e-01, 2.1825e-05],
        [1.0594e-03, 9.9894e-01],
        [2.3085e-07, 1.0000e+00],
        [3.7415e-05, 9.9996e-01],
        [9.9929e-01, 7.1102e-04],
        [2.1061e-08, 1.0000e+00],
        [9.9982e-01, 1.7732e-04],
        [2.2148e-08, 1.0000e+00],
        [7.6759e-09, 1.0000e+00],
        [6.1637e-09, 1.0000e+00],
        [2.4158e-03, 9.9758e-01],
        [9.2404e-01, 7.5958e-02],
        [1.1940e-06, 1.0000e+00],
        [8.0399e-07, 1.0000e+00],
        [9.9994e-01, 6.1449e-05],
        [9.0205e-09, 1.0000e+00],
        [9.9976e-01, 2.3567e-04],
        [9.9984e-01, 1.6181e-04],
        [1.0024e-01, 8.9976e-01],
        [9.9734e-05, 9.9990e-01],
        [9.2211e-01, 7.7887e-02],
        [1.4467e-04, 9.9986e-01],
        [3.4232e-05, 9.9997e-01],
        [1.0901e-07, 1.0000e+00],
        [8.9517e-05, 9.9991e-01],
        [8.6482e-01, 1.3518e-01],
        [5.9507e-04, 9.9940e-01],
        [4.0180e-03, 9.9598e-01],
        [2.0485e-09, 1.0000e+00],
        [9.9981e-01, 1.8780e-04],
        [5.0996e-04, 9.9949e-01],
        [9.9983e-01, 1.7097e-04],
        [9.9993e-01, 6.6869e-05],
        [1.7973e-08, 1.0000e+00],
        [8.5459e-07, 1.0000e+00],
        [1.1328e-07, 1.0000e+00],
        [1.9369e-08, 1.0000e+00],
        [8.7172e-09, 1.0000e+00],
        [9.9991e-01, 8.8455e-05],
        [7.9736e-04, 9.9920e-01],
        [2.5162e-06, 1.0000e+00],
        [1.3051e-04, 9.9987e-01],
        [2.5040e-06, 1.0000e+00],
        [8.7420e-04, 9.9913e-01],
        [1.4075e-07, 1.0000e+00],
        [9.9968e-01, 3.1848e-04],
        [1.0966e-06, 1.0000e+00],
        [2.2489e-01, 7.7511e-01],
        [9.9959e-01, 4.1340e-04],
        [9.0182e-08, 1.0000e+00],
        [5.0711e-05, 9.9995e-01],
        [3.5796e-06, 1.0000e+00],
        [8.7015e-01, 1.2985e-01],
        [7.2670e-08, 1.0000e+00],
        [9.9998e-01, 1.9634e-05],
        [9.9980e-01, 2.0113e-04],
        [9.9989e-01, 1.1151e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4611e-06, 1.0000e+00],
        [4.9397e-07, 1.0000e+00],
        [5.5861e-05, 9.9994e-01],
        [4.4600e-04, 9.9955e-01],
        [9.9489e-01, 5.1100e-03],
        [6.4650e-06, 9.9999e-01],
        [6.9826e-06, 9.9999e-01],
        [6.0011e-05, 9.9994e-01],
        [4.4818e-08, 1.0000e+00],
        [1.3204e-02, 9.8680e-01],
        [5.3843e-04, 9.9946e-01],
        [2.3130e-08, 1.0000e+00],
        [9.9970e-01, 3.0170e-04],
        [9.9999e-01, 9.9125e-06],
        [1.5296e-08, 1.0000e+00],
        [1.2198e-03, 9.9878e-01],
        [9.8350e-01, 1.6498e-02],
        [6.9461e-06, 9.9999e-01],
        [9.9973e-01, 2.6894e-04],
        [6.7732e-10, 1.0000e+00],
        [4.2470e-07, 1.0000e+00],
        [9.9990e-01, 1.0054e-04],
        [3.7195e-06, 1.0000e+00],
        [3.1694e-03, 9.9683e-01],
        [5.6114e-07, 1.0000e+00],
        [7.6306e-05, 9.9992e-01],
        [1.5055e-08, 1.0000e+00],
        [3.0634e-07, 1.0000e+00],
        [9.2497e-06, 9.9999e-01],
        [1.9868e-05, 9.9998e-01],
        [2.5427e-04, 9.9975e-01],
        [1.9885e-02, 9.8012e-01],
        [9.9149e-01, 8.5055e-03],
        [3.2296e-04, 9.9968e-01],
        [9.4632e-04, 9.9905e-01],
        [6.6073e-07, 1.0000e+00],
        [9.9992e-01, 7.8020e-05],
        [1.0894e-01, 8.9106e-01],
        [1.3353e-05, 9.9999e-01],
        [8.1286e-06, 9.9999e-01],
        [5.0975e-07, 1.0000e+00],
        [1.1906e-03, 9.9881e-01],
        [1.9404e-08, 1.0000e+00],
        [9.9989e-01, 1.1022e-04],
        [9.8906e-01, 1.0942e-02],
        [9.8185e-01, 1.8148e-02],
        [9.9978e-01, 2.1984e-04],
        [2.0082e-05, 9.9998e-01],
        [9.9990e-01, 9.5601e-05],
        [9.9928e-01, 7.1614e-04],
        [5.2661e-06, 9.9999e-01],
        [1.7728e-04, 9.9982e-01],
        [8.0426e-02, 9.1957e-01],
        [5.2743e-06, 9.9999e-01],
        [2.5867e-04, 9.9974e-01],
        [9.4632e-04, 9.9905e-01],
        [9.9933e-01, 6.6976e-04],
        [6.4505e-03, 9.9355e-01],
        [2.1488e-02, 9.7851e-01],
        [3.4585e-10, 1.0000e+00],
        [2.3622e-05, 9.9998e-01],
        [1.4015e-05, 9.9999e-01],
        [9.9989e-01, 1.1329e-04],
        [8.1788e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9983e-01, 1.6715e-04],
        [5.0718e-04, 9.9949e-01],
        [3.8867e-10, 1.0000e+00],
        [1.0086e-07, 1.0000e+00],
        [4.8524e-06, 1.0000e+00],
        [2.5668e-10, 1.0000e+00],
        [9.9991e-01, 8.5408e-05],
        [9.9996e-01, 4.4261e-05],
        [8.3650e-11, 1.0000e+00],
        [9.9991e-01, 9.3727e-05],
        [4.5973e-03, 9.9540e-01],
        [2.4252e-07, 1.0000e+00],
        [3.6403e-05, 9.9996e-01],
        [1.3368e-07, 1.0000e+00],
        [9.9905e-01, 9.4990e-04],
        [2.0239e-07, 1.0000e+00],
        [3.0425e-07, 1.0000e+00],
        [1.6400e-05, 9.9998e-01],
        [1.4997e-02, 9.8500e-01],
        [9.9989e-01, 1.1380e-04],
        [1.7366e-06, 1.0000e+00],
        [9.9989e-01, 1.0647e-04],
        [6.9504e-06, 9.9999e-01],
        [8.0637e-04, 9.9919e-01],
        [6.0990e-08, 1.0000e+00],
        [3.1865e-09, 1.0000e+00],
        [4.8613e-08, 1.0000e+00],
        [4.8293e-08, 1.0000e+00],
        [2.8166e-04, 9.9972e-01],
        [9.9777e-01, 2.2346e-03],
        [3.9614e-07, 1.0000e+00],
        [6.4525e-07, 1.0000e+00],
        [2.7431e-06, 1.0000e+00],
        [9.9993e-01, 6.8581e-05],
        [1.2303e-03, 9.9877e-01],
        [9.9976e-01, 2.4221e-04],
        [1.6175e-08, 1.0000e+00],
        [6.2137e-07, 1.0000e+00],
        [1.3869e-02, 9.8613e-01],
        [1.1546e-02, 9.8845e-01],
        [9.9922e-01, 7.8241e-04],
        [9.0063e-08, 1.0000e+00],
        [9.9949e-01, 5.0689e-04],
        [4.4440e-06, 1.0000e+00],
        [1.2096e-06, 1.0000e+00],
        [1.0008e-06, 1.0000e+00],
        [9.7557e-01, 2.4426e-02],
        [9.9978e-01, 2.2382e-04],
        [9.9964e-01, 3.5691e-04],
        [3.6387e-09, 1.0000e+00],
        [4.0533e-10, 1.0000e+00],
        [5.1182e-02, 9.4882e-01],
        [9.9994e-01, 6.0163e-05],
        [1.6039e-02, 9.8396e-01],
        [4.3485e-04, 9.9957e-01],
        [5.6956e-05, 9.9994e-01],
        [1.2910e-07, 1.0000e+00],
        [9.9996e-01, 3.8432e-05],
        [1.1191e-05, 9.9999e-01],
        [2.1689e-04, 9.9978e-01],
        [6.0259e-07, 1.0000e+00],
        [9.9980e-01, 2.0169e-04],
        [1.7041e-07, 1.0000e+00],
        [9.9979e-01, 2.0857e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1326e-04, 9.9989e-01],
        [9.9979e-01, 2.1136e-04],
        [9.9996e-01, 4.3138e-05],
        [9.9931e-01, 6.8925e-04],
        [3.2614e-04, 9.9967e-01],
        [9.2210e-01, 7.7904e-02],
        [9.9995e-01, 5.1431e-05],
        [7.8824e-01, 2.1176e-01],
        [1.0000e+00, 3.6752e-06],
        [6.3134e-06, 9.9999e-01],
        [9.9986e-01, 1.4142e-04],
        [9.9987e-01, 1.2993e-04],
        [5.4748e-04, 9.9945e-01],
        [9.9991e-01, 8.6949e-05],
        [2.4731e-06, 1.0000e+00],
        [7.7078e-01, 2.2922e-01],
        [9.9974e-01, 2.5822e-04],
        [9.9972e-01, 2.8129e-04],
        [4.9149e-05, 9.9995e-01],
        [1.8625e-05, 9.9998e-01],
        [9.9982e-01, 1.8031e-04],
        [4.2756e-10, 1.0000e+00],
        [3.4383e-06, 1.0000e+00],
        [9.9993e-01, 6.8690e-05],
        [1.9069e-07, 1.0000e+00],
        [3.0021e-05, 9.9997e-01],
        [3.7308e-05, 9.9996e-01],
        [5.6744e-04, 9.9943e-01],
        [1.6664e-04, 9.9983e-01],
        [6.7625e-07, 1.0000e+00],
        [9.9995e-01, 4.5473e-05],
        [9.5431e-07, 1.0000e+00],
        [9.9928e-01, 7.2489e-04],
        [9.9563e-01, 4.3692e-03],
        [1.9826e-09, 1.0000e+00],
        [9.7900e-01, 2.1002e-02],
        [9.5214e-04, 9.9905e-01],
        [1.0152e-04, 9.9990e-01],
        [9.9995e-01, 5.4227e-05],
        [2.3870e-06, 1.0000e+00],
        [1.5451e-03, 9.9845e-01],
        [8.9610e-01, 1.0390e-01],
        [2.2363e-06, 1.0000e+00],
        [1.0867e-01, 8.9133e-01],
        [5.4241e-08, 1.0000e+00],
        [1.2716e-01, 8.7284e-01],
        [3.4146e-09, 1.0000e+00],
        [4.9585e-06, 9.9999e-01],
        [4.1833e-03, 9.9582e-01],
        [6.9583e-09, 1.0000e+00],
        [8.7925e-04, 9.9912e-01],
        [9.9996e-01, 3.9894e-05],
        [9.9973e-01, 2.7110e-04],
        [9.9998e-01, 2.3419e-05],
        [9.9963e-01, 3.6640e-04],
        [2.8634e-06, 1.0000e+00],
        [9.9977e-01, 2.3314e-04],
        [9.9984e-01, 1.5575e-04],
        [9.9957e-01, 4.3176e-04],
        [9.9983e-01, 1.6908e-04],
        [1.4267e-06, 1.0000e+00],
        [5.7056e-06, 9.9999e-01],
        [1.8652e-03, 9.9813e-01],
        [3.5326e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9994e-01, 6.4820e-05],
        [1.6350e-04, 9.9984e-01],
        [3.5845e-09, 1.0000e+00],
        [1.3387e-04, 9.9987e-01],
        [1.1590e-06, 1.0000e+00],
        [9.9939e-01, 6.1176e-04],
        [9.9967e-01, 3.2855e-04],
        [6.5876e-04, 9.9934e-01],
        [2.9302e-08, 1.0000e+00],
        [1.2593e-09, 1.0000e+00],
        [9.9933e-01, 6.6747e-04],
        [1.5635e-08, 1.0000e+00],
        [9.9973e-01, 2.6599e-04],
        [5.1943e-05, 9.9995e-01],
        [1.3616e-06, 1.0000e+00],
        [4.3901e-04, 9.9956e-01],
        [1.7263e-04, 9.9983e-01],
        [1.0532e-08, 1.0000e+00],
        [8.6274e-05, 9.9991e-01],
        [8.7172e-05, 9.9991e-01],
        [1.8687e-05, 9.9998e-01],
        [1.5785e-06, 1.0000e+00],
        [6.5249e-05, 9.9993e-01],
        [9.9857e-01, 1.4288e-03],
        [9.9972e-01, 2.8284e-04],
        [6.0535e-06, 9.9999e-01],
        [2.9987e-06, 1.0000e+00],
        [3.9889e-02, 9.6011e-01],
        [7.3655e-11, 1.0000e+00],
        [5.8390e-08, 1.0000e+00],
        [3.9772e-07, 1.0000e+00],
        [9.9987e-01, 1.2719e-04],
        [3.0684e-02, 9.6932e-01],
        [9.9996e-01, 4.4644e-05],
        [1.3124e-04, 9.9987e-01],
        [4.2205e-08, 1.0000e+00],
        [9.9989e-01, 1.0707e-04],
        [9.9316e-01, 6.8360e-03],
        [1.1938e-04, 9.9988e-01],
        [2.8647e-01, 7.1353e-01],
        [9.3219e-04, 9.9907e-01],
        [2.7472e-04, 9.9973e-01],
        [1.8912e-05, 9.9998e-01],
        [9.5863e-02, 9.0414e-01],
        [2.2734e-08, 1.0000e+00],
        [3.4894e-02, 9.6511e-01],
        [9.9989e-01, 1.0712e-04],
        [9.9845e-01, 1.5497e-03],
        [6.8730e-06, 9.9999e-01],
        [3.0851e-09, 1.0000e+00],
        [7.3142e-01, 2.6858e-01],
        [3.8459e-10, 1.0000e+00],
        [9.9988e-01, 1.1862e-04],
        [1.6834e-06, 1.0000e+00],
        [9.1913e-09, 1.0000e+00],
        [5.4483e-08, 1.0000e+00],
        [9.9635e-01, 3.6525e-03],
        [9.9772e-01, 2.2798e-03],
        [1.1608e-02, 9.8839e-01],
        [9.9993e-01, 7.3956e-05],
        [5.3828e-02, 9.4617e-01],
        [7.9990e-01, 2.0010e-01],
        [2.2629e-09, 1.0000e+00],
        [1.4128e-02, 9.8587e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0953e-07, 1.0000e+00],
        [9.9959e-01, 4.1480e-04],
        [3.2277e-07, 1.0000e+00],
        [8.5324e-04, 9.9915e-01],
        [3.2090e-04, 9.9968e-01],
        [9.9985e-01, 1.5106e-04],
        [2.2105e-07, 1.0000e+00],
        [3.0676e-06, 1.0000e+00],
        [4.1888e-05, 9.9996e-01],
        [2.3844e-05, 9.9998e-01],
        [1.4008e-04, 9.9986e-01],
        [9.9926e-01, 7.4204e-04],
        [4.4177e-03, 9.9558e-01],
        [9.9558e-07, 1.0000e+00],
        [9.9972e-01, 2.7522e-04],
        [4.3852e-04, 9.9956e-01],
        [9.9988e-01, 1.2211e-04],
        [9.9981e-01, 1.9336e-04],
        [9.9971e-01, 2.8602e-04],
        [9.9670e-01, 3.2991e-03],
        [8.9853e-01, 1.0147e-01],
        [2.1177e-05, 9.9998e-01],
        [4.1325e-06, 1.0000e+00],
        [9.8483e-05, 9.9990e-01],
        [9.8187e-01, 1.8126e-02],
        [9.9848e-01, 1.5180e-03],
        [8.9022e-08, 1.0000e+00],
        [4.7736e-02, 9.5226e-01],
        [9.9983e-01, 1.6623e-04],
        [1.3640e-05, 9.9999e-01],
        [1.2161e-02, 9.8784e-01],
        [2.1850e-06, 1.0000e+00],
        [7.2999e-03, 9.9270e-01],
        [9.9999e-01, 8.2368e-06],
        [7.1664e-07, 1.0000e+00],
        [1.5139e-03, 9.9849e-01],
        [2.8918e-06, 1.0000e+00],
        [1.8424e-04, 9.9982e-01],
        [6.2701e-03, 9.9373e-01],
        [6.6793e-01, 3.3207e-01],
        [3.2210e-06, 1.0000e+00],
        [2.9993e-08, 1.0000e+00],
        [6.1747e-01, 3.8253e-01],
        [9.9969e-01, 3.0922e-04],
        [9.4629e-01, 5.3705e-02],
        [4.7566e-03, 9.9524e-01],
        [4.0956e-02, 9.5904e-01],
        [5.0447e-07, 1.0000e+00],
        [9.9993e-01, 6.7286e-05],
        [6.4956e-04, 9.9935e-01],
        [4.9877e-05, 9.9995e-01],
        [1.9095e-06, 1.0000e+00],
        [9.9993e-01, 6.5151e-05],
        [9.9981e-01, 1.8719e-04],
        [9.9992e-01, 8.2755e-05],
        [8.0785e-06, 9.9999e-01],
        [8.1513e-01, 1.8487e-01],
        [1.2649e-04, 9.9987e-01],
        [1.0202e-06, 1.0000e+00],
        [2.1095e-08, 1.0000e+00],
        [9.9993e-01, 7.1863e-05],
        [2.3216e-05, 9.9998e-01],
        [6.4078e-06, 9.9999e-01],
        [9.9995e-01, 4.5212e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9597e-01, 4.0256e-03],
        [9.5518e-01, 4.4816e-02],
        [1.1068e-06, 1.0000e+00],
        [1.2296e-04, 9.9988e-01],
        [2.8318e-05, 9.9997e-01],
        [6.9361e-04, 9.9931e-01],
        [1.6480e-09, 1.0000e+00],
        [9.9984e-01, 1.6201e-04],
        [2.8319e-06, 1.0000e+00],
        [1.8715e-04, 9.9981e-01],
        [1.1171e-03, 9.9888e-01],
        [9.9077e-01, 9.2311e-03],
        [1.2158e-08, 1.0000e+00],
        [7.3422e-05, 9.9993e-01],
        [1.3075e-05, 9.9999e-01],
        [6.7050e-04, 9.9933e-01],
        [5.3748e-04, 9.9946e-01],
        [9.9996e-01, 3.5751e-05],
        [1.9732e-03, 9.9803e-01],
        [2.2970e-05, 9.9998e-01],
        [9.9966e-01, 3.4412e-04],
        [3.4394e-07, 1.0000e+00],
        [6.4909e-05, 9.9994e-01],
        [7.0411e-09, 1.0000e+00],
        [9.7246e-01, 2.7540e-02],
        [7.4381e-07, 1.0000e+00],
        [1.6627e-01, 8.3373e-01],
        [1.7707e-06, 1.0000e+00],
        [9.9972e-01, 2.7980e-04],
        [9.9975e-01, 2.4982e-04],
        [2.4332e-02, 9.7567e-01],
        [1.3722e-07, 1.0000e+00],
        [2.2765e-05, 9.9998e-01],
        [1.5842e-01, 8.4158e-01],
        [9.5303e-06, 9.9999e-01],
        [3.5941e-08, 1.0000e+00],
        [9.9907e-01, 9.2992e-04],
        [1.7728e-01, 8.2272e-01],
        [1.0570e-08, 1.0000e+00],
        [3.6197e-07, 1.0000e+00],
        [3.3596e-04, 9.9966e-01],
        [1.5468e-06, 1.0000e+00],
        [1.3148e-05, 9.9999e-01],
        [1.3142e-05, 9.9999e-01],
        [4.8535e-04, 9.9951e-01],
        [4.3908e-08, 1.0000e+00],
        [1.0332e-04, 9.9990e-01],
        [3.0742e-08, 1.0000e+00],
        [9.9993e-01, 7.0389e-05],
        [4.4955e-08, 1.0000e+00],
        [3.2944e-05, 9.9997e-01],
        [2.5457e-08, 1.0000e+00],
        [2.5829e-05, 9.9997e-01],
        [4.2724e-05, 9.9996e-01],
        [1.2685e-06, 1.0000e+00],
        [7.7792e-04, 9.9922e-01],
        [1.6959e-05, 9.9998e-01],
        [2.2765e-05, 9.9998e-01],
        [9.3301e-06, 9.9999e-01],
        [9.9931e-01, 6.9204e-04],
        [1.1530e-04, 9.9988e-01],
        [9.8267e-01, 1.7329e-02],
        [2.6540e-03, 9.9735e-01],
        [2.0247e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7219e-03, 9.9828e-01],
        [9.9989e-01, 1.0676e-04],
        [9.8792e-09, 1.0000e+00],
        [9.9896e-01, 1.0440e-03],
        [2.7225e-02, 9.7278e-01],
        [2.9237e-05, 9.9997e-01],
        [9.6259e-07, 1.0000e+00],
        [9.9983e-01, 1.6950e-04],
        [4.4362e-06, 1.0000e+00],
        [9.9894e-01, 1.0569e-03],
        [9.9979e-01, 2.1118e-04],
        [1.7921e-06, 1.0000e+00],
        [9.9970e-01, 3.0424e-04],
        [2.2636e-08, 1.0000e+00],
        [8.3374e-07, 1.0000e+00],
        [9.9992e-01, 7.5463e-05],
        [9.3261e-05, 9.9991e-01],
        [6.7599e-06, 9.9999e-01],
        [9.9993e-01, 7.2591e-05],
        [1.0000e+00, 1.1342e-06],
        [1.4683e-05, 9.9999e-01],
        [1.1295e-05, 9.9999e-01],
        [9.9055e-01, 9.4532e-03],
        [3.2962e-06, 1.0000e+00],
        [4.9829e-03, 9.9502e-01],
        [4.6232e-02, 9.5377e-01],
        [2.5436e-07, 1.0000e+00],
        [3.7527e-05, 9.9996e-01],
        [2.1404e-02, 9.7860e-01],
        [3.6764e-02, 9.6324e-01],
        [2.2611e-05, 9.9998e-01],
        [7.8318e-05, 9.9992e-01],
        [5.8026e-02, 9.4197e-01],
        [1.0111e-08, 1.0000e+00],
        [1.3267e-06, 1.0000e+00],
        [9.9999e-01, 9.3551e-06],
        [4.1762e-09, 1.0000e+00],
        [7.4714e-07, 1.0000e+00],
        [9.9997e-01, 3.4192e-05],
        [2.2089e-02, 9.7791e-01],
        [2.3475e-07, 1.0000e+00],
        [9.8391e-01, 1.6085e-02],
        [3.6483e-05, 9.9996e-01],
        [2.1499e-07, 1.0000e+00],
        [4.8303e-02, 9.5170e-01],
        [6.7382e-04, 9.9933e-01],
        [9.5505e-01, 4.4952e-02],
        [2.6321e-04, 9.9974e-01],
        [4.8411e-03, 9.9516e-01],
        [1.7093e-06, 1.0000e+00],
        [3.5505e-07, 1.0000e+00],
        [9.9983e-01, 1.6848e-04],
        [9.9993e-01, 7.1637e-05],
        [1.3988e-03, 9.9860e-01],
        [9.9980e-01, 1.9597e-04],
        [3.0989e-06, 1.0000e+00],
        [9.9969e-01, 3.0973e-04],
        [1.8936e-05, 9.9998e-01],
        [1.2535e-07, 1.0000e+00],
        [2.7613e-02, 9.7239e-01],
        [2.1034e-06, 1.0000e+00],
        [9.9984e-01, 1.6086e-04],
        [1.1273e-05, 9.9999e-01],
        [5.3019e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9990e-01, 1.0395e-04],
        [9.7697e-08, 1.0000e+00],
        [2.3947e-02, 9.7605e-01],
        [3.5438e-07, 1.0000e+00],
        [6.3057e-09, 1.0000e+00],
        [1.8557e-04, 9.9981e-01],
        [2.8694e-09, 1.0000e+00],
        [5.5073e-04, 9.9945e-01],
        [2.9654e-04, 9.9970e-01],
        [7.5040e-06, 9.9999e-01],
        [2.0430e-05, 9.9998e-01],
        [1.0397e-03, 9.9896e-01],
        [1.6090e-02, 9.8391e-01],
        [5.1785e-01, 4.8215e-01],
        [1.8980e-05, 9.9998e-01],
        [3.0356e-06, 1.0000e+00],
        [1.2520e-06, 1.0000e+00],
        [2.8564e-01, 7.1436e-01],
        [9.9988e-01, 1.1814e-04],
        [8.5404e-04, 9.9915e-01],
        [2.8302e-02, 9.7170e-01],
        [1.7460e-05, 9.9998e-01],
        [2.5395e-04, 9.9975e-01],
        [9.9954e-01, 4.5915e-04],
        [6.1123e-04, 9.9939e-01],
        [9.6171e-01, 3.8294e-02],
        [2.5082e-05, 9.9997e-01],
        [1.5503e-07, 1.0000e+00],
        [1.0629e-08, 1.0000e+00],
        [9.9994e-01, 5.5885e-05],
        [9.9998e-01, 1.6775e-05],
        [9.3131e-06, 9.9999e-01],
        [1.5059e-05, 9.9998e-01],
        [9.9965e-01, 3.5384e-04],
        [2.1743e-04, 9.9978e-01],
        [7.7085e-08, 1.0000e+00],
        [7.0908e-06, 9.9999e-01],
        [3.5410e-04, 9.9965e-01],
        [2.0874e-06, 1.0000e+00],
        [1.8152e-07, 1.0000e+00],
        [2.8950e-07, 1.0000e+00],
        [4.4938e-05, 9.9996e-01],
        [1.0504e-06, 1.0000e+00],
        [9.9960e-01, 4.0196e-04],
        [5.9238e-07, 1.0000e+00],
        [5.4687e-07, 1.0000e+00],
        [1.0527e-04, 9.9989e-01],
        [8.7395e-07, 1.0000e+00],
        [5.5696e-06, 9.9999e-01],
        [9.1026e-06, 9.9999e-01],
        [1.3835e-06, 1.0000e+00],
        [9.9998e-01, 1.7579e-05],
        [7.9061e-07, 1.0000e+00],
        [1.8689e-08, 1.0000e+00],
        [5.0184e-06, 9.9999e-01],
        [8.9149e-12, 1.0000e+00],
        [5.8979e-08, 1.0000e+00],
        [9.9917e-01, 8.3279e-04],
        [9.9358e-01, 6.4216e-03],
        [3.1840e-03, 9.9682e-01],
        [7.8147e-10, 1.0000e+00],
        [8.0152e-05, 9.9992e-01],
        [2.1013e-02, 9.7899e-01],
        [9.9911e-01, 8.8676e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5692e-05, 9.9996e-01],
        [1.3081e-07, 1.0000e+00],
        [9.9894e-01, 1.0590e-03],
        [2.4391e-07, 1.0000e+00],
        [9.9363e-01, 6.3704e-03],
        [9.7727e-01, 2.2730e-02],
        [5.4215e-03, 9.9458e-01],
        [2.4110e-05, 9.9998e-01],
        [9.9074e-01, 9.2592e-03],
        [9.9984e-01, 1.5659e-04],
        [1.1391e-10, 1.0000e+00],
        [4.1214e-09, 1.0000e+00],
        [9.9846e-01, 1.5443e-03],
        [4.0313e-05, 9.9996e-01],
        [1.3020e-04, 9.9987e-01],
        [5.8849e-07, 1.0000e+00],
        [2.6027e-06, 1.0000e+00],
        [8.4347e-07, 1.0000e+00],
        [9.9987e-01, 1.3099e-04],
        [3.8845e-02, 9.6115e-01],
        [1.6790e-05, 9.9998e-01],
        [1.5108e-10, 1.0000e+00],
        [1.0995e-07, 1.0000e+00],
        [1.7600e-04, 9.9982e-01],
        [9.9995e-01, 4.6518e-05],
        [3.6498e-08, 1.0000e+00],
        [9.9978e-01, 2.1652e-04],
        [7.8700e-02, 9.2130e-01],
        [4.2213e-08, 1.0000e+00],
        [1.0000e+00, 2.8740e-07],
        [6.0395e-05, 9.9994e-01],
        [9.9981e-01, 1.8803e-04],
        [7.0807e-08, 1.0000e+00],
        [1.0634e-07, 1.0000e+00],
        [9.9971e-01, 2.9048e-04],
        [4.0552e-03, 9.9594e-01],
        [3.2722e-07, 1.0000e+00],
        [7.6173e-07, 1.0000e+00],
        [8.3056e-01, 1.6944e-01],
        [3.0452e-07, 1.0000e+00],
        [4.3572e-06, 1.0000e+00],
        [1.2823e-01, 8.7177e-01],
        [2.5274e-07, 1.0000e+00],
        [9.9988e-01, 1.1983e-04],
        [1.1530e-08, 1.0000e+00],
        [9.5440e-07, 1.0000e+00],
        [3.0157e-02, 9.6984e-01],
        [3.0671e-04, 9.9969e-01],
        [6.6840e-06, 9.9999e-01],
        [9.9999e-01, 1.2186e-05],
        [1.0346e-03, 9.9897e-01],
        [2.0604e-03, 9.9794e-01],
        [9.9994e-01, 6.2654e-05],
        [5.1546e-02, 9.4845e-01],
        [9.9983e-01, 1.7062e-04],
        [9.7171e-01, 2.8285e-02],
        [9.9991e-01, 8.6130e-05],
        [3.6025e-08, 1.0000e+00],
        [4.4862e-02, 9.5514e-01],
        [7.6860e-01, 2.3140e-01],
        [9.9991e-01, 8.9746e-05],
        [1.4600e-08, 1.0000e+00],
        [9.9981e-01, 1.9394e-04],
        [9.9954e-01, 4.5665e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0219e-09, 1.0000e+00],
        [2.6312e-10, 1.0000e+00],
        [9.9980e-01, 2.0071e-04],
        [9.9590e-01, 4.0959e-03],
        [9.9926e-01, 7.4030e-04],
        [2.9819e-03, 9.9702e-01],
        [3.7712e-03, 9.9623e-01],
        [9.9837e-01, 1.6343e-03],
        [9.9259e-01, 7.4076e-03],
        [9.8920e-09, 1.0000e+00],
        [4.6287e-12, 1.0000e+00],
        [3.8359e-07, 1.0000e+00],
        [9.4195e-05, 9.9991e-01],
        [2.9311e-07, 1.0000e+00],
        [4.2347e-04, 9.9958e-01],
        [4.2874e-06, 1.0000e+00],
        [1.2240e-06, 1.0000e+00],
        [1.8741e-04, 9.9981e-01],
        [4.4627e-04, 9.9955e-01],
        [7.3936e-09, 1.0000e+00],
        [1.4621e-05, 9.9999e-01],
        [7.4435e-06, 9.9999e-01],
        [9.9987e-01, 1.3214e-04],
        [9.9998e-01, 1.9387e-05],
        [8.6429e-01, 1.3571e-01],
        [6.0011e-06, 9.9999e-01],
        [5.8855e-05, 9.9994e-01],
        [9.1188e-01, 8.8120e-02],
        [9.9328e-01, 6.7236e-03],
        [1.6049e-05, 9.9998e-01],
        [1.0853e-05, 9.9999e-01],
        [4.1060e-06, 1.0000e+00],
        [2.1410e-05, 9.9998e-01],
        [3.2961e-06, 1.0000e+00],
        [4.4485e-06, 1.0000e+00],
        [9.9960e-01, 3.9850e-04],
        [1.8918e-07, 1.0000e+00],
        [1.8349e-08, 1.0000e+00],
        [8.0886e-07, 1.0000e+00],
        [2.7476e-08, 1.0000e+00],
        [8.8203e-09, 1.0000e+00],
        [6.6755e-07, 1.0000e+00],
        [1.0571e-04, 9.9989e-01],
        [3.0986e-04, 9.9969e-01],
        [9.9983e-01, 1.7188e-04],
        [7.7040e-09, 1.0000e+00],
        [6.1293e-11, 1.0000e+00],
        [3.9996e-06, 1.0000e+00],
        [1.1213e-05, 9.9999e-01],
        [2.4962e-10, 1.0000e+00],
        [3.7173e-08, 1.0000e+00],
        [9.9528e-01, 4.7187e-03],
        [2.2338e-01, 7.7662e-01],
        [9.9898e-01, 1.0167e-03],
        [3.0503e-07, 1.0000e+00],
        [8.3367e-07, 1.0000e+00],
        [9.9997e-01, 2.9120e-05],
        [1.2560e-08, 1.0000e+00],
        [1.9644e-10, 1.0000e+00],
        [5.2130e-06, 9.9999e-01],
        [3.7390e-03, 9.9626e-01],
        [5.1166e-03, 9.9488e-01],
        [9.9806e-01, 1.9381e-03],
        [1.0067e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.8007e-05, 9.9996e-01],
        [2.5181e-05, 9.9997e-01],
        [9.9975e-01, 2.4554e-04],
        [7.2652e-05, 9.9993e-01],
        [2.3275e-08, 1.0000e+00],
        [2.6836e-10, 1.0000e+00],
        [7.6968e-05, 9.9992e-01],
        [9.9985e-01, 1.4812e-04],
        [8.8869e-10, 1.0000e+00],
        [9.9981e-01, 1.9261e-04],
        [1.8882e-04, 9.9981e-01],
        [9.9964e-07, 1.0000e+00],
        [9.9980e-01, 2.0062e-04],
        [9.5352e-06, 9.9999e-01],
        [4.4235e-05, 9.9996e-01],
        [3.1766e-05, 9.9997e-01],
        [3.8856e-11, 1.0000e+00],
        [2.2794e-06, 1.0000e+00],
        [8.1958e-09, 1.0000e+00],
        [2.1441e-05, 9.9998e-01],
        [2.2920e-05, 9.9998e-01],
        [6.3875e-08, 1.0000e+00],
        [1.8858e-06, 1.0000e+00],
        [2.3472e-06, 1.0000e+00],
        [7.1225e-05, 9.9993e-01],
        [5.1258e-04, 9.9949e-01],
        [4.5997e-06, 1.0000e+00],
        [9.5783e-01, 4.2171e-02],
        [6.0645e-05, 9.9994e-01],
        [9.9993e-01, 7.0280e-05],
        [4.5906e-07, 1.0000e+00],
        [9.3616e-10, 1.0000e+00],
        [2.8476e-06, 1.0000e+00],
        [1.4003e-10, 1.0000e+00],
        [1.0986e-09, 1.0000e+00],
        [7.6162e-06, 9.9999e-01],
        [2.6257e-06, 1.0000e+00],
        [4.0661e-05, 9.9996e-01],
        [9.0233e-09, 1.0000e+00],
        [9.9918e-01, 8.2429e-04],
        [5.3112e-06, 9.9999e-01],
        [1.3217e-07, 1.0000e+00],
        [1.1290e-05, 9.9999e-01],
        [9.9976e-01, 2.3506e-04],
        [4.6566e-01, 5.3434e-01],
        [2.1781e-04, 9.9978e-01],
        [1.3014e-08, 1.0000e+00],
        [9.9991e-01, 9.4839e-05],
        [4.6775e-07, 1.0000e+00],
        [5.7224e-10, 1.0000e+00],
        [1.4098e-07, 1.0000e+00],
        [3.7373e-04, 9.9963e-01],
        [1.6289e-08, 1.0000e+00],
        [4.0995e-10, 1.0000e+00],
        [4.8401e-07, 1.0000e+00],
        [1.9514e-04, 9.9980e-01],
        [9.9962e-01, 3.8351e-04],
        [9.9540e-01, 4.6036e-03],
        [2.1007e-08, 1.0000e+00],
        [9.9967e-01, 3.3181e-04],
        [9.9955e-01, 4.5327e-04],
        [4.8749e-06, 1.0000e+00],
        [3.2126e-03, 9.9679e-01],
        [1.5170e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3777e-08, 1.0000e+00],
        [5.1032e-08, 1.0000e+00],
        [9.9995e-01, 4.6098e-05],
        [1.6968e-12, 1.0000e+00],
        [9.9102e-01, 8.9848e-03],
        [2.9611e-08, 1.0000e+00],
        [9.2172e-07, 1.0000e+00],
        [4.0559e-06, 1.0000e+00],
        [2.1385e-08, 1.0000e+00],
        [9.9074e-01, 9.2579e-03],
        [6.8371e-06, 9.9999e-01],
        [2.0541e-09, 1.0000e+00],
        [4.5543e-05, 9.9995e-01],
        [7.5823e-08, 1.0000e+00],
        [6.4924e-01, 3.5076e-01],
        [3.3701e-06, 1.0000e+00],
        [9.3163e-05, 9.9991e-01],
        [9.2703e-01, 7.2969e-02],
        [8.0013e-10, 1.0000e+00],
        [3.9754e-09, 1.0000e+00],
        [9.9995e-01, 4.6303e-05],
        [9.3393e-10, 1.0000e+00],
        [3.0046e-03, 9.9700e-01],
        [9.9841e-01, 1.5880e-03],
        [1.4146e-07, 1.0000e+00],
        [9.9966e-01, 3.4244e-04],
        [3.7562e-03, 9.9624e-01],
        [1.5094e-07, 1.0000e+00],
        [1.5498e-06, 1.0000e+00],
        [3.0736e-09, 1.0000e+00],
        [2.4493e-06, 1.0000e+00],
        [3.5686e-05, 9.9996e-01],
        [1.4089e-09, 1.0000e+00],
        [9.9951e-01, 4.8501e-04],
        [5.0832e-09, 1.0000e+00],
        [1.6474e-07, 1.0000e+00],
        [6.6032e-06, 9.9999e-01],
        [2.9765e-02, 9.7023e-01],
        [9.9967e-01, 3.3336e-04],
        [2.9692e-08, 1.0000e+00],
        [3.5243e-06, 1.0000e+00],
        [4.7744e-05, 9.9995e-01],
        [1.2879e-07, 1.0000e+00],
        [8.9571e-01, 1.0429e-01],
        [9.0580e-04, 9.9909e-01],
        [9.9994e-01, 5.6659e-05],
        [1.7980e-05, 9.9998e-01],
        [9.7680e-01, 2.3196e-02],
        [3.1732e-11, 1.0000e+00],
        [1.7436e-09, 1.0000e+00],
        [6.9196e-04, 9.9931e-01],
        [3.5579e-05, 9.9996e-01],
        [3.4060e-03, 9.9659e-01],
        [7.0047e-07, 1.0000e+00],
        [6.4806e-10, 1.0000e+00],
        [2.7352e-12, 1.0000e+00],
        [1.2930e-05, 9.9999e-01],
        [6.7055e-04, 9.9933e-01],
        [9.9973e-01, 2.6986e-04],
        [9.9767e-03, 9.9002e-01],
        [9.9990e-01, 9.7792e-05],
        [1.7186e-05, 9.9998e-01],
        [5.7088e-05, 9.9994e-01],
        [3.1541e-04, 9.9968e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5440e-02, 9.5456e-01],
        [9.9960e-01, 4.0223e-04],
        [6.0145e-07, 1.0000e+00],
        [1.9800e-05, 9.9998e-01],
        [2.2541e-06, 1.0000e+00],
        [2.1781e-04, 9.9978e-01],
        [9.9972e-01, 2.7953e-04],
        [9.9967e-01, 3.3409e-04],
        [9.9995e-01, 4.5286e-05],
        [7.4339e-05, 9.9993e-01],
        [9.9984e-01, 1.5837e-04],
        [8.1901e-08, 1.0000e+00],
        [3.3330e-06, 1.0000e+00],
        [9.8987e-01, 1.0129e-02],
        [2.1017e-07, 1.0000e+00],
        [5.2114e-03, 9.9479e-01],
        [9.9987e-01, 1.2767e-04],
        [2.5528e-09, 1.0000e+00],
        [9.9914e-01, 8.5832e-04],
        [2.2910e-06, 1.0000e+00],
        [2.8735e-06, 1.0000e+00],
        [6.5028e-07, 1.0000e+00],
        [4.0044e-06, 1.0000e+00],
        [4.4592e-08, 1.0000e+00],
        [9.3874e-01, 6.1259e-02],
        [9.9843e-01, 1.5708e-03],
        [5.3404e-10, 1.0000e+00],
        [8.9076e-04, 9.9911e-01],
        [1.2089e-05, 9.9999e-01],
        [2.4943e-06, 1.0000e+00],
        [2.3537e-03, 9.9765e-01],
        [9.9355e-01, 6.4528e-03],
        [9.9792e-01, 2.0787e-03],
        [6.0819e-07, 1.0000e+00],
        [6.3459e-03, 9.9365e-01],
        [1.1439e-09, 1.0000e+00],
        [2.9140e-05, 9.9997e-01],
        [4.7786e-03, 9.9522e-01],
        [3.4257e-02, 9.6574e-01],
        [4.9075e-08, 1.0000e+00],
        [6.4337e-04, 9.9936e-01],
        [2.2149e-04, 9.9978e-01],
        [5.0570e-05, 9.9995e-01],
        [3.0321e-08, 1.0000e+00],
        [4.6687e-07, 1.0000e+00],
        [9.9950e-01, 4.9835e-04],
        [1.0593e-05, 9.9999e-01],
        [1.0718e-04, 9.9989e-01],
        [2.2980e-09, 1.0000e+00],
        [1.7211e-05, 9.9998e-01],
        [1.5282e-07, 1.0000e+00],
        [3.2600e-09, 1.0000e+00],
        [2.1094e-09, 1.0000e+00],
        [9.7780e-08, 1.0000e+00],
        [9.9959e-01, 4.1179e-04],
        [9.9976e-01, 2.4187e-04],
        [4.3611e-09, 1.0000e+00],
        [4.7762e-05, 9.9995e-01],
        [4.8536e-08, 1.0000e+00],
        [4.9075e-08, 1.0000e+00],
        [2.8797e-11, 1.0000e+00],
        [9.9916e-01, 8.3881e-04],
        [5.5061e-05, 9.9994e-01],
        [9.9984e-01, 1.5763e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2658e-03, 9.9873e-01],
        [3.5387e-03, 9.9646e-01],
        [1.0841e-02, 9.8916e-01],
        [2.2640e-05, 9.9998e-01],
        [5.0391e-10, 1.0000e+00],
        [9.9998e-01, 2.0648e-05],
        [6.9561e-06, 9.9999e-01],
        [2.3477e-04, 9.9977e-01],
        [2.2725e-10, 1.0000e+00],
        [1.7779e-07, 1.0000e+00],
        [7.8528e-01, 2.1472e-01],
        [9.9397e-01, 6.0301e-03],
        [1.4029e-07, 1.0000e+00],
        [5.6194e-05, 9.9994e-01],
        [3.2113e-06, 1.0000e+00],
        [4.8305e-07, 1.0000e+00],
        [4.5644e-05, 9.9995e-01],
        [1.0131e-05, 9.9999e-01],
        [1.1008e-05, 9.9999e-01],
        [1.5627e-06, 1.0000e+00],
        [1.7436e-06, 1.0000e+00],
        [2.8177e-04, 9.9972e-01],
        [1.0088e-01, 8.9912e-01],
        [1.5749e-05, 9.9998e-01],
        [2.0581e-07, 1.0000e+00],
        [7.4013e-03, 9.9260e-01],
        [1.3491e-07, 1.0000e+00],
        [2.4331e-04, 9.9976e-01],
        [1.2647e-05, 9.9999e-01],
        [4.3187e-06, 1.0000e+00],
        [1.5849e-11, 1.0000e+00],
        [4.3876e-09, 1.0000e+00],
        [6.9809e-08, 1.0000e+00],
        [9.4880e-01, 5.1199e-02],
        [9.9972e-01, 2.7812e-04],
        [9.8232e-01, 1.7678e-02],
        [9.9991e-01, 9.3088e-05],
        [9.9898e-01, 1.0193e-03],
        [9.9993e-01, 7.0641e-05],
        [1.7980e-09, 1.0000e+00],
        [8.7876e-08, 1.0000e+00],
        [1.4771e-06, 1.0000e+00],
        [8.9058e-01, 1.0942e-01],
        [8.3593e-02, 9.1641e-01],
        [9.0482e-06, 9.9999e-01],
        [5.7455e-02, 9.4255e-01],
        [1.1155e-02, 9.8885e-01],
        [4.2402e-06, 1.0000e+00],
        [3.6086e-04, 9.9964e-01],
        [9.9633e-01, 3.6743e-03],
        [6.5274e-09, 1.0000e+00],
        [2.7159e-07, 1.0000e+00],
        [3.6371e-05, 9.9996e-01],
        [2.9818e-03, 9.9702e-01],
        [2.7059e-04, 9.9973e-01],
        [1.3489e-08, 1.0000e+00],
        [8.7102e-04, 9.9913e-01],
        [1.3956e-05, 9.9999e-01],
        [9.9988e-01, 1.2306e-04],
        [1.8438e-09, 1.0000e+00],
        [4.9738e-01, 5.0262e-01],
        [4.5108e-04, 9.9955e-01],
        [2.6423e-04, 9.9974e-01],
        [2.0058e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.6175e-04, 9.9904e-01],
        [9.9993e-01, 7.1108e-05],
        [9.9991e-01, 9.4272e-05],
        [9.9156e-01, 8.4444e-03],
        [3.3331e-10, 1.0000e+00],
        [1.4511e-07, 1.0000e+00],
        [1.5011e-04, 9.9985e-01],
        [5.2311e-09, 1.0000e+00],
        [2.9392e-06, 1.0000e+00],
        [9.9998e-01, 2.2729e-05],
        [9.9882e-01, 1.1837e-03],
        [9.9996e-01, 3.7813e-05],
        [2.8481e-05, 9.9997e-01],
        [2.2882e-06, 1.0000e+00],
        [1.0600e-07, 1.0000e+00],
        [9.7896e-01, 2.1039e-02],
        [5.3905e-05, 9.9995e-01],
        [1.4439e-04, 9.9986e-01],
        [4.4770e-06, 1.0000e+00],
        [5.5097e-06, 9.9999e-01],
        [9.9314e-01, 6.8556e-03],
        [1.1027e-01, 8.8973e-01],
        [2.4778e-08, 1.0000e+00],
        [4.8081e-08, 1.0000e+00],
        [1.7648e-05, 9.9998e-01],
        [2.4951e-09, 1.0000e+00],
        [3.1871e-08, 1.0000e+00],
        [2.4380e-07, 1.0000e+00],
        [1.0065e-03, 9.9899e-01],
        [9.9921e-01, 7.8742e-04],
        [9.9989e-01, 1.1088e-04],
        [1.1174e-05, 9.9999e-01],
        [9.9982e-01, 1.7731e-04],
        [6.9156e-05, 9.9993e-01],
        [5.4723e-09, 1.0000e+00],
        [4.7361e-04, 9.9953e-01],
        [5.3057e-08, 1.0000e+00],
        [3.3312e-07, 1.0000e+00],
        [4.7771e-06, 1.0000e+00],
        [4.0435e-06, 1.0000e+00],
        [9.2677e-01, 7.3235e-02],
        [9.9859e-01, 1.4147e-03],
        [3.0651e-06, 1.0000e+00],
        [2.7761e-04, 9.9972e-01],
        [1.6531e-04, 9.9983e-01],
        [1.1034e-06, 1.0000e+00],
        [9.9992e-01, 7.9920e-05],
        [8.4431e-07, 1.0000e+00],
        [1.9935e-06, 1.0000e+00],
        [1.6888e-08, 1.0000e+00],
        [1.9655e-01, 8.0345e-01],
        [1.2442e-07, 1.0000e+00],
        [1.5516e-01, 8.4484e-01],
        [7.3635e-09, 1.0000e+00],
        [1.7431e-05, 9.9998e-01],
        [9.9979e-01, 2.0755e-04],
        [5.3866e-06, 9.9999e-01],
        [8.3364e-05, 9.9992e-01],
        [2.8472e-07, 1.0000e+00],
        [9.4143e-01, 5.8566e-02],
        [1.0508e-08, 1.0000e+00],
        [3.3929e-10, 1.0000e+00],
        [2.5095e-09, 1.0000e+00],
        [2.8096e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.8095e-01, 1.1905e-01],
        [1.0734e-07, 1.0000e+00],
        [9.9989e-01, 1.0584e-04],
        [9.9986e-01, 1.4115e-04],
        [5.4414e-06, 9.9999e-01],
        [3.2111e-01, 6.7889e-01],
        [2.2223e-05, 9.9998e-01],
        [2.9129e-06, 1.0000e+00],
        [1.5738e-05, 9.9998e-01],
        [9.9989e-01, 1.0845e-04],
        [9.9984e-01, 1.6463e-04],
        [4.5714e-06, 1.0000e+00],
        [1.0292e-05, 9.9999e-01],
        [9.9985e-01, 1.5027e-04],
        [3.7995e-09, 1.0000e+00],
        [2.2658e-01, 7.7342e-01],
        [4.6426e-07, 1.0000e+00],
        [5.5729e-08, 1.0000e+00],
        [2.3629e-04, 9.9976e-01],
        [4.9467e-06, 1.0000e+00],
        [3.3939e-02, 9.6606e-01],
        [3.4168e-02, 9.6583e-01],
        [9.9991e-01, 8.5757e-05],
        [6.8319e-04, 9.9932e-01],
        [5.3372e-03, 9.9466e-01],
        [9.9985e-01, 1.4902e-04],
        [3.6231e-05, 9.9996e-01],
        [1.4147e-06, 1.0000e+00],
        [7.5320e-07, 1.0000e+00],
        [1.1595e-08, 1.0000e+00],
        [1.0034e-03, 9.9900e-01],
        [8.5122e-07, 1.0000e+00],
        [3.2972e-08, 1.0000e+00],
        [7.0287e-05, 9.9993e-01],
        [9.9988e-01, 1.2321e-04],
        [9.9900e-01, 9.9706e-04],
        [1.7798e-08, 1.0000e+00],
        [3.5358e-08, 1.0000e+00],
        [9.2944e-01, 7.0558e-02],
        [2.8211e-05, 9.9997e-01],
        [3.5295e-06, 1.0000e+00],
        [9.9994e-01, 5.8993e-05],
        [1.2561e-06, 1.0000e+00],
        [2.0145e-02, 9.7986e-01],
        [1.0127e-02, 9.8987e-01],
        [7.6668e-03, 9.9233e-01],
        [9.9870e-06, 9.9999e-01],
        [9.9990e-01, 9.6735e-05],
        [9.9997e-01, 2.9685e-05],
        [6.3663e-10, 1.0000e+00],
        [9.9954e-01, 4.6425e-04],
        [2.2719e-06, 1.0000e+00],
        [1.2587e-06, 1.0000e+00],
        [3.1925e-07, 1.0000e+00],
        [1.8695e-07, 1.0000e+00],
        [1.0796e-06, 1.0000e+00],
        [5.7648e-04, 9.9942e-01],
        [3.0376e-08, 1.0000e+00],
        [2.1050e-07, 1.0000e+00],
        [6.1893e-06, 9.9999e-01],
        [1.0277e-07, 1.0000e+00],
        [1.1464e-05, 9.9999e-01],
        [1.8827e-05, 9.9998e-01],
        [6.9123e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.4081e-04, 9.9936e-01],
        [3.4414e-05, 9.9997e-01],
        [1.2016e-06, 1.0000e+00],
        [6.4356e-01, 3.5644e-01],
        [4.2613e-07, 1.0000e+00],
        [9.9984e-01, 1.5650e-04],
        [3.4868e-06, 1.0000e+00],
        [9.0591e-01, 9.4089e-02],
        [5.1979e-07, 1.0000e+00],
        [3.5445e-06, 1.0000e+00],
        [1.3258e-04, 9.9987e-01],
        [5.1532e-10, 1.0000e+00],
        [1.5527e-07, 1.0000e+00],
        [8.4758e-07, 1.0000e+00],
        [2.1040e-03, 9.9790e-01],
        [3.0270e-09, 1.0000e+00],
        [8.3891e-09, 1.0000e+00],
        [4.4649e-05, 9.9996e-01],
        [7.7847e-06, 9.9999e-01],
        [1.4282e-09, 1.0000e+00],
        [9.9785e-01, 2.1513e-03],
        [3.8001e-02, 9.6200e-01],
        [9.9997e-01, 3.0345e-05],
        [9.9989e-01, 1.1184e-04],
        [9.9856e-01, 1.4387e-03],
        [1.4906e-07, 1.0000e+00],
        [2.0865e-04, 9.9979e-01],
        [7.1318e-03, 9.9287e-01],
        [2.1336e-05, 9.9998e-01],
        [4.3616e-06, 1.0000e+00],
        [1.5602e-07, 1.0000e+00],
        [2.1193e-05, 9.9998e-01],
        [5.7160e-05, 9.9994e-01],
        [1.0347e-05, 9.9999e-01],
        [9.9994e-01, 6.4525e-05],
        [1.9390e-07, 1.0000e+00],
        [4.5433e-03, 9.9546e-01],
        [2.1893e-04, 9.9978e-01],
        [2.7005e-03, 9.9730e-01],
        [1.0757e-05, 9.9999e-01],
        [9.8522e-01, 1.4777e-02],
        [1.8027e-10, 1.0000e+00],
        [9.5985e-03, 9.9040e-01],
        [2.2228e-07, 1.0000e+00],
        [9.9861e-03, 9.9001e-01],
        [1.5738e-03, 9.9843e-01],
        [9.5248e-03, 9.9048e-01],
        [2.2678e-05, 9.9998e-01],
        [9.9991e-01, 8.8343e-05],
        [7.3836e-04, 9.9926e-01],
        [9.9994e-01, 5.9978e-05],
        [9.9989e-01, 1.1411e-04],
        [7.4591e-06, 9.9999e-01],
        [9.9952e-01, 4.8433e-04],
        [4.4572e-05, 9.9996e-01],
        [1.2967e-06, 1.0000e+00],
        [3.7425e-07, 1.0000e+00],
        [9.9857e-01, 1.4301e-03],
        [3.1012e-09, 1.0000e+00],
        [9.9992e-01, 7.8891e-05],
        [2.6973e-05, 9.9997e-01],
        [3.5851e-11, 1.0000e+00],
        [9.9957e-01, 4.2996e-04],
        [9.9976e-01, 2.3752e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2049e-06, 1.0000e+00],
        [9.9995e-01, 4.9040e-05],
        [8.6373e-03, 9.9136e-01],
        [9.7291e-01, 2.7090e-02],
        [8.2206e-03, 9.9178e-01],
        [1.0262e-06, 1.0000e+00],
        [1.3179e-06, 1.0000e+00],
        [9.9951e-01, 4.9308e-04],
        [9.9876e-01, 1.2384e-03],
        [2.4282e-08, 1.0000e+00],
        [6.7581e-03, 9.9324e-01],
        [8.9904e-01, 1.0096e-01],
        [2.3108e-08, 1.0000e+00],
        [5.0338e-07, 1.0000e+00],
        [8.6518e-08, 1.0000e+00],
        [3.2537e-06, 1.0000e+00],
        [1.0000e+00, 1.7831e-06],
        [9.9017e-09, 1.0000e+00],
        [1.0000e+00, 3.3298e-06],
        [9.9961e-01, 3.8822e-04],
        [1.0931e-01, 8.9069e-01],
        [9.9998e-01, 2.4596e-05],
        [9.9382e-05, 9.9990e-01],
        [1.4040e-08, 1.0000e+00],
        [9.9981e-01, 1.9195e-04],
        [1.7467e-03, 9.9825e-01],
        [9.9842e-01, 1.5841e-03],
        [6.7647e-06, 9.9999e-01],
        [9.0831e-01, 9.1692e-02],
        [2.8162e-04, 9.9972e-01],
        [1.4858e-04, 9.9985e-01],
        [4.8924e-09, 1.0000e+00],
        [1.0689e-08, 1.0000e+00],
        [1.0987e-04, 9.9989e-01],
        [4.0616e-07, 1.0000e+00],
        [2.8354e-04, 9.9972e-01],
        [3.6219e-08, 1.0000e+00],
        [1.4171e-12, 1.0000e+00],
        [9.8745e-03, 9.9013e-01],
        [9.9721e-01, 2.7850e-03],
        [4.2682e-07, 1.0000e+00],
        [3.9158e-05, 9.9996e-01],
        [1.9829e-03, 9.9802e-01],
        [1.0145e-07, 1.0000e+00],
        [9.9974e-01, 2.6497e-04],
        [1.8756e-04, 9.9981e-01],
        [1.4255e-07, 1.0000e+00],
        [1.0789e-03, 9.9892e-01],
        [2.5604e-07, 1.0000e+00],
        [9.5824e-03, 9.9042e-01],
        [1.0000e+00, 5.9589e-07],
        [1.2262e-03, 9.9877e-01],
        [2.6617e-05, 9.9997e-01],
        [5.9552e-06, 9.9999e-01],
        [1.0468e-06, 1.0000e+00],
        [2.6121e-08, 1.0000e+00],
        [9.9503e-01, 4.9699e-03],
        [5.0082e-04, 9.9950e-01],
        [9.9949e-01, 5.0639e-04],
        [2.4935e-05, 9.9998e-01],
        [1.7968e-04, 9.9982e-01],
        [8.5140e-01, 1.4860e-01],
        [9.9997e-01, 3.0067e-05],
        [1.5696e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2498e-07, 1.0000e+00],
        [2.7453e-05, 9.9997e-01],
        [2.6872e-06, 1.0000e+00],
        [2.8432e-05, 9.9997e-01],
        [1.9019e-05, 9.9998e-01],
        [3.7837e-04, 9.9962e-01],
        [5.9939e-08, 1.0000e+00],
        [6.2336e-07, 1.0000e+00],
        [1.0000e+00, 9.7757e-08],
        [2.0354e-08, 1.0000e+00],
        [6.6543e-08, 1.0000e+00],
        [4.0737e-05, 9.9996e-01],
        [1.1903e-04, 9.9988e-01],
        [9.9989e-01, 1.0506e-04],
        [2.3358e-08, 1.0000e+00],
        [2.2769e-04, 9.9977e-01],
        [9.9982e-01, 1.7936e-04],
        [9.9981e-01, 1.8572e-04],
        [8.4296e-05, 9.9992e-01],
        [4.9361e-09, 1.0000e+00],
        [5.1488e-07, 1.0000e+00],
        [6.1689e-01, 3.8311e-01],
        [1.6363e-07, 1.0000e+00],
        [9.9573e-01, 4.2703e-03],
        [9.9930e-01, 6.9737e-04],
        [1.6132e-04, 9.9984e-01],
        [1.4307e-06, 1.0000e+00],
        [2.5792e-05, 9.9997e-01],
        [5.1008e-03, 9.9490e-01],
        [2.6559e-08, 1.0000e+00],
        [7.2645e-06, 9.9999e-01],
        [9.9991e-01, 8.7175e-05],
        [9.9955e-01, 4.5022e-04],
        [1.9116e-06, 1.0000e+00],
        [1.3316e-06, 1.0000e+00],
        [4.8085e-02, 9.5192e-01],
        [1.9979e-03, 9.9800e-01],
        [7.4615e-06, 9.9999e-01],
        [9.8374e-01, 1.6263e-02],
        [7.8918e-07, 1.0000e+00],
        [6.7288e-03, 9.9327e-01],
        [9.9995e-01, 5.3112e-05],
        [1.1582e-03, 9.9884e-01],
        [1.5142e-08, 1.0000e+00],
        [7.6878e-04, 9.9923e-01],
        [9.9914e-01, 8.5785e-04],
        [5.0169e-11, 1.0000e+00],
        [1.6340e-04, 9.9984e-01],
        [2.4330e-03, 9.9757e-01],
        [9.9980e-01, 2.0207e-04],
        [1.3827e-04, 9.9986e-01],
        [1.5593e-06, 1.0000e+00],
        [1.1656e-01, 8.8344e-01],
        [2.9123e-01, 7.0877e-01],
        [9.9994e-01, 5.9960e-05],
        [5.9732e-04, 9.9940e-01],
        [5.5724e-07, 1.0000e+00],
        [9.9996e-01, 3.9402e-05],
        [9.8546e-05, 9.9990e-01],
        [3.0014e-04, 9.9970e-01],
        [1.5877e-06, 1.0000e+00],
        [1.9801e-08, 1.0000e+00],
        [5.2907e-05, 9.9995e-01],
        [1.1329e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.8248e-02, 9.5175e-01],
        [9.9978e-01, 2.1806e-04],
        [3.5711e-07, 1.0000e+00],
        [9.9979e-01, 2.0508e-04],
        [1.3165e-06, 1.0000e+00],
        [9.4589e-05, 9.9991e-01],
        [2.3785e-07, 1.0000e+00],
        [5.6455e-07, 1.0000e+00],
        [9.9973e-01, 2.6903e-04],
        [9.9994e-01, 5.9961e-05],
        [8.8221e-05, 9.9991e-01],
        [1.8233e-03, 9.9818e-01],
        [9.9988e-01, 1.1532e-04],
        [1.1738e-05, 9.9999e-01],
        [8.9368e-10, 1.0000e+00],
        [3.3629e-07, 1.0000e+00],
        [2.3124e-04, 9.9977e-01],
        [8.2272e-08, 1.0000e+00],
        [5.9017e-09, 1.0000e+00],
        [2.6963e-03, 9.9730e-01],
        [9.9939e-01, 6.0870e-04],
        [1.3078e-05, 9.9999e-01],
        [9.9678e-01, 3.2175e-03],
        [1.5488e-06, 1.0000e+00],
        [3.7331e-04, 9.9963e-01],
        [9.9945e-01, 5.5464e-04],
        [9.9965e-01, 3.4891e-04],
        [1.2303e-07, 1.0000e+00],
        [1.0300e-06, 1.0000e+00],
        [9.9978e-01, 2.2159e-04],
        [9.9992e-01, 7.7206e-05],
        [6.3445e-09, 1.0000e+00],
        [2.7000e-06, 1.0000e+00],
        [7.4852e-06, 9.9999e-01],
        [1.2755e-04, 9.9987e-01],
        [4.3341e-04, 9.9957e-01],
        [1.1554e-05, 9.9999e-01],
        [9.9946e-01, 5.4291e-04],
        [3.4720e-09, 1.0000e+00],
        [3.8860e-11, 1.0000e+00],
        [1.1392e-05, 9.9999e-01],
        [9.9986e-01, 1.4433e-04],
        [9.9977e-01, 2.2717e-04],
        [9.9979e-01, 2.1055e-04],
        [1.9920e-06, 1.0000e+00],
        [5.2377e-06, 9.9999e-01],
        [1.5484e-07, 1.0000e+00],
        [1.5484e-04, 9.9985e-01],
        [1.2382e-04, 9.9988e-01],
        [1.8125e-02, 9.8188e-01],
        [6.3936e-06, 9.9999e-01],
        [3.6066e-07, 1.0000e+00],
        [4.7682e-07, 1.0000e+00],
        [4.6817e-08, 1.0000e+00],
        [2.4604e-05, 9.9998e-01],
        [4.9969e-05, 9.9995e-01],
        [9.9955e-01, 4.5394e-04],
        [1.4520e-06, 1.0000e+00],
        [3.8683e-06, 1.0000e+00],
        [6.9663e-04, 9.9930e-01],
        [6.4551e-07, 1.0000e+00],
        [1.4649e-05, 9.9999e-01],
        [9.9998e-01, 2.4881e-05],
        [4.0026e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9081e-07, 1.0000e+00],
        [1.4803e-07, 1.0000e+00],
        [9.9993e-01, 7.1073e-05],
        [3.2791e-03, 9.9672e-01],
        [8.8791e-04, 9.9911e-01],
        [4.7558e-03, 9.9524e-01],
        [1.0886e-02, 9.8911e-01],
        [2.2053e-04, 9.9978e-01],
        [9.9672e-01, 3.2762e-03],
        [1.9888e-08, 1.0000e+00],
        [7.1177e-06, 9.9999e-01],
        [3.3996e-07, 1.0000e+00],
        [8.4971e-07, 1.0000e+00],
        [1.9947e-05, 9.9998e-01],
        [6.2018e-07, 1.0000e+00],
        [1.2235e-05, 9.9999e-01],
        [2.9759e-07, 1.0000e+00],
        [6.1584e-07, 1.0000e+00],
        [6.5131e-06, 9.9999e-01],
        [9.6191e-01, 3.8092e-02],
        [3.7292e-07, 1.0000e+00],
        [8.6966e-04, 9.9913e-01],
        [6.6944e-01, 3.3056e-01],
        [9.2449e-03, 9.9076e-01],
        [3.3242e-05, 9.9997e-01],
        [9.7425e-01, 2.5753e-02],
        [1.8824e-06, 1.0000e+00],
        [1.4363e-08, 1.0000e+00],
        [2.0288e-07, 1.0000e+00],
        [1.6175e-03, 9.9838e-01],
        [2.4457e-05, 9.9998e-01],
        [9.9989e-01, 1.1289e-04],
        [9.4029e-01, 5.9710e-02],
        [6.5692e-07, 1.0000e+00],
        [9.9998e-01, 1.6721e-05],
        [2.3767e-08, 1.0000e+00],
        [7.7838e-10, 1.0000e+00],
        [9.9991e-01, 8.9425e-05],
        [7.6071e-05, 9.9992e-01],
        [1.0187e-05, 9.9999e-01],
        [3.5606e-06, 1.0000e+00],
        [6.6956e-05, 9.9993e-01],
        [7.6883e-07, 1.0000e+00],
        [1.1011e-04, 9.9989e-01],
        [3.7088e-04, 9.9963e-01],
        [2.3323e-07, 1.0000e+00],
        [1.4691e-05, 9.9999e-01],
        [1.9127e-05, 9.9998e-01],
        [1.0631e-03, 9.9894e-01],
        [9.9984e-01, 1.6048e-04],
        [2.7697e-03, 9.9723e-01],
        [9.9980e-01, 1.9667e-04],
        [9.9992e-01, 8.4276e-05],
        [1.2574e-06, 1.0000e+00],
        [2.2007e-02, 9.7799e-01],
        [9.5849e-09, 1.0000e+00],
        [1.0198e-06, 1.0000e+00],
        [2.2410e-07, 1.0000e+00],
        [9.9969e-01, 3.1188e-04],
        [1.1056e-04, 9.9989e-01],
        [2.4918e-09, 1.0000e+00],
        [9.3571e-05, 9.9991e-01],
        [1.9326e-08, 1.0000e+00],
        [9.9909e-01, 9.1117e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1413e-02, 9.8859e-01],
        [5.1581e-07, 1.0000e+00],
        [2.3117e-02, 9.7688e-01],
        [7.3680e-01, 2.6320e-01],
        [1.6700e-05, 9.9998e-01],
        [9.5487e-09, 1.0000e+00],
        [3.8198e-07, 1.0000e+00],
        [1.1104e-06, 1.0000e+00],
        [9.6127e-06, 9.9999e-01],
        [9.9879e-01, 1.2055e-03],
        [6.6309e-05, 9.9993e-01],
        [4.6155e-08, 1.0000e+00],
        [3.1614e-01, 6.8386e-01],
        [1.3053e-02, 9.8695e-01],
        [4.3483e-10, 1.0000e+00],
        [8.6310e-05, 9.9991e-01],
        [9.9949e-01, 5.1225e-04],
        [1.6377e-10, 1.0000e+00],
        [5.8011e-03, 9.9420e-01],
        [6.5020e-09, 1.0000e+00],
        [9.9851e-08, 1.0000e+00],
        [9.8652e-08, 1.0000e+00],
        [9.9697e-01, 3.0349e-03],
        [9.9995e-01, 5.4024e-05],
        [9.9979e-01, 2.1477e-04],
        [2.3532e-05, 9.9998e-01],
        [1.2276e-05, 9.9999e-01],
        [9.9998e-01, 2.4774e-05],
        [2.2091e-07, 1.0000e+00],
        [9.7175e-01, 2.8247e-02],
        [8.8360e-09, 1.0000e+00],
        [9.9986e-01, 1.3612e-04],
        [9.9757e-01, 2.4319e-03],
        [9.9938e-01, 6.1706e-04],
        [9.9885e-01, 1.1480e-03],
        [9.9995e-01, 5.2526e-05],
        [8.8030e-04, 9.9912e-01],
        [1.3332e-07, 1.0000e+00],
        [1.0000e+00, 8.3039e-07],
        [1.3697e-09, 1.0000e+00],
        [1.0979e-03, 9.9890e-01],
        [2.8322e-07, 1.0000e+00],
        [1.7173e-02, 9.8283e-01],
        [1.0980e-08, 1.0000e+00],
        [8.0302e-04, 9.9920e-01],
        [9.9981e-01, 1.9045e-04],
        [9.8067e-01, 1.9328e-02],
        [2.4726e-03, 9.9753e-01],
        [9.9991e-01, 9.1126e-05],
        [2.4799e-06, 1.0000e+00],
        [9.9992e-01, 8.4035e-05],
        [7.7576e-09, 1.0000e+00],
        [9.9986e-01, 1.4451e-04],
        [9.5779e-01, 4.2212e-02],
        [7.8027e-04, 9.9922e-01],
        [9.9706e-01, 2.9363e-03],
        [1.0428e-02, 9.8957e-01],
        [5.0521e-05, 9.9995e-01],
        [2.3615e-08, 1.0000e+00],
        [4.5696e-07, 1.0000e+00],
        [2.6510e-05, 9.9997e-01],
        [2.5203e-04, 9.9975e-01],
        [7.9147e-04, 9.9921e-01],
        [2.7515e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.1244e-01, 2.8756e-01],
        [5.4327e-10, 1.0000e+00],
        [9.9945e-01, 5.4919e-04],
        [9.9206e-01, 7.9449e-03],
        [5.3706e-08, 1.0000e+00],
        [1.7752e-07, 1.0000e+00],
        [7.8955e-09, 1.0000e+00],
        [7.5646e-06, 9.9999e-01],
        [1.8518e-06, 1.0000e+00],
        [1.2261e-04, 9.9988e-01],
        [8.7563e-08, 1.0000e+00],
        [1.8886e-05, 9.9998e-01],
        [3.1412e-03, 9.9686e-01],
        [7.7171e-05, 9.9992e-01],
        [9.8684e-01, 1.3155e-02],
        [1.8295e-09, 1.0000e+00],
        [5.4449e-04, 9.9946e-01],
        [8.5894e-01, 1.4106e-01],
        [9.8571e-01, 1.4290e-02],
        [9.9440e-01, 5.6016e-03],
        [7.2023e-04, 9.9928e-01],
        [3.0875e-08, 1.0000e+00],
        [1.3110e-05, 9.9999e-01],
        [9.9885e-01, 1.1465e-03],
        [9.9978e-01, 2.1958e-04],
        [2.4415e-02, 9.7559e-01],
        [4.6382e-07, 1.0000e+00],
        [5.1296e-09, 1.0000e+00],
        [1.1016e-06, 1.0000e+00],
        [3.4899e-04, 9.9965e-01],
        [5.9154e-09, 1.0000e+00],
        [2.0764e-07, 1.0000e+00],
        [5.0068e-07, 1.0000e+00],
        [9.9845e-01, 1.5510e-03],
        [3.5402e-04, 9.9965e-01],
        [1.7757e-09, 1.0000e+00],
        [1.1638e-04, 9.9988e-01],
        [1.5654e-02, 9.8435e-01],
        [2.2507e-05, 9.9998e-01],
        [3.2151e-09, 1.0000e+00],
        [1.2690e-03, 9.9873e-01],
        [7.5266e-08, 1.0000e+00],
        [2.9580e-07, 1.0000e+00],
        [2.4724e-03, 9.9753e-01],
        [1.1793e-08, 1.0000e+00],
        [3.9086e-08, 1.0000e+00],
        [2.3615e-06, 1.0000e+00],
        [6.3730e-01, 3.6270e-01],
        [9.9458e-01, 5.4158e-03],
        [9.9971e-01, 2.8517e-04],
        [5.1296e-09, 1.0000e+00],
        [2.7707e-03, 9.9723e-01],
        [2.2898e-07, 1.0000e+00],
        [8.3472e-09, 1.0000e+00],
        [7.7056e-07, 1.0000e+00],
        [1.9655e-07, 1.0000e+00],
        [9.5489e-07, 1.0000e+00],
        [4.3168e-07, 1.0000e+00],
        [1.3135e-06, 1.0000e+00],
        [1.6045e-09, 1.0000e+00],
        [9.3087e-05, 9.9991e-01],
        [2.4428e-05, 9.9998e-01],
        [1.3783e-04, 9.9986e-01],
        [9.9987e-01, 1.3277e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[5.1691e-09, 1.0000e+00],
        [6.3618e-03, 9.9364e-01],
        [1.9744e-06, 1.0000e+00],
        [9.9476e-01, 5.2389e-03],
        [1.7600e-07, 1.0000e+00],
        [5.8419e-05, 9.9994e-01],
        [7.8597e-05, 9.9992e-01],
        [9.9999e-01, 1.0381e-05],
        [8.6133e-07, 1.0000e+00],
        [2.1352e-04, 9.9979e-01],
        [2.0009e-01, 7.9991e-01],
        [1.1750e-07, 1.0000e+00],
        [3.3083e-07, 1.0000e+00],
        [9.9990e-01, 1.0083e-04],
        [1.0581e-04, 9.9989e-01],
        [5.0659e-09, 1.0000e+00],
        [7.9086e-05, 9.9992e-01],
        [9.9868e-01, 1.3248e-03],
        [3.3168e-03, 9.9668e-01],
        [9.9989e-01, 1.1248e-04],
        [9.9918e-05, 9.9990e-01],
        [3.1485e-03, 9.9685e-01],
        [1.2729e-03, 9.9873e-01],
        [2.4427e-06, 1.0000e+00],
        [6.5271e-09, 1.0000e+00],
        [8.0565e-08, 1.0000e+00],
        [9.3590e-05, 9.9991e-01],
        [6.2708e-06, 9.9999e-01],
        [2.9785e-06, 1.0000e+00],
        [3.6336e-06, 1.0000e+00],
        [3.0171e-08, 1.0000e+00],
        [6.0898e-08, 1.0000e+00],
        [1.3872e-03, 9.9861e-01],
        [1.8462e-09, 1.0000e+00],
        [3.5152e-10, 1.0000e+00],
        [1.2218e-05, 9.9999e-01],
        [2.0245e-06, 1.0000e+00],
        [5.5326e-04, 9.9945e-01],
        [1.2008e-06, 1.0000e+00],
        [3.7538e-07, 1.0000e+00],
        [2.6983e-06, 1.0000e+00],
        [9.9984e-01, 1.5616e-04],
        [2.2069e-05, 9.9998e-01],
        [2.5690e-04, 9.9974e-01],
        [7.6177e-05, 9.9992e-01],
        [1.0502e-03, 9.9895e-01],
        [9.4607e-01, 5.3934e-02],
        [1.2349e-07, 1.0000e+00],
        [9.9994e-01, 6.2787e-05],
        [5.8723e-08, 1.0000e+00],
        [2.3813e-10, 1.0000e+00],
        [9.0972e-06, 9.9999e-01],
        [3.1255e-01, 6.8745e-01],
        [2.6338e-05, 9.9997e-01],
        [9.9961e-01, 3.9003e-04],
        [9.9981e-01, 1.8699e-04],
        [3.4141e-07, 1.0000e+00],
        [2.9075e-13, 1.0000e+00],
        [9.9980e-01, 2.0246e-04],
        [4.8026e-09, 1.0000e+00],
        [3.1537e-07, 1.0000e+00],
        [9.9912e-01, 8.7587e-04],
        [1.2350e-04, 9.9988e-01],
        [7.2177e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9211e-07, 1.0000e+00],
        [9.9982e-01, 1.8357e-04],
        [4.7124e-06, 1.0000e+00],
        [1.5931e-04, 9.9984e-01],
        [1.9688e-03, 9.9803e-01],
        [6.6000e-04, 9.9934e-01],
        [9.9902e-01, 9.8190e-04],
        [9.9978e-01, 2.2464e-04],
        [1.0228e-05, 9.9999e-01],
        [1.6502e-05, 9.9998e-01],
        [9.9961e-01, 3.9027e-04],
        [1.3649e-04, 9.9986e-01],
        [1.7974e-05, 9.9998e-01],
        [5.5801e-05, 9.9994e-01],
        [1.3000e-08, 1.0000e+00],
        [1.4484e-05, 9.9999e-01],
        [5.7708e-09, 1.0000e+00],
        [7.7803e-08, 1.0000e+00],
        [2.3665e-04, 9.9976e-01],
        [2.9640e-04, 9.9970e-01],
        [9.9986e-01, 1.3920e-04],
        [9.9968e-01, 3.2192e-04],
        [9.9838e-01, 1.6240e-03],
        [1.0461e-08, 1.0000e+00],
        [2.5286e-09, 1.0000e+00],
        [1.9762e-13, 1.0000e+00],
        [3.5549e-04, 9.9964e-01],
        [9.9979e-01, 2.1332e-04],
        [9.9970e-01, 3.0000e-04],
        [9.3413e-08, 1.0000e+00],
        [6.4471e-02, 9.3553e-01],
        [6.3944e-04, 9.9936e-01],
        [9.9850e-01, 1.5025e-03],
        [5.9478e-09, 1.0000e+00],
        [6.3932e-02, 9.3607e-01],
        [2.7716e-03, 9.9723e-01],
        [1.9686e-05, 9.9998e-01],
        [6.7094e-04, 9.9933e-01],
        [6.4198e-06, 9.9999e-01],
        [2.9847e-04, 9.9970e-01],
        [2.8968e-09, 1.0000e+00],
        [2.0087e-07, 1.0000e+00],
        [3.4859e-05, 9.9997e-01],
        [9.9969e-01, 3.1271e-04],
        [9.9754e-10, 1.0000e+00],
        [9.9993e-01, 6.7388e-05],
        [7.9584e-01, 2.0416e-01],
        [3.8776e-05, 9.9996e-01],
        [6.2110e-08, 1.0000e+00],
        [9.9791e-01, 2.0925e-03],
        [1.5020e-08, 1.0000e+00],
        [9.9863e-01, 1.3693e-03],
        [4.1129e-05, 9.9996e-01],
        [3.9001e-05, 9.9996e-01],
        [4.4840e-07, 1.0000e+00],
        [5.5233e-05, 9.9994e-01],
        [7.7269e-08, 1.0000e+00],
        [1.8150e-06, 1.0000e+00],
        [1.4469e-06, 1.0000e+00],
        [9.9985e-01, 1.5189e-04],
        [3.0302e-04, 9.9970e-01],
        [5.5706e-07, 1.0000e+00],
        [1.1237e-03, 9.9888e-01],
        [7.1453e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2938e-05, 9.9999e-01],
        [9.9991e-01, 9.1978e-05],
        [9.0330e-09, 1.0000e+00],
        [9.9992e-01, 7.6179e-05],
        [3.0825e-06, 1.0000e+00],
        [4.0887e-10, 1.0000e+00],
        [2.9959e-04, 9.9970e-01],
        [3.0253e-06, 1.0000e+00],
        [2.1784e-09, 1.0000e+00],
        [6.2572e-03, 9.9374e-01],
        [5.7688e-05, 9.9994e-01],
        [9.9995e-01, 4.5718e-05],
        [8.2885e-08, 1.0000e+00],
        [9.9990e-01, 9.8521e-05],
        [1.0133e-06, 1.0000e+00],
        [3.6938e-05, 9.9996e-01],
        [1.1075e-04, 9.9989e-01],
        [9.9989e-01, 1.1139e-04],
        [3.1923e-06, 1.0000e+00],
        [5.6447e-04, 9.9944e-01],
        [3.3391e-01, 6.6609e-01],
        [7.3668e-08, 1.0000e+00],
        [2.7322e-05, 9.9997e-01],
        [9.1889e-05, 9.9991e-01],
        [5.4499e-08, 1.0000e+00],
        [7.3377e-12, 1.0000e+00],
        [3.1094e-04, 9.9969e-01],
        [1.2421e-08, 1.0000e+00],
        [9.9996e-01, 4.3049e-05],
        [9.9996e-01, 4.0918e-05],
        [7.9176e-06, 9.9999e-01],
        [9.9546e-01, 4.5389e-03],
        [2.5894e-08, 1.0000e+00],
        [1.3307e-04, 9.9987e-01],
        [9.9980e-01, 2.0068e-04],
        [9.9995e-01, 5.2593e-05],
        [7.2040e-01, 2.7960e-01],
        [2.8420e-04, 9.9972e-01],
        [1.3048e-07, 1.0000e+00],
        [5.0935e-10, 1.0000e+00],
        [1.3296e-06, 1.0000e+00],
        [9.9978e-01, 2.1781e-04],
        [3.1739e-08, 1.0000e+00],
        [1.9398e-05, 9.9998e-01],
        [9.9984e-01, 1.6130e-04],
        [1.3499e-08, 1.0000e+00],
        [9.9932e-01, 6.8445e-04],
        [1.4216e-04, 9.9986e-01],
        [6.1544e-05, 9.9994e-01],
        [4.7342e-07, 1.0000e+00],
        [1.3931e-04, 9.9986e-01],
        [1.3115e-08, 1.0000e+00],
        [2.0465e-06, 1.0000e+00],
        [3.3211e-02, 9.6679e-01],
        [6.9764e-06, 9.9999e-01],
        [4.3464e-05, 9.9996e-01],
        [1.2295e-10, 1.0000e+00],
        [2.2277e-03, 9.9777e-01],
        [9.9537e-01, 4.6308e-03],
        [9.9363e-01, 6.3701e-03],
        [7.6537e-06, 9.9999e-01],
        [1.2022e-08, 1.0000e+00],
        [4.8319e-08, 1.0000e+00],
        [1.1912e-03, 9.9881e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.3397e-05, 9.9997e-01],
        [1.9924e-01, 8.0076e-01],
        [1.0343e-09, 1.0000e+00],
        [1.3429e-04, 9.9987e-01],
        [5.9898e-03, 9.9401e-01],
        [2.9387e-10, 1.0000e+00],
        [2.7731e-04, 9.9972e-01],
        [9.9970e-01, 2.9653e-04],
        [8.3321e-08, 1.0000e+00],
        [9.9994e-01, 5.9457e-05],
        [5.7387e-09, 1.0000e+00],
        [5.9777e-03, 9.9402e-01],
        [1.9884e-10, 1.0000e+00],
        [1.0052e-06, 1.0000e+00],
        [9.9905e-01, 9.5096e-04],
        [9.9972e-01, 2.8233e-04],
        [1.9114e-08, 1.0000e+00],
        [1.3238e-04, 9.9987e-01],
        [1.0000e+00, 3.5448e-07],
        [8.3771e-06, 9.9999e-01],
        [6.7539e-04, 9.9932e-01],
        [5.1415e-06, 9.9999e-01],
        [1.1758e-03, 9.9882e-01],
        [9.9993e-01, 6.9016e-05],
        [1.2769e-05, 9.9999e-01],
        [1.1290e-06, 1.0000e+00],
        [2.6267e-06, 1.0000e+00],
        [9.6187e-05, 9.9990e-01],
        [1.6078e-07, 1.0000e+00],
        [8.0433e-04, 9.9920e-01],
        [1.6130e-03, 9.9839e-01],
        [1.5996e-07, 1.0000e+00],
        [1.3355e-03, 9.9866e-01],
        [6.7232e-07, 1.0000e+00],
        [6.2574e-01, 3.7426e-01],
        [9.9500e-01, 4.9990e-03],
        [9.9989e-01, 1.0618e-04],
        [3.4402e-09, 1.0000e+00],
        [2.3578e-03, 9.9764e-01],
        [3.0442e-03, 9.9696e-01],
        [2.4417e-05, 9.9998e-01],
        [1.3317e-09, 1.0000e+00],
        [7.2059e-06, 9.9999e-01],
        [4.1137e-10, 1.0000e+00],
        [2.0235e-06, 1.0000e+00],
        [6.4238e-04, 9.9936e-01],
        [9.9752e-01, 2.4772e-03],
        [3.0307e-06, 1.0000e+00],
        [9.9971e-01, 2.9104e-04],
        [4.9471e-02, 9.5053e-01],
        [9.9684e-01, 3.1630e-03],
        [3.6019e-08, 1.0000e+00],
        [8.3147e-05, 9.9992e-01],
        [9.9904e-01, 9.5676e-04],
        [3.8166e-07, 1.0000e+00],
        [2.7208e-09, 1.0000e+00],
        [8.0759e-05, 9.9992e-01],
        [3.0575e-04, 9.9969e-01],
        [6.1543e-11, 1.0000e+00],
        [9.9985e-01, 1.5058e-04],
        [2.9870e-07, 1.0000e+00],
        [2.4081e-04, 9.9976e-01],
        [7.7367e-07, 1.0000e+00],
        [1.4446e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.9254e-05, 9.9995e-01],
        [9.9995e-01, 4.9694e-05],
        [9.9976e-01, 2.3901e-04],
        [1.6833e-05, 9.9998e-01],
        [3.7988e-06, 1.0000e+00],
        [1.0818e-05, 9.9999e-01],
        [3.8397e-07, 1.0000e+00],
        [1.5555e-07, 1.0000e+00],
        [9.7312e-01, 2.6885e-02],
        [6.1899e-01, 3.8101e-01],
        [5.6777e-01, 4.3223e-01],
        [9.9983e-01, 1.6937e-04],
        [9.9988e-01, 1.1505e-04],
        [7.2478e-03, 9.9275e-01],
        [9.2755e-06, 9.9999e-01],
        [1.6759e-07, 1.0000e+00],
        [1.1091e-09, 1.0000e+00],
        [1.6438e-02, 9.8356e-01],
        [9.9686e-01, 3.1438e-03],
        [1.4370e-08, 1.0000e+00],
        [1.6158e-02, 9.8384e-01],
        [8.2677e-06, 9.9999e-01],
        [3.6734e-06, 1.0000e+00],
        [1.1473e-03, 9.9885e-01],
        [7.0329e-13, 1.0000e+00],
        [5.1570e-03, 9.9484e-01],
        [4.7152e-08, 1.0000e+00],
        [5.4526e-06, 9.9999e-01],
        [7.8429e-01, 2.1571e-01],
        [2.4131e-10, 1.0000e+00],
        [8.7244e-01, 1.2756e-01],
        [3.1334e-06, 1.0000e+00],
        [7.6820e-05, 9.9992e-01],
        [2.0163e-06, 1.0000e+00],
        [9.9986e-01, 1.3678e-04],
        [2.9663e-09, 1.0000e+00],
        [1.1613e-08, 1.0000e+00],
        [2.9894e-05, 9.9997e-01],
        [1.1053e-09, 1.0000e+00],
        [2.4172e-04, 9.9976e-01],
        [9.9974e-01, 2.5642e-04],
        [6.5599e-03, 9.9344e-01],
        [1.9953e-06, 1.0000e+00],
        [4.0246e-04, 9.9960e-01],
        [1.3698e-09, 1.0000e+00],
        [9.5402e-02, 9.0460e-01],
        [9.1956e-01, 8.0443e-02],
        [9.9984e-01, 1.6381e-04],
        [1.6155e-09, 1.0000e+00],
        [1.4767e-06, 1.0000e+00],
        [9.5141e-06, 9.9999e-01],
        [1.0455e-03, 9.9895e-01],
        [1.0286e-10, 1.0000e+00],
        [9.9987e-01, 1.3384e-04],
        [7.1229e-09, 1.0000e+00],
        [1.8116e-06, 1.0000e+00],
        [6.9906e-06, 9.9999e-01],
        [3.1804e-04, 9.9968e-01],
        [1.2378e-03, 9.9876e-01],
        [9.9730e-01, 2.7017e-03],
        [3.3680e-07, 1.0000e+00],
        [2.0271e-05, 9.9998e-01],
        [3.4575e-12, 1.0000e+00],
        [9.9979e-01, 2.0973e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.1548e-08, 1.0000e+00],
        [9.9975e-01, 2.5302e-04],
        [8.4125e-01, 1.5875e-01],
        [4.8534e-08, 1.0000e+00],
        [1.5537e-06, 1.0000e+00],
        [9.9982e-01, 1.7607e-04],
        [3.1201e-05, 9.9997e-01],
        [2.2897e-10, 1.0000e+00],
        [3.9458e-09, 1.0000e+00],
        [5.3969e-07, 1.0000e+00],
        [9.9946e-01, 5.3573e-04],
        [1.8019e-06, 1.0000e+00],
        [8.8845e-05, 9.9991e-01],
        [7.9801e-07, 1.0000e+00],
        [2.2792e-08, 1.0000e+00],
        [4.9414e-06, 1.0000e+00],
        [2.0031e-09, 1.0000e+00],
        [1.3336e-06, 1.0000e+00],
        [1.0925e-04, 9.9989e-01],
        [4.9950e-08, 1.0000e+00],
        [1.0000e+00, 1.1348e-06],
        [2.1216e-10, 1.0000e+00],
        [1.5152e-01, 8.4848e-01],
        [1.0484e-06, 1.0000e+00],
        [8.6098e-08, 1.0000e+00],
        [1.4591e-05, 9.9999e-01],
        [1.8395e-04, 9.9982e-01],
        [2.3798e-11, 1.0000e+00],
        [3.7583e-05, 9.9996e-01],
        [9.9944e-01, 5.5920e-04],
        [1.0497e-05, 9.9999e-01],
        [2.3995e-08, 1.0000e+00],
        [6.8116e-10, 1.0000e+00],
        [6.0913e-05, 9.9994e-01],
        [1.3136e-06, 1.0000e+00],
        [1.0867e-07, 1.0000e+00],
        [9.9869e-01, 1.3145e-03],
        [3.7296e-08, 1.0000e+00],
        [5.2755e-08, 1.0000e+00],
        [9.9989e-01, 1.0957e-04],
        [2.2699e-01, 7.7301e-01],
        [5.4715e-06, 9.9999e-01],
        [9.9979e-01, 2.1045e-04],
        [9.0978e-07, 1.0000e+00],
        [9.9827e-01, 1.7336e-03],
        [9.9230e-01, 7.6952e-03],
        [4.5443e-11, 1.0000e+00],
        [9.9927e-01, 7.3448e-04],
        [1.6167e-11, 1.0000e+00],
        [1.5578e-06, 1.0000e+00],
        [9.6350e-01, 3.6495e-02],
        [6.7335e-08, 1.0000e+00],
        [7.8897e-06, 9.9999e-01],
        [1.3336e-06, 1.0000e+00],
        [1.0579e-09, 1.0000e+00],
        [1.8675e-06, 1.0000e+00],
        [9.9997e-01, 2.6311e-05],
        [7.7161e-03, 9.9228e-01],
        [6.6895e-06, 9.9999e-01],
        [1.6505e-03, 9.9835e-01],
        [9.9988e-01, 1.2174e-04],
        [9.8512e-01, 1.4883e-02],
        [8.2134e-08, 1.0000e+00],
        [7.7427e-10, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9668e-01, 3.3187e-03],
        [1.0714e-04, 9.9989e-01],
        [5.5520e-10, 1.0000e+00],
        [7.5568e-06, 9.9999e-01],
        [2.6424e-05, 9.9997e-01],
        [5.3333e-07, 1.0000e+00],
        [3.5932e-05, 9.9996e-01],
        [1.2989e-05, 9.9999e-01],
        [9.9983e-01, 1.6821e-04],
        [2.0517e-06, 1.0000e+00],
        [1.2524e-09, 1.0000e+00],
        [2.2584e-10, 1.0000e+00],
        [7.1044e-06, 9.9999e-01],
        [5.9416e-10, 1.0000e+00],
        [9.9998e-01, 2.4019e-05],
        [5.9253e-05, 9.9994e-01],
        [9.9966e-01, 3.3988e-04],
        [3.5882e-03, 9.9641e-01],
        [4.6863e-09, 1.0000e+00],
        [4.9462e-08, 1.0000e+00],
        [4.9324e-07, 1.0000e+00],
        [4.2824e-10, 1.0000e+00],
        [6.2361e-08, 1.0000e+00],
        [3.0143e-09, 1.0000e+00],
        [9.1585e-09, 1.0000e+00],
        [5.9077e-09, 1.0000e+00],
        [1.2165e-06, 1.0000e+00],
        [9.9972e-01, 2.7643e-04],
        [1.2287e-08, 1.0000e+00],
        [9.5783e-05, 9.9990e-01],
        [1.4750e-08, 1.0000e+00],
        [2.6961e-08, 1.0000e+00],
        [1.5896e-10, 1.0000e+00],
        [2.7814e-04, 9.9972e-01],
        [2.2486e-03, 9.9775e-01],
        [9.8974e-06, 9.9999e-01],
        [9.9980e-01, 1.9656e-04],
        [9.9979e-01, 2.0709e-04],
        [3.6268e-11, 1.0000e+00],
        [8.5211e-12, 1.0000e+00],
        [9.9961e-01, 3.8758e-04],
        [7.4182e-07, 1.0000e+00],
        [4.9744e-08, 1.0000e+00],
        [4.9646e-07, 1.0000e+00],
        [1.2996e-08, 1.0000e+00],
        [2.3659e-10, 1.0000e+00],
        [9.5738e-01, 4.2616e-02],
        [7.1968e-09, 1.0000e+00],
        [4.4832e-04, 9.9955e-01],
        [7.0701e-09, 1.0000e+00],
        [9.9997e-01, 3.0814e-05],
        [9.9797e-01, 2.0344e-03],
        [3.6319e-09, 1.0000e+00],
        [1.4901e-04, 9.9985e-01],
        [2.9661e-09, 1.0000e+00],
        [1.1318e-06, 1.0000e+00],
        [1.2434e-05, 9.9999e-01],
        [8.8569e-01, 1.1431e-01],
        [4.9640e-09, 1.0000e+00],
        [1.3914e-08, 1.0000e+00],
        [2.2451e-06, 1.0000e+00],
        [9.9986e-01, 1.3627e-04],
        [9.8610e-01, 1.3904e-02],
        [2.0013e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1604e-09, 1.0000e+00],
        [5.8670e-05, 9.9994e-01],
        [1.9805e-07, 1.0000e+00],
        [9.9967e-01, 3.3067e-04],
        [4.6606e-07, 1.0000e+00],
        [6.5989e-07, 1.0000e+00],
        [8.3852e-07, 1.0000e+00],
        [2.8941e-09, 1.0000e+00],
        [1.4263e-09, 1.0000e+00],
        [5.7081e-08, 1.0000e+00],
        [3.2509e-07, 1.0000e+00],
        [8.6373e-05, 9.9991e-01],
        [8.0900e-10, 1.0000e+00],
        [9.4798e-07, 1.0000e+00],
        [7.4023e-08, 1.0000e+00],
        [2.5722e-07, 1.0000e+00],
        [8.9432e-09, 1.0000e+00],
        [7.5873e-10, 1.0000e+00],
        [1.4291e-09, 1.0000e+00],
        [1.4998e-05, 9.9998e-01],
        [4.3350e-07, 1.0000e+00],
        [6.9274e-08, 1.0000e+00],
        [1.3217e-03, 9.9868e-01],
        [2.7968e-08, 1.0000e+00],
        [8.9432e-09, 1.0000e+00],
        [4.5291e-07, 1.0000e+00],
        [7.3803e-08, 1.0000e+00],
        [3.2173e-05, 9.9997e-01],
        [7.5483e-07, 1.0000e+00],
        [3.9059e-10, 1.0000e+00],
        [9.9989e-01, 1.1324e-04],
        [9.9795e-01, 2.0529e-03],
        [4.9468e-08, 1.0000e+00],
        [3.6853e-09, 1.0000e+00],
        [4.0032e-05, 9.9996e-01],
        [9.4144e-06, 9.9999e-01],
        [2.8169e-01, 7.1831e-01],
        [1.3369e-09, 1.0000e+00],
        [9.9982e-01, 1.7921e-04],
        [1.8505e-08, 1.0000e+00],
        [4.3367e-12, 1.0000e+00],
        [2.5744e-06, 1.0000e+00],
        [2.5821e-07, 1.0000e+00],
        [7.9649e-10, 1.0000e+00],
        [3.7404e-06, 1.0000e+00],
        [1.8299e-06, 1.0000e+00],
        [7.9633e-01, 2.0367e-01],
        [9.9987e-01, 1.3464e-04],
        [1.5658e-04, 9.9984e-01],
        [1.3821e-06, 1.0000e+00],
        [5.0999e-01, 4.9001e-01],
        [2.0865e-08, 1.0000e+00],
        [1.2630e-08, 1.0000e+00],
        [2.1427e-09, 1.0000e+00],
        [6.3944e-08, 1.0000e+00],
        [3.2018e-05, 9.9997e-01],
        [9.9990e-01, 1.0297e-04],
        [9.9983e-01, 1.7252e-04],
        [4.8620e-01, 5.1380e-01],
        [1.8727e-10, 1.0000e+00],
        [3.7501e-07, 1.0000e+00],
        [4.1793e-10, 1.0000e+00],
        [9.9987e-01, 1.3406e-04],
        [5.8569e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.8469e-07, 1.0000e+00],
        [1.0788e-08, 1.0000e+00],
        [8.8375e-10, 1.0000e+00],
        [7.8876e-04, 9.9921e-01],
        [9.9948e-01, 5.1850e-04],
        [1.1141e-07, 1.0000e+00],
        [2.1818e-07, 1.0000e+00],
        [3.2676e-07, 1.0000e+00],
        [3.5503e-10, 1.0000e+00],
        [2.0417e-05, 9.9998e-01],
        [2.4663e-05, 9.9998e-01],
        [5.3298e-07, 1.0000e+00],
        [9.9992e-01, 8.1818e-05],
        [7.9178e-07, 1.0000e+00],
        [9.9963e-01, 3.7195e-04],
        [8.6873e-07, 1.0000e+00],
        [1.6520e-08, 1.0000e+00],
        [9.7732e-01, 2.2681e-02],
        [8.7221e-08, 1.0000e+00],
        [5.0318e-09, 1.0000e+00],
        [1.6867e-11, 1.0000e+00],
        [4.5554e-04, 9.9954e-01],
        [9.9473e-01, 5.2670e-03],
        [9.9973e-01, 2.7455e-04],
        [9.9400e-01, 6.0037e-03],
        [1.0604e-06, 1.0000e+00],
        [6.4761e-08, 1.0000e+00],
        [3.5433e-05, 9.9996e-01],
        [7.4997e-10, 1.0000e+00],
        [8.6838e-01, 1.3162e-01],
        [8.5408e-01, 1.4592e-01],
        [1.1745e-08, 1.0000e+00],
        [9.9994e-01, 6.0098e-05],
        [6.1891e-04, 9.9938e-01],
        [5.4118e-06, 9.9999e-01],
        [9.9995e-01, 5.0983e-05],
        [5.2613e-05, 9.9995e-01],
        [5.9371e-07, 1.0000e+00],
        [9.9986e-01, 1.3982e-04],
        [5.9371e-07, 1.0000e+00],
        [3.4639e-06, 1.0000e+00],
        [8.8822e-10, 1.0000e+00],
        [4.0155e-07, 1.0000e+00],
        [4.5481e-10, 1.0000e+00],
        [8.4299e-01, 1.5701e-01],
        [2.0326e-07, 1.0000e+00],
        [6.0412e-03, 9.9396e-01],
        [5.9887e-07, 1.0000e+00],
        [8.3992e-10, 1.0000e+00],
        [4.4767e-07, 1.0000e+00],
        [2.2633e-05, 9.9998e-01],
        [1.8072e-09, 1.0000e+00],
        [4.8370e-10, 1.0000e+00],
        [9.5690e-01, 4.3097e-02],
        [1.5414e-10, 1.0000e+00],
        [6.8455e-06, 9.9999e-01],
        [2.1614e-12, 1.0000e+00],
        [2.7874e-07, 1.0000e+00],
        [4.0809e-07, 1.0000e+00],
        [6.4746e-08, 1.0000e+00],
        [9.9974e-01, 2.6423e-04],
        [1.9206e-07, 1.0000e+00],
        [6.9112e-04, 9.9931e-01],
        [2.3824e-04, 9.9976e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.1105e-09, 1.0000e+00],
        [2.8457e-03, 9.9715e-01],
        [1.3007e-01, 8.6993e-01],
        [8.2333e-12, 1.0000e+00],
        [4.0882e-06, 1.0000e+00],
        [4.2507e-06, 1.0000e+00],
        [8.3815e-03, 9.9162e-01],
        [7.8782e-09, 1.0000e+00],
        [9.2678e-07, 1.0000e+00],
        [4.7233e-08, 1.0000e+00],
        [4.6732e-04, 9.9953e-01],
        [9.9689e-01, 3.1150e-03],
        [1.1790e-02, 9.8821e-01],
        [1.0417e-06, 1.0000e+00],
        [3.1776e-01, 6.8224e-01],
        [3.7005e-08, 1.0000e+00],
        [9.9513e-01, 4.8739e-03],
        [9.9986e-01, 1.4187e-04],
        [9.8800e-01, 1.1995e-02],
        [4.0203e-12, 1.0000e+00],
        [4.0514e-07, 1.0000e+00],
        [1.4220e-03, 9.9858e-01],
        [2.6463e-08, 1.0000e+00],
        [9.7409e-07, 1.0000e+00],
        [5.5761e-12, 1.0000e+00],
        [2.9700e-07, 1.0000e+00],
        [1.1560e-10, 1.0000e+00],
        [1.0000e+00, 2.0234e-06],
        [3.3811e-09, 1.0000e+00],
        [1.9235e-07, 1.0000e+00],
        [1.1081e-07, 1.0000e+00],
        [9.9327e-01, 6.7283e-03],
        [9.9990e-01, 9.8028e-05],
        [3.2981e-06, 1.0000e+00],
        [9.9898e-01, 1.0160e-03],
        [9.9976e-01, 2.3613e-04],
        [4.9164e-10, 1.0000e+00],
        [8.2701e-05, 9.9992e-01],
        [7.0353e-08, 1.0000e+00],
        [9.6073e-01, 3.9274e-02],
        [1.6999e-08, 1.0000e+00],
        [9.3007e-06, 9.9999e-01],
        [6.6276e-12, 1.0000e+00],
        [5.1041e-06, 9.9999e-01],
        [4.0585e-06, 1.0000e+00],
        [9.9986e-01, 1.3840e-04],
        [9.9646e-01, 3.5367e-03],
        [2.8853e-03, 9.9711e-01],
        [9.6008e-08, 1.0000e+00],
        [1.3071e-07, 1.0000e+00],
        [1.1912e-08, 1.0000e+00],
        [4.0117e-07, 1.0000e+00],
        [1.9655e-09, 1.0000e+00],
        [1.1751e-01, 8.8249e-01],
        [4.2723e-07, 1.0000e+00],
        [9.8299e-01, 1.7008e-02],
        [9.1247e-10, 1.0000e+00],
        [3.1443e-09, 1.0000e+00],
        [1.3455e-09, 1.0000e+00],
        [7.3115e-07, 1.0000e+00],
        [3.1387e-07, 1.0000e+00],
        [4.1827e-10, 1.0000e+00],
        [5.7412e-08, 1.0000e+00],
        [7.0692e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.4523e-02, 9.7548e-01],
        [1.8053e-05, 9.9998e-01],
        [8.2947e-04, 9.9917e-01],
        [1.0557e-05, 9.9999e-01],
        [9.9999e-01, 1.1234e-05],
        [9.9689e-01, 3.1150e-03],
        [1.7515e-05, 9.9998e-01],
        [4.3421e-05, 9.9996e-01],
        [3.2769e-08, 1.0000e+00],
        [1.9862e-05, 9.9998e-01],
        [1.4320e-08, 1.0000e+00],
        [4.0243e-09, 1.0000e+00],
        [9.9997e-01, 3.2378e-05],
        [2.7398e-07, 1.0000e+00],
        [9.9694e-01, 3.0582e-03],
        [6.0450e-07, 1.0000e+00],
        [8.8857e-08, 1.0000e+00],
        [1.2660e-05, 9.9999e-01],
        [3.5902e-11, 1.0000e+00],
        [1.2200e-04, 9.9988e-01],
        [9.9983e-01, 1.7106e-04],
        [1.0185e-06, 1.0000e+00],
        [2.9498e-07, 1.0000e+00],
        [6.7214e-07, 1.0000e+00],
        [8.4754e-08, 1.0000e+00],
        [1.6168e-06, 1.0000e+00],
        [1.9012e-03, 9.9810e-01],
        [2.7414e-07, 1.0000e+00],
        [9.9787e-01, 2.1342e-03],
        [3.1852e-05, 9.9997e-01],
        [5.4352e-09, 1.0000e+00],
        [9.9905e-01, 9.4540e-04],
        [5.5595e-07, 1.0000e+00],
        [9.9995e-01, 5.2012e-05],
        [4.7002e-04, 9.9953e-01],
        [1.7655e-05, 9.9998e-01],
        [9.0939e-05, 9.9991e-01],
        [1.2070e-01, 8.7930e-01],
        [9.9987e-01, 1.3404e-04],
        [6.5883e-01, 3.4117e-01],
        [7.1566e-10, 1.0000e+00],
        [3.1923e-01, 6.8077e-01],
        [3.0736e-08, 1.0000e+00],
        [9.3154e-07, 1.0000e+00],
        [2.1674e-06, 1.0000e+00],
        [6.4652e-06, 9.9999e-01],
        [1.0844e-06, 1.0000e+00],
        [1.8863e-09, 1.0000e+00],
        [9.9951e-01, 4.9413e-04],
        [9.2101e-07, 1.0000e+00],
        [9.9969e-01, 3.0579e-04],
        [3.4808e-07, 1.0000e+00],
        [7.3944e-07, 1.0000e+00],
        [6.6119e-08, 1.0000e+00],
        [2.5651e-10, 1.0000e+00],
        [9.9975e-01, 2.5298e-04],
        [9.9984e-01, 1.6429e-04],
        [3.1076e-07, 1.0000e+00],
        [3.9936e-02, 9.6006e-01],
        [1.0397e-09, 1.0000e+00],
        [9.7103e-01, 2.8974e-02],
        [4.7002e-01, 5.2998e-01],
        [8.8880e-09, 1.0000e+00],
        [4.5710e-04, 9.9954e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.9706e-01, 2.0294e-01],
        [9.3128e-01, 6.8717e-02],
        [1.2324e-04, 9.9988e-01],
        [1.8303e-10, 1.0000e+00],
        [8.8697e-01, 1.1303e-01],
        [2.6055e-10, 1.0000e+00],
        [3.2754e-05, 9.9997e-01],
        [1.0427e-07, 1.0000e+00],
        [9.9962e-01, 3.8135e-04],
        [8.1273e-07, 1.0000e+00],
        [9.9994e-01, 5.5552e-05],
        [1.3911e-08, 1.0000e+00],
        [3.5141e-05, 9.9996e-01],
        [9.9962e-01, 3.7844e-04],
        [2.0637e-06, 1.0000e+00],
        [1.1710e-06, 1.0000e+00],
        [3.1420e-01, 6.8580e-01],
        [9.9955e-01, 4.5099e-04],
        [3.8467e-06, 1.0000e+00],
        [8.7972e-06, 9.9999e-01],
        [1.8182e-07, 1.0000e+00],
        [1.6269e-10, 1.0000e+00],
        [2.2692e-10, 1.0000e+00],
        [9.5952e-01, 4.0479e-02],
        [1.0049e-04, 9.9990e-01],
        [1.9677e-11, 1.0000e+00],
        [9.0017e-09, 1.0000e+00],
        [9.9984e-01, 1.6364e-04],
        [3.0496e-08, 1.0000e+00],
        [9.9992e-01, 8.2921e-05],
        [9.9968e-01, 3.1510e-04],
        [9.9991e-01, 9.2410e-05],
        [1.6364e-09, 1.0000e+00],
        [3.8444e-04, 9.9962e-01],
        [3.5082e-06, 1.0000e+00],
        [9.9991e-01, 9.3351e-05],
        [3.9445e-11, 1.0000e+00],
        [4.4382e-07, 1.0000e+00],
        [9.9992e-01, 7.9949e-05],
        [8.8697e-01, 1.1303e-01],
        [1.5752e-04, 9.9984e-01],
        [3.9287e-08, 1.0000e+00],
        [2.6432e-02, 9.7357e-01],
        [4.4959e-06, 1.0000e+00],
        [1.6155e-02, 9.8385e-01],
        [1.3033e-06, 1.0000e+00],
        [1.1556e-07, 1.0000e+00],
        [1.1719e-08, 1.0000e+00],
        [9.4337e-01, 5.6633e-02],
        [9.9988e-01, 1.1707e-04],
        [2.9775e-05, 9.9997e-01],
        [9.3454e-01, 6.5458e-02],
        [1.2037e-08, 1.0000e+00],
        [1.7365e-08, 1.0000e+00],
        [2.4582e-08, 1.0000e+00],
        [4.8413e-08, 1.0000e+00],
        [1.8031e-11, 1.0000e+00],
        [2.3175e-09, 1.0000e+00],
        [3.0605e-06, 1.0000e+00],
        [2.9115e-06, 1.0000e+00],
        [9.9969e-01, 3.1078e-04],
        [6.2652e-05, 9.9994e-01],
        [9.9584e-10, 1.0000e+00],
        [9.9721e-01, 2.7879e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8375e-10, 1.0000e+00],
        [1.9665e-07, 1.0000e+00],
        [3.0321e-05, 9.9997e-01],
        [4.0749e-08, 1.0000e+00],
        [8.4775e-04, 9.9915e-01],
        [3.1407e-10, 1.0000e+00],
        [2.9128e-08, 1.0000e+00],
        [3.4515e-07, 1.0000e+00],
        [5.5346e-08, 1.0000e+00],
        [9.9988e-01, 1.1757e-04],
        [9.9982e-01, 1.7945e-04],
        [9.9977e-01, 2.2497e-04],
        [6.7960e-04, 9.9932e-01],
        [6.5472e-03, 9.9345e-01],
        [2.0650e-08, 1.0000e+00],
        [7.5535e-09, 1.0000e+00],
        [1.5487e-10, 1.0000e+00],
        [9.9970e-01, 3.0435e-04],
        [3.6672e-07, 1.0000e+00],
        [1.1799e-07, 1.0000e+00],
        [1.5960e-03, 9.9840e-01],
        [9.9608e-01, 3.9170e-03],
        [2.0502e-03, 9.9795e-01],
        [1.1711e-08, 1.0000e+00],
        [5.5374e-06, 9.9999e-01],
        [2.1769e-04, 9.9978e-01],
        [9.9980e-01, 2.0031e-04],
        [1.7991e-05, 9.9998e-01],
        [4.7376e-08, 1.0000e+00],
        [1.4065e-04, 9.9986e-01],
        [1.5660e-05, 9.9998e-01],
        [9.9876e-01, 1.2433e-03],
        [2.7468e-04, 9.9973e-01],
        [1.0242e-08, 1.0000e+00],
        [3.2736e-08, 1.0000e+00],
        [9.9118e-01, 8.8211e-03],
        [3.1251e-05, 9.9997e-01],
        [5.7468e-05, 9.9994e-01],
        [2.2102e-06, 1.0000e+00],
        [9.7050e-09, 1.0000e+00],
        [9.9949e-01, 5.1087e-04],
        [1.6630e-04, 9.9983e-01],
        [6.4261e-12, 1.0000e+00],
        [1.1203e-01, 8.8797e-01],
        [9.9858e-01, 1.4160e-03],
        [4.3421e-07, 1.0000e+00],
        [9.9987e-01, 1.3183e-04],
        [2.7141e-09, 1.0000e+00],
        [1.2705e-05, 9.9999e-01],
        [4.9638e-05, 9.9995e-01],
        [1.1733e-06, 1.0000e+00],
        [4.0521e-10, 1.0000e+00],
        [6.3572e-05, 9.9994e-01],
        [5.0844e-03, 9.9492e-01],
        [7.4340e-04, 9.9926e-01],
        [1.8004e-07, 1.0000e+00],
        [9.9987e-01, 1.3017e-04],
        [7.9254e-06, 9.9999e-01],
        [9.4273e-03, 9.9057e-01],
        [7.4066e-07, 1.0000e+00],
        [1.6680e-06, 1.0000e+00],
        [1.2537e-11, 1.0000e+00],
        [4.5523e-05, 9.9995e-01],
        [1.5984e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.6795e-08, 1.0000e+00],
        [7.3432e-05, 9.9993e-01],
        [9.8769e-01, 1.2306e-02],
        [7.6566e-02, 9.2343e-01],
        [9.9282e-01, 7.1768e-03],
        [9.9901e-01, 9.8764e-04],
        [6.4664e-04, 9.9935e-01],
        [3.4953e-02, 9.6505e-01],
        [8.9528e-01, 1.0472e-01],
        [1.2850e-05, 9.9999e-01],
        [1.7368e-07, 1.0000e+00],
        [2.4431e-03, 9.9756e-01],
        [4.7940e-08, 1.0000e+00],
        [1.6409e-07, 1.0000e+00],
        [2.0706e-06, 1.0000e+00],
        [3.9993e-07, 1.0000e+00],
        [1.9474e-05, 9.9998e-01],
        [1.5760e-06, 1.0000e+00],
        [1.5309e-06, 1.0000e+00],
        [9.9966e-01, 3.3511e-04],
        [9.9969e-01, 3.1001e-04],
        [2.8336e-04, 9.9972e-01],
        [6.1337e-07, 1.0000e+00],
        [1.1667e-10, 1.0000e+00],
        [5.7952e-09, 1.0000e+00],
        [9.9984e-01, 1.6205e-04],
        [7.4700e-07, 1.0000e+00],
        [1.0640e-06, 1.0000e+00],
        [6.7383e-01, 3.2617e-01],
        [2.0309e-05, 9.9998e-01],
        [1.1682e-08, 1.0000e+00],
        [2.1165e-07, 1.0000e+00],
        [1.9419e-06, 1.0000e+00],
        [9.9992e-01, 7.7117e-05],
        [4.5737e-06, 1.0000e+00],
        [5.6179e-09, 1.0000e+00],
        [7.3274e-07, 1.0000e+00],
        [9.9982e-01, 1.8294e-04],
        [6.1744e-08, 1.0000e+00],
        [1.6814e-07, 1.0000e+00],
        [2.3573e-10, 1.0000e+00],
        [1.3628e-10, 1.0000e+00],
        [4.0879e-09, 1.0000e+00],
        [8.1111e-06, 9.9999e-01],
        [9.9976e-01, 2.4143e-04],
        [8.3665e-07, 1.0000e+00],
        [6.9952e-08, 1.0000e+00],
        [3.1726e-06, 1.0000e+00],
        [9.9976e-01, 2.4144e-04],
        [2.9216e-01, 7.0784e-01],
        [2.0161e-07, 1.0000e+00],
        [4.9704e-10, 1.0000e+00],
        [6.5987e-08, 1.0000e+00],
        [8.3732e-01, 1.6268e-01],
        [1.4355e-01, 8.5645e-01],
        [9.9997e-01, 3.1025e-05],
        [2.7296e-06, 1.0000e+00],
        [7.5340e-05, 9.9992e-01],
        [8.2272e-08, 1.0000e+00],
        [1.7756e-10, 1.0000e+00],
        [4.7646e-01, 5.2354e-01],
        [2.2912e-06, 1.0000e+00],
        [1.1525e-08, 1.0000e+00],
        [1.4193e-03, 9.9858e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9998e-01, 1.7278e-05],
        [1.0717e-07, 1.0000e+00],
        [1.6897e-07, 1.0000e+00],
        [4.7183e-04, 9.9953e-01],
        [6.5115e-05, 9.9993e-01],
        [1.3007e-05, 9.9999e-01],
        [4.4065e-08, 1.0000e+00],
        [9.0256e-05, 9.9991e-01],
        [1.5829e-09, 1.0000e+00],
        [2.5441e-06, 1.0000e+00],
        [7.0312e-08, 1.0000e+00],
        [9.9569e-01, 4.3049e-03],
        [2.2694e-08, 1.0000e+00],
        [7.5829e-06, 9.9999e-01],
        [2.4058e-08, 1.0000e+00],
        [2.3765e-05, 9.9998e-01],
        [1.1880e-06, 1.0000e+00],
        [1.3328e-06, 1.0000e+00],
        [1.8388e-06, 1.0000e+00],
        [4.2227e-10, 1.0000e+00],
        [7.9589e-10, 1.0000e+00],
        [9.9994e-01, 6.4087e-05],
        [3.3634e-06, 1.0000e+00],
        [2.4058e-08, 1.0000e+00],
        [9.8244e-01, 1.7557e-02],
        [2.4124e-04, 9.9976e-01],
        [9.4525e-04, 9.9905e-01],
        [4.8296e-07, 1.0000e+00],
        [1.7599e-04, 9.9982e-01],
        [4.8198e-01, 5.1802e-01],
        [8.5141e-04, 9.9915e-01],
        [1.9938e-05, 9.9998e-01],
        [9.9994e-01, 6.4152e-05],
        [5.9908e-07, 1.0000e+00],
        [1.5648e-07, 1.0000e+00],
        [2.7323e-03, 9.9727e-01],
        [9.9909e-01, 9.0845e-04],
        [9.9980e-01, 1.9912e-04],
        [2.2008e-05, 9.9998e-01],
        [8.7659e-03, 9.9123e-01],
        [9.9991e-01, 9.2381e-05],
        [1.4301e-12, 1.0000e+00],
        [9.9982e-01, 1.7951e-04],
        [1.3161e-06, 1.0000e+00],
        [6.3217e-08, 1.0000e+00],
        [1.0689e-05, 9.9999e-01],
        [9.9968e-01, 3.1681e-04],
        [4.7228e-09, 1.0000e+00],
        [8.7593e-08, 1.0000e+00],
        [9.2654e-05, 9.9991e-01],
        [8.8172e-07, 1.0000e+00],
        [9.9967e-01, 3.3151e-04],
        [6.5519e-08, 1.0000e+00],
        [5.9766e-06, 9.9999e-01],
        [8.2100e-01, 1.7900e-01],
        [3.2835e-03, 9.9672e-01],
        [8.8633e-05, 9.9991e-01],
        [4.9051e-01, 5.0949e-01],
        [1.5544e-08, 1.0000e+00],
        [9.8815e-01, 1.1854e-02],
        [1.0000e+00, 4.8029e-06],
        [1.0710e-06, 1.0000e+00],
        [1.8298e-03, 9.9817e-01],
        [1.2409e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.0243e-01, 1.9757e-01],
        [8.5885e-05, 9.9991e-01],
        [3.6165e-05, 9.9996e-01],
        [9.9985e-01, 1.4658e-04],
        [3.8406e-08, 1.0000e+00],
        [1.3635e-04, 9.9986e-01],
        [6.9141e-16, 1.0000e+00],
        [7.9975e-01, 2.0025e-01],
        [1.1157e-06, 1.0000e+00],
        [9.9975e-01, 2.4746e-04],
        [2.2024e-04, 9.9978e-01],
        [2.2318e-05, 9.9998e-01],
        [9.9270e-01, 7.3031e-03],
        [2.7274e-05, 9.9997e-01],
        [4.4206e-08, 1.0000e+00],
        [5.7951e-08, 1.0000e+00],
        [2.5041e-05, 9.9997e-01],
        [2.5515e-09, 1.0000e+00],
        [9.9990e-01, 9.9354e-05],
        [1.0945e-09, 1.0000e+00],
        [5.6735e-01, 4.3265e-01],
        [5.0265e-01, 4.9735e-01],
        [1.5572e-07, 1.0000e+00],
        [2.4427e-07, 1.0000e+00],
        [6.3252e-05, 9.9994e-01],
        [1.4208e-05, 9.9999e-01],
        [9.6173e-01, 3.8272e-02],
        [3.3682e-01, 6.6318e-01],
        [9.9998e-01, 1.6656e-05],
        [5.5095e-01, 4.4905e-01],
        [6.2724e-10, 1.0000e+00],
        [3.6420e-04, 9.9964e-01],
        [9.7353e-01, 2.6474e-02],
        [2.3292e-07, 1.0000e+00],
        [2.9264e-03, 9.9707e-01],
        [3.2147e-05, 9.9997e-01],
        [1.6197e-04, 9.9984e-01],
        [9.9952e-01, 4.8389e-04],
        [4.6750e-03, 9.9532e-01],
        [8.3670e-09, 1.0000e+00],
        [9.8992e-01, 1.0080e-02],
        [4.9383e-07, 1.0000e+00],
        [2.6399e-06, 1.0000e+00],
        [9.3727e-07, 1.0000e+00],
        [1.6684e-05, 9.9998e-01],
        [1.3614e-06, 1.0000e+00],
        [9.9991e-01, 9.2239e-05],
        [9.9612e-01, 3.8751e-03],
        [6.4045e-03, 9.9360e-01],
        [3.7417e-07, 1.0000e+00],
        [9.0912e-09, 1.0000e+00],
        [9.8134e-09, 1.0000e+00],
        [9.9990e-01, 9.7910e-05],
        [2.1436e-06, 1.0000e+00],
        [9.2668e-08, 1.0000e+00],
        [5.2484e-03, 9.9475e-01],
        [3.6881e-03, 9.9631e-01],
        [9.9981e-01, 1.9388e-04],
        [9.2417e-01, 7.5828e-02],
        [1.1350e-08, 1.0000e+00],
        [2.9115e-03, 9.9709e-01],
        [5.3892e-08, 1.0000e+00],
        [2.1440e-03, 9.9786e-01],
        [2.4243e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3237e-08, 1.0000e+00],
        [7.5832e-07, 1.0000e+00],
        [7.3355e-07, 1.0000e+00],
        [1.6971e-12, 1.0000e+00],
        [2.1852e-06, 1.0000e+00],
        [1.8570e-05, 9.9998e-01],
        [2.0485e-07, 1.0000e+00],
        [9.9976e-01, 2.3734e-04],
        [1.3697e-04, 9.9986e-01],
        [7.4810e-04, 9.9925e-01],
        [9.9986e-01, 1.3507e-04],
        [3.7410e-05, 9.9996e-01],
        [4.2767e-08, 1.0000e+00],
        [1.3392e-04, 9.9987e-01],
        [1.4623e-06, 1.0000e+00],
        [7.8665e-06, 9.9999e-01],
        [4.3876e-07, 1.0000e+00],
        [1.6868e-04, 9.9983e-01],
        [9.9987e-01, 1.2694e-04],
        [6.3619e-08, 1.0000e+00],
        [9.8750e-01, 1.2501e-02],
        [9.9976e-01, 2.3568e-04],
        [6.7984e-04, 9.9932e-01],
        [2.0524e-05, 9.9998e-01],
        [4.2094e-09, 1.0000e+00],
        [5.0187e-10, 1.0000e+00],
        [1.0895e-05, 9.9999e-01],
        [1.0550e-07, 1.0000e+00],
        [2.7327e-07, 1.0000e+00],
        [9.9972e-01, 2.7842e-04],
        [4.1264e-04, 9.9959e-01],
        [6.8400e-06, 9.9999e-01],
        [9.9969e-01, 3.1109e-04],
        [7.2929e-08, 1.0000e+00],
        [4.0729e-04, 9.9959e-01],
        [7.0552e-06, 9.9999e-01],
        [1.0000e+00, 1.1428e-07],
        [9.9981e-01, 1.9388e-04],
        [1.0812e-03, 9.9892e-01],
        [1.1182e-03, 9.9888e-01],
        [7.1972e-05, 9.9993e-01],
        [2.4133e-07, 1.0000e+00],
        [4.6244e-03, 9.9538e-01],
        [1.5338e-06, 1.0000e+00],
        [9.9966e-01, 3.3831e-04],
        [5.1033e-07, 1.0000e+00],
        [9.9988e-01, 1.2213e-04],
        [1.2856e-07, 1.0000e+00],
        [4.4472e-06, 1.0000e+00],
        [3.4353e-03, 9.9656e-01],
        [9.9996e-01, 3.6531e-05],
        [4.5687e-02, 9.5431e-01],
        [9.9986e-01, 1.3762e-04],
        [8.4257e-02, 9.1574e-01],
        [9.9965e-01, 3.5117e-04],
        [6.5863e-04, 9.9934e-01],
        [8.7363e-03, 9.9126e-01],
        [1.2902e-08, 1.0000e+00],
        [5.5880e-09, 1.0000e+00],
        [9.8072e-07, 1.0000e+00],
        [9.9990e-01, 1.0471e-04],
        [9.9852e-01, 1.4781e-03],
        [1.4025e-07, 1.0000e+00],
        [1.7640e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.1588e-04, 9.9928e-01],
        [1.0421e-08, 1.0000e+00],
        [3.2044e-07, 1.0000e+00],
        [9.9998e-01, 1.7732e-05],
        [9.9986e-01, 1.3615e-04],
        [1.4673e-03, 9.9853e-01],
        [9.9984e-01, 1.6040e-04],
        [8.3288e-08, 1.0000e+00],
        [9.9989e-01, 1.0542e-04],
        [1.6448e-08, 1.0000e+00],
        [1.0530e-09, 1.0000e+00],
        [1.3748e-08, 1.0000e+00],
        [8.1457e-07, 1.0000e+00],
        [9.9988e-01, 1.1687e-04],
        [2.2716e-06, 1.0000e+00],
        [1.8521e-06, 1.0000e+00],
        [4.9074e-01, 5.0926e-01],
        [1.8620e-09, 1.0000e+00],
        [6.4519e-05, 9.9994e-01],
        [8.5444e-02, 9.1456e-01],
        [5.2034e-03, 9.9480e-01],
        [9.7223e-01, 2.7766e-02],
        [6.9837e-07, 1.0000e+00],
        [9.4944e-05, 9.9990e-01],
        [5.2796e-07, 1.0000e+00],
        [4.0545e-07, 1.0000e+00],
        [1.8697e-05, 9.9998e-01],
        [7.7658e-03, 9.9223e-01],
        [2.1087e-04, 9.9979e-01],
        [9.9985e-01, 1.5450e-04],
        [9.9969e-01, 3.1222e-04],
        [4.6276e-05, 9.9995e-01],
        [1.4015e-08, 1.0000e+00],
        [8.9030e-07, 1.0000e+00],
        [1.7588e-04, 9.9982e-01],
        [1.0000e+00, 2.1552e-06],
        [3.4525e-02, 9.6548e-01],
        [3.4098e-01, 6.5902e-01],
        [4.0931e-05, 9.9996e-01],
        [5.6096e-06, 9.9999e-01],
        [3.5131e-02, 9.6487e-01],
        [2.1521e-07, 1.0000e+00],
        [1.4357e-02, 9.8564e-01],
        [3.4497e-07, 1.0000e+00],
        [4.5480e-04, 9.9955e-01],
        [3.2378e-06, 1.0000e+00],
        [2.7361e-04, 9.9973e-01],
        [1.4346e-01, 8.5654e-01],
        [9.9992e-01, 8.2555e-05],
        [4.6900e-08, 1.0000e+00],
        [2.3699e-07, 1.0000e+00],
        [9.6158e-07, 1.0000e+00],
        [1.6008e-07, 1.0000e+00],
        [3.1482e-04, 9.9969e-01],
        [3.7492e-05, 9.9996e-01],
        [2.4197e-03, 9.9758e-01],
        [9.9837e-01, 1.6323e-03],
        [9.9995e-01, 5.4069e-05],
        [9.9879e-01, 1.2051e-03],
        [9.9993e-01, 6.8333e-05],
        [2.5026e-08, 1.0000e+00],
        [1.0893e-05, 9.9999e-01],
        [9.9998e-01, 2.3441e-05],
        [1.3909e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1398e-05, 9.9998e-01],
        [2.2479e-07, 1.0000e+00],
        [4.8060e-08, 1.0000e+00],
        [9.9995e-01, 5.1621e-05],
        [3.4369e-06, 1.0000e+00],
        [2.4258e-06, 1.0000e+00],
        [1.6628e-05, 9.9998e-01],
        [2.6726e-04, 9.9973e-01],
        [3.6543e-04, 9.9963e-01],
        [3.1414e-01, 6.8586e-01],
        [7.2797e-06, 9.9999e-01],
        [1.7295e-02, 9.8271e-01],
        [5.4610e-06, 9.9999e-01],
        [1.2719e-05, 9.9999e-01],
        [1.1331e-03, 9.9887e-01],
        [2.9546e-06, 1.0000e+00],
        [1.8469e-06, 1.0000e+00],
        [5.3665e-04, 9.9946e-01],
        [2.9362e-12, 1.0000e+00],
        [4.8155e-03, 9.9518e-01],
        [9.9997e-01, 2.6603e-05],
        [2.8933e-06, 1.0000e+00],
        [2.4865e-05, 9.9998e-01],
        [9.9096e-01, 9.0410e-03],
        [4.5437e-03, 9.9546e-01],
        [3.2500e-03, 9.9675e-01],
        [1.5902e-10, 1.0000e+00],
        [1.5482e-05, 9.9998e-01],
        [7.2434e-05, 9.9993e-01],
        [1.8285e-07, 1.0000e+00],
        [7.0327e-08, 1.0000e+00],
        [4.4109e-04, 9.9956e-01],
        [9.9969e-01, 3.0703e-04],
        [4.5281e-06, 1.0000e+00],
        [9.9989e-01, 1.1378e-04],
        [4.5321e-04, 9.9955e-01],
        [4.0507e-05, 9.9996e-01],
        [1.7167e-03, 9.9828e-01],
        [5.2087e-04, 9.9948e-01],
        [1.4916e-02, 9.8508e-01],
        [3.8516e-08, 1.0000e+00],
        [7.8862e-04, 9.9921e-01],
        [4.0351e-04, 9.9960e-01],
        [8.9679e-04, 9.9910e-01],
        [1.0239e-01, 8.9761e-01],
        [2.1726e-04, 9.9978e-01],
        [5.4610e-06, 9.9999e-01],
        [9.9874e-01, 1.2641e-03],
        [3.1928e-09, 1.0000e+00],
        [2.1148e-04, 9.9979e-01],
        [9.9710e-05, 9.9990e-01],
        [3.7636e-06, 1.0000e+00],
        [9.9989e-01, 1.1369e-04],
        [9.9981e-01, 1.8632e-04],
        [1.0873e-04, 9.9989e-01],
        [7.1515e-07, 1.0000e+00],
        [6.3256e-04, 9.9937e-01],
        [5.3004e-09, 1.0000e+00],
        [9.7070e-06, 9.9999e-01],
        [9.9949e-01, 5.0628e-04],
        [1.3330e-03, 9.9867e-01],
        [7.0797e-05, 9.9993e-01],
        [8.1849e-07, 1.0000e+00],
        [1.6586e-04, 9.9983e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9954e-01, 4.5683e-04],
        [2.0819e-07, 1.0000e+00],
        [1.6796e-03, 9.9832e-01],
        [9.1308e-06, 9.9999e-01],
        [8.6986e-07, 1.0000e+00],
        [7.8990e-08, 1.0000e+00],
        [5.4191e-06, 9.9999e-01],
        [8.3022e-01, 1.6978e-01],
        [7.7230e-05, 9.9992e-01],
        [1.4348e-06, 1.0000e+00],
        [9.9951e-01, 4.8580e-04],
        [2.4295e-01, 7.5705e-01],
        [8.1935e-06, 9.9999e-01],
        [7.4227e-02, 9.2577e-01],
        [1.6575e-01, 8.3425e-01],
        [4.5574e-03, 9.9544e-01],
        [3.2909e-06, 1.0000e+00],
        [3.1487e-09, 1.0000e+00],
        [4.1831e-05, 9.9996e-01],
        [2.3593e-06, 1.0000e+00],
        [1.7963e-03, 9.9820e-01],
        [2.6977e-06, 1.0000e+00],
        [7.6051e-07, 1.0000e+00],
        [8.9018e-04, 9.9911e-01],
        [2.5523e-09, 1.0000e+00],
        [9.9896e-01, 1.0403e-03],
        [6.0475e-06, 9.9999e-01],
        [9.7006e-07, 1.0000e+00],
        [2.9198e-06, 1.0000e+00],
        [9.9973e-01, 2.6654e-04],
        [1.4624e-01, 8.5376e-01],
        [3.8892e-07, 1.0000e+00],
        [1.5027e-05, 9.9998e-01],
        [1.5046e-07, 1.0000e+00],
        [3.8483e-09, 1.0000e+00],
        [7.5263e-01, 2.4737e-01],
        [1.5027e-05, 9.9998e-01],
        [1.9471e-02, 9.8053e-01],
        [8.2174e-06, 9.9999e-01],
        [9.9976e-01, 2.3554e-04],
        [1.5628e-04, 9.9984e-01],
        [6.7535e-01, 3.2465e-01],
        [1.3192e-10, 1.0000e+00],
        [1.8054e-07, 1.0000e+00],
        [9.9993e-01, 6.6737e-05],
        [2.8776e-04, 9.9971e-01],
        [1.4070e-04, 9.9986e-01],
        [6.9167e-09, 1.0000e+00],
        [9.9988e-01, 1.1799e-04],
        [9.9989e-01, 1.1491e-04],
        [3.3465e-06, 1.0000e+00],
        [9.9939e-01, 6.1226e-04],
        [1.7449e-06, 1.0000e+00],
        [9.9942e-01, 5.8465e-04],
        [3.8622e-09, 1.0000e+00],
        [6.8426e-05, 9.9993e-01],
        [9.9997e-01, 2.6741e-05],
        [3.9792e-04, 9.9960e-01],
        [6.3248e-10, 1.0000e+00],
        [9.9976e-01, 2.3977e-04],
        [6.2612e-06, 9.9999e-01],
        [2.2934e-02, 9.7707e-01],
        [1.3569e-06, 1.0000e+00],
        [7.5974e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9993e-01, 6.7694e-05],
        [1.4969e-04, 9.9985e-01],
        [7.4751e-07, 1.0000e+00],
        [1.8833e-06, 1.0000e+00],
        [8.8382e-04, 9.9912e-01],
        [6.0524e-06, 9.9999e-01],
        [7.8560e-01, 2.1440e-01],
        [1.0611e-09, 1.0000e+00],
        [9.9966e-01, 3.3557e-04],
        [8.3945e-09, 1.0000e+00],
        [6.8165e-06, 9.9999e-01],
        [2.2188e-10, 1.0000e+00],
        [5.6940e-07, 1.0000e+00],
        [1.1211e-08, 1.0000e+00],
        [9.9924e-01, 7.6426e-04],
        [4.6723e-04, 9.9953e-01],
        [3.6573e-02, 9.6343e-01],
        [1.0007e-04, 9.9990e-01],
        [8.2697e-04, 9.9917e-01],
        [9.9990e-01, 9.6325e-05],
        [9.9941e-01, 5.9143e-04],
        [9.5362e-01, 4.6383e-02],
        [2.1782e-06, 1.0000e+00],
        [7.0560e-01, 2.9440e-01],
        [9.9961e-01, 3.8822e-04],
        [2.3264e-08, 1.0000e+00],
        [9.9982e-01, 1.8138e-04],
        [8.6695e-06, 9.9999e-01],
        [1.6521e-01, 8.3479e-01],
        [4.6396e-06, 1.0000e+00],
        [9.9869e-01, 1.3110e-03],
        [5.7946e-04, 9.9942e-01],
        [1.3975e-07, 1.0000e+00],
        [2.1287e-10, 1.0000e+00],
        [9.9983e-01, 1.6936e-04],
        [2.8427e-05, 9.9997e-01],
        [9.9987e-01, 1.3027e-04],
        [2.8303e-07, 1.0000e+00],
        [9.9996e-01, 3.9022e-05],
        [2.8824e-06, 1.0000e+00],
        [1.6889e-03, 9.9831e-01],
        [2.6479e-06, 1.0000e+00],
        [7.6171e-02, 9.2383e-01],
        [4.3027e-05, 9.9996e-01],
        [3.3285e-03, 9.9667e-01],
        [1.5077e-07, 1.0000e+00],
        [7.2371e-05, 9.9993e-01],
        [3.6368e-07, 1.0000e+00],
        [3.6027e-07, 1.0000e+00],
        [3.1363e-01, 6.8637e-01],
        [9.9898e-08, 1.0000e+00],
        [3.9333e-02, 9.6067e-01],
        [1.2570e-05, 9.9999e-01],
        [7.1877e-07, 1.0000e+00],
        [2.8223e-09, 1.0000e+00],
        [8.6048e-07, 1.0000e+00],
        [2.1099e-02, 9.7890e-01],
        [1.8661e-05, 9.9998e-01],
        [6.5360e-05, 9.9993e-01],
        [6.2635e-04, 9.9937e-01],
        [8.0847e-10, 1.0000e+00],
        [8.0723e-05, 9.9992e-01],
        [2.2140e-07, 1.0000e+00],
        [1.1169e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9983e-01, 1.6809e-04],
        [6.4815e-08, 1.0000e+00],
        [6.2684e-09, 1.0000e+00],
        [9.9917e-01, 8.2979e-04],
        [8.6292e-09, 1.0000e+00],
        [2.0710e-10, 1.0000e+00],
        [8.4791e-09, 1.0000e+00],
        [9.9983e-01, 1.6605e-04],
        [9.9969e-01, 3.0596e-04],
        [7.3148e-07, 1.0000e+00],
        [4.8402e-07, 1.0000e+00],
        [4.1897e-09, 1.0000e+00],
        [1.8406e-07, 1.0000e+00],
        [4.5997e-07, 1.0000e+00],
        [1.4664e-06, 1.0000e+00],
        [1.9260e-05, 9.9998e-01],
        [1.5130e-04, 9.9985e-01],
        [4.9634e-10, 1.0000e+00],
        [3.8122e-03, 9.9619e-01],
        [2.1991e-10, 1.0000e+00],
        [1.5335e-02, 9.8466e-01],
        [1.1603e-10, 1.0000e+00],
        [7.4730e-04, 9.9925e-01],
        [2.1337e-04, 9.9979e-01],
        [2.0210e-06, 1.0000e+00],
        [6.3764e-09, 1.0000e+00],
        [1.9790e-07, 1.0000e+00],
        [9.9980e-01, 2.0046e-04],
        [9.9896e-01, 1.0449e-03],
        [3.2275e-09, 1.0000e+00],
        [1.5798e-06, 1.0000e+00],
        [9.9973e-01, 2.7323e-04],
        [9.9042e-01, 9.5757e-03],
        [2.0243e-07, 1.0000e+00],
        [9.9926e-01, 7.4060e-04],
        [3.7003e-06, 1.0000e+00],
        [9.9846e-01, 1.5379e-03],
        [9.9982e-01, 1.7559e-04],
        [2.4909e-07, 1.0000e+00],
        [3.8171e-06, 1.0000e+00],
        [1.1221e-03, 9.9888e-01],
        [6.4989e-05, 9.9994e-01],
        [9.9812e-01, 1.8755e-03],
        [4.2756e-03, 9.9572e-01],
        [4.2057e-09, 1.0000e+00],
        [1.0895e-06, 1.0000e+00],
        [2.3660e-07, 1.0000e+00],
        [6.2996e-01, 3.7004e-01],
        [9.3549e-01, 6.4508e-02],
        [5.2296e-07, 1.0000e+00],
        [2.9825e-08, 1.0000e+00],
        [1.5716e-07, 1.0000e+00],
        [1.0849e-05, 9.9999e-01],
        [4.5083e-09, 1.0000e+00],
        [9.9997e-01, 3.1508e-05],
        [9.6769e-01, 3.2311e-02],
        [9.9507e-03, 9.9005e-01],
        [9.9997e-01, 3.2929e-05],
        [5.4843e-06, 9.9999e-01],
        [1.8865e-07, 1.0000e+00],
        [2.0860e-11, 1.0000e+00],
        [1.0159e-04, 9.9990e-01],
        [1.0467e-02, 9.8953e-01],
        [9.9993e-01, 7.2544e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4998e-03, 9.9850e-01],
        [9.9985e-01, 1.4571e-04],
        [6.1467e-07, 1.0000e+00],
        [5.4461e-08, 1.0000e+00],
        [7.7971e-08, 1.0000e+00],
        [9.9537e-01, 4.6329e-03],
        [1.0340e-08, 1.0000e+00],
        [3.0237e-10, 1.0000e+00],
        [5.8058e-07, 1.0000e+00],
        [6.7364e-10, 1.0000e+00],
        [6.7508e-11, 1.0000e+00],
        [9.9488e-01, 5.1196e-03],
        [4.3196e-08, 1.0000e+00],
        [5.5405e-04, 9.9945e-01],
        [3.8769e-11, 1.0000e+00],
        [2.8187e-03, 9.9718e-01],
        [2.1460e-08, 1.0000e+00],
        [2.0091e-02, 9.7991e-01],
        [3.7031e-06, 1.0000e+00],
        [5.2867e-03, 9.9471e-01],
        [1.0358e-09, 1.0000e+00],
        [7.1682e-07, 1.0000e+00],
        [9.9917e-01, 8.2639e-04],
        [1.9157e-04, 9.9981e-01],
        [9.4822e-08, 1.0000e+00],
        [1.5620e-04, 9.9984e-01],
        [1.0290e-03, 9.9897e-01],
        [1.0446e-08, 1.0000e+00],
        [3.2701e-08, 1.0000e+00],
        [8.3492e-07, 1.0000e+00],
        [6.5981e-06, 9.9999e-01],
        [4.6713e-08, 1.0000e+00],
        [9.9985e-01, 1.4650e-04],
        [8.5684e-06, 9.9999e-01],
        [3.7508e-06, 1.0000e+00],
        [9.9989e-01, 1.1351e-04],
        [4.5090e-09, 1.0000e+00],
        [1.2095e-07, 1.0000e+00],
        [3.3862e-10, 1.0000e+00],
        [1.3235e-07, 1.0000e+00],
        [8.7668e-05, 9.9991e-01],
        [1.3287e-07, 1.0000e+00],
        [9.9992e-01, 7.6656e-05],
        [9.6511e-01, 3.4886e-02],
        [9.9991e-01, 8.7513e-05],
        [9.9995e-01, 5.2766e-05],
        [5.7412e-05, 9.9994e-01],
        [2.3682e-07, 1.0000e+00],
        [4.5315e-01, 5.4685e-01],
        [1.8498e-03, 9.9815e-01],
        [5.4782e-08, 1.0000e+00],
        [1.6270e-04, 9.9984e-01],
        [8.2164e-01, 1.7836e-01],
        [1.0789e-06, 1.0000e+00],
        [6.3415e-07, 1.0000e+00],
        [9.9986e-01, 1.4462e-04],
        [8.5380e-06, 9.9999e-01],
        [1.5559e-06, 1.0000e+00],
        [1.1269e-02, 9.8873e-01],
        [7.9937e-07, 1.0000e+00],
        [3.7549e-05, 9.9996e-01],
        [9.9987e-01, 1.3460e-04],
        [3.5681e-08, 1.0000e+00],
        [4.4192e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9700e-01, 3.0016e-03],
        [9.9427e-01, 5.7308e-03],
        [2.1744e-04, 9.9978e-01],
        [5.8952e-07, 1.0000e+00],
        [1.0000e+00, 1.9173e-06],
        [1.3550e-06, 1.0000e+00],
        [9.9045e-01, 9.5508e-03],
        [9.8996e-11, 1.0000e+00],
        [4.3983e-08, 1.0000e+00],
        [1.9556e-08, 1.0000e+00],
        [9.9606e-01, 3.9426e-03],
        [3.8432e-07, 1.0000e+00],
        [9.9966e-01, 3.4029e-04],
        [1.9352e-07, 1.0000e+00],
        [1.9892e-07, 1.0000e+00],
        [4.8182e-11, 1.0000e+00],
        [4.6649e-01, 5.3351e-01],
        [2.1062e-09, 1.0000e+00],
        [9.9952e-01, 4.7571e-04],
        [8.4476e-10, 1.0000e+00],
        [1.5815e-07, 1.0000e+00],
        [9.9956e-01, 4.3803e-04],
        [1.2329e-05, 9.9999e-01],
        [8.5900e-01, 1.4100e-01],
        [1.4786e-08, 1.0000e+00],
        [9.7247e-07, 1.0000e+00],
        [9.9801e-01, 1.9872e-03],
        [1.3937e-04, 9.9986e-01],
        [2.1955e-07, 1.0000e+00],
        [2.1829e-05, 9.9998e-01],
        [9.9837e-01, 1.6330e-03],
        [1.8923e-08, 1.0000e+00],
        [5.6500e-10, 1.0000e+00],
        [7.6680e-06, 9.9999e-01],
        [6.2059e-11, 1.0000e+00],
        [9.9974e-01, 2.6279e-04],
        [9.9957e-01, 4.3169e-04],
        [9.9989e-01, 1.0594e-04],
        [5.8379e-06, 9.9999e-01],
        [1.5890e-06, 1.0000e+00],
        [9.9999e-01, 5.5131e-06],
        [2.1806e-08, 1.0000e+00],
        [5.3839e-01, 4.6161e-01],
        [2.5963e-02, 9.7404e-01],
        [5.5922e-09, 1.0000e+00],
        [1.9212e-07, 1.0000e+00],
        [2.1490e-05, 9.9998e-01],
        [4.8822e-08, 1.0000e+00],
        [1.5629e-07, 1.0000e+00],
        [3.6405e-08, 1.0000e+00],
        [9.9400e-01, 5.9995e-03],
        [9.9959e-01, 4.0589e-04],
        [4.6801e-05, 9.9995e-01],
        [8.2024e-07, 1.0000e+00],
        [4.8409e-08, 1.0000e+00],
        [1.7260e-03, 9.9827e-01],
        [5.5592e-02, 9.4441e-01],
        [9.9960e-01, 3.9613e-04],
        [2.2268e-06, 1.0000e+00],
        [4.4800e-05, 9.9996e-01],
        [1.2884e-07, 1.0000e+00],
        [9.9920e-01, 7.9627e-04],
        [8.9168e-09, 1.0000e+00],
        [1.4547e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.6963e-08, 1.0000e+00],
        [1.4244e-08, 1.0000e+00],
        [1.1509e-07, 1.0000e+00],
        [9.9675e-01, 3.2531e-03],
        [9.9990e-01, 9.5599e-05],
        [4.6208e-07, 1.0000e+00],
        [1.3098e-06, 1.0000e+00],
        [2.1867e-05, 9.9998e-01],
        [7.6542e-07, 1.0000e+00],
        [6.6018e-12, 1.0000e+00],
        [7.0852e-09, 1.0000e+00],
        [1.1715e-07, 1.0000e+00],
        [3.2204e-03, 9.9678e-01],
        [9.9212e-01, 7.8757e-03],
        [8.6822e-05, 9.9991e-01],
        [1.2013e-04, 9.9988e-01],
        [9.4972e-01, 5.0283e-02],
        [8.6011e-03, 9.9140e-01],
        [2.5530e-07, 1.0000e+00],
        [5.1505e-02, 9.4850e-01],
        [2.8582e-01, 7.1418e-01],
        [3.0860e-07, 1.0000e+00],
        [5.3880e-02, 9.4612e-01],
        [3.0082e-08, 1.0000e+00],
        [9.9986e-01, 1.4305e-04],
        [2.4606e-04, 9.9975e-01],
        [3.2356e-07, 1.0000e+00],
        [2.6283e-07, 1.0000e+00],
        [2.1837e-07, 1.0000e+00],
        [9.8749e-06, 9.9999e-01],
        [8.5924e-10, 1.0000e+00],
        [8.3244e-02, 9.1676e-01],
        [7.7959e-08, 1.0000e+00],
        [6.1378e-10, 1.0000e+00],
        [4.5779e-07, 1.0000e+00],
        [6.7523e-11, 1.0000e+00],
        [3.7981e-05, 9.9996e-01],
        [7.4585e-09, 1.0000e+00],
        [1.3357e-06, 1.0000e+00],
        [9.9996e-01, 3.7662e-05],
        [2.8291e-05, 9.9997e-01],
        [1.2258e-04, 9.9988e-01],
        [9.9981e-01, 1.8683e-04],
        [2.3746e-07, 1.0000e+00],
        [9.9921e-01, 7.9038e-04],
        [3.9176e-11, 1.0000e+00],
        [3.7346e-06, 1.0000e+00],
        [8.5907e-01, 1.4093e-01],
        [9.9975e-01, 2.4891e-04],
        [3.0436e-08, 1.0000e+00],
        [9.2332e-04, 9.9908e-01],
        [9.7849e-01, 2.1508e-02],
        [1.8266e-10, 1.0000e+00],
        [7.9306e-07, 1.0000e+00],
        [3.0321e-10, 1.0000e+00],
        [1.0229e-06, 1.0000e+00],
        [3.6731e-06, 1.0000e+00],
        [1.7079e-05, 9.9998e-01],
        [2.1800e-08, 1.0000e+00],
        [5.2881e-11, 1.0000e+00],
        [9.0607e-10, 1.0000e+00],
        [6.5390e-06, 9.9999e-01],
        [7.9580e-07, 1.0000e+00],
        [1.3812e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.1281e-03, 9.9287e-01],
        [9.9967e-01, 3.3269e-04],
        [9.0007e-06, 9.9999e-01],
        [9.9984e-01, 1.6496e-04],
        [6.3327e-09, 1.0000e+00],
        [2.9850e-07, 1.0000e+00],
        [9.9855e-01, 1.4528e-03],
        [1.8800e-08, 1.0000e+00],
        [9.9641e-01, 3.5895e-03],
        [9.9995e-01, 4.8675e-05],
        [1.5998e-08, 1.0000e+00],
        [9.9870e-01, 1.3047e-03],
        [4.8476e-08, 1.0000e+00],
        [5.5543e-08, 1.0000e+00],
        [1.9121e-04, 9.9981e-01],
        [9.9550e-01, 4.5011e-03],
        [1.7272e-02, 9.8273e-01],
        [2.2856e-03, 9.9771e-01],
        [8.2275e-08, 1.0000e+00],
        [4.1438e-04, 9.9959e-01],
        [2.0774e-06, 1.0000e+00],
        [9.9975e-01, 2.5065e-04],
        [1.5941e-08, 1.0000e+00],
        [9.6356e-06, 9.9999e-01],
        [1.2413e-11, 1.0000e+00],
        [2.9204e-07, 1.0000e+00],
        [8.8596e-04, 9.9911e-01],
        [5.6647e-02, 9.4335e-01],
        [1.3747e-08, 1.0000e+00],
        [3.7294e-06, 1.0000e+00],
        [3.9269e-05, 9.9996e-01],
        [4.3756e-07, 1.0000e+00],
        [2.5888e-07, 1.0000e+00],
        [9.9990e-01, 1.0457e-04],
        [4.5984e-10, 1.0000e+00],
        [9.9939e-01, 6.0572e-04],
        [3.2068e-11, 1.0000e+00],
        [9.6829e-08, 1.0000e+00],
        [3.4747e-08, 1.0000e+00],
        [1.4160e-08, 1.0000e+00],
        [2.9485e-11, 1.0000e+00],
        [1.5257e-07, 1.0000e+00],
        [1.3135e-07, 1.0000e+00],
        [1.6603e-09, 1.0000e+00],
        [9.9894e-01, 1.0579e-03],
        [9.9974e-01, 2.6452e-04],
        [1.4329e-07, 1.0000e+00],
        [2.0912e-07, 1.0000e+00],
        [1.8331e-09, 1.0000e+00],
        [2.2443e-08, 1.0000e+00],
        [9.9983e-01, 1.7271e-04],
        [8.5995e-07, 1.0000e+00],
        [3.1539e-06, 1.0000e+00],
        [8.6304e-01, 1.3696e-01],
        [2.5027e-07, 1.0000e+00],
        [1.0360e-08, 1.0000e+00],
        [1.1026e-08, 1.0000e+00],
        [9.8879e-01, 1.1210e-02],
        [1.2413e-11, 1.0000e+00],
        [9.8070e-01, 1.9299e-02],
        [9.9874e-01, 1.2573e-03],
        [1.3048e-04, 9.9987e-01],
        [3.8586e-09, 1.0000e+00],
        [1.2132e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.1906e-05, 9.9992e-01],
        [3.6061e-09, 1.0000e+00],
        [1.9037e-08, 1.0000e+00],
        [9.9964e-01, 3.6494e-04],
        [5.8511e-06, 9.9999e-01],
        [9.9991e-01, 8.5402e-05],
        [7.6123e-11, 1.0000e+00],
        [5.3393e-05, 9.9995e-01],
        [7.2311e-04, 9.9928e-01],
        [9.3390e-01, 6.6096e-02],
        [1.6100e-06, 1.0000e+00],
        [9.9980e-01, 2.0240e-04],
        [2.8676e-03, 9.9713e-01],
        [8.3887e-07, 1.0000e+00],
        [3.1867e-10, 1.0000e+00],
        [3.2697e-08, 1.0000e+00],
        [6.6416e-10, 1.0000e+00],
        [6.3360e-09, 1.0000e+00],
        [2.1611e-06, 1.0000e+00],
        [2.2255e-09, 1.0000e+00],
        [2.3535e-07, 1.0000e+00],
        [6.0293e-10, 1.0000e+00],
        [4.1390e-06, 1.0000e+00],
        [9.1401e-05, 9.9991e-01],
        [3.3192e-09, 1.0000e+00],
        [1.3594e-08, 1.0000e+00],
        [2.5838e-05, 9.9997e-01],
        [2.6192e-06, 1.0000e+00],
        [6.2059e-07, 1.0000e+00],
        [4.8804e-01, 5.1196e-01],
        [9.9897e-01, 1.0290e-03],
        [1.9696e-06, 1.0000e+00],
        [6.5960e-06, 9.9999e-01],
        [9.9989e-01, 1.0676e-04],
        [9.9997e-01, 2.8907e-05],
        [9.9348e-01, 6.5179e-03],
        [6.1921e-05, 9.9994e-01],
        [4.8005e-05, 9.9995e-01],
        [3.9451e-08, 1.0000e+00],
        [1.4565e-01, 8.5435e-01],
        [2.2792e-07, 1.0000e+00],
        [4.8810e-11, 1.0000e+00],
        [2.2285e-05, 9.9998e-01],
        [9.9972e-01, 2.8346e-04],
        [9.9984e-01, 1.6383e-04],
        [2.2551e-05, 9.9998e-01],
        [3.7143e-04, 9.9963e-01],
        [9.1320e-03, 9.9087e-01],
        [5.1975e-03, 9.9480e-01],
        [4.7171e-09, 1.0000e+00],
        [1.6237e-08, 1.0000e+00],
        [2.2446e-06, 1.0000e+00],
        [9.9972e-01, 2.8270e-04],
        [3.1136e-06, 1.0000e+00],
        [9.4074e-01, 5.9257e-02],
        [3.0672e-07, 1.0000e+00],
        [9.9999e-01, 6.0519e-06],
        [3.4978e-06, 1.0000e+00],
        [1.3142e-03, 9.9869e-01],
        [2.5628e-07, 1.0000e+00],
        [2.4696e-05, 9.9998e-01],
        [9.9969e-01, 3.1076e-04],
        [4.6109e-09, 1.0000e+00],
        [4.2631e-02, 9.5737e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.9020e-06, 9.9999e-01],
        [8.8869e-07, 1.0000e+00],
        [4.2981e-05, 9.9996e-01],
        [8.5084e-01, 1.4916e-01],
        [3.4859e-06, 1.0000e+00],
        [9.9971e-01, 2.8886e-04],
        [3.7289e-03, 9.9627e-01],
        [4.2702e-06, 1.0000e+00],
        [3.7609e-08, 1.0000e+00],
        [1.0858e-05, 9.9999e-01],
        [3.0530e-08, 1.0000e+00],
        [4.9751e-04, 9.9950e-01],
        [8.6838e-02, 9.1316e-01],
        [3.3579e-04, 9.9966e-01],
        [1.1054e-10, 1.0000e+00],
        [2.5970e-07, 1.0000e+00],
        [1.2877e-08, 1.0000e+00],
        [1.0664e-04, 9.9989e-01],
        [1.9873e-11, 1.0000e+00],
        [7.9066e-01, 2.0934e-01],
        [1.1719e-09, 1.0000e+00],
        [5.7581e-05, 9.9994e-01],
        [4.4726e-06, 1.0000e+00],
        [9.9981e-01, 1.9023e-04],
        [9.2147e-09, 1.0000e+00],
        [9.5936e-01, 4.0644e-02],
        [8.6521e-04, 9.9913e-01],
        [9.9990e-01, 9.8735e-05],
        [9.5494e-03, 9.9045e-01],
        [1.2230e-04, 9.9988e-01],
        [3.2456e-11, 1.0000e+00],
        [2.7104e-05, 9.9997e-01],
        [9.9989e-01, 1.1432e-04],
        [1.9108e-08, 1.0000e+00],
        [1.2525e-08, 1.0000e+00],
        [8.4192e-09, 1.0000e+00],
        [7.8566e-01, 2.1434e-01],
        [2.5323e-04, 9.9975e-01],
        [9.9817e-01, 1.8283e-03],
        [5.3632e-08, 1.0000e+00],
        [2.4386e-07, 1.0000e+00],
        [3.0726e-12, 1.0000e+00],
        [9.9952e-01, 4.8271e-04],
        [9.2847e-01, 7.1532e-02],
        [4.1918e-01, 5.8082e-01],
        [3.4603e-05, 9.9997e-01],
        [1.5758e-06, 1.0000e+00],
        [5.5969e-09, 1.0000e+00],
        [1.1991e-03, 9.9880e-01],
        [3.0979e-06, 1.0000e+00],
        [9.9920e-01, 8.0012e-04],
        [7.3356e-07, 1.0000e+00],
        [1.0958e-10, 1.0000e+00],
        [6.9779e-01, 3.0221e-01],
        [9.9357e-01, 6.4257e-03],
        [9.1772e-01, 8.2283e-02],
        [5.0826e-09, 1.0000e+00],
        [1.0996e-04, 9.9989e-01],
        [1.1334e-06, 1.0000e+00],
        [5.5383e-06, 9.9999e-01],
        [9.9999e-01, 1.4117e-05],
        [5.5641e-07, 1.0000e+00],
        [9.9869e-01, 1.3061e-03],
        [1.0000e+00, 2.1382e-07]], grad_fn=<SoftmaxBackward0>)
tensor([[8.0520e-10, 1.0000e+00],
        [9.9949e-01, 5.0859e-04],
        [3.1160e-02, 9.6884e-01],
        [1.1756e-09, 1.0000e+00],
        [9.9982e-01, 1.7849e-04],
        [1.1560e-07, 1.0000e+00],
        [9.8355e-01, 1.6449e-02],
        [5.0000e-07, 1.0000e+00],
        [9.9991e-01, 8.6874e-05],
        [2.9180e-06, 1.0000e+00],
        [9.8509e-05, 9.9990e-01],
        [9.8508e-01, 1.4917e-02],
        [1.7852e-10, 1.0000e+00],
        [9.9999e-01, 1.0327e-05],
        [7.3515e-09, 1.0000e+00],
        [3.5117e-09, 1.0000e+00],
        [9.9975e-01, 2.4673e-04],
        [7.4525e-04, 9.9925e-01],
        [9.9957e-01, 4.3493e-04],
        [5.8409e-05, 9.9994e-01],
        [9.9993e-01, 6.7215e-05],
        [9.8130e-01, 1.8700e-02],
        [1.0619e-03, 9.9894e-01],
        [3.8301e-10, 1.0000e+00],
        [2.2462e-05, 9.9998e-01],
        [9.9993e-01, 7.3209e-05],
        [9.9959e-01, 4.1145e-04],
        [1.6939e-06, 1.0000e+00],
        [7.5945e-07, 1.0000e+00],
        [4.3474e-07, 1.0000e+00],
        [9.9999e-01, 1.1106e-05],
        [9.9162e-01, 8.3837e-03],
        [3.2064e-04, 9.9968e-01],
        [7.5616e-06, 9.9999e-01],
        [7.4094e-05, 9.9993e-01],
        [2.4387e-09, 1.0000e+00],
        [2.1650e-05, 9.9998e-01],
        [9.9674e-01, 3.2602e-03],
        [3.6886e-08, 1.0000e+00],
        [1.3189e-08, 1.0000e+00],
        [4.1028e-06, 1.0000e+00],
        [1.8140e-06, 1.0000e+00],
        [2.0592e-04, 9.9979e-01],
        [9.9989e-01, 1.0992e-04],
        [8.4265e-04, 9.9916e-01],
        [1.7705e-05, 9.9998e-01],
        [6.4611e-08, 1.0000e+00],
        [1.7660e-08, 1.0000e+00],
        [9.1177e-04, 9.9909e-01],
        [8.6817e-07, 1.0000e+00],
        [7.0236e-06, 9.9999e-01],
        [3.0285e-04, 9.9970e-01],
        [3.8217e-04, 9.9962e-01],
        [6.1352e-03, 9.9386e-01],
        [9.9471e-01, 5.2856e-03],
        [1.5062e-07, 1.0000e+00],
        [2.6678e-05, 9.9997e-01],
        [1.2909e-06, 1.0000e+00],
        [1.1977e-06, 1.0000e+00],
        [9.9572e-01, 4.2808e-03],
        [3.7341e-02, 9.6266e-01],
        [6.5512e-06, 9.9999e-01],
        [8.8999e-08, 1.0000e+00],
        [8.6740e-05, 9.9991e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.7155e-05, 9.9990e-01],
        [5.0165e-09, 1.0000e+00],
        [1.0904e-08, 1.0000e+00],
        [1.0852e-04, 9.9989e-01],
        [2.7076e-04, 9.9973e-01],
        [1.2573e-04, 9.9987e-01],
        [1.9759e-05, 9.9998e-01],
        [3.0101e-03, 9.9699e-01],
        [6.4652e-03, 9.9353e-01],
        [8.4861e-08, 1.0000e+00],
        [6.8146e-06, 9.9999e-01],
        [4.8107e-10, 1.0000e+00],
        [8.6388e-08, 1.0000e+00],
        [9.5434e-10, 1.0000e+00],
        [8.7003e-01, 1.2997e-01],
        [1.2795e-09, 1.0000e+00],
        [1.1679e-01, 8.8321e-01],
        [2.4128e-05, 9.9998e-01],
        [1.0161e-03, 9.9898e-01],
        [8.1792e-06, 9.9999e-01],
        [7.2537e-01, 2.7463e-01],
        [9.9964e-01, 3.6436e-04],
        [1.3239e-08, 1.0000e+00],
        [6.4144e-07, 1.0000e+00],
        [7.7932e-01, 2.2068e-01],
        [1.0030e-06, 1.0000e+00],
        [3.9622e-07, 1.0000e+00],
        [7.4243e-08, 1.0000e+00],
        [1.0535e-04, 9.9989e-01],
        [5.2735e-03, 9.9473e-01],
        [4.6532e-08, 1.0000e+00],
        [1.7843e-05, 9.9998e-01],
        [4.6373e-07, 1.0000e+00],
        [9.9990e-01, 9.8228e-05],
        [9.6348e-06, 9.9999e-01],
        [9.9983e-01, 1.7009e-04],
        [6.4603e-01, 3.5397e-01],
        [3.1605e-06, 1.0000e+00],
        [4.0745e-10, 1.0000e+00],
        [1.3512e-03, 9.9865e-01],
        [8.9368e-06, 9.9999e-01],
        [9.9985e-01, 1.5321e-04],
        [2.1442e-06, 1.0000e+00],
        [9.9988e-01, 1.1529e-04],
        [6.6098e-01, 3.3902e-01],
        [9.9994e-01, 6.1145e-05],
        [6.7789e-07, 1.0000e+00],
        [1.6842e-09, 1.0000e+00],
        [9.8790e-01, 1.2102e-02],
        [6.1915e-04, 9.9938e-01],
        [1.1186e-04, 9.9989e-01],
        [2.5775e-07, 1.0000e+00],
        [9.9979e-01, 2.0605e-04],
        [9.9999e-01, 9.1867e-06],
        [2.9223e-06, 1.0000e+00],
        [1.8075e-06, 1.0000e+00],
        [7.9837e-06, 9.9999e-01],
        [2.8947e-04, 9.9971e-01],
        [9.9989e-01, 1.1320e-04],
        [3.7266e-09, 1.0000e+00],
        [9.9958e-01, 4.1672e-04],
        [9.9763e-01, 2.3680e-03],
        [2.1090e-06, 1.0000e+00],
        [2.7747e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1442e-08, 1.0000e+00],
        [1.2318e-04, 9.9988e-01],
        [3.4571e-05, 9.9997e-01],
        [1.2548e-07, 1.0000e+00],
        [9.9993e-01, 7.3656e-05],
        [9.9996e-01, 3.8813e-05],
        [4.0441e-06, 1.0000e+00],
        [4.5306e-08, 1.0000e+00],
        [2.4382e-06, 1.0000e+00],
        [7.7341e-06, 9.9999e-01],
        [9.9997e-01, 3.3491e-05],
        [9.9989e-01, 1.0740e-04],
        [8.0973e-04, 9.9919e-01],
        [6.6360e-09, 1.0000e+00],
        [2.6056e-05, 9.9997e-01],
        [9.4022e-01, 5.9782e-02],
        [3.1768e-07, 1.0000e+00],
        [9.9984e-01, 1.6287e-04],
        [2.3925e-05, 9.9998e-01],
        [5.0473e-06, 9.9999e-01],
        [1.1983e-04, 9.9988e-01],
        [6.3873e-05, 9.9994e-01],
        [1.1739e-06, 1.0000e+00],
        [9.9899e-01, 1.0149e-03],
        [1.0659e-03, 9.9893e-01],
        [1.2925e-03, 9.9871e-01],
        [3.6523e-05, 9.9996e-01],
        [1.6359e-02, 9.8364e-01],
        [9.7440e-01, 2.5599e-02],
        [2.5571e-07, 1.0000e+00],
        [1.3627e-01, 8.6373e-01],
        [6.0723e-04, 9.9939e-01],
        [4.9950e-04, 9.9950e-01],
        [9.6620e-07, 1.0000e+00],
        [4.4889e-07, 1.0000e+00],
        [8.4144e-05, 9.9992e-01],
        [1.1840e-03, 9.9882e-01],
        [8.7159e-07, 1.0000e+00],
        [3.3245e-02, 9.6675e-01],
        [5.2247e-03, 9.9478e-01],
        [9.9989e-01, 1.0904e-04],
        [1.3166e-05, 9.9999e-01],
        [9.3269e-07, 1.0000e+00],
        [1.0543e-03, 9.9895e-01],
        [1.3168e-03, 9.9868e-01],
        [9.0726e-01, 9.2742e-02],
        [4.6311e-08, 1.0000e+00],
        [7.7267e-06, 9.9999e-01],
        [3.3387e-06, 1.0000e+00],
        [1.1777e-04, 9.9988e-01],
        [9.9993e-01, 6.6160e-05],
        [1.7753e-06, 1.0000e+00],
        [1.0522e-08, 1.0000e+00],
        [2.0094e-06, 1.0000e+00],
        [2.9728e-04, 9.9970e-01],
        [3.9289e-07, 1.0000e+00],
        [2.7256e-05, 9.9997e-01],
        [9.9997e-01, 3.1676e-05],
        [1.3150e-07, 1.0000e+00],
        [2.3844e-03, 9.9762e-01],
        [9.9600e-01, 4.0005e-03],
        [8.0075e-06, 9.9999e-01],
        [9.2638e-04, 9.9907e-01],
        [2.0913e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.3562e-05, 9.9997e-01],
        [4.2343e-07, 1.0000e+00],
        [3.7562e-06, 1.0000e+00],
        [8.2268e-05, 9.9992e-01],
        [9.9889e-01, 1.1071e-03],
        [1.5996e-06, 1.0000e+00],
        [6.7271e-09, 1.0000e+00],
        [1.0458e-05, 9.9999e-01],
        [1.7788e-10, 1.0000e+00],
        [2.2979e-09, 1.0000e+00],
        [2.6143e-05, 9.9997e-01],
        [6.4457e-07, 1.0000e+00],
        [9.5776e-05, 9.9990e-01],
        [9.9994e-01, 5.8233e-05],
        [2.4267e-03, 9.9757e-01],
        [8.7385e-07, 1.0000e+00],
        [9.9992e-01, 8.2516e-05],
        [3.5365e-05, 9.9996e-01],
        [3.1855e-06, 1.0000e+00],
        [9.9968e-01, 3.2248e-04],
        [9.5605e-01, 4.3948e-02],
        [3.7329e-04, 9.9963e-01],
        [9.9990e-01, 1.0495e-04],
        [1.6786e-02, 9.8321e-01],
        [7.6560e-05, 9.9992e-01],
        [9.9925e-01, 7.4755e-04],
        [6.1403e-05, 9.9994e-01],
        [3.3360e-04, 9.9967e-01],
        [9.9988e-01, 1.1907e-04],
        [9.5226e-08, 1.0000e+00],
        [1.4669e-01, 8.5331e-01],
        [6.6186e-08, 1.0000e+00],
        [1.9185e-05, 9.9998e-01],
        [9.9996e-01, 4.1204e-05],
        [1.3810e-06, 1.0000e+00],
        [5.8109e-06, 9.9999e-01],
        [1.2742e-07, 1.0000e+00],
        [6.3440e-07, 1.0000e+00],
        [5.0955e-03, 9.9490e-01],
        [1.5313e-05, 9.9998e-01],
        [4.7313e-05, 9.9995e-01],
        [3.9431e-06, 1.0000e+00],
        [9.8341e-05, 9.9990e-01],
        [9.9991e-01, 8.9955e-05],
        [1.6949e-06, 1.0000e+00],
        [2.1698e-02, 9.7830e-01],
        [9.9999e-01, 7.5613e-06],
        [9.9976e-01, 2.3762e-04],
        [1.2836e-05, 9.9999e-01],
        [6.2608e-06, 9.9999e-01],
        [9.9952e-01, 4.8284e-04],
        [9.7019e-05, 9.9990e-01],
        [1.6281e-02, 9.8372e-01],
        [7.5427e-05, 9.9992e-01],
        [2.8263e-04, 9.9972e-01],
        [4.0187e-03, 9.9598e-01],
        [4.7677e-10, 1.0000e+00],
        [9.9927e-01, 7.2507e-04],
        [7.9702e-07, 1.0000e+00],
        [8.8425e-02, 9.1157e-01],
        [9.0872e-01, 9.1277e-02],
        [9.9715e-01, 2.8522e-03],
        [9.9993e-01, 7.4961e-05],
        [9.9985e-01, 1.5058e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[6.3606e-07, 1.0000e+00],
        [9.9994e-01, 6.4085e-05],
        [1.1075e-03, 9.9889e-01],
        [2.2589e-08, 1.0000e+00],
        [9.9995e-01, 4.5440e-05],
        [9.9038e-01, 9.6244e-03],
        [1.7028e-07, 1.0000e+00],
        [6.4299e-01, 3.5701e-01],
        [1.9120e-06, 1.0000e+00],
        [4.9396e-06, 1.0000e+00],
        [1.5660e-02, 9.8434e-01],
        [9.9986e-01, 1.3758e-04],
        [2.0198e-02, 9.7980e-01],
        [5.2210e-06, 9.9999e-01],
        [9.9994e-01, 6.2426e-05],
        [2.0932e-05, 9.9998e-01],
        [3.0477e-07, 1.0000e+00],
        [9.8435e-01, 1.5653e-02],
        [3.6833e-06, 1.0000e+00],
        [1.5022e-06, 1.0000e+00],
        [9.9995e-01, 4.7853e-05],
        [5.1826e-03, 9.9482e-01],
        [4.5064e-05, 9.9995e-01],
        [3.4331e-05, 9.9997e-01],
        [1.6881e-04, 9.9983e-01],
        [9.9968e-01, 3.1551e-04],
        [4.8800e-07, 1.0000e+00],
        [5.4732e-03, 9.9453e-01],
        [1.2003e-05, 9.9999e-01],
        [9.9994e-01, 6.1121e-05],
        [1.6592e-03, 9.9834e-01],
        [9.9989e-01, 1.1378e-04],
        [3.6835e-05, 9.9996e-01],
        [8.5079e-07, 1.0000e+00],
        [1.5716e-03, 9.9843e-01],
        [9.5292e-06, 9.9999e-01],
        [5.4045e-07, 1.0000e+00],
        [9.9989e-01, 1.1163e-04],
        [7.5194e-02, 9.2481e-01],
        [2.6134e-01, 7.3866e-01],
        [9.9991e-01, 9.4256e-05],
        [9.9999e-01, 1.3759e-05],
        [3.5129e-05, 9.9996e-01],
        [4.4046e-02, 9.5595e-01],
        [9.9996e-01, 4.3412e-05],
        [9.8716e-06, 9.9999e-01],
        [1.1719e-05, 9.9999e-01],
        [1.6408e-02, 9.8359e-01],
        [9.9994e-01, 6.0387e-05],
        [9.9988e-01, 1.2320e-04],
        [9.9950e-01, 5.0344e-04],
        [9.9973e-01, 2.7147e-04],
        [2.1138e-03, 9.9789e-01],
        [1.0000e+00, 2.9969e-07],
        [1.2976e-04, 9.9987e-01],
        [4.1379e-05, 9.9996e-01],
        [1.0490e-07, 1.0000e+00],
        [7.1857e-02, 9.2814e-01],
        [9.9596e-01, 4.0433e-03],
        [1.8240e-06, 1.0000e+00],
        [2.3017e-06, 1.0000e+00],
        [3.2816e-06, 1.0000e+00],
        [2.8746e-07, 1.0000e+00],
        [4.9035e-04, 9.9951e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.4041e-05, 9.9998e-01],
        [9.8181e-03, 9.9018e-01],
        [7.0532e-06, 9.9999e-01],
        [1.3318e-07, 1.0000e+00],
        [3.2521e-04, 9.9967e-01],
        [3.0009e-07, 1.0000e+00],
        [3.1202e-06, 1.0000e+00],
        [2.6972e-05, 9.9997e-01],
        [6.2126e-04, 9.9938e-01],
        [8.5538e-06, 9.9999e-01],
        [9.9979e-01, 2.1361e-04],
        [2.3285e-03, 9.9767e-01],
        [1.2538e-03, 9.9875e-01],
        [6.8936e-04, 9.9931e-01],
        [1.8360e-05, 9.9998e-01],
        [1.9677e-02, 9.8032e-01],
        [3.1281e-05, 9.9997e-01],
        [5.8058e-03, 9.9419e-01],
        [1.6651e-07, 1.0000e+00],
        [3.6102e-01, 6.3898e-01],
        [4.3515e-05, 9.9996e-01],
        [1.2431e-02, 9.8757e-01],
        [1.1413e-03, 9.9886e-01]], grad_fn=<SoftmaxBackward0>)
Epoch 7, loss: 0.3315
tensor([[9.9988e-01, 1.2321e-04],
        [9.9985e-01, 1.4784e-04],
        [3.1778e-05, 9.9997e-01],
        [4.3707e-08, 1.0000e+00],
        [2.9752e-06, 1.0000e+00],
        [1.1836e-06, 1.0000e+00],
        [9.9992e-01, 7.9868e-05],
        [9.9988e-01, 1.1619e-04],
        [9.9751e-01, 2.4920e-03],
        [9.4520e-07, 1.0000e+00],
        [9.9959e-01, 4.0868e-04],
        [9.9995e-01, 5.3875e-05],
        [1.4071e-04, 9.9986e-01],
        [2.0522e-05, 9.9998e-01],
        [9.9947e-01, 5.2863e-04],
        [9.9986e-01, 1.3637e-04],
        [9.9969e-01, 3.0662e-04],
        [2.7083e-08, 1.0000e+00],
        [1.2340e-06, 1.0000e+00],
        [1.0253e-03, 9.9897e-01],
        [9.9962e-01, 3.7957e-04],
        [8.3671e-07, 1.0000e+00],
        [8.2773e-07, 1.0000e+00],
        [1.4809e-04, 9.9985e-01],
        [9.9996e-01, 3.5236e-05],
        [9.9992e-01, 7.9124e-05],
        [5.0607e-05, 9.9995e-01],
        [9.9998e-01, 2.0914e-05],
        [1.7351e-06, 1.0000e+00],
        [2.8335e-05, 9.9997e-01],
        [9.9984e-01, 1.6206e-04],
        [7.4606e-01, 2.5394e-01],
        [9.9989e-01, 1.1175e-04],
        [9.9972e-01, 2.7957e-04],
        [1.3795e-03, 9.9862e-01],
        [9.9590e-01, 4.1002e-03],
        [3.8966e-08, 1.0000e+00],
        [1.5154e-03, 9.9848e-01],
        [1.0593e-07, 1.0000e+00],
        [8.6844e-03, 9.9132e-01],
        [1.6728e-06, 1.0000e+00],
        [9.4945e-05, 9.9990e-01],
        [3.8872e-05, 9.9996e-01],
        [4.4944e-07, 1.0000e+00],
        [9.9986e-01, 1.3968e-04],
        [1.3922e-02, 9.8608e-01],
        [7.7570e-07, 1.0000e+00],
        [1.3585e-02, 9.8641e-01],
        [2.1393e-08, 1.0000e+00],
        [9.9988e-01, 1.1872e-04],
        [9.3126e-08, 1.0000e+00],
        [3.7247e-05, 9.9996e-01],
        [6.8620e-04, 9.9931e-01],
        [3.5880e-07, 1.0000e+00],
        [8.0987e-07, 1.0000e+00],
        [4.6407e-03, 9.9536e-01],
        [9.9825e-01, 1.7451e-03],
        [1.3508e-06, 1.0000e+00],
        [9.9995e-01, 4.6926e-05],
        [1.8218e-07, 1.0000e+00],
        [1.8156e-03, 9.9818e-01],
        [9.9997e-01, 3.4984e-05],
        [9.9978e-01, 2.1507e-04],
        [9.9987e-01, 1.2951e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9152e-01, 8.0848e-01],
        [2.7128e-07, 1.0000e+00],
        [2.0681e-03, 9.9793e-01],
        [3.5506e-05, 9.9996e-01],
        [1.1628e-08, 1.0000e+00],
        [4.2521e-02, 9.5748e-01],
        [4.1628e-03, 9.9584e-01],
        [9.9992e-01, 8.3193e-05],
        [9.5832e-01, 4.1676e-02],
        [1.7765e-05, 9.9998e-01],
        [3.4463e-05, 9.9997e-01],
        [1.6299e-05, 9.9998e-01],
        [8.0840e-04, 9.9919e-01],
        [4.7140e-04, 9.9953e-01],
        [9.9980e-01, 1.9677e-04],
        [9.9989e-01, 1.0840e-04],
        [1.9363e-04, 9.9981e-01],
        [7.6420e-05, 9.9992e-01],
        [2.1103e-04, 9.9979e-01],
        [4.7724e-04, 9.9952e-01],
        [4.1519e-06, 1.0000e+00],
        [2.3180e-06, 1.0000e+00],
        [3.1241e-06, 1.0000e+00],
        [6.0154e-10, 1.0000e+00],
        [5.0574e-08, 1.0000e+00],
        [9.2209e-06, 9.9999e-01],
        [3.2895e-03, 9.9671e-01],
        [5.1746e-06, 9.9999e-01],
        [8.5629e-08, 1.0000e+00],
        [1.1753e-05, 9.9999e-01],
        [8.6224e-02, 9.1378e-01],
        [1.6133e-09, 1.0000e+00],
        [8.2435e-04, 9.9918e-01],
        [1.8011e-06, 1.0000e+00],
        [1.7470e-02, 9.8253e-01],
        [3.9218e-05, 9.9996e-01],
        [4.1622e-05, 9.9996e-01],
        [4.3544e-04, 9.9956e-01],
        [3.2358e-06, 1.0000e+00],
        [9.9973e-01, 2.7298e-04],
        [4.4062e-07, 1.0000e+00],
        [1.4617e-04, 9.9985e-01],
        [8.3223e-02, 9.1678e-01],
        [9.9997e-01, 2.8493e-05],
        [2.4371e-05, 9.9998e-01],
        [4.2261e-07, 1.0000e+00],
        [6.7433e-01, 3.2567e-01],
        [9.9973e-01, 2.6681e-04],
        [5.4275e-08, 1.0000e+00],
        [1.2231e-04, 9.9988e-01],
        [9.9997e-01, 3.3198e-05],
        [5.1367e-07, 1.0000e+00],
        [9.9999e-01, 1.4673e-05],
        [2.2478e-04, 9.9978e-01],
        [2.9749e-05, 9.9997e-01],
        [4.8917e-08, 1.0000e+00],
        [1.9333e-05, 9.9998e-01],
        [1.2587e-06, 1.0000e+00],
        [1.3288e-05, 9.9999e-01],
        [5.8533e-02, 9.4147e-01],
        [1.1114e-02, 9.8889e-01],
        [2.9149e-05, 9.9997e-01],
        [1.0341e-06, 1.0000e+00],
        [4.1723e-03, 9.9583e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5440e-10, 1.0000e+00],
        [9.9992e-01, 7.6841e-05],
        [5.5848e-05, 9.9994e-01],
        [9.9989e-01, 1.0762e-04],
        [9.9080e-01, 9.1958e-03],
        [6.3513e-01, 3.6487e-01],
        [9.9995e-01, 5.3762e-05],
        [7.6870e-02, 9.2313e-01],
        [1.8993e-06, 1.0000e+00],
        [8.7146e-01, 1.2854e-01],
        [3.8976e-03, 9.9610e-01],
        [9.9997e-01, 2.6397e-05],
        [4.7459e-09, 1.0000e+00],
        [9.9989e-01, 1.0620e-04],
        [9.7046e-07, 1.0000e+00],
        [9.9994e-01, 6.4081e-05],
        [1.0262e-05, 9.9999e-01],
        [1.3604e-02, 9.8640e-01],
        [9.9710e-01, 2.8996e-03],
        [1.9032e-03, 9.9810e-01],
        [9.7490e-01, 2.5097e-02],
        [2.8809e-04, 9.9971e-01],
        [5.3445e-05, 9.9995e-01],
        [2.3102e-04, 9.9977e-01],
        [3.2234e-06, 1.0000e+00],
        [4.0157e-05, 9.9996e-01],
        [9.9979e-01, 2.1179e-04],
        [9.9990e-01, 9.6643e-05],
        [2.0544e-03, 9.9795e-01],
        [9.0855e-05, 9.9991e-01],
        [1.4100e-02, 9.8590e-01],
        [4.8429e-03, 9.9516e-01],
        [9.9988e-01, 1.1873e-04],
        [9.7731e-05, 9.9990e-01],
        [9.9987e-01, 1.3435e-04],
        [5.8350e-04, 9.9942e-01],
        [2.2616e-02, 9.7738e-01],
        [2.2225e-05, 9.9998e-01],
        [9.9993e-01, 6.7033e-05],
        [2.8607e-05, 9.9997e-01],
        [4.3143e-07, 1.0000e+00],
        [9.9964e-01, 3.5972e-04],
        [3.6966e-04, 9.9963e-01],
        [1.4376e-05, 9.9999e-01],
        [9.9986e-01, 1.4012e-04],
        [2.9644e-03, 9.9704e-01],
        [1.6541e-06, 1.0000e+00],
        [2.6878e-05, 9.9997e-01],
        [3.6262e-10, 1.0000e+00],
        [1.7183e-05, 9.9998e-01],
        [1.1910e-06, 1.0000e+00],
        [9.9996e-01, 3.9130e-05],
        [6.8090e-08, 1.0000e+00],
        [4.2511e-04, 9.9957e-01],
        [3.6080e-06, 1.0000e+00],
        [3.3289e-04, 9.9967e-01],
        [1.0162e-05, 9.9999e-01],
        [9.9994e-01, 6.1507e-05],
        [1.0444e-06, 1.0000e+00],
        [2.5345e-06, 1.0000e+00],
        [8.8652e-06, 9.9999e-01],
        [1.3557e-07, 1.0000e+00],
        [2.5351e-04, 9.9975e-01],
        [9.9996e-01, 3.7921e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[2.6281e-04, 9.9974e-01],
        [1.3691e-05, 9.9999e-01],
        [2.6865e-07, 1.0000e+00],
        [2.5167e-05, 9.9997e-01],
        [3.2887e-05, 9.9997e-01],
        [4.5076e-10, 1.0000e+00],
        [3.7566e-06, 1.0000e+00],
        [9.5224e-01, 4.7758e-02],
        [1.6057e-08, 1.0000e+00],
        [6.4100e-03, 9.9359e-01],
        [6.6161e-10, 1.0000e+00],
        [4.1869e-08, 1.0000e+00],
        [2.5359e-08, 1.0000e+00],
        [9.9981e-01, 1.9498e-04],
        [8.7985e-04, 9.9912e-01],
        [9.8342e-01, 1.6579e-02],
        [9.9984e-01, 1.5959e-04],
        [9.9988e-01, 1.1783e-04],
        [5.2919e-07, 1.0000e+00],
        [1.0485e-01, 8.9515e-01],
        [3.6265e-09, 1.0000e+00],
        [9.9982e-01, 1.7616e-04],
        [1.1530e-04, 9.9988e-01],
        [3.0071e-07, 1.0000e+00],
        [1.1443e-03, 9.9886e-01],
        [9.9991e-01, 8.8975e-05],
        [1.5980e-08, 1.0000e+00],
        [9.9980e-01, 2.0051e-04],
        [9.9992e-01, 7.6707e-05],
        [1.0243e-03, 9.9898e-01],
        [1.6502e-10, 1.0000e+00],
        [9.9870e-01, 1.2966e-03],
        [5.9892e-06, 9.9999e-01],
        [1.3367e-06, 1.0000e+00],
        [3.9229e-05, 9.9996e-01],
        [9.9989e-01, 1.1108e-04],
        [1.9481e-07, 1.0000e+00],
        [3.8023e-10, 1.0000e+00],
        [3.1961e-05, 9.9997e-01],
        [3.5692e-09, 1.0000e+00],
        [2.2475e-07, 1.0000e+00],
        [9.9975e-01, 2.4695e-04],
        [4.3541e-08, 1.0000e+00],
        [9.4557e-10, 1.0000e+00],
        [5.5272e-05, 9.9994e-01],
        [2.5560e-04, 9.9974e-01],
        [3.4093e-07, 1.0000e+00],
        [1.9294e-06, 1.0000e+00],
        [4.0839e-07, 1.0000e+00],
        [1.4797e-05, 9.9999e-01],
        [1.7133e-02, 9.8287e-01],
        [3.9792e-03, 9.9602e-01],
        [1.7715e-04, 9.9982e-01],
        [9.9960e-01, 4.0227e-04],
        [3.8583e-06, 1.0000e+00],
        [8.8814e-08, 1.0000e+00],
        [2.1182e-08, 1.0000e+00],
        [1.2095e-04, 9.9988e-01],
        [7.0841e-03, 9.9292e-01],
        [1.0966e-04, 9.9989e-01],
        [1.4812e-06, 1.0000e+00],
        [1.4520e-03, 9.9855e-01],
        [3.6300e-07, 1.0000e+00],
        [9.9994e-01, 5.5066e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0374e-07, 1.0000e+00],
        [5.2403e-05, 9.9995e-01],
        [9.9989e-01, 1.0768e-04],
        [1.8423e-07, 1.0000e+00],
        [7.9413e-04, 9.9921e-01],
        [9.9982e-01, 1.8029e-04],
        [2.3009e-07, 1.0000e+00],
        [2.6266e-06, 1.0000e+00],
        [3.5907e-05, 9.9996e-01],
        [9.9978e-01, 2.1760e-04],
        [3.8767e-06, 1.0000e+00],
        [7.2196e-03, 9.9278e-01],
        [1.5298e-04, 9.9985e-01],
        [2.2874e-09, 1.0000e+00],
        [1.6407e-06, 1.0000e+00],
        [4.0347e-07, 1.0000e+00],
        [2.8950e-02, 9.7105e-01],
        [6.8179e-09, 1.0000e+00],
        [9.9748e-01, 2.5175e-03],
        [5.9159e-05, 9.9994e-01],
        [1.1382e-02, 9.8862e-01],
        [1.7560e-05, 9.9998e-01],
        [9.9995e-01, 5.3746e-05],
        [1.2866e-02, 9.8713e-01],
        [3.8525e-13, 1.0000e+00],
        [6.2393e-09, 1.0000e+00],
        [1.0017e-02, 9.8998e-01],
        [1.0000e+00, 2.2625e-06],
        [1.2299e-08, 1.0000e+00],
        [3.1524e-08, 1.0000e+00],
        [1.4372e-03, 9.9856e-01],
        [5.7416e-07, 1.0000e+00],
        [9.9986e-01, 1.4163e-04],
        [7.0337e-04, 9.9930e-01],
        [6.0228e-07, 1.0000e+00],
        [7.7413e-01, 2.2587e-01],
        [6.5645e-06, 9.9999e-01],
        [3.1112e-08, 1.0000e+00],
        [9.9866e-01, 1.3407e-03],
        [2.9863e-03, 9.9701e-01],
        [4.7735e-04, 9.9952e-01],
        [9.9984e-01, 1.5898e-04],
        [1.7944e-04, 9.9982e-01],
        [1.7066e-07, 1.0000e+00],
        [8.6122e-01, 1.3878e-01],
        [6.9731e-05, 9.9993e-01],
        [6.8746e-04, 9.9931e-01],
        [1.0002e-01, 8.9999e-01],
        [9.9984e-01, 1.5948e-04],
        [9.9992e-01, 8.1167e-05],
        [9.9995e-01, 5.1546e-05],
        [9.9993e-01, 6.7911e-05],
        [1.0000e+00, 3.4875e-07],
        [1.0522e-07, 1.0000e+00],
        [2.0830e-08, 1.0000e+00],
        [6.9401e-03, 9.9306e-01],
        [1.0265e-06, 1.0000e+00],
        [1.2081e-05, 9.9999e-01],
        [1.2554e-05, 9.9999e-01],
        [9.9995e-01, 4.5328e-05],
        [3.4144e-08, 1.0000e+00],
        [2.4386e-06, 1.0000e+00],
        [2.2600e-07, 1.0000e+00],
        [9.9994e-01, 5.7887e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[7.9039e-10, 1.0000e+00],
        [2.3002e-04, 9.9977e-01],
        [6.3073e-10, 1.0000e+00],
        [9.9710e-01, 2.9043e-03],
        [4.0675e-08, 1.0000e+00],
        [9.9754e-01, 2.4632e-03],
        [2.8578e-06, 1.0000e+00],
        [9.9988e-01, 1.1928e-04],
        [7.1530e-07, 1.0000e+00],
        [1.9025e-07, 1.0000e+00],
        [1.8114e-06, 1.0000e+00],
        [2.3746e-08, 1.0000e+00],
        [1.4293e-10, 1.0000e+00],
        [9.8177e-01, 1.8229e-02],
        [1.6944e-08, 1.0000e+00],
        [8.8802e-01, 1.1198e-01],
        [3.9676e-08, 1.0000e+00],
        [1.2728e-05, 9.9999e-01],
        [9.9994e-01, 5.5987e-05],
        [9.9979e-01, 2.1456e-04],
        [4.0921e-08, 1.0000e+00],
        [9.9982e-01, 1.8235e-04],
        [7.4776e-06, 9.9999e-01],
        [4.9479e-08, 1.0000e+00],
        [1.3290e-09, 1.0000e+00],
        [5.7249e-06, 9.9999e-01],
        [9.9997e-01, 2.8268e-05],
        [4.9954e-06, 9.9999e-01],
        [2.8210e-06, 1.0000e+00],
        [9.1736e-07, 1.0000e+00],
        [3.2302e-04, 9.9968e-01],
        [4.0515e-06, 1.0000e+00],
        [6.3283e-07, 1.0000e+00],
        [9.9787e-01, 2.1304e-03],
        [1.0088e-05, 9.9999e-01],
        [9.9984e-01, 1.6488e-04],
        [3.2792e-05, 9.9997e-01],
        [7.1534e-06, 9.9999e-01],
        [7.5275e-05, 9.9992e-01],
        [9.9992e-01, 8.0047e-05],
        [1.1340e-10, 1.0000e+00],
        [9.9981e-01, 1.9291e-04],
        [1.7700e-03, 9.9823e-01],
        [9.9871e-01, 1.2899e-03],
        [6.7408e-07, 1.0000e+00],
        [6.9365e-11, 1.0000e+00],
        [2.6940e-08, 1.0000e+00],
        [2.4521e-07, 1.0000e+00],
        [2.0948e-02, 9.7905e-01],
        [2.6271e-09, 1.0000e+00],
        [7.7541e-07, 1.0000e+00],
        [1.9540e-06, 1.0000e+00],
        [4.2110e-06, 1.0000e+00],
        [9.9497e-05, 9.9990e-01],
        [3.9859e-04, 9.9960e-01],
        [6.6910e-05, 9.9993e-01],
        [3.8172e-05, 9.9996e-01],
        [3.5735e-06, 1.0000e+00],
        [3.1163e-07, 1.0000e+00],
        [9.9992e-01, 8.3348e-05],
        [9.9938e-01, 6.1878e-04],
        [5.8743e-10, 1.0000e+00],
        [3.4159e-07, 1.0000e+00],
        [9.9757e-01, 2.4320e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[3.6941e-08, 1.0000e+00],
        [9.1718e-08, 1.0000e+00],
        [1.7409e-05, 9.9998e-01],
        [6.5063e-03, 9.9349e-01],
        [8.5098e-07, 1.0000e+00],
        [1.7092e-05, 9.9998e-01],
        [9.9983e-01, 1.6762e-04],
        [4.9430e-05, 9.9995e-01],
        [2.7323e-06, 1.0000e+00],
        [6.0883e-10, 1.0000e+00],
        [4.2387e-09, 1.0000e+00],
        [9.9982e-01, 1.7696e-04],
        [2.5031e-03, 9.9750e-01],
        [9.2080e-12, 1.0000e+00],
        [1.3463e-04, 9.9987e-01],
        [9.2121e-10, 1.0000e+00],
        [4.7611e-07, 1.0000e+00],
        [2.0628e-05, 9.9998e-01],
        [9.9991e-01, 9.3093e-05],
        [1.8105e-07, 1.0000e+00],
        [9.9999e-01, 5.0199e-06],
        [5.1370e-04, 9.9949e-01],
        [2.3974e-08, 1.0000e+00],
        [9.9745e-01, 2.5535e-03],
        [9.9880e-01, 1.2046e-03],
        [9.9980e-01, 2.0090e-04],
        [5.6382e-01, 4.3618e-01],
        [9.7985e-01, 2.0147e-02],
        [7.0011e-02, 9.2999e-01],
        [3.0958e-09, 1.0000e+00],
        [1.3836e-07, 1.0000e+00],
        [1.2901e-02, 9.8710e-01],
        [3.1152e-07, 1.0000e+00],
        [1.7906e-03, 9.9821e-01],
        [1.1118e-05, 9.9999e-01],
        [9.7596e-01, 2.4041e-02],
        [6.2815e-08, 1.0000e+00],
        [8.4601e-07, 1.0000e+00],
        [2.8494e-04, 9.9972e-01],
        [2.0682e-07, 1.0000e+00],
        [1.1924e-10, 1.0000e+00],
        [9.9957e-01, 4.2609e-04],
        [2.3899e-06, 1.0000e+00],
        [9.9687e-01, 3.1317e-03],
        [9.9965e-01, 3.4961e-04],
        [1.9386e-06, 1.0000e+00],
        [9.9999e-01, 1.4889e-05],
        [1.9525e-10, 1.0000e+00],
        [3.1365e-02, 9.6863e-01],
        [2.3959e-09, 1.0000e+00],
        [2.7066e-06, 1.0000e+00],
        [1.2996e-10, 1.0000e+00],
        [1.2694e-05, 9.9999e-01],
        [9.9891e-01, 1.0876e-03],
        [1.0051e-10, 1.0000e+00],
        [5.9364e-08, 1.0000e+00],
        [3.7697e-06, 1.0000e+00],
        [9.9998e-01, 2.2062e-05],
        [3.0639e-07, 1.0000e+00],
        [1.9300e-08, 1.0000e+00],
        [1.3308e-07, 1.0000e+00],
        [8.7944e-05, 9.9991e-01],
        [4.1359e-06, 1.0000e+00],
        [3.5891e-05, 9.9996e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.2299e-06, 9.9999e-01],
        [1.0967e-06, 1.0000e+00],
        [1.6457e-10, 1.0000e+00],
        [1.3120e-11, 1.0000e+00],
        [1.2931e-04, 9.9987e-01],
        [1.7253e-08, 1.0000e+00],
        [3.7603e-04, 9.9962e-01],
        [3.0622e-10, 1.0000e+00],
        [1.9021e-10, 1.0000e+00],
        [2.8843e-08, 1.0000e+00],
        [9.9999e-01, 1.3303e-05],
        [2.8682e-06, 1.0000e+00],
        [4.7630e-04, 9.9952e-01],
        [9.9898e-01, 1.0237e-03],
        [1.9493e-05, 9.9998e-01],
        [2.0560e-10, 1.0000e+00],
        [6.0752e-09, 1.0000e+00],
        [4.7440e-06, 1.0000e+00],
        [7.5126e-08, 1.0000e+00],
        [9.9996e-01, 4.4447e-05],
        [3.0086e-07, 1.0000e+00],
        [9.9989e-01, 1.0763e-04],
        [6.6843e-10, 1.0000e+00],
        [6.1255e-08, 1.0000e+00],
        [2.6020e-09, 1.0000e+00],
        [1.9787e-02, 9.8021e-01],
        [1.5296e-05, 9.9998e-01],
        [2.3128e-07, 1.0000e+00],
        [6.7977e-04, 9.9932e-01],
        [5.1125e-02, 9.4888e-01],
        [1.0000e+00, 1.6451e-06],
        [9.2563e-01, 7.4371e-02],
        [1.8514e-08, 1.0000e+00],
        [4.4727e-08, 1.0000e+00],
        [4.3015e-05, 9.9996e-01],
        [2.5530e-01, 7.4470e-01],
        [9.8199e-01, 1.8012e-02],
        [9.9395e-01, 6.0511e-03],
        [1.2414e-04, 9.9988e-01],
        [9.3561e-01, 6.4391e-02],
        [5.5104e-06, 9.9999e-01],
        [9.9995e-01, 5.4855e-05],
        [3.8526e-09, 1.0000e+00],
        [9.8643e-01, 1.3570e-02],
        [7.4560e-10, 1.0000e+00],
        [5.2313e-08, 1.0000e+00],
        [9.8847e-01, 1.1533e-02],
        [6.1928e-06, 9.9999e-01],
        [5.7889e-07, 1.0000e+00],
        [5.9547e-07, 1.0000e+00],
        [3.5712e-13, 1.0000e+00],
        [2.8867e-07, 1.0000e+00],
        [9.9867e-01, 1.3327e-03],
        [3.3701e-08, 1.0000e+00],
        [9.9993e-01, 7.0731e-05],
        [7.4861e-09, 1.0000e+00],
        [7.3031e-10, 1.0000e+00],
        [4.0313e-07, 1.0000e+00],
        [9.9933e-01, 6.7493e-04],
        [2.1818e-08, 1.0000e+00],
        [8.7836e-07, 1.0000e+00],
        [9.9993e-01, 7.0587e-05],
        [2.1788e-06, 1.0000e+00],
        [9.9968e-01, 3.2011e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0021e-10, 1.0000e+00],
        [1.1820e-08, 1.0000e+00],
        [1.7303e-08, 1.0000e+00],
        [2.8575e-10, 1.0000e+00],
        [1.3150e-04, 9.9987e-01],
        [6.5935e-05, 9.9993e-01],
        [2.4900e-04, 9.9975e-01],
        [4.9171e-10, 1.0000e+00],
        [2.0465e-02, 9.7953e-01],
        [1.3127e-10, 1.0000e+00],
        [9.9763e-01, 2.3726e-03],
        [3.9826e-09, 1.0000e+00],
        [3.8218e-09, 1.0000e+00],
        [8.2691e-01, 1.7309e-01],
        [9.9999e-01, 1.3803e-05],
        [9.9997e-01, 3.1437e-05],
        [9.9979e-01, 2.1034e-04],
        [9.9708e-01, 2.9156e-03],
        [1.0962e-04, 9.9989e-01],
        [9.3505e-01, 6.4955e-02],
        [1.0399e-06, 1.0000e+00],
        [4.3624e-08, 1.0000e+00],
        [2.7597e-07, 1.0000e+00],
        [1.3207e-07, 1.0000e+00],
        [9.8600e-01, 1.4001e-02],
        [1.1754e-04, 9.9988e-01],
        [4.7274e-08, 1.0000e+00],
        [4.2868e-10, 1.0000e+00],
        [1.3307e-10, 1.0000e+00],
        [1.0248e-04, 9.9990e-01],
        [3.9200e-06, 1.0000e+00],
        [6.6200e-08, 1.0000e+00],
        [9.7610e-11, 1.0000e+00],
        [9.9994e-01, 6.1800e-05],
        [8.6508e-01, 1.3492e-01],
        [7.9795e-07, 1.0000e+00],
        [9.9272e-01, 7.2812e-03],
        [1.4224e-06, 1.0000e+00],
        [5.5456e-10, 1.0000e+00],
        [9.9961e-01, 3.8958e-04],
        [1.3934e-07, 1.0000e+00],
        [3.2271e-06, 1.0000e+00],
        [4.6823e-09, 1.0000e+00],
        [1.8786e-07, 1.0000e+00],
        [2.3005e-08, 1.0000e+00],
        [3.8672e-09, 1.0000e+00],
        [7.5643e-08, 1.0000e+00],
        [3.9131e-06, 1.0000e+00],
        [3.2252e-05, 9.9997e-01],
        [9.9978e-01, 2.2033e-04],
        [9.9929e-01, 7.0832e-04],
        [8.9144e-09, 1.0000e+00],
        [1.5693e-04, 9.9984e-01],
        [7.2020e-07, 1.0000e+00],
        [7.8304e-08, 1.0000e+00],
        [7.7334e-14, 1.0000e+00],
        [4.8472e-08, 1.0000e+00],
        [2.4582e-07, 1.0000e+00],
        [9.9999e-01, 5.9984e-06],
        [8.4922e-11, 1.0000e+00],
        [9.9992e-01, 8.1105e-05],
        [4.1408e-05, 9.9996e-01],
        [5.2285e-07, 1.0000e+00],
        [4.1941e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9981e-01, 1.9433e-04],
        [6.1586e-09, 1.0000e+00],
        [1.7648e-08, 1.0000e+00],
        [1.2488e-09, 1.0000e+00],
        [1.1697e-04, 9.9988e-01],
        [2.7905e-06, 1.0000e+00],
        [1.4575e-08, 1.0000e+00],
        [3.6369e-08, 1.0000e+00],
        [3.3846e-05, 9.9997e-01],
        [9.9456e-01, 5.4399e-03],
        [9.4208e-01, 5.7921e-02],
        [7.8912e-05, 9.9992e-01],
        [1.0146e-05, 9.9999e-01],
        [9.9870e-01, 1.2959e-03],
        [1.5184e-08, 1.0000e+00],
        [9.9907e-01, 9.2897e-04],
        [9.9797e-01, 2.0346e-03],
        [5.9901e-06, 9.9999e-01],
        [3.9714e-06, 1.0000e+00],
        [2.3733e-05, 9.9998e-01],
        [1.4787e-07, 1.0000e+00],
        [1.1988e-02, 9.8801e-01],
        [4.7184e-12, 1.0000e+00],
        [6.9882e-05, 9.9993e-01],
        [1.9216e-02, 9.8078e-01],
        [9.9595e-09, 1.0000e+00],
        [9.8528e-07, 1.0000e+00],
        [6.5472e-07, 1.0000e+00],
        [3.7487e-08, 1.0000e+00],
        [6.7676e-06, 9.9999e-01],
        [4.7278e-05, 9.9995e-01],
        [5.1495e-06, 9.9999e-01],
        [2.0713e-07, 1.0000e+00],
        [9.9970e-01, 3.0099e-04],
        [5.1568e-03, 9.9484e-01],
        [1.7508e-07, 1.0000e+00],
        [2.1349e-10, 1.0000e+00],
        [3.6891e-07, 1.0000e+00],
        [4.1255e-09, 1.0000e+00],
        [6.4494e-05, 9.9994e-01],
        [9.9983e-01, 1.7259e-04],
        [5.0107e-07, 1.0000e+00],
        [1.0319e-07, 1.0000e+00],
        [3.4248e-11, 1.0000e+00],
        [1.0000e+00, 9.2839e-07],
        [1.2118e-05, 9.9999e-01],
        [8.8783e-05, 9.9991e-01],
        [2.0135e-07, 1.0000e+00],
        [9.9982e-01, 1.7579e-04],
        [8.3699e-12, 1.0000e+00],
        [4.0841e-06, 1.0000e+00],
        [8.7020e-05, 9.9991e-01],
        [5.3904e-03, 9.9461e-01],
        [9.7945e-11, 1.0000e+00],
        [5.7902e-07, 1.0000e+00],
        [1.4259e-02, 9.8574e-01],
        [9.2459e-08, 1.0000e+00],
        [5.0956e-09, 1.0000e+00],
        [9.9826e-01, 1.7426e-03],
        [1.7846e-07, 1.0000e+00],
        [7.0654e-02, 9.2935e-01],
        [9.4717e-04, 9.9905e-01],
        [2.3719e-09, 1.0000e+00],
        [3.7829e-05, 9.9996e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9167e-07, 1.0000e+00],
        [3.7157e-08, 1.0000e+00],
        [2.6389e-07, 1.0000e+00],
        [5.5971e-07, 1.0000e+00],
        [2.0703e-07, 1.0000e+00],
        [9.3201e-08, 1.0000e+00],
        [9.5545e-01, 4.4546e-02],
        [6.2958e-07, 1.0000e+00],
        [1.9125e-13, 1.0000e+00],
        [6.7622e-08, 1.0000e+00],
        [3.3125e-09, 1.0000e+00],
        [3.0827e-03, 9.9692e-01],
        [7.0291e-09, 1.0000e+00],
        [2.0692e-08, 1.0000e+00],
        [5.1991e-09, 1.0000e+00],
        [9.9827e-01, 1.7339e-03],
        [4.3137e-11, 1.0000e+00],
        [9.7895e-01, 2.1048e-02],
        [1.7652e-09, 1.0000e+00],
        [9.9982e-01, 1.8352e-04],
        [1.0608e-05, 9.9999e-01],
        [9.9763e-01, 2.3663e-03],
        [1.1017e-05, 9.9999e-01],
        [2.4310e-07, 1.0000e+00],
        [5.2790e-08, 1.0000e+00],
        [9.5155e-05, 9.9990e-01],
        [5.8404e-08, 1.0000e+00],
        [5.7954e-08, 1.0000e+00],
        [1.1270e-06, 1.0000e+00],
        [2.2903e-10, 1.0000e+00],
        [1.0858e-06, 1.0000e+00],
        [8.8639e-05, 9.9991e-01],
        [1.6893e-04, 9.9983e-01],
        [1.6557e-09, 1.0000e+00],
        [4.0177e-08, 1.0000e+00],
        [9.9851e-01, 1.4877e-03],
        [6.2603e-08, 1.0000e+00],
        [2.3076e-07, 1.0000e+00],
        [1.2162e-06, 1.0000e+00],
        [9.9919e-01, 8.1358e-04],
        [9.9958e-01, 4.1607e-04],
        [1.4074e-04, 9.9986e-01],
        [2.2587e-09, 1.0000e+00],
        [3.0348e-10, 1.0000e+00],
        [9.9991e-01, 8.6760e-05],
        [2.9312e-06, 1.0000e+00],
        [9.9970e-01, 3.0072e-04],
        [7.4081e-07, 1.0000e+00],
        [3.5313e-07, 1.0000e+00],
        [2.5917e-08, 1.0000e+00],
        [1.2165e-09, 1.0000e+00],
        [3.3229e-08, 1.0000e+00],
        [9.9718e-01, 2.8178e-03],
        [1.3436e-07, 1.0000e+00],
        [5.1756e-01, 4.8244e-01],
        [9.2806e-08, 1.0000e+00],
        [1.4490e-06, 1.0000e+00],
        [1.7142e-11, 1.0000e+00],
        [5.4530e-06, 9.9999e-01],
        [9.5615e-03, 9.9044e-01],
        [7.0170e-08, 1.0000e+00],
        [1.4750e-05, 9.9999e-01],
        [9.9983e-01, 1.7309e-04],
        [7.3484e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.8014e-09, 1.0000e+00],
        [9.9975e-01, 2.5123e-04],
        [4.4809e-06, 1.0000e+00],
        [8.3664e-05, 9.9992e-01],
        [8.8096e-01, 1.1904e-01],
        [4.1150e-03, 9.9588e-01],
        [9.9999e-01, 5.7212e-06],
        [1.5512e-09, 1.0000e+00],
        [1.5203e-07, 1.0000e+00],
        [8.6443e-06, 9.9999e-01],
        [8.4289e-10, 1.0000e+00],
        [9.9973e-01, 2.6787e-04],
        [7.5999e-05, 9.9992e-01],
        [1.1887e-06, 1.0000e+00],
        [2.4384e-07, 1.0000e+00],
        [8.8057e-06, 9.9999e-01],
        [2.6805e-06, 1.0000e+00],
        [1.0807e-09, 1.0000e+00],
        [1.6177e-10, 1.0000e+00],
        [2.7498e-07, 1.0000e+00],
        [3.6577e-11, 1.0000e+00],
        [4.4214e-03, 9.9558e-01],
        [9.9986e-01, 1.3888e-04],
        [9.9971e-01, 2.9458e-04],
        [8.5225e-09, 1.0000e+00],
        [9.9181e-01, 8.1949e-03],
        [2.7856e-08, 1.0000e+00],
        [9.9988e-01, 1.2265e-04],
        [3.5231e-05, 9.9996e-01],
        [9.9874e-01, 1.2561e-03],
        [9.8551e-06, 9.9999e-01],
        [9.9967e-01, 3.2599e-04],
        [2.8907e-06, 1.0000e+00],
        [8.4180e-11, 1.0000e+00],
        [4.7200e-08, 1.0000e+00],
        [9.9984e-01, 1.5814e-04],
        [2.2792e-06, 1.0000e+00],
        [1.3293e-08, 1.0000e+00],
        [3.0615e-05, 9.9997e-01],
        [9.9907e-01, 9.3037e-04],
        [1.8862e-08, 1.0000e+00],
        [9.9980e-01, 1.9606e-04],
        [2.4148e-07, 1.0000e+00],
        [5.3687e-10, 1.0000e+00],
        [1.7991e-06, 1.0000e+00],
        [1.5632e-10, 1.0000e+00],
        [9.5683e-02, 9.0432e-01],
        [8.6919e-11, 1.0000e+00],
        [8.9846e-10, 1.0000e+00],
        [8.8242e-04, 9.9912e-01],
        [9.9965e-01, 3.4621e-04],
        [1.4673e-06, 1.0000e+00],
        [9.9992e-01, 7.9841e-05],
        [9.5233e-06, 9.9999e-01],
        [3.3803e-09, 1.0000e+00],
        [2.6178e-04, 9.9974e-01],
        [1.4853e-03, 9.9851e-01],
        [2.4937e-05, 9.9998e-01],
        [1.7320e-08, 1.0000e+00],
        [1.0807e-09, 1.0000e+00],
        [5.0812e-04, 9.9949e-01],
        [1.1332e-06, 1.0000e+00],
        [3.3605e-08, 1.0000e+00],
        [1.6552e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.0431e-05, 9.9994e-01],
        [9.9988e-01, 1.1867e-04],
        [7.3175e-08, 1.0000e+00],
        [2.1399e-06, 1.0000e+00],
        [5.1960e-01, 4.8040e-01],
        [5.0490e-08, 1.0000e+00],
        [1.8904e-09, 1.0000e+00],
        [1.7068e-08, 1.0000e+00],
        [3.1070e-08, 1.0000e+00],
        [1.5885e-09, 1.0000e+00],
        [1.1247e-06, 1.0000e+00],
        [3.5040e-05, 9.9996e-01],
        [1.9495e-03, 9.9805e-01],
        [7.8403e-10, 1.0000e+00],
        [3.7335e-06, 1.0000e+00],
        [8.2681e-10, 1.0000e+00],
        [1.0868e-06, 1.0000e+00],
        [2.5194e-10, 1.0000e+00],
        [9.9996e-01, 4.1680e-05],
        [9.9434e-09, 1.0000e+00],
        [7.5774e-10, 1.0000e+00],
        [2.9486e-06, 1.0000e+00],
        [1.0415e-06, 1.0000e+00],
        [9.7145e-01, 2.8550e-02],
        [3.6878e-07, 1.0000e+00],
        [4.3926e-06, 1.0000e+00],
        [2.0271e-03, 9.9797e-01],
        [1.2692e-05, 9.9999e-01],
        [9.5414e-07, 1.0000e+00],
        [8.7129e-01, 1.2871e-01],
        [9.9776e-01, 2.2404e-03],
        [5.2277e-05, 9.9995e-01],
        [1.3134e-07, 1.0000e+00],
        [7.1798e-06, 9.9999e-01],
        [2.2131e-07, 1.0000e+00],
        [2.9226e-03, 9.9708e-01],
        [4.3420e-05, 9.9996e-01],
        [9.9991e-01, 8.6356e-05],
        [6.6636e-11, 1.0000e+00],
        [6.3277e-04, 9.9937e-01],
        [5.0931e-12, 1.0000e+00],
        [1.2033e-05, 9.9999e-01],
        [9.9990e-01, 1.0038e-04],
        [3.9908e-09, 1.0000e+00],
        [1.4783e-04, 9.9985e-01],
        [3.0638e-08, 1.0000e+00],
        [1.4206e-08, 1.0000e+00],
        [9.9948e-01, 5.2304e-04],
        [9.9976e-01, 2.3600e-04],
        [1.8824e-09, 1.0000e+00],
        [5.3666e-01, 4.6334e-01],
        [8.5362e-07, 1.0000e+00],
        [2.1902e-07, 1.0000e+00],
        [9.9974e-01, 2.5595e-04],
        [9.5078e-08, 1.0000e+00],
        [1.9675e-09, 1.0000e+00],
        [1.6262e-05, 9.9998e-01],
        [2.7986e-06, 1.0000e+00],
        [6.2199e-05, 9.9994e-01],
        [4.3450e-08, 1.0000e+00],
        [9.9804e-01, 1.9633e-03],
        [3.7240e-08, 1.0000e+00],
        [9.9974e-01, 2.5994e-04],
        [4.1468e-03, 9.9585e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.3009e-04, 9.9907e-01],
        [9.5953e-08, 1.0000e+00],
        [6.5314e-10, 1.0000e+00],
        [9.9969e-01, 3.0882e-04],
        [1.0290e-04, 9.9990e-01],
        [5.2753e-07, 1.0000e+00],
        [1.0638e-05, 9.9999e-01],
        [1.6503e-09, 1.0000e+00],
        [2.7269e-04, 9.9973e-01],
        [1.2289e-04, 9.9988e-01],
        [1.5955e-04, 9.9984e-01],
        [2.8148e-08, 1.0000e+00],
        [1.4379e-08, 1.0000e+00],
        [1.1834e-07, 1.0000e+00],
        [1.6141e-07, 1.0000e+00],
        [3.6241e-07, 1.0000e+00],
        [9.9992e-01, 8.2621e-05],
        [8.9320e-01, 1.0680e-01],
        [4.8910e-06, 1.0000e+00],
        [1.7914e-04, 9.9982e-01],
        [4.7185e-08, 1.0000e+00],
        [5.0996e-06, 9.9999e-01],
        [9.9991e-01, 9.4766e-05],
        [1.0522e-08, 1.0000e+00],
        [9.9981e-01, 1.9109e-04],
        [4.5157e-10, 1.0000e+00],
        [2.0056e-07, 1.0000e+00],
        [9.9965e-01, 3.5190e-04],
        [9.9569e-01, 4.3051e-03],
        [7.4697e-07, 1.0000e+00],
        [1.1947e-04, 9.9988e-01],
        [9.5489e-01, 4.5114e-02],
        [9.9970e-01, 3.0440e-04],
        [1.2003e-07, 1.0000e+00],
        [7.5447e-10, 1.0000e+00],
        [1.4435e-09, 1.0000e+00],
        [1.4838e-14, 1.0000e+00],
        [9.9284e-08, 1.0000e+00],
        [6.4821e-07, 1.0000e+00],
        [2.5515e-08, 1.0000e+00],
        [4.9330e-06, 1.0000e+00],
        [4.3568e-03, 9.9564e-01],
        [5.8988e-04, 9.9941e-01],
        [1.5675e-09, 1.0000e+00],
        [9.8836e-01, 1.1635e-02],
        [2.3226e-05, 9.9998e-01],
        [3.8756e-10, 1.0000e+00],
        [9.9905e-01, 9.5278e-04],
        [8.6814e-07, 1.0000e+00],
        [1.0123e-05, 9.9999e-01],
        [9.9966e-01, 3.3712e-04],
        [6.8187e-03, 9.9318e-01],
        [7.5665e-09, 1.0000e+00],
        [9.9944e-01, 5.6162e-04],
        [9.9979e-01, 2.1072e-04],
        [6.8360e-07, 1.0000e+00],
        [9.9948e-01, 5.2253e-04],
        [6.7960e-08, 1.0000e+00],
        [7.3927e-06, 9.9999e-01],
        [1.7495e-05, 9.9998e-01],
        [1.2638e-05, 9.9999e-01],
        [6.1646e-10, 1.0000e+00],
        [9.9914e-01, 8.6420e-04],
        [9.9906e-01, 9.4439e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0564e-08, 1.0000e+00],
        [3.9212e-04, 9.9961e-01],
        [4.4841e-03, 9.9552e-01],
        [2.2240e-05, 9.9998e-01],
        [8.0126e-03, 9.9199e-01],
        [7.5639e-04, 9.9924e-01],
        [2.5667e-09, 1.0000e+00],
        [3.6755e-07, 1.0000e+00],
        [9.9992e-01, 8.2938e-05],
        [5.9522e-03, 9.9405e-01],
        [2.9156e-04, 9.9971e-01],
        [9.9990e-01, 1.0181e-04],
        [6.4000e-04, 9.9936e-01],
        [9.9972e-01, 2.7527e-04],
        [4.7720e-07, 1.0000e+00],
        [4.4774e-08, 1.0000e+00],
        [1.1740e-05, 9.9999e-01],
        [1.4536e-10, 1.0000e+00],
        [9.9991e-01, 8.7575e-05],
        [9.9986e-01, 1.3809e-04],
        [2.4807e-07, 1.0000e+00],
        [1.1973e-08, 1.0000e+00],
        [1.5167e-04, 9.9985e-01],
        [8.5848e-07, 1.0000e+00],
        [1.6234e-03, 9.9838e-01],
        [1.8030e-07, 1.0000e+00],
        [9.9980e-01, 1.9780e-04],
        [9.9993e-01, 6.5234e-05],
        [7.7934e-07, 1.0000e+00],
        [9.1041e-09, 1.0000e+00],
        [1.6570e-07, 1.0000e+00],
        [1.3369e-02, 9.8663e-01],
        [8.8353e-09, 1.0000e+00],
        [9.9091e-01, 9.0947e-03],
        [9.9996e-01, 3.9324e-05],
        [1.0924e-09, 1.0000e+00],
        [1.7288e-07, 1.0000e+00],
        [9.9589e-01, 4.1063e-03],
        [1.2829e-04, 9.9987e-01],
        [1.1799e-05, 9.9999e-01],
        [1.3276e-01, 8.6724e-01],
        [1.8745e-05, 9.9998e-01],
        [9.9754e-01, 2.4604e-03],
        [1.1555e-03, 9.9884e-01],
        [9.9992e-01, 8.0760e-05],
        [5.6695e-08, 1.0000e+00],
        [5.3568e-07, 1.0000e+00],
        [6.1263e-10, 1.0000e+00],
        [1.1273e-06, 1.0000e+00],
        [7.9596e-09, 1.0000e+00],
        [2.5261e-08, 1.0000e+00],
        [5.8513e-08, 1.0000e+00],
        [9.9964e-01, 3.5912e-04],
        [9.9981e-01, 1.8832e-04],
        [1.3977e-06, 1.0000e+00],
        [7.7363e-06, 9.9999e-01],
        [6.2564e-09, 1.0000e+00],
        [8.7353e-01, 1.2647e-01],
        [2.0271e-08, 1.0000e+00],
        [7.1073e-08, 1.0000e+00],
        [9.9709e-01, 2.9058e-03],
        [3.9745e-11, 1.0000e+00],
        [2.5519e-07, 1.0000e+00],
        [7.9268e-04, 9.9921e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1532e-09, 1.0000e+00],
        [9.9992e-01, 8.2032e-05],
        [3.3722e-08, 1.0000e+00],
        [9.9976e-01, 2.3961e-04],
        [1.0155e-04, 9.9990e-01],
        [5.0764e-09, 1.0000e+00],
        [1.2678e-04, 9.9987e-01],
        [3.9617e-06, 1.0000e+00],
        [1.6381e-06, 1.0000e+00],
        [1.0000e+00, 4.5778e-06],
        [1.0382e-09, 1.0000e+00],
        [9.9857e-01, 1.4270e-03],
        [9.9994e-01, 6.0985e-05],
        [4.2065e-06, 1.0000e+00],
        [2.2537e-05, 9.9998e-01],
        [2.5488e-12, 1.0000e+00],
        [1.1575e-07, 1.0000e+00],
        [8.1658e-05, 9.9992e-01],
        [9.9743e-01, 2.5747e-03],
        [1.7419e-06, 1.0000e+00],
        [1.3159e-03, 9.9868e-01],
        [3.8400e-05, 9.9996e-01],
        [3.5526e-07, 1.0000e+00],
        [6.2697e-08, 1.0000e+00],
        [1.3329e-02, 9.8667e-01],
        [1.3180e-04, 9.9987e-01],
        [2.3879e-03, 9.9761e-01],
        [1.3870e-01, 8.6130e-01],
        [5.6261e-06, 9.9999e-01],
        [9.9998e-01, 2.0307e-05],
        [1.6678e-09, 1.0000e+00],
        [9.9999e-01, 1.0053e-05],
        [9.9985e-01, 1.4517e-04],
        [9.9967e-01, 3.2571e-04],
        [8.9968e-09, 1.0000e+00],
        [1.0581e-10, 1.0000e+00],
        [9.9994e-01, 5.7918e-05],
        [3.5050e-03, 9.9650e-01],
        [9.9983e-01, 1.7345e-04],
        [9.9990e-01, 9.8183e-05],
        [1.9274e-04, 9.9981e-01],
        [1.4646e-02, 9.8535e-01],
        [1.5055e-01, 8.4945e-01],
        [9.9978e-01, 2.1631e-04],
        [7.4246e-08, 1.0000e+00],
        [9.9991e-01, 9.2484e-05],
        [1.9910e-06, 1.0000e+00],
        [2.2360e-06, 1.0000e+00],
        [2.1421e-08, 1.0000e+00],
        [2.5821e-10, 1.0000e+00],
        [4.7714e-06, 1.0000e+00],
        [9.9866e-01, 1.3443e-03],
        [9.9465e-01, 5.3499e-03],
        [2.7982e-07, 1.0000e+00],
        [5.8238e-04, 9.9942e-01],
        [1.1723e-07, 1.0000e+00],
        [4.9287e-03, 9.9507e-01],
        [9.0628e-06, 9.9999e-01],
        [8.6270e-08, 1.0000e+00],
        [1.2215e-05, 9.9999e-01],
        [9.9627e-01, 3.7344e-03],
        [2.8954e-06, 1.0000e+00],
        [3.1085e-05, 9.9997e-01],
        [9.9997e-01, 2.6696e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9846e-01, 1.5439e-03],
        [1.8458e-04, 9.9982e-01],
        [3.4634e-06, 1.0000e+00],
        [9.9931e-01, 6.8653e-04],
        [7.2868e-06, 9.9999e-01],
        [6.4383e-09, 1.0000e+00],
        [5.1883e-06, 9.9999e-01],
        [4.9407e-02, 9.5059e-01],
        [9.9947e-01, 5.3077e-04],
        [1.0175e-03, 9.9898e-01],
        [6.3162e-07, 1.0000e+00],
        [9.9997e-01, 3.2612e-05],
        [8.3384e-06, 9.9999e-01],
        [9.9981e-01, 1.8736e-04],
        [9.9545e-01, 4.5478e-03],
        [9.9990e-01, 1.0480e-04],
        [5.9116e-08, 1.0000e+00],
        [2.5583e-04, 9.9974e-01],
        [9.9680e-01, 3.1990e-03],
        [1.6896e-03, 9.9831e-01],
        [4.2074e-06, 1.0000e+00],
        [6.8772e-07, 1.0000e+00],
        [9.9308e-01, 6.9186e-03],
        [2.3531e-05, 9.9998e-01],
        [3.3536e-07, 1.0000e+00],
        [9.4884e-01, 5.1158e-02],
        [9.3587e-10, 1.0000e+00],
        [2.0280e-05, 9.9998e-01],
        [2.5243e-08, 1.0000e+00],
        [2.8883e-03, 9.9711e-01],
        [2.3626e-05, 9.9998e-01],
        [3.8312e-09, 1.0000e+00],
        [3.9759e-02, 9.6024e-01],
        [1.2749e-07, 1.0000e+00],
        [3.6917e-04, 9.9963e-01],
        [1.3072e-06, 1.0000e+00],
        [9.0217e-01, 9.7829e-02],
        [1.3320e-08, 1.0000e+00],
        [4.2599e-06, 1.0000e+00],
        [9.4686e-05, 9.9991e-01],
        [1.8673e-05, 9.9998e-01],
        [1.8990e-01, 8.1010e-01],
        [9.9932e-01, 6.8455e-04],
        [1.0839e-05, 9.9999e-01],
        [9.9981e-01, 1.9363e-04],
        [1.1599e-03, 9.9884e-01],
        [3.2692e-05, 9.9997e-01],
        [9.9978e-01, 2.2180e-04],
        [9.9958e-01, 4.2278e-04],
        [3.6037e-10, 1.0000e+00],
        [1.5462e-08, 1.0000e+00],
        [1.9321e-04, 9.9981e-01],
        [2.5430e-03, 9.9746e-01],
        [3.4532e-08, 1.0000e+00],
        [1.3054e-07, 1.0000e+00],
        [2.3451e-05, 9.9998e-01],
        [9.9980e-01, 2.0165e-04],
        [1.2908e-08, 1.0000e+00],
        [9.9977e-01, 2.2577e-04],
        [3.5859e-08, 1.0000e+00],
        [2.4942e-09, 1.0000e+00],
        [3.0495e-05, 9.9997e-01],
        [2.4418e-01, 7.5582e-01],
        [2.5087e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9964e-01, 3.6480e-04],
        [9.9924e-01, 7.5616e-04],
        [7.2772e-08, 1.0000e+00],
        [9.9993e-01, 6.8018e-05],
        [1.3066e-06, 1.0000e+00],
        [9.9973e-01, 2.6656e-04],
        [1.0480e-06, 1.0000e+00],
        [1.9938e-07, 1.0000e+00],
        [2.2408e-09, 1.0000e+00],
        [9.5617e-01, 4.3835e-02],
        [3.1408e-08, 1.0000e+00],
        [9.9378e-01, 6.2190e-03],
        [3.6543e-06, 1.0000e+00],
        [2.1376e-03, 9.9786e-01],
        [5.1047e-05, 9.9995e-01],
        [4.3662e-04, 9.9956e-01],
        [1.2236e-06, 1.0000e+00],
        [3.4926e-06, 1.0000e+00],
        [1.6949e-05, 9.9998e-01],
        [1.9334e-04, 9.9981e-01],
        [5.8339e-06, 9.9999e-01],
        [9.7547e-11, 1.0000e+00],
        [1.9274e-08, 1.0000e+00],
        [7.7801e-05, 9.9992e-01],
        [9.9996e-01, 3.6215e-05],
        [2.1990e-07, 1.0000e+00],
        [1.6493e-05, 9.9998e-01],
        [1.4363e-05, 9.9999e-01],
        [4.7332e-03, 9.9527e-01],
        [1.6289e-06, 1.0000e+00],
        [4.4984e-06, 1.0000e+00],
        [9.9715e-01, 2.8499e-03],
        [1.6144e-07, 1.0000e+00],
        [1.7405e-07, 1.0000e+00],
        [1.2007e-05, 9.9999e-01],
        [2.0798e-08, 1.0000e+00],
        [8.5405e-02, 9.1459e-01],
        [4.7492e-05, 9.9995e-01],
        [4.7301e-08, 1.0000e+00],
        [9.9996e-01, 3.7099e-05],
        [6.1251e-03, 9.9387e-01],
        [5.7250e-06, 9.9999e-01],
        [1.5659e-07, 1.0000e+00],
        [6.3223e-09, 1.0000e+00],
        [8.6062e-04, 9.9914e-01],
        [9.8401e-01, 1.5995e-02],
        [9.9981e-01, 1.9369e-04],
        [1.2991e-06, 1.0000e+00],
        [7.6331e-01, 2.3669e-01],
        [1.4951e-03, 9.9850e-01],
        [7.5776e-04, 9.9924e-01],
        [9.9983e-01, 1.7319e-04],
        [9.9998e-01, 1.8029e-05],
        [3.9020e-04, 9.9961e-01],
        [9.9981e-01, 1.9446e-04],
        [1.1980e-04, 9.9988e-01],
        [7.4287e-01, 2.5713e-01],
        [3.8063e-09, 1.0000e+00],
        [9.9978e-01, 2.1970e-04],
        [2.6644e-07, 1.0000e+00],
        [8.8443e-07, 1.0000e+00],
        [1.1050e-03, 9.9889e-01],
        [6.0255e-03, 9.9397e-01],
        [7.2816e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.7664e-07, 1.0000e+00],
        [3.7551e-08, 1.0000e+00],
        [3.3627e-08, 1.0000e+00],
        [9.9957e-01, 4.3221e-04],
        [9.9915e-01, 8.5021e-04],
        [1.4746e-06, 1.0000e+00],
        [9.9972e-01, 2.8024e-04],
        [4.1525e-05, 9.9996e-01],
        [1.1816e-05, 9.9999e-01],
        [8.7984e-01, 1.2016e-01],
        [1.1541e-01, 8.8459e-01],
        [7.6069e-04, 9.9924e-01],
        [9.9988e-01, 1.1525e-04],
        [3.0268e-07, 1.0000e+00],
        [1.0612e-04, 9.9989e-01],
        [1.0908e-05, 9.9999e-01],
        [4.1366e-04, 9.9959e-01],
        [2.9335e-06, 1.0000e+00],
        [9.9999e-01, 1.4053e-05],
        [8.2076e-02, 9.1792e-01],
        [9.9956e-01, 4.4071e-04],
        [9.9975e-01, 2.4992e-04],
        [2.6273e-07, 1.0000e+00],
        [2.8033e-02, 9.7197e-01],
        [9.2114e-06, 9.9999e-01],
        [9.9988e-01, 1.2171e-04],
        [9.9131e-01, 8.6938e-03],
        [5.0295e-09, 1.0000e+00],
        [5.1188e-11, 1.0000e+00],
        [5.5248e-06, 9.9999e-01],
        [4.2063e-04, 9.9958e-01],
        [2.0998e-04, 9.9979e-01],
        [5.8404e-07, 1.0000e+00],
        [8.9782e-11, 1.0000e+00],
        [1.0971e-06, 1.0000e+00],
        [2.8387e-01, 7.1613e-01],
        [1.1000e-08, 1.0000e+00],
        [9.9983e-01, 1.7212e-04],
        [7.0343e-05, 9.9993e-01],
        [6.9143e-06, 9.9999e-01],
        [9.3233e-03, 9.9068e-01],
        [1.0000e+00, 1.6441e-06],
        [2.6809e-04, 9.9973e-01],
        [9.0236e-05, 9.9991e-01],
        [4.6799e-07, 1.0000e+00],
        [5.4242e-10, 1.0000e+00],
        [7.6442e-06, 9.9999e-01],
        [4.1104e-05, 9.9996e-01],
        [1.1203e-08, 1.0000e+00],
        [1.7995e-09, 1.0000e+00],
        [9.9985e-01, 1.4740e-04],
        [3.7693e-06, 1.0000e+00],
        [9.9981e-01, 1.9397e-04],
        [1.0080e-04, 9.9990e-01],
        [9.9982e-01, 1.8039e-04],
        [3.5674e-06, 1.0000e+00],
        [9.9982e-01, 1.7826e-04],
        [2.8424e-06, 1.0000e+00],
        [9.9581e-01, 4.1938e-03],
        [1.9605e-09, 1.0000e+00],
        [9.1901e-01, 8.0988e-02],
        [1.6223e-04, 9.9984e-01],
        [9.9978e-01, 2.1900e-04],
        [2.1315e-04, 9.9979e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9970e-01, 2.9857e-04],
        [2.4013e-06, 1.0000e+00],
        [2.1356e-01, 7.8644e-01],
        [9.9445e-01, 5.5504e-03],
        [1.0222e-09, 1.0000e+00],
        [9.9982e-01, 1.7745e-04],
        [2.2599e-10, 1.0000e+00],
        [9.9568e-08, 1.0000e+00],
        [9.9359e-01, 6.4099e-03],
        [2.6001e-06, 1.0000e+00],
        [3.0150e-08, 1.0000e+00],
        [3.7875e-06, 1.0000e+00],
        [6.2147e-06, 9.9999e-01],
        [9.9656e-01, 3.4428e-03],
        [1.3842e-05, 9.9999e-01],
        [9.9990e-01, 1.0249e-04],
        [9.6884e-01, 3.1159e-02],
        [9.9186e-01, 8.1431e-03],
        [8.6684e-06, 9.9999e-01],
        [8.0330e-10, 1.0000e+00],
        [5.3333e-05, 9.9995e-01],
        [3.7698e-03, 9.9623e-01],
        [1.4635e-04, 9.9985e-01],
        [1.7091e-05, 9.9998e-01],
        [9.6064e-10, 1.0000e+00],
        [2.3620e-05, 9.9998e-01],
        [4.1819e-04, 9.9958e-01],
        [2.4592e-04, 9.9975e-01],
        [3.3554e-03, 9.9664e-01],
        [1.6102e-07, 1.0000e+00],
        [8.1719e-07, 1.0000e+00],
        [3.2775e-02, 9.6722e-01],
        [1.9940e-01, 8.0060e-01],
        [3.7656e-07, 1.0000e+00],
        [9.9980e-01, 1.9560e-04],
        [9.8519e-04, 9.9901e-01],
        [6.4444e-09, 1.0000e+00],
        [3.3173e-05, 9.9997e-01],
        [5.5740e-09, 1.0000e+00],
        [3.4682e-07, 1.0000e+00],
        [9.9933e-01, 6.6992e-04],
        [9.8512e-01, 1.4876e-02],
        [9.9994e-01, 6.1390e-05],
        [5.9075e-08, 1.0000e+00],
        [6.7004e-08, 1.0000e+00],
        [9.9381e-01, 6.1908e-03],
        [2.0428e-01, 7.9572e-01],
        [2.1737e-09, 1.0000e+00],
        [2.2977e-07, 1.0000e+00],
        [4.0464e-05, 9.9996e-01],
        [9.9980e-01, 1.9524e-04],
        [9.9990e-01, 1.0104e-04],
        [7.8647e-03, 9.9214e-01],
        [1.9879e-09, 1.0000e+00],
        [6.2367e-10, 1.0000e+00],
        [8.1508e-04, 9.9918e-01],
        [3.0095e-07, 1.0000e+00],
        [8.0330e-10, 1.0000e+00],
        [1.3346e-08, 1.0000e+00],
        [1.9944e-06, 1.0000e+00],
        [1.2479e-03, 9.9875e-01],
        [3.0330e-07, 1.0000e+00],
        [9.6706e-03, 9.9033e-01],
        [6.6654e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1998e-05, 9.9999e-01],
        [1.3841e-04, 9.9986e-01],
        [9.9991e-01, 9.2932e-05],
        [9.9367e-01, 6.3338e-03],
        [8.4564e-09, 1.0000e+00],
        [9.5649e-03, 9.9044e-01],
        [1.0006e-03, 9.9900e-01],
        [5.5980e-08, 1.0000e+00],
        [1.6045e-06, 1.0000e+00],
        [9.9947e-01, 5.2674e-04],
        [1.1935e-01, 8.8065e-01],
        [2.5373e-08, 1.0000e+00],
        [4.5694e-05, 9.9995e-01],
        [9.9980e-01, 1.9881e-04],
        [9.9963e-06, 9.9999e-01],
        [5.6950e-05, 9.9994e-01],
        [8.7933e-04, 9.9912e-01],
        [9.9889e-01, 1.1123e-03],
        [2.1763e-08, 1.0000e+00],
        [3.2471e-07, 1.0000e+00],
        [9.9207e-01, 7.9336e-03],
        [9.9984e-01, 1.6313e-04],
        [7.9675e-01, 2.0325e-01],
        [8.8947e-01, 1.1053e-01],
        [1.1615e-09, 1.0000e+00],
        [5.5348e-06, 9.9999e-01],
        [1.6858e-05, 9.9998e-01],
        [6.0158e-06, 9.9999e-01],
        [4.5686e-07, 1.0000e+00],
        [9.9616e-01, 3.8401e-03],
        [1.2101e-01, 8.7899e-01],
        [1.5881e-08, 1.0000e+00],
        [2.0615e-05, 9.9998e-01],
        [1.0199e-08, 1.0000e+00],
        [9.1281e-05, 9.9991e-01],
        [3.2438e-04, 9.9968e-01],
        [3.6081e-01, 6.3919e-01],
        [9.9999e-01, 6.5938e-06],
        [3.2296e-04, 9.9968e-01],
        [2.6219e-10, 1.0000e+00],
        [6.8881e-08, 1.0000e+00],
        [9.4980e-01, 5.0201e-02],
        [4.1698e-06, 1.0000e+00],
        [9.9998e-01, 1.7185e-05],
        [3.3634e-06, 1.0000e+00],
        [9.9997e-01, 3.4886e-05],
        [9.9994e-01, 5.8033e-05],
        [6.9702e-05, 9.9993e-01],
        [9.9989e-01, 1.1140e-04],
        [1.2404e-08, 1.0000e+00],
        [2.6921e-08, 1.0000e+00],
        [9.9991e-01, 9.0327e-05],
        [4.1198e-06, 1.0000e+00],
        [9.9983e-01, 1.6802e-04],
        [1.6694e-08, 1.0000e+00],
        [6.5457e-06, 9.9999e-01],
        [9.9980e-01, 1.9749e-04],
        [1.6596e-05, 9.9998e-01],
        [9.0746e-01, 9.2537e-02],
        [1.3943e-07, 1.0000e+00],
        [9.9978e-01, 2.2053e-04],
        [1.9557e-03, 9.9804e-01],
        [9.9439e-01, 5.6148e-03],
        [9.9991e-01, 8.6839e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[2.8506e-04, 9.9971e-01],
        [7.0104e-05, 9.9993e-01],
        [5.5219e-09, 1.0000e+00],
        [1.6348e-06, 1.0000e+00],
        [6.9855e-08, 1.0000e+00],
        [3.2459e-09, 1.0000e+00],
        [4.7701e-08, 1.0000e+00],
        [1.4942e-04, 9.9985e-01],
        [7.6831e-08, 1.0000e+00],
        [1.3994e-06, 1.0000e+00],
        [3.1267e-06, 1.0000e+00],
        [4.9286e-02, 9.5071e-01],
        [1.6510e-05, 9.9998e-01],
        [1.9705e-03, 9.9803e-01],
        [7.7818e-06, 9.9999e-01],
        [8.2184e-01, 1.7816e-01],
        [8.2784e-07, 1.0000e+00],
        [9.9986e-01, 1.3966e-04],
        [2.0457e-06, 1.0000e+00],
        [9.8970e-01, 1.0299e-02],
        [5.0555e-08, 1.0000e+00],
        [1.5733e-04, 9.9984e-01],
        [9.9987e-01, 1.2888e-04],
        [7.9457e-04, 9.9921e-01],
        [6.3199e-09, 1.0000e+00],
        [8.2044e-06, 9.9999e-01],
        [1.0228e-07, 1.0000e+00],
        [3.0579e-03, 9.9694e-01],
        [1.2449e-05, 9.9999e-01],
        [6.5832e-04, 9.9934e-01],
        [9.9247e-01, 7.5276e-03],
        [9.9933e-01, 6.6729e-04],
        [1.6651e-04, 9.9983e-01],
        [9.9972e-01, 2.7789e-04],
        [9.9964e-01, 3.6084e-04],
        [9.9970e-01, 3.0010e-04],
        [9.9381e-01, 6.1904e-03],
        [1.2421e-03, 9.9876e-01],
        [2.4457e-04, 9.9976e-01],
        [2.7468e-07, 1.0000e+00],
        [1.5723e-05, 9.9998e-01],
        [6.2978e-06, 9.9999e-01],
        [8.0435e-05, 9.9992e-01],
        [1.3622e-04, 9.9986e-01],
        [4.8167e-06, 1.0000e+00],
        [2.3917e-04, 9.9976e-01],
        [7.1532e-04, 9.9928e-01],
        [2.1296e-08, 1.0000e+00],
        [9.9989e-01, 1.0553e-04],
        [9.7152e-03, 9.9028e-01],
        [2.0226e-06, 1.0000e+00],
        [9.9991e-01, 8.7969e-05],
        [1.8550e-04, 9.9981e-01],
        [7.8448e-06, 9.9999e-01],
        [5.1595e-04, 9.9948e-01],
        [4.2001e-03, 9.9580e-01],
        [1.1330e-08, 1.0000e+00],
        [2.3480e-08, 1.0000e+00],
        [9.4650e-01, 5.3504e-02],
        [8.7468e-07, 1.0000e+00],
        [6.4405e-09, 1.0000e+00],
        [2.4726e-03, 9.9753e-01],
        [5.7222e-03, 9.9428e-01],
        [2.0759e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3340e-04, 9.9977e-01],
        [2.1245e-10, 1.0000e+00],
        [9.9982e-01, 1.8197e-04],
        [9.8975e-04, 9.9901e-01],
        [9.9984e-01, 1.5816e-04],
        [1.4452e-08, 1.0000e+00],
        [1.9425e-08, 1.0000e+00],
        [6.1335e-03, 9.9387e-01],
        [1.1706e-09, 1.0000e+00],
        [8.9539e-06, 9.9999e-01],
        [4.9714e-09, 1.0000e+00],
        [5.5088e-04, 9.9945e-01],
        [2.4918e-04, 9.9975e-01],
        [4.0720e-04, 9.9959e-01],
        [7.3466e-09, 1.0000e+00],
        [9.8635e-01, 1.3652e-02],
        [2.8722e-06, 1.0000e+00],
        [9.8562e-01, 1.4383e-02],
        [1.9479e-03, 9.9805e-01],
        [5.6375e-07, 1.0000e+00],
        [3.6079e-07, 1.0000e+00],
        [9.9886e-01, 1.1388e-03],
        [1.5527e-03, 9.9845e-01],
        [7.8935e-01, 2.1065e-01],
        [1.5209e-06, 1.0000e+00],
        [8.5030e-01, 1.4970e-01],
        [1.1413e-09, 1.0000e+00],
        [1.9489e-04, 9.9981e-01],
        [2.0702e-07, 1.0000e+00],
        [1.8484e-02, 9.8152e-01],
        [1.2909e-07, 1.0000e+00],
        [2.7400e-05, 9.9997e-01],
        [8.2037e-09, 1.0000e+00],
        [9.6690e-01, 3.3100e-02],
        [9.9997e-01, 2.6675e-05],
        [1.7614e-07, 1.0000e+00],
        [1.8150e-04, 9.9982e-01],
        [2.9662e-03, 9.9703e-01],
        [7.7166e-05, 9.9992e-01],
        [1.5064e-06, 1.0000e+00],
        [4.6696e-09, 1.0000e+00],
        [1.3644e-05, 9.9999e-01],
        [2.6067e-05, 9.9997e-01],
        [9.9727e-01, 2.7265e-03],
        [6.6056e-01, 3.3944e-01],
        [3.5775e-04, 9.9964e-01],
        [1.8967e-04, 9.9981e-01],
        [7.9777e-07, 1.0000e+00],
        [9.9170e-01, 8.2995e-03],
        [7.4599e-02, 9.2540e-01],
        [3.4009e-06, 1.0000e+00],
        [1.1030e-05, 9.9999e-01],
        [9.9992e-01, 7.9553e-05],
        [9.9982e-01, 1.8079e-04],
        [1.5655e-05, 9.9998e-01],
        [9.7809e-01, 2.1913e-02],
        [7.1186e-05, 9.9993e-01],
        [4.2828e-05, 9.9996e-01],
        [1.2119e-03, 9.9879e-01],
        [1.1050e-04, 9.9989e-01],
        [1.0764e-09, 1.0000e+00],
        [3.6905e-01, 6.3095e-01],
        [8.8629e-06, 9.9999e-01],
        [9.9997e-01, 3.0615e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6802e-03, 9.9832e-01],
        [3.7008e-12, 1.0000e+00],
        [9.9858e-01, 1.4154e-03],
        [4.4095e-08, 1.0000e+00],
        [4.9430e-10, 1.0000e+00],
        [9.9994e-01, 5.9305e-05],
        [8.1735e-08, 1.0000e+00],
        [9.9946e-01, 5.4076e-04],
        [2.4643e-07, 1.0000e+00],
        [1.7993e-07, 1.0000e+00],
        [6.3882e-10, 1.0000e+00],
        [9.9993e-01, 7.3950e-05],
        [2.0573e-07, 1.0000e+00],
        [1.4273e-04, 9.9986e-01],
        [9.9990e-01, 9.7348e-05],
        [2.8573e-03, 9.9714e-01],
        [8.8193e-06, 9.9999e-01],
        [6.3670e-05, 9.9994e-01],
        [3.0719e-02, 9.6928e-01],
        [9.9983e-01, 1.7142e-04],
        [9.0967e-04, 9.9909e-01],
        [5.8968e-09, 1.0000e+00],
        [9.9995e-01, 5.4636e-05],
        [9.9660e-05, 9.9990e-01],
        [2.0186e-08, 1.0000e+00],
        [5.4124e-03, 9.9459e-01],
        [5.7132e-02, 9.4287e-01],
        [9.9996e-01, 4.3854e-05],
        [5.5025e-03, 9.9450e-01],
        [1.1108e-04, 9.9989e-01],
        [1.4562e-03, 9.9854e-01],
        [2.0824e-05, 9.9998e-01],
        [3.2322e-05, 9.9997e-01],
        [4.1568e-03, 9.9584e-01],
        [1.0505e-09, 1.0000e+00],
        [9.9901e-01, 9.8821e-04],
        [4.9816e-02, 9.5018e-01],
        [1.2787e-07, 1.0000e+00],
        [1.7641e-08, 1.0000e+00],
        [2.0536e-06, 1.0000e+00],
        [1.4622e-06, 1.0000e+00],
        [1.2611e-10, 1.0000e+00],
        [9.8842e-03, 9.9012e-01],
        [7.9210e-07, 1.0000e+00],
        [3.8865e-06, 1.0000e+00],
        [3.5263e-06, 1.0000e+00],
        [5.9160e-06, 9.9999e-01],
        [3.3333e-06, 1.0000e+00],
        [9.9404e-06, 9.9999e-01],
        [4.9292e-05, 9.9995e-01],
        [1.8011e-04, 9.9982e-01],
        [9.2945e-09, 1.0000e+00],
        [9.7773e-01, 2.2274e-02],
        [9.1208e-01, 8.7916e-02],
        [2.8441e-02, 9.7156e-01],
        [1.5948e-07, 1.0000e+00],
        [1.2289e-03, 9.9877e-01],
        [2.1071e-07, 1.0000e+00],
        [8.9556e-07, 1.0000e+00],
        [9.9989e-01, 1.0760e-04],
        [2.5726e-05, 9.9997e-01],
        [1.0951e-01, 8.9049e-01],
        [2.2303e-06, 1.0000e+00],
        [5.7614e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9946e-01, 5.4025e-04],
        [9.9999e-01, 1.1388e-05],
        [3.3927e-05, 9.9997e-01],
        [9.9996e-01, 4.2920e-05],
        [1.0258e-08, 1.0000e+00],
        [3.7979e-09, 1.0000e+00],
        [9.9980e-01, 1.9761e-04],
        [9.9995e-01, 4.8570e-05],
        [1.0702e-05, 9.9999e-01],
        [2.6404e-08, 1.0000e+00],
        [9.9935e-01, 6.4715e-04],
        [4.1199e-05, 9.9996e-01],
        [2.7547e-07, 1.0000e+00],
        [7.5847e-07, 1.0000e+00],
        [1.7241e-05, 9.9998e-01],
        [4.4710e-08, 1.0000e+00],
        [3.4877e-09, 1.0000e+00],
        [8.4375e-06, 9.9999e-01],
        [2.4696e-03, 9.9753e-01],
        [1.6251e-06, 1.0000e+00],
        [5.7928e-12, 1.0000e+00],
        [8.2487e-07, 1.0000e+00],
        [4.1123e-05, 9.9996e-01],
        [9.9981e-01, 1.9303e-04],
        [1.0818e-06, 1.0000e+00],
        [2.7969e-07, 1.0000e+00],
        [9.9996e-01, 3.6293e-05],
        [1.5334e-06, 1.0000e+00],
        [9.9990e-01, 1.0118e-04],
        [1.0787e-09, 1.0000e+00],
        [1.4502e-07, 1.0000e+00],
        [7.3068e-06, 9.9999e-01],
        [9.9789e-01, 2.1106e-03],
        [1.1926e-05, 9.9999e-01],
        [2.3844e-08, 1.0000e+00],
        [3.3824e-03, 9.9662e-01],
        [9.9994e-01, 6.2156e-05],
        [9.9991e-01, 9.0011e-05],
        [2.3332e-02, 9.7667e-01],
        [9.9995e-01, 5.0582e-05],
        [2.7910e-06, 1.0000e+00],
        [5.5278e-04, 9.9945e-01],
        [9.9994e-01, 5.9563e-05],
        [9.9914e-01, 8.5972e-04],
        [7.8743e-06, 9.9999e-01],
        [5.4935e-10, 1.0000e+00],
        [3.2195e-08, 1.0000e+00],
        [1.5304e-03, 9.9847e-01],
        [8.8663e-06, 9.9999e-01],
        [1.0368e-06, 1.0000e+00],
        [3.3369e-07, 1.0000e+00],
        [1.7817e-10, 1.0000e+00],
        [9.9949e-01, 5.0795e-04],
        [2.8171e-06, 1.0000e+00],
        [3.6347e-08, 1.0000e+00],
        [1.3460e-05, 9.9999e-01],
        [1.6104e-07, 1.0000e+00],
        [1.1420e-05, 9.9999e-01],
        [5.3210e-06, 9.9999e-01],
        [7.3871e-05, 9.9993e-01],
        [2.7115e-04, 9.9973e-01],
        [3.6578e-05, 9.9996e-01],
        [2.5803e-06, 1.0000e+00],
        [9.9996e-01, 3.7053e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[6.6431e-08, 1.0000e+00],
        [4.5632e-01, 5.4368e-01],
        [1.8507e-05, 9.9998e-01],
        [2.2013e-04, 9.9978e-01],
        [4.8733e-10, 1.0000e+00],
        [7.0473e-09, 1.0000e+00],
        [3.3068e-05, 9.9997e-01],
        [2.7621e-04, 9.9972e-01],
        [9.3860e-01, 6.1404e-02],
        [9.9968e-01, 3.2250e-04],
        [3.2178e-09, 1.0000e+00],
        [9.8418e-01, 1.5815e-02],
        [9.8353e-05, 9.9990e-01],
        [1.1313e-09, 1.0000e+00],
        [9.9998e-01, 2.2682e-05],
        [6.6559e-08, 1.0000e+00],
        [9.9976e-01, 2.4342e-04],
        [4.3377e-12, 1.0000e+00],
        [5.5363e-01, 4.4637e-01],
        [2.1869e-09, 1.0000e+00],
        [2.1688e-07, 1.0000e+00],
        [4.7612e-08, 1.0000e+00],
        [8.5249e-09, 1.0000e+00],
        [1.4279e-09, 1.0000e+00],
        [1.2465e-06, 1.0000e+00],
        [9.9988e-01, 1.2457e-04],
        [9.9969e-01, 3.0763e-04],
        [2.5019e-06, 1.0000e+00],
        [2.3032e-05, 9.9998e-01],
        [2.4556e-10, 1.0000e+00],
        [2.6256e-01, 7.3744e-01],
        [1.4025e-07, 1.0000e+00],
        [1.7877e-06, 1.0000e+00],
        [1.3282e-07, 1.0000e+00],
        [5.8198e-07, 1.0000e+00],
        [2.9367e-05, 9.9997e-01],
        [4.7480e-06, 1.0000e+00],
        [2.1675e-06, 1.0000e+00],
        [1.9785e-04, 9.9980e-01],
        [1.0496e-11, 1.0000e+00],
        [9.7049e-04, 9.9903e-01],
        [9.9974e-01, 2.5557e-04],
        [1.0831e-07, 1.0000e+00],
        [2.6115e-07, 1.0000e+00],
        [7.2696e-05, 9.9993e-01],
        [1.6125e-09, 1.0000e+00],
        [3.0109e-02, 9.6989e-01],
        [1.1997e-09, 1.0000e+00],
        [2.0718e-07, 1.0000e+00],
        [1.8216e-04, 9.9982e-01],
        [3.9430e-08, 1.0000e+00],
        [2.3355e-05, 9.9998e-01],
        [1.2898e-08, 1.0000e+00],
        [9.8862e-11, 1.0000e+00],
        [9.9649e-01, 3.5090e-03],
        [1.3226e-04, 9.9987e-01],
        [9.9989e-01, 1.1281e-04],
        [1.6766e-06, 1.0000e+00],
        [1.8620e-08, 1.0000e+00],
        [9.9990e-01, 1.0283e-04],
        [9.6032e-04, 9.9904e-01],
        [4.0798e-08, 1.0000e+00],
        [9.9016e-05, 9.9990e-01],
        [9.9429e-01, 5.7122e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9989e-01, 1.1044e-04],
        [1.4212e-05, 9.9999e-01],
        [1.3483e-10, 1.0000e+00],
        [1.3837e-04, 9.9986e-01],
        [1.7469e-04, 9.9983e-01],
        [9.9985e-01, 1.4614e-04],
        [3.4439e-06, 1.0000e+00],
        [2.5312e-07, 1.0000e+00],
        [7.2492e-10, 1.0000e+00],
        [6.1572e-09, 1.0000e+00],
        [9.9950e-01, 4.9634e-04],
        [2.4956e-06, 1.0000e+00],
        [2.3733e-05, 9.9998e-01],
        [9.9981e-01, 1.8851e-04],
        [3.9495e-08, 1.0000e+00],
        [1.4062e-03, 9.9859e-01],
        [1.7168e-08, 1.0000e+00],
        [2.5597e-03, 9.9744e-01],
        [9.9309e-01, 6.9064e-03],
        [3.5741e-06, 1.0000e+00],
        [1.7824e-03, 9.9822e-01],
        [6.2426e-08, 1.0000e+00],
        [8.4190e-08, 1.0000e+00],
        [9.9967e-01, 3.2883e-04],
        [9.4352e-10, 1.0000e+00],
        [1.0661e-04, 9.9989e-01],
        [9.9988e-01, 1.1841e-04],
        [4.0867e-06, 1.0000e+00],
        [9.9114e-01, 8.8646e-03],
        [6.3534e-01, 3.6466e-01],
        [9.9245e-01, 7.5531e-03],
        [3.7153e-06, 1.0000e+00],
        [2.8602e-09, 1.0000e+00],
        [1.8565e-08, 1.0000e+00],
        [9.9993e-01, 7.2478e-05],
        [8.9774e-01, 1.0226e-01],
        [4.9738e-09, 1.0000e+00],
        [2.3426e-06, 1.0000e+00],
        [1.4705e-06, 1.0000e+00],
        [5.3091e-05, 9.9995e-01],
        [3.1507e-09, 1.0000e+00],
        [9.8713e-01, 1.2869e-02],
        [9.9516e-07, 1.0000e+00],
        [4.6422e-06, 1.0000e+00],
        [7.1958e-11, 1.0000e+00],
        [8.5416e-10, 1.0000e+00],
        [1.6520e-07, 1.0000e+00],
        [9.0336e-01, 9.6642e-02],
        [8.8118e-05, 9.9991e-01],
        [9.8130e-01, 1.8703e-02],
        [1.2350e-09, 1.0000e+00],
        [9.9813e-01, 1.8716e-03],
        [2.1823e-03, 9.9782e-01],
        [6.7885e-07, 1.0000e+00],
        [4.5521e-06, 1.0000e+00],
        [1.6858e-03, 9.9831e-01],
        [9.9963e-01, 3.6814e-04],
        [2.9686e-08, 1.0000e+00],
        [5.4968e-05, 9.9995e-01],
        [2.8607e-10, 1.0000e+00],
        [2.6707e-03, 9.9733e-01],
        [1.0858e-03, 9.9891e-01],
        [3.7641e-05, 9.9996e-01],
        [9.9985e-01, 1.4859e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5085e-03, 9.9849e-01],
        [4.4252e-10, 1.0000e+00],
        [9.9974e-01, 2.6023e-04],
        [3.1728e-04, 9.9968e-01],
        [1.9238e-07, 1.0000e+00],
        [1.0370e-01, 8.9630e-01],
        [5.4197e-07, 1.0000e+00],
        [9.9987e-01, 1.3203e-04],
        [9.9103e-06, 9.9999e-01],
        [6.5143e-01, 3.4857e-01],
        [9.9985e-01, 1.4668e-04],
        [3.2250e-08, 1.0000e+00],
        [9.9577e-01, 4.2279e-03],
        [3.4620e-01, 6.5380e-01],
        [9.8230e-06, 9.9999e-01],
        [3.1982e-06, 1.0000e+00],
        [1.1769e-07, 1.0000e+00],
        [8.1749e-10, 1.0000e+00],
        [9.2557e-01, 7.4432e-02],
        [4.3455e-01, 5.6545e-01],
        [4.0596e-09, 1.0000e+00],
        [3.9650e-04, 9.9960e-01],
        [1.2931e-06, 1.0000e+00],
        [9.8184e-05, 9.9990e-01],
        [1.1820e-08, 1.0000e+00],
        [9.9986e-01, 1.4356e-04],
        [3.9650e-04, 9.9960e-01],
        [1.2328e-03, 9.9877e-01],
        [9.0818e-05, 9.9991e-01],
        [1.3460e-05, 9.9999e-01],
        [5.3441e-05, 9.9995e-01],
        [6.7501e-06, 9.9999e-01],
        [5.8009e-06, 9.9999e-01],
        [2.9205e-10, 1.0000e+00],
        [2.9157e-02, 9.7084e-01],
        [1.0147e-07, 1.0000e+00],
        [9.9622e-01, 3.7790e-03],
        [9.9992e-01, 8.1321e-05],
        [1.7948e-07, 1.0000e+00],
        [9.9998e-01, 1.7848e-05],
        [9.7196e-11, 1.0000e+00],
        [2.2021e-09, 1.0000e+00],
        [1.7812e-07, 1.0000e+00],
        [2.1755e-08, 1.0000e+00],
        [9.9983e-01, 1.6693e-04],
        [3.5269e-08, 1.0000e+00],
        [6.9363e-01, 3.0637e-01],
        [7.9889e-09, 1.0000e+00],
        [9.9999e-01, 1.3532e-05],
        [7.8360e-06, 9.9999e-01],
        [1.8573e-09, 1.0000e+00],
        [6.5968e-05, 9.9993e-01],
        [9.9983e-01, 1.6570e-04],
        [9.9967e-01, 3.2645e-04],
        [2.1816e-10, 1.0000e+00],
        [2.9234e-04, 9.9971e-01],
        [8.5929e-06, 9.9999e-01],
        [3.3078e-10, 1.0000e+00],
        [9.9998e-01, 2.1710e-05],
        [7.6823e-04, 9.9923e-01],
        [4.8375e-11, 1.0000e+00],
        [9.9974e-01, 2.6349e-04],
        [3.1114e-10, 1.0000e+00],
        [9.9948e-01, 5.1803e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[7.4808e-06, 9.9999e-01],
        [2.2983e-08, 1.0000e+00],
        [9.9976e-01, 2.3827e-04],
        [9.9997e-01, 2.7588e-05],
        [2.1700e-05, 9.9998e-01],
        [8.0670e-07, 1.0000e+00],
        [1.5951e-05, 9.9998e-01],
        [9.9471e-06, 9.9999e-01],
        [9.4501e-06, 9.9999e-01],
        [1.2566e-03, 9.9874e-01],
        [3.8698e-06, 1.0000e+00],
        [1.4213e-04, 9.9986e-01],
        [9.9990e-01, 9.7990e-05],
        [2.1339e-04, 9.9979e-01],
        [6.7798e-04, 9.9932e-01],
        [9.9977e-01, 2.3157e-04],
        [9.5520e-01, 4.4800e-02],
        [9.9991e-01, 8.5128e-05],
        [3.4240e-04, 9.9966e-01],
        [7.9979e-08, 1.0000e+00],
        [2.9335e-08, 1.0000e+00],
        [6.1914e-07, 1.0000e+00],
        [2.7997e-04, 9.9972e-01],
        [9.9970e-01, 3.0114e-04],
        [7.6491e-01, 2.3509e-01],
        [4.8794e-05, 9.9995e-01],
        [8.5825e-05, 9.9991e-01],
        [9.9987e-01, 1.2716e-04],
        [9.9999e-01, 7.4774e-06],
        [2.0079e-06, 1.0000e+00],
        [7.6552e-01, 2.3448e-01],
        [6.6714e-05, 9.9993e-01],
        [2.6253e-07, 1.0000e+00],
        [1.5769e-04, 9.9984e-01],
        [2.0211e-09, 1.0000e+00],
        [9.9991e-01, 9.2597e-05],
        [3.0109e-10, 1.0000e+00],
        [9.9970e-01, 2.9985e-04],
        [1.6472e-08, 1.0000e+00],
        [1.4881e-06, 1.0000e+00],
        [6.5109e-07, 1.0000e+00],
        [7.6158e-02, 9.2384e-01],
        [2.7723e-09, 1.0000e+00],
        [4.2261e-08, 1.0000e+00],
        [9.8761e-03, 9.9012e-01],
        [9.3733e-01, 6.2668e-02],
        [8.9374e-08, 1.0000e+00],
        [2.3482e-05, 9.9998e-01],
        [2.1926e-02, 9.7807e-01],
        [1.4215e-06, 1.0000e+00],
        [1.0191e-07, 1.0000e+00],
        [8.8320e-11, 1.0000e+00],
        [9.9997e-01, 3.2516e-05],
        [3.9235e-09, 1.0000e+00],
        [4.9129e-04, 9.9951e-01],
        [1.3169e-06, 1.0000e+00],
        [9.9996e-01, 4.4514e-05],
        [4.2984e-05, 9.9996e-01],
        [1.8944e-07, 1.0000e+00],
        [9.0420e-01, 9.5796e-02],
        [1.3833e-04, 9.9986e-01],
        [1.5780e-04, 9.9984e-01],
        [1.3555e-07, 1.0000e+00],
        [2.1026e-03, 9.9790e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7036e-05, 9.9998e-01],
        [5.1725e-07, 1.0000e+00],
        [3.5427e-05, 9.9996e-01],
        [3.3999e-11, 1.0000e+00],
        [9.9988e-01, 1.1570e-04],
        [8.2846e-09, 1.0000e+00],
        [7.2070e-05, 9.9993e-01],
        [1.0000e+00, 6.1444e-07],
        [3.4425e-06, 1.0000e+00],
        [9.4574e-09, 1.0000e+00],
        [1.3989e-03, 9.9860e-01],
        [2.2812e-09, 1.0000e+00],
        [7.0423e-06, 9.9999e-01],
        [1.0250e-05, 9.9999e-01],
        [9.9999e-01, 1.3072e-05],
        [4.8110e-07, 1.0000e+00],
        [4.4341e-04, 9.9956e-01],
        [1.3928e-05, 9.9999e-01],
        [9.6152e-07, 1.0000e+00],
        [9.9994e-01, 5.8903e-05],
        [1.4995e-10, 1.0000e+00],
        [2.0055e-07, 1.0000e+00],
        [9.9994e-01, 6.4923e-05],
        [6.3168e-04, 9.9937e-01],
        [5.1962e-06, 9.9999e-01],
        [2.6932e-02, 9.7307e-01],
        [9.9534e-01, 4.6567e-03],
        [1.6336e-06, 1.0000e+00],
        [5.5095e-10, 1.0000e+00],
        [8.5698e-06, 9.9999e-01],
        [4.4507e-01, 5.5493e-01],
        [9.6155e-04, 9.9904e-01],
        [2.5669e-09, 1.0000e+00],
        [1.4642e-06, 1.0000e+00],
        [3.6870e-04, 9.9963e-01],
        [9.9976e-01, 2.4491e-04],
        [9.9990e-01, 1.0412e-04],
        [8.9554e-10, 1.0000e+00],
        [1.0900e-03, 9.9891e-01],
        [1.8030e-01, 8.1970e-01],
        [9.9276e-01, 7.2384e-03],
        [1.4218e-06, 1.0000e+00],
        [8.5659e-01, 1.4341e-01],
        [6.5521e-07, 1.0000e+00],
        [5.9345e-07, 1.0000e+00],
        [9.9985e-01, 1.4866e-04],
        [7.9330e-01, 2.0670e-01],
        [2.3634e-11, 1.0000e+00],
        [3.7398e-05, 9.9996e-01],
        [9.9901e-01, 9.9130e-04],
        [7.4889e-08, 1.0000e+00],
        [3.2198e-09, 1.0000e+00],
        [9.0383e-05, 9.9991e-01],
        [1.2349e-06, 1.0000e+00],
        [1.9656e-10, 1.0000e+00],
        [2.6376e-06, 1.0000e+00],
        [2.1225e-05, 9.9998e-01],
        [1.9532e-08, 1.0000e+00],
        [9.9991e-01, 8.9563e-05],
        [2.8718e-05, 9.9997e-01],
        [9.9987e-01, 1.3199e-04],
        [4.4804e-09, 1.0000e+00],
        [3.4964e-07, 1.0000e+00],
        [1.5858e-02, 9.8414e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0908e-05, 9.9999e-01],
        [9.4560e-01, 5.4396e-02],
        [4.5131e-07, 1.0000e+00],
        [7.2453e-02, 9.2755e-01],
        [3.3146e-08, 1.0000e+00],
        [5.3118e-03, 9.9469e-01],
        [1.7046e-09, 1.0000e+00],
        [1.4903e-05, 9.9999e-01],
        [1.0000e+00, 8.1202e-08],
        [1.8652e-08, 1.0000e+00],
        [1.0051e-09, 1.0000e+00],
        [5.0578e-11, 1.0000e+00],
        [9.9997e-01, 3.1153e-05],
        [2.7985e-07, 1.0000e+00],
        [4.6086e-07, 1.0000e+00],
        [2.1273e-03, 9.9787e-01],
        [6.1585e-09, 1.0000e+00],
        [9.9932e-01, 6.8241e-04],
        [3.9839e-08, 1.0000e+00],
        [9.9997e-01, 3.4392e-05],
        [3.0074e-04, 9.9970e-01],
        [9.9999e-01, 6.5789e-06],
        [3.0244e-04, 9.9970e-01],
        [6.1727e-05, 9.9994e-01],
        [9.6582e-09, 1.0000e+00],
        [2.6568e-03, 9.9734e-01],
        [1.1490e-03, 9.9885e-01],
        [9.2505e-01, 7.4948e-02],
        [1.5562e-05, 9.9998e-01],
        [5.2147e-05, 9.9995e-01],
        [7.4191e-02, 9.2581e-01],
        [1.1080e-04, 9.9989e-01],
        [9.9979e-01, 2.0539e-04],
        [9.9994e-01, 5.7324e-05],
        [1.5329e-05, 9.9998e-01],
        [2.4405e-06, 1.0000e+00],
        [9.9946e-01, 5.4229e-04],
        [9.9977e-01, 2.3225e-04],
        [9.9955e-01, 4.4904e-04],
        [1.0929e-07, 1.0000e+00],
        [1.3154e-02, 9.8685e-01],
        [1.8903e-11, 1.0000e+00],
        [3.8870e-06, 1.0000e+00],
        [9.9968e-01, 3.2271e-04],
        [1.2493e-03, 9.9875e-01],
        [2.6373e-07, 1.0000e+00],
        [1.2527e-04, 9.9987e-01],
        [9.9994e-01, 6.1034e-05],
        [8.5112e-01, 1.4888e-01],
        [3.9179e-04, 9.9961e-01],
        [3.1871e-06, 1.0000e+00],
        [7.3296e-08, 1.0000e+00],
        [3.9400e-05, 9.9996e-01],
        [7.5135e-09, 1.0000e+00],
        [9.9987e-01, 1.3147e-04],
        [1.6310e-10, 1.0000e+00],
        [1.0953e-08, 1.0000e+00],
        [3.5574e-07, 1.0000e+00],
        [3.6718e-04, 9.9963e-01],
        [1.1095e-07, 1.0000e+00],
        [2.1702e-03, 9.9783e-01],
        [2.8837e-04, 9.9971e-01],
        [9.9968e-01, 3.1700e-04],
        [9.9999e-01, 1.0422e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9931e-01, 6.9401e-04],
        [1.4805e-05, 9.9999e-01],
        [3.8860e-06, 1.0000e+00],
        [1.5246e-05, 9.9998e-01],
        [9.9982e-01, 1.8344e-04],
        [9.9994e-01, 5.7872e-05],
        [1.5360e-08, 1.0000e+00],
        [2.2161e-04, 9.9978e-01],
        [1.3115e-05, 9.9999e-01],
        [1.2431e-06, 1.0000e+00],
        [9.9987e-01, 1.3328e-04],
        [4.6899e-05, 9.9995e-01],
        [4.2513e-02, 9.5749e-01],
        [9.9980e-01, 1.9621e-04],
        [2.5363e-08, 1.0000e+00],
        [2.2269e-07, 1.0000e+00],
        [4.5461e-10, 1.0000e+00],
        [9.9986e-01, 1.3727e-04],
        [1.9679e-02, 9.8032e-01],
        [9.9972e-01, 2.8199e-04],
        [1.4757e-05, 9.9999e-01],
        [3.2946e-03, 9.9671e-01],
        [2.3991e-06, 1.0000e+00],
        [1.4572e-01, 8.5428e-01],
        [9.9996e-01, 4.2661e-05],
        [9.9998e-01, 2.0708e-05],
        [3.9694e-08, 1.0000e+00],
        [3.2805e-06, 1.0000e+00],
        [1.7571e-08, 1.0000e+00],
        [1.7718e-05, 9.9998e-01],
        [1.1596e-08, 1.0000e+00],
        [3.9308e-04, 9.9961e-01],
        [9.6884e-09, 1.0000e+00],
        [1.1212e-08, 1.0000e+00],
        [9.9970e-01, 3.0136e-04],
        [1.1095e-09, 1.0000e+00],
        [1.7667e-10, 1.0000e+00],
        [5.7989e-06, 9.9999e-01],
        [9.7481e-11, 1.0000e+00],
        [9.9798e-01, 2.0164e-03],
        [7.0355e-07, 1.0000e+00],
        [7.4678e-06, 9.9999e-01],
        [4.5011e-06, 1.0000e+00],
        [9.9985e-01, 1.4926e-04],
        [9.9655e-01, 3.4522e-03],
        [9.9994e-01, 6.1659e-05],
        [4.5058e-05, 9.9995e-01],
        [1.0824e-03, 9.9892e-01],
        [1.3932e-03, 9.9861e-01],
        [3.6179e-09, 1.0000e+00],
        [9.9996e-01, 3.9039e-05],
        [2.7024e-09, 1.0000e+00],
        [9.9467e-01, 5.3323e-03],
        [4.0566e-07, 1.0000e+00],
        [9.9976e-01, 2.4373e-04],
        [9.9968e-01, 3.2360e-04],
        [1.2369e-07, 1.0000e+00],
        [3.7289e-08, 1.0000e+00],
        [9.9995e-01, 4.5306e-05],
        [7.1010e-08, 1.0000e+00],
        [1.4475e-05, 9.9999e-01],
        [3.6399e-03, 9.9636e-01],
        [2.8219e-02, 9.7178e-01],
        [6.3685e-03, 9.9363e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.4357e-05, 9.9991e-01],
        [9.4404e-05, 9.9991e-01],
        [1.0332e-06, 1.0000e+00],
        [9.9977e-01, 2.3175e-04],
        [9.9760e-01, 2.3959e-03],
        [5.3631e-04, 9.9946e-01],
        [3.2161e-04, 9.9968e-01],
        [1.8369e-03, 9.9816e-01],
        [1.6810e-05, 9.9998e-01],
        [9.9994e-01, 6.2751e-05],
        [4.9169e-04, 9.9951e-01],
        [1.4183e-02, 9.8582e-01],
        [3.9681e-06, 1.0000e+00],
        [3.1511e-04, 9.9968e-01],
        [1.1043e-04, 9.9989e-01],
        [6.4014e-06, 9.9999e-01],
        [5.8975e-05, 9.9994e-01],
        [1.0839e-03, 9.9892e-01],
        [1.9918e-08, 1.0000e+00],
        [1.8830e-07, 1.0000e+00],
        [1.1232e-06, 1.0000e+00],
        [1.5614e-01, 8.4386e-01],
        [5.6968e-04, 9.9943e-01],
        [9.9984e-01, 1.6368e-04],
        [1.0519e-03, 9.9895e-01],
        [4.1101e-04, 9.9959e-01],
        [1.1926e-05, 9.9999e-01],
        [2.2194e-06, 1.0000e+00],
        [4.3470e-03, 9.9565e-01],
        [5.0969e-04, 9.9949e-01],
        [2.5857e-04, 9.9974e-01],
        [3.6240e-05, 9.9996e-01],
        [3.6435e-04, 9.9964e-01],
        [9.1905e-02, 9.0809e-01],
        [9.9935e-01, 6.5025e-04],
        [8.9001e-07, 1.0000e+00],
        [2.4044e-03, 9.9760e-01],
        [5.3709e-02, 9.4629e-01],
        [2.5904e-01, 7.4096e-01],
        [2.1442e-09, 1.0000e+00],
        [9.9711e-01, 2.8922e-03],
        [1.0625e-06, 1.0000e+00],
        [3.3723e-04, 9.9966e-01],
        [7.3241e-01, 2.6759e-01],
        [1.0000e+00, 2.4256e-06],
        [2.4701e-06, 1.0000e+00],
        [7.3054e-06, 9.9999e-01],
        [3.5666e-10, 1.0000e+00],
        [1.5945e-04, 9.9984e-01],
        [3.2358e-06, 1.0000e+00],
        [9.9992e-01, 7.6103e-05],
        [1.1914e-07, 1.0000e+00],
        [9.9990e-01, 9.6607e-05],
        [9.9996e-01, 4.3990e-05],
        [3.8965e-03, 9.9610e-01],
        [8.8929e-03, 9.9111e-01],
        [1.7881e-04, 9.9982e-01],
        [3.3606e-06, 1.0000e+00],
        [9.9911e-01, 8.9169e-04],
        [2.5295e-07, 1.0000e+00],
        [9.9992e-01, 7.5416e-05],
        [3.2709e-05, 9.9997e-01],
        [5.6472e-05, 9.9994e-01],
        [3.0829e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.4092e-01, 7.5908e-01],
        [9.9995e-01, 4.6329e-05],
        [2.4550e-03, 9.9755e-01],
        [7.7731e-05, 9.9992e-01],
        [9.9992e-01, 8.0904e-05],
        [7.6747e-06, 9.9999e-01],
        [8.8374e-04, 9.9912e-01],
        [1.8084e-05, 9.9998e-01],
        [9.9993e-01, 6.5744e-05],
        [1.9720e-07, 1.0000e+00],
        [2.2259e-09, 1.0000e+00],
        [1.2124e-05, 9.9999e-01],
        [5.2069e-03, 9.9479e-01],
        [3.1370e-07, 1.0000e+00],
        [3.3912e-06, 1.0000e+00],
        [9.9996e-01, 3.5256e-05],
        [4.2863e-06, 1.0000e+00],
        [9.9984e-01, 1.6453e-04],
        [9.9983e-01, 1.6604e-04],
        [4.7296e-09, 1.0000e+00],
        [1.3339e-01, 8.6661e-01],
        [7.1745e-11, 1.0000e+00],
        [2.7761e-10, 1.0000e+00],
        [8.6641e-06, 9.9999e-01],
        [1.0143e-05, 9.9999e-01],
        [7.8466e-07, 1.0000e+00],
        [8.6347e-02, 9.1365e-01],
        [5.4658e-06, 9.9999e-01],
        [9.8536e-01, 1.4636e-02],
        [3.6690e-06, 1.0000e+00],
        [9.9986e-01, 1.3904e-04],
        [9.9946e-01, 5.3876e-04],
        [9.9978e-01, 2.2455e-04],
        [8.8978e-04, 9.9911e-01],
        [2.8936e-06, 1.0000e+00],
        [9.8771e-01, 1.2291e-02],
        [9.9995e-01, 5.3497e-05],
        [9.8154e-01, 1.8459e-02],
        [8.3524e-06, 9.9999e-01],
        [9.9988e-01, 1.1575e-04],
        [1.5159e-05, 9.9998e-01],
        [1.0082e-04, 9.9990e-01],
        [9.9974e-01, 2.6299e-04],
        [2.4107e-09, 1.0000e+00],
        [6.7028e-05, 9.9993e-01],
        [9.9953e-01, 4.6828e-04],
        [1.1479e-03, 9.9885e-01],
        [4.0155e-05, 9.9996e-01],
        [1.0162e-06, 1.0000e+00],
        [1.4809e-01, 8.5191e-01],
        [2.5011e-02, 9.7499e-01],
        [4.0660e-04, 9.9959e-01],
        [1.1596e-06, 1.0000e+00],
        [2.5571e-06, 1.0000e+00],
        [9.9890e-01, 1.0954e-03],
        [9.9881e-01, 1.1930e-03],
        [7.4371e-02, 9.2563e-01],
        [1.7440e-04, 9.9983e-01],
        [1.5028e-05, 9.9998e-01],
        [3.3276e-02, 9.6672e-01],
        [1.2019e-07, 1.0000e+00],
        [9.9984e-01, 1.6322e-04],
        [9.9938e-01, 6.2427e-04],
        [6.3674e-01, 3.6326e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9932e-01, 6.7624e-04],
        [9.9989e-01, 1.1300e-04],
        [1.1202e-09, 1.0000e+00],
        [2.1834e-07, 1.0000e+00],
        [9.3506e-08, 1.0000e+00],
        [2.9851e-05, 9.9997e-01],
        [9.9983e-01, 1.6724e-04],
        [1.3059e-10, 1.0000e+00],
        [1.4736e-04, 9.9985e-01],
        [1.6924e-09, 1.0000e+00],
        [9.0859e-06, 9.9999e-01],
        [1.5901e-10, 1.0000e+00],
        [9.9995e-01, 4.6468e-05],
        [6.1181e-07, 1.0000e+00],
        [1.9659e-09, 1.0000e+00],
        [1.5516e-08, 1.0000e+00],
        [2.1350e-03, 9.9787e-01],
        [1.3176e-03, 9.9868e-01],
        [9.9872e-01, 1.2789e-03],
        [9.9764e-01, 2.3603e-03],
        [9.9984e-01, 1.5934e-04],
        [3.2380e-06, 1.0000e+00],
        [9.9986e-01, 1.3510e-04],
        [1.0970e-03, 9.9890e-01],
        [1.4085e-03, 9.9859e-01],
        [5.7474e-06, 9.9999e-01],
        [3.1053e-03, 9.9689e-01],
        [6.0550e-07, 1.0000e+00],
        [3.7982e-08, 1.0000e+00],
        [1.0000e+00, 2.8455e-09],
        [9.6954e-01, 3.0458e-02],
        [7.5440e-05, 9.9992e-01],
        [3.3990e-07, 1.0000e+00],
        [9.9949e-01, 5.1221e-04],
        [4.6679e-07, 1.0000e+00],
        [4.4251e-07, 1.0000e+00],
        [1.1923e-07, 1.0000e+00],
        [3.8563e-07, 1.0000e+00],
        [9.6109e-03, 9.9039e-01],
        [1.7331e-08, 1.0000e+00],
        [9.9959e-01, 4.1074e-04],
        [5.5287e-06, 9.9999e-01],
        [6.8471e-08, 1.0000e+00],
        [1.8004e-04, 9.9982e-01],
        [8.2772e-04, 9.9917e-01],
        [3.8187e-05, 9.9996e-01],
        [2.3388e-02, 9.7661e-01],
        [6.6308e-06, 9.9999e-01],
        [9.9999e-01, 1.2030e-05],
        [3.5907e-04, 9.9964e-01],
        [2.6201e-06, 1.0000e+00],
        [8.6565e-06, 9.9999e-01],
        [1.2649e-04, 9.9987e-01],
        [5.6452e-04, 9.9944e-01],
        [9.9978e-01, 2.2417e-04],
        [5.3085e-05, 9.9995e-01],
        [9.9993e-01, 6.9755e-05],
        [2.3136e-03, 9.9769e-01],
        [1.6697e-10, 1.0000e+00],
        [8.8853e-08, 1.0000e+00],
        [1.6201e-08, 1.0000e+00],
        [1.5405e-03, 9.9846e-01],
        [3.7150e-09, 1.0000e+00],
        [1.5694e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9995e-01, 5.0398e-05],
        [7.4941e-06, 9.9999e-01],
        [9.9981e-01, 1.8940e-04],
        [1.7234e-08, 1.0000e+00],
        [3.4297e-06, 1.0000e+00],
        [5.5997e-05, 9.9994e-01],
        [1.7171e-09, 1.0000e+00],
        [9.8293e-05, 9.9990e-01],
        [1.6637e-06, 1.0000e+00],
        [1.6836e-11, 1.0000e+00],
        [9.2073e-04, 9.9908e-01],
        [5.0027e-06, 9.9999e-01],
        [4.0696e-08, 1.0000e+00],
        [6.5114e-08, 1.0000e+00],
        [9.9988e-01, 1.1897e-04],
        [2.4593e-05, 9.9998e-01],
        [1.8241e-08, 1.0000e+00],
        [7.2213e-05, 9.9993e-01],
        [3.3374e-02, 9.6663e-01],
        [1.0787e-08, 1.0000e+00],
        [9.9986e-01, 1.4088e-04],
        [1.4501e-04, 9.9985e-01],
        [9.7833e-01, 2.1667e-02],
        [9.9993e-01, 7.5030e-05],
        [5.2566e-02, 9.4743e-01],
        [3.8941e-09, 1.0000e+00],
        [9.9985e-01, 1.4808e-04],
        [2.0077e-07, 1.0000e+00],
        [8.6996e-01, 1.3004e-01],
        [9.9987e-01, 1.3223e-04],
        [1.1218e-05, 9.9999e-01],
        [6.5170e-02, 9.3483e-01],
        [9.9983e-01, 1.7086e-04],
        [7.9038e-05, 9.9992e-01],
        [3.0971e-05, 9.9997e-01],
        [9.9998e-01, 2.3526e-05],
        [9.5846e-10, 1.0000e+00],
        [4.1318e-03, 9.9587e-01],
        [2.1537e-08, 1.0000e+00],
        [2.1744e-06, 1.0000e+00],
        [3.6790e-11, 1.0000e+00],
        [1.0000e+00, 2.9784e-07],
        [1.6501e-01, 8.3499e-01],
        [9.9988e-01, 1.2179e-04],
        [2.6777e-07, 1.0000e+00],
        [9.5323e-06, 9.9999e-01],
        [9.9724e-01, 2.7641e-03],
        [3.3566e-05, 9.9997e-01],
        [9.1171e-07, 1.0000e+00],
        [4.3791e-04, 9.9956e-01],
        [2.0629e-05, 9.9998e-01],
        [4.3758e-07, 1.0000e+00],
        [4.6899e-08, 1.0000e+00],
        [9.9983e-01, 1.7248e-04],
        [4.7684e-07, 1.0000e+00],
        [9.9938e-01, 6.1825e-04],
        [1.8067e-05, 9.9998e-01],
        [3.4787e-06, 1.0000e+00],
        [9.5001e-01, 4.9988e-02],
        [9.9994e-01, 6.4836e-05],
        [5.1099e-04, 9.9949e-01],
        [5.6693e-03, 9.9433e-01],
        [1.1338e-08, 1.0000e+00],
        [3.3000e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.3626e-01, 1.6374e-01],
        [1.7947e-01, 8.2053e-01],
        [9.9994e-01, 6.3235e-05],
        [1.7648e-10, 1.0000e+00],
        [2.0276e-03, 9.9797e-01],
        [9.9969e-01, 3.0817e-04],
        [1.2772e-05, 9.9999e-01],
        [9.5676e-04, 9.9904e-01],
        [9.9994e-01, 6.0688e-05],
        [1.7322e-02, 9.8268e-01],
        [3.3410e-05, 9.9997e-01],
        [1.5422e-04, 9.9985e-01],
        [2.7555e-05, 9.9997e-01],
        [3.9174e-07, 1.0000e+00],
        [9.9976e-01, 2.3987e-04],
        [2.7489e-04, 9.9973e-01],
        [1.2820e-06, 1.0000e+00],
        [4.6549e-05, 9.9995e-01],
        [9.3326e-08, 1.0000e+00],
        [9.9266e-01, 7.3416e-03],
        [4.6697e-05, 9.9995e-01],
        [2.1517e-06, 1.0000e+00],
        [2.2442e-07, 1.0000e+00],
        [1.7136e-08, 1.0000e+00],
        [1.4228e-08, 1.0000e+00],
        [3.0565e-08, 1.0000e+00],
        [1.0165e-04, 9.9990e-01],
        [9.9992e-01, 7.8245e-05],
        [3.0321e-02, 9.6968e-01],
        [9.9951e-01, 4.9376e-04],
        [9.4171e-01, 5.8285e-02],
        [9.7929e-01, 2.0712e-02],
        [2.4506e-07, 1.0000e+00],
        [1.6252e-03, 9.9837e-01],
        [5.6564e-03, 9.9434e-01],
        [9.9992e-01, 7.5232e-05],
        [9.9973e-01, 2.7312e-04],
        [8.7468e-11, 1.0000e+00],
        [4.0462e-05, 9.9996e-01],
        [5.3067e-09, 1.0000e+00],
        [9.9987e-01, 1.3043e-04],
        [8.2631e-07, 1.0000e+00],
        [4.2006e-07, 1.0000e+00],
        [9.9985e-01, 1.5420e-04],
        [3.7917e-06, 1.0000e+00],
        [9.9993e-01, 7.1503e-05],
        [3.4314e-05, 9.9997e-01],
        [2.6328e-07, 1.0000e+00],
        [1.6264e-04, 9.9984e-01],
        [4.8478e-05, 9.9995e-01],
        [8.8055e-08, 1.0000e+00],
        [5.1413e-08, 1.0000e+00],
        [2.1153e-04, 9.9979e-01],
        [8.4011e-09, 1.0000e+00],
        [2.8583e-05, 9.9997e-01],
        [3.1510e-03, 9.9685e-01],
        [9.9999e-01, 1.2595e-05],
        [9.9987e-01, 1.2781e-04],
        [8.5621e-07, 1.0000e+00],
        [1.9460e-06, 1.0000e+00],
        [1.7579e-06, 1.0000e+00],
        [1.5762e-08, 1.0000e+00],
        [9.5497e-09, 1.0000e+00],
        [2.8280e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.9636e-01, 1.0364e-01],
        [8.4052e-01, 1.5948e-01],
        [2.4122e-05, 9.9998e-01],
        [8.7736e-07, 1.0000e+00],
        [9.9939e-01, 6.1359e-04],
        [6.8958e-05, 9.9993e-01],
        [2.6906e-06, 1.0000e+00],
        [2.6652e-08, 1.0000e+00],
        [9.9994e-01, 5.7520e-05],
        [7.4016e-06, 9.9999e-01],
        [2.1372e-04, 9.9979e-01],
        [5.3022e-06, 9.9999e-01],
        [1.0241e-08, 1.0000e+00],
        [2.5088e-09, 1.0000e+00],
        [1.1446e-01, 8.8554e-01],
        [3.5081e-07, 1.0000e+00],
        [1.0695e-06, 1.0000e+00],
        [1.6828e-08, 1.0000e+00],
        [7.1258e-11, 1.0000e+00],
        [3.0486e-07, 1.0000e+00],
        [1.1188e-07, 1.0000e+00],
        [2.0393e-05, 9.9998e-01],
        [2.4949e-03, 9.9751e-01],
        [9.9991e-01, 9.2117e-05],
        [1.6231e-08, 1.0000e+00],
        [6.7856e-06, 9.9999e-01],
        [3.4440e-01, 6.5560e-01],
        [1.7633e-05, 9.9998e-01],
        [6.2565e-07, 1.0000e+00],
        [9.9239e-01, 7.6116e-03],
        [2.5670e-05, 9.9997e-01],
        [9.9988e-01, 1.2189e-04],
        [4.3718e-08, 1.0000e+00],
        [1.6839e-07, 1.0000e+00],
        [1.1008e-07, 1.0000e+00],
        [1.7636e-07, 1.0000e+00],
        [1.1471e-06, 1.0000e+00],
        [1.8197e-06, 1.0000e+00],
        [7.1660e-03, 9.9283e-01],
        [4.4556e-07, 1.0000e+00],
        [4.0107e-05, 9.9996e-01],
        [9.9990e-01, 1.0090e-04],
        [9.9994e-01, 5.5892e-05],
        [6.2052e-02, 9.3795e-01],
        [1.4734e-06, 1.0000e+00],
        [2.5750e-07, 1.0000e+00],
        [8.8014e-09, 1.0000e+00],
        [9.9984e-01, 1.6017e-04],
        [5.4616e-05, 9.9995e-01],
        [6.9664e-06, 9.9999e-01],
        [2.3038e-04, 9.9977e-01],
        [4.1065e-05, 9.9996e-01],
        [3.1685e-07, 1.0000e+00],
        [9.9937e-01, 6.2518e-04],
        [9.9967e-01, 3.2863e-04],
        [9.8457e-08, 1.0000e+00],
        [1.1607e-01, 8.8393e-01],
        [7.2431e-09, 1.0000e+00],
        [6.1769e-02, 9.3823e-01],
        [9.9992e-01, 8.0259e-05],
        [6.0742e-06, 9.9999e-01],
        [5.1011e-05, 9.9995e-01],
        [3.9294e-03, 9.9607e-01],
        [5.3251e-04, 9.9947e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0193e-08, 1.0000e+00],
        [4.3005e-05, 9.9996e-01],
        [1.8402e-07, 1.0000e+00],
        [9.9995e-01, 5.3413e-05],
        [3.0550e-03, 9.9695e-01],
        [1.8928e-09, 1.0000e+00],
        [1.2957e-07, 1.0000e+00],
        [3.0861e-06, 1.0000e+00],
        [9.9987e-01, 1.2667e-04],
        [9.9861e-01, 1.3913e-03],
        [9.9969e-01, 3.1462e-04],
        [4.3377e-05, 9.9996e-01],
        [4.3772e-08, 1.0000e+00],
        [5.6147e-10, 1.0000e+00],
        [1.6650e-02, 9.8335e-01],
        [9.9986e-01, 1.3523e-04],
        [1.9417e-04, 9.9981e-01],
        [3.9807e-03, 9.9602e-01],
        [9.9912e-01, 8.8287e-04],
        [2.6618e-08, 1.0000e+00],
        [9.9937e-01, 6.3288e-04],
        [2.4112e-07, 1.0000e+00],
        [4.6804e-07, 1.0000e+00],
        [9.9994e-01, 6.0748e-05],
        [1.0322e-03, 9.9897e-01],
        [1.9059e-07, 1.0000e+00],
        [3.3188e-05, 9.9997e-01],
        [5.7364e-05, 9.9994e-01],
        [1.6192e-12, 1.0000e+00],
        [9.9084e-01, 9.1592e-03],
        [9.9065e-06, 9.9999e-01],
        [9.2021e-05, 9.9991e-01],
        [6.3591e-07, 1.0000e+00],
        [2.4544e-05, 9.9998e-01],
        [1.1180e-08, 1.0000e+00],
        [9.9911e-01, 8.8967e-04],
        [3.3957e-06, 1.0000e+00],
        [1.7004e-05, 9.9998e-01],
        [9.9926e-01, 7.4006e-04],
        [1.8556e-05, 9.9998e-01],
        [3.3478e-04, 9.9967e-01],
        [2.6583e-06, 1.0000e+00],
        [5.5735e-09, 1.0000e+00],
        [9.7489e-09, 1.0000e+00],
        [1.6092e-08, 1.0000e+00],
        [1.2661e-08, 1.0000e+00],
        [9.9986e-01, 1.4240e-04],
        [5.5964e-08, 1.0000e+00],
        [2.3504e-06, 1.0000e+00],
        [2.6639e-05, 9.9997e-01],
        [9.9989e-01, 1.1132e-04],
        [3.1567e-08, 1.0000e+00],
        [6.6523e-01, 3.3477e-01],
        [3.9422e-05, 9.9996e-01],
        [6.1637e-06, 9.9999e-01],
        [7.1772e-07, 1.0000e+00],
        [1.7325e-08, 1.0000e+00],
        [3.9991e-02, 9.6001e-01],
        [9.9958e-01, 4.1608e-04],
        [3.1658e-09, 1.0000e+00],
        [1.0790e-05, 9.9999e-01],
        [2.1545e-03, 9.9785e-01],
        [8.7976e-05, 9.9991e-01],
        [3.5215e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2407e-05, 9.9998e-01],
        [6.2670e-10, 1.0000e+00],
        [1.5844e-06, 1.0000e+00],
        [1.1741e-08, 1.0000e+00],
        [4.7969e-05, 9.9995e-01],
        [2.8036e-07, 1.0000e+00],
        [9.8134e-05, 9.9990e-01],
        [2.8179e-08, 1.0000e+00],
        [3.3544e-05, 9.9997e-01],
        [3.2982e-04, 9.9967e-01],
        [9.9994e-01, 5.9665e-05],
        [9.9977e-01, 2.2657e-04],
        [1.1673e-10, 1.0000e+00],
        [1.3995e-06, 1.0000e+00],
        [9.8014e-01, 1.9857e-02],
        [3.0923e-05, 9.9997e-01],
        [2.2741e-09, 1.0000e+00],
        [5.6647e-04, 9.9943e-01],
        [1.7731e-07, 1.0000e+00],
        [5.7647e-06, 9.9999e-01],
        [1.2790e-02, 9.8721e-01],
        [9.6755e-03, 9.9032e-01],
        [9.9985e-01, 1.5088e-04],
        [7.4240e-01, 2.5760e-01],
        [3.1410e-08, 1.0000e+00],
        [8.6551e-04, 9.9913e-01],
        [9.9990e-01, 9.9705e-05],
        [4.6266e-10, 1.0000e+00],
        [6.1854e-06, 9.9999e-01],
        [2.9918e-05, 9.9997e-01],
        [5.5091e-09, 1.0000e+00],
        [1.4135e-07, 1.0000e+00],
        [2.4871e-04, 9.9975e-01],
        [8.6403e-07, 1.0000e+00],
        [9.8860e-01, 1.1405e-02],
        [7.6657e-03, 9.9233e-01],
        [4.0625e-06, 1.0000e+00],
        [3.1020e-03, 9.9690e-01],
        [9.9971e-01, 2.9045e-04],
        [5.8203e-06, 9.9999e-01],
        [6.3497e-05, 9.9994e-01],
        [9.9968e-01, 3.1652e-04],
        [9.9998e-01, 1.5163e-05],
        [9.9893e-01, 1.0686e-03],
        [4.7238e-07, 1.0000e+00],
        [5.3284e-09, 1.0000e+00],
        [2.6081e-04, 9.9974e-01],
        [9.9992e-01, 7.6479e-05],
        [9.9990e-01, 1.0294e-04],
        [6.2839e-10, 1.0000e+00],
        [1.2058e-03, 9.9879e-01],
        [2.2037e-07, 1.0000e+00],
        [9.9999e-01, 7.7616e-06],
        [1.2670e-07, 1.0000e+00],
        [9.9990e-01, 9.5075e-05],
        [1.6935e-05, 9.9998e-01],
        [7.5694e-02, 9.2431e-01],
        [9.9879e-01, 1.2103e-03],
        [2.4125e-07, 1.0000e+00],
        [1.1231e-04, 9.9989e-01],
        [7.4573e-07, 1.0000e+00],
        [3.1051e-06, 1.0000e+00],
        [9.9952e-01, 4.8359e-04],
        [9.9996e-01, 4.0868e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[8.1455e-08, 1.0000e+00],
        [5.6941e-09, 1.0000e+00],
        [1.4607e-04, 9.9985e-01],
        [1.8831e-08, 1.0000e+00],
        [8.4082e-09, 1.0000e+00],
        [8.2519e-09, 1.0000e+00],
        [9.9988e-01, 1.2122e-04],
        [9.3356e-08, 1.0000e+00],
        [1.4295e-06, 1.0000e+00],
        [9.9963e-01, 3.7470e-04],
        [2.1281e-04, 9.9979e-01],
        [3.7561e-09, 1.0000e+00],
        [7.2345e-08, 1.0000e+00],
        [3.7789e-04, 9.9962e-01],
        [4.8923e-05, 9.9995e-01],
        [2.1731e-07, 1.0000e+00],
        [4.1338e-01, 5.8662e-01],
        [5.0070e-04, 9.9950e-01],
        [9.8958e-01, 1.0415e-02],
        [7.5067e-08, 1.0000e+00],
        [1.5491e-06, 1.0000e+00],
        [3.0527e-05, 9.9997e-01],
        [7.8743e-08, 1.0000e+00],
        [7.3309e-05, 9.9993e-01],
        [5.7676e-05, 9.9994e-01],
        [3.6366e-06, 1.0000e+00],
        [6.5452e-08, 1.0000e+00],
        [6.8868e-03, 9.9311e-01],
        [9.9990e-01, 1.0297e-04],
        [2.3885e-07, 1.0000e+00],
        [1.7102e-09, 1.0000e+00],
        [2.7634e-06, 1.0000e+00],
        [6.7256e-09, 1.0000e+00],
        [4.8997e-03, 9.9510e-01],
        [1.6938e-09, 1.0000e+00],
        [6.3021e-01, 3.6979e-01],
        [6.6143e-05, 9.9993e-01],
        [2.4929e-04, 9.9975e-01],
        [1.1503e-05, 9.9999e-01],
        [4.8754e-07, 1.0000e+00],
        [9.6284e-08, 1.0000e+00],
        [1.3411e-04, 9.9987e-01],
        [1.1444e-06, 1.0000e+00],
        [9.9934e-01, 6.5582e-04],
        [9.9890e-01, 1.1030e-03],
        [6.4742e-01, 3.5258e-01],
        [1.4513e-05, 9.9999e-01],
        [5.1182e-10, 1.0000e+00],
        [1.0530e-03, 9.9895e-01],
        [1.5519e-08, 1.0000e+00],
        [1.9895e-07, 1.0000e+00],
        [1.0479e-05, 9.9999e-01],
        [1.4594e-08, 1.0000e+00],
        [9.9988e-01, 1.2086e-04],
        [4.6811e-04, 9.9953e-01],
        [4.4135e-04, 9.9956e-01],
        [2.6029e-06, 1.0000e+00],
        [1.5249e-05, 9.9998e-01],
        [2.0854e-07, 1.0000e+00],
        [9.9990e-01, 9.6929e-05],
        [3.1764e-01, 6.8236e-01],
        [2.4705e-03, 9.9753e-01],
        [1.6111e-05, 9.9998e-01],
        [9.9988e-01, 1.1672e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0007e-07, 1.0000e+00],
        [2.6216e-06, 1.0000e+00],
        [9.9987e-01, 1.3214e-04],
        [2.6627e-06, 1.0000e+00],
        [1.9277e-10, 1.0000e+00],
        [9.9989e-01, 1.1134e-04],
        [2.4137e-09, 1.0000e+00],
        [9.9990e-01, 9.6344e-05],
        [9.9995e-01, 5.4844e-05],
        [9.0330e-07, 1.0000e+00],
        [1.4305e-05, 9.9999e-01],
        [8.4860e-04, 9.9915e-01],
        [9.9342e-01, 6.5839e-03],
        [2.1143e-05, 9.9998e-01],
        [4.2300e-01, 5.7700e-01],
        [9.7465e-01, 2.5351e-02],
        [9.9990e-01, 9.9216e-05],
        [8.5218e-09, 1.0000e+00],
        [4.4972e-07, 1.0000e+00],
        [3.1352e-05, 9.9997e-01],
        [9.9942e-01, 5.7608e-04],
        [2.7846e-04, 9.9972e-01],
        [5.4384e-09, 1.0000e+00],
        [9.9998e-01, 1.5675e-05],
        [1.1922e-08, 1.0000e+00],
        [6.0829e-05, 9.9994e-01],
        [5.6235e-06, 9.9999e-01],
        [1.1297e-06, 1.0000e+00],
        [6.3699e-08, 1.0000e+00],
        [1.3690e-06, 1.0000e+00],
        [3.3933e-04, 9.9966e-01],
        [4.0985e-02, 9.5902e-01],
        [9.9974e-01, 2.5544e-04],
        [9.8278e-07, 1.0000e+00],
        [2.5725e-05, 9.9997e-01],
        [5.3221e-07, 1.0000e+00],
        [6.8547e-07, 1.0000e+00],
        [9.9942e-01, 5.7529e-04],
        [9.8061e-01, 1.9390e-02],
        [1.9062e-05, 9.9998e-01],
        [5.1484e-03, 9.9485e-01],
        [8.9433e-11, 1.0000e+00],
        [2.9835e-06, 1.0000e+00],
        [1.1747e-03, 9.9883e-01],
        [1.9130e-08, 1.0000e+00],
        [9.9991e-01, 9.0099e-05],
        [9.9748e-01, 2.5225e-03],
        [9.2036e-11, 1.0000e+00],
        [5.6857e-05, 9.9994e-01],
        [9.9986e-01, 1.3534e-04],
        [5.3400e-07, 1.0000e+00],
        [9.9693e-11, 1.0000e+00],
        [1.5126e-12, 1.0000e+00],
        [2.9121e-01, 7.0879e-01],
        [6.7258e-01, 3.2742e-01],
        [6.3687e-05, 9.9994e-01],
        [7.1235e-05, 9.9993e-01],
        [9.9958e-01, 4.2465e-04],
        [1.2651e-05, 9.9999e-01],
        [1.4281e-04, 9.9986e-01],
        [1.5494e-05, 9.9998e-01],
        [9.9998e-01, 1.9778e-05],
        [1.5413e-04, 9.9985e-01],
        [9.9993e-01, 6.9007e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[3.3608e-07, 1.0000e+00],
        [4.0656e-07, 1.0000e+00],
        [3.0835e-04, 9.9969e-01],
        [1.1074e-06, 1.0000e+00],
        [2.3586e-07, 1.0000e+00],
        [4.2380e-08, 1.0000e+00],
        [7.7571e-05, 9.9992e-01],
        [1.3881e-08, 1.0000e+00],
        [2.8400e-11, 1.0000e+00],
        [9.9999e-01, 1.0447e-05],
        [3.2137e-09, 1.0000e+00],
        [1.0897e-06, 1.0000e+00],
        [1.4423e-05, 9.9999e-01],
        [1.0641e-12, 1.0000e+00],
        [9.9294e-01, 7.0611e-03],
        [1.6872e-07, 1.0000e+00],
        [2.6086e-11, 1.0000e+00],
        [1.7614e-06, 1.0000e+00],
        [9.9989e-01, 1.0879e-04],
        [5.5849e-12, 1.0000e+00],
        [2.4949e-08, 1.0000e+00],
        [3.2024e-02, 9.6798e-01],
        [9.9991e-01, 9.4256e-05],
        [2.0913e-04, 9.9979e-01],
        [1.6397e-06, 1.0000e+00],
        [9.9978e-01, 2.1587e-04],
        [9.8298e-01, 1.7017e-02],
        [1.0564e-06, 1.0000e+00],
        [6.1693e-05, 9.9994e-01],
        [1.5183e-04, 9.9985e-01],
        [1.0000e+00, 3.6468e-06],
        [1.7043e-09, 1.0000e+00],
        [1.0012e-07, 1.0000e+00],
        [6.0565e-01, 3.9435e-01],
        [4.8236e-10, 1.0000e+00],
        [1.1068e-05, 9.9999e-01],
        [3.5720e-04, 9.9964e-01],
        [9.9732e-01, 2.6829e-03],
        [1.4496e-09, 1.0000e+00],
        [7.6599e-08, 1.0000e+00],
        [5.2466e-01, 4.7534e-01],
        [9.9988e-01, 1.1645e-04],
        [1.3984e-05, 9.9999e-01],
        [3.5383e-04, 9.9965e-01],
        [9.4153e-01, 5.8472e-02],
        [2.3083e-06, 1.0000e+00],
        [3.8920e-08, 1.0000e+00],
        [1.5194e-05, 9.9998e-01],
        [1.6524e-08, 1.0000e+00],
        [1.8899e-05, 9.9998e-01],
        [2.4065e-06, 1.0000e+00],
        [3.1433e-08, 1.0000e+00],
        [5.6465e-10, 1.0000e+00],
        [6.1145e-08, 1.0000e+00],
        [4.9820e-02, 9.5018e-01],
        [1.3249e-05, 9.9999e-01],
        [1.5284e-04, 9.9985e-01],
        [9.9972e-01, 2.8345e-04],
        [1.3987e-10, 1.0000e+00],
        [8.2875e-06, 9.9999e-01],
        [2.2418e-06, 1.0000e+00],
        [3.2137e-08, 1.0000e+00],
        [9.9851e-01, 1.4904e-03],
        [6.8873e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9976e-01, 2.3895e-04],
        [1.6111e-10, 1.0000e+00],
        [8.4946e-01, 1.5054e-01],
        [1.1353e-06, 1.0000e+00],
        [3.7450e-11, 1.0000e+00],
        [4.6036e-05, 9.9995e-01],
        [9.9975e-01, 2.4544e-04],
        [2.6621e-02, 9.7338e-01],
        [6.3115e-06, 9.9999e-01],
        [5.6971e-04, 9.9943e-01],
        [1.9496e-06, 1.0000e+00],
        [3.7790e-11, 1.0000e+00],
        [2.0645e-03, 9.9794e-01],
        [9.9988e-01, 1.1889e-04],
        [5.4427e-12, 1.0000e+00],
        [9.9963e-01, 3.6527e-04],
        [9.9892e-01, 1.0835e-03],
        [1.1845e-09, 1.0000e+00],
        [2.3787e-12, 1.0000e+00],
        [6.6467e-03, 9.9335e-01],
        [1.5209e-07, 1.0000e+00],
        [1.5380e-10, 1.0000e+00],
        [2.9451e-07, 1.0000e+00],
        [2.1718e-10, 1.0000e+00],
        [1.9778e-07, 1.0000e+00],
        [1.5122e-07, 1.0000e+00],
        [9.9115e-01, 8.8512e-03],
        [3.1202e-07, 1.0000e+00],
        [1.1693e-07, 1.0000e+00],
        [6.9454e-03, 9.9305e-01],
        [6.7184e-05, 9.9993e-01],
        [6.8167e-05, 9.9993e-01],
        [9.9970e-01, 3.0258e-04],
        [9.9968e-01, 3.2028e-04],
        [9.9510e-08, 1.0000e+00],
        [8.6483e-10, 1.0000e+00],
        [9.8056e-01, 1.9440e-02],
        [3.5875e-06, 1.0000e+00],
        [7.9462e-05, 9.9992e-01],
        [9.9974e-01, 2.6005e-04],
        [1.4407e-06, 1.0000e+00],
        [4.9306e-04, 9.9951e-01],
        [8.0159e-01, 1.9841e-01],
        [3.0492e-07, 1.0000e+00],
        [4.0398e-03, 9.9596e-01],
        [7.2679e-09, 1.0000e+00],
        [5.5279e-07, 1.0000e+00],
        [1.8376e-03, 9.9816e-01],
        [2.7477e-08, 1.0000e+00],
        [2.2116e-10, 1.0000e+00],
        [9.2198e-06, 9.9999e-01],
        [1.4003e-09, 1.0000e+00],
        [1.2095e-08, 1.0000e+00],
        [8.2531e-03, 9.9175e-01],
        [4.4926e-05, 9.9996e-01],
        [1.8876e-06, 1.0000e+00],
        [9.4411e-07, 1.0000e+00],
        [5.4508e-04, 9.9945e-01],
        [1.3371e-04, 9.9987e-01],
        [1.7674e-06, 1.0000e+00],
        [5.4960e-10, 1.0000e+00],
        [9.9958e-01, 4.2269e-04],
        [2.4436e-08, 1.0000e+00],
        [3.1583e-03, 9.9684e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5792e-08, 1.0000e+00],
        [8.7661e-05, 9.9991e-01],
        [9.9989e-01, 1.1193e-04],
        [9.9975e-01, 2.5192e-04],
        [8.0504e-08, 1.0000e+00],
        [3.7275e-04, 9.9963e-01],
        [9.6081e-01, 3.9186e-02],
        [5.2024e-05, 9.9995e-01],
        [9.9982e-01, 1.8297e-04],
        [1.6999e-03, 9.9830e-01],
        [1.0016e-06, 1.0000e+00],
        [5.6414e-08, 1.0000e+00],
        [9.9995e-01, 5.3419e-05],
        [1.7604e-09, 1.0000e+00],
        [1.7556e-05, 9.9998e-01],
        [4.5212e-06, 1.0000e+00],
        [4.5725e-06, 1.0000e+00],
        [1.8785e-08, 1.0000e+00],
        [1.5532e-07, 1.0000e+00],
        [9.9986e-01, 1.4163e-04],
        [9.9961e-01, 3.9422e-04],
        [9.9801e-01, 1.9921e-03],
        [2.2738e-07, 1.0000e+00],
        [5.2274e-07, 1.0000e+00],
        [6.4174e-06, 9.9999e-01],
        [1.7165e-05, 9.9998e-01],
        [2.1318e-08, 1.0000e+00],
        [9.1326e-01, 8.6743e-02],
        [8.2232e-08, 1.0000e+00],
        [5.4817e-11, 1.0000e+00],
        [9.9999e-01, 4.9528e-06],
        [1.8850e-06, 1.0000e+00],
        [8.1075e-07, 1.0000e+00],
        [4.5253e-11, 1.0000e+00],
        [9.6120e-01, 3.8804e-02],
        [5.9660e-03, 9.9403e-01],
        [1.1648e-10, 1.0000e+00],
        [9.4439e-09, 1.0000e+00],
        [1.5484e-09, 1.0000e+00],
        [5.0911e-06, 9.9999e-01],
        [2.7825e-07, 1.0000e+00],
        [2.5525e-05, 9.9997e-01],
        [9.8929e-09, 1.0000e+00],
        [9.9983e-01, 1.7018e-04],
        [9.5687e-07, 1.0000e+00],
        [2.3991e-09, 1.0000e+00],
        [6.7826e-06, 9.9999e-01],
        [1.9391e-05, 9.9998e-01],
        [6.5606e-07, 1.0000e+00],
        [8.6452e-07, 1.0000e+00],
        [3.3586e-06, 1.0000e+00],
        [4.2068e-02, 9.5793e-01],
        [4.5554e-09, 1.0000e+00],
        [9.9980e-01, 1.9802e-04],
        [2.8867e-11, 1.0000e+00],
        [9.9994e-01, 5.9819e-05],
        [1.5062e-02, 9.8494e-01],
        [5.7317e-07, 1.0000e+00],
        [2.9077e-04, 9.9971e-01],
        [1.3926e-06, 1.0000e+00],
        [9.9976e-01, 2.4333e-04],
        [1.7710e-10, 1.0000e+00],
        [5.2130e-10, 1.0000e+00],
        [9.9993e-01, 6.8639e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9994e-01, 5.6939e-05],
        [8.0195e-07, 1.0000e+00],
        [1.8732e-09, 1.0000e+00],
        [2.4665e-02, 9.7533e-01],
        [9.9804e-01, 1.9550e-03],
        [9.2278e-07, 1.0000e+00],
        [5.6712e-09, 1.0000e+00],
        [5.7210e-07, 1.0000e+00],
        [2.5081e-05, 9.9997e-01],
        [2.8611e-05, 9.9997e-01],
        [3.9613e-11, 1.0000e+00],
        [9.0276e-06, 9.9999e-01],
        [9.9990e-01, 9.9723e-05],
        [2.5633e-08, 1.0000e+00],
        [5.6026e-08, 1.0000e+00],
        [1.4073e-07, 1.0000e+00],
        [3.8257e-07, 1.0000e+00],
        [9.9999e-01, 5.2352e-06],
        [2.1448e-08, 1.0000e+00],
        [1.2465e-04, 9.9988e-01],
        [1.8593e-12, 1.0000e+00],
        [8.7551e-07, 1.0000e+00],
        [3.2911e-04, 9.9967e-01],
        [1.6465e-07, 1.0000e+00],
        [9.9980e-01, 1.9771e-04],
        [3.7141e-03, 9.9629e-01],
        [2.6852e-11, 1.0000e+00],
        [1.4616e-08, 1.0000e+00],
        [2.7968e-07, 1.0000e+00],
        [1.8703e-12, 1.0000e+00],
        [9.9987e-01, 1.3168e-04],
        [2.0330e-07, 1.0000e+00],
        [1.0290e-07, 1.0000e+00],
        [3.0920e-10, 1.0000e+00],
        [2.7800e-08, 1.0000e+00],
        [1.7093e-06, 1.0000e+00],
        [9.6857e-06, 9.9999e-01],
        [7.9503e-07, 1.0000e+00],
        [7.5921e-08, 1.0000e+00],
        [8.5672e-01, 1.4328e-01],
        [2.7968e-07, 1.0000e+00],
        [8.2781e-01, 1.7219e-01],
        [8.9302e-11, 1.0000e+00],
        [1.7031e-04, 9.9983e-01],
        [9.9927e-01, 7.3085e-04],
        [1.2845e-02, 9.8715e-01],
        [1.8686e-07, 1.0000e+00],
        [9.9981e-01, 1.9101e-04],
        [1.5070e-08, 1.0000e+00],
        [2.6561e-07, 1.0000e+00],
        [7.6223e-11, 1.0000e+00],
        [1.3250e-08, 1.0000e+00],
        [7.3753e-01, 2.6247e-01],
        [1.0683e-09, 1.0000e+00],
        [8.1011e-06, 9.9999e-01],
        [9.9988e-01, 1.1946e-04],
        [8.3827e-01, 1.6173e-01],
        [4.8684e-08, 1.0000e+00],
        [9.9988e-01, 1.2236e-04],
        [1.4380e-05, 9.9999e-01],
        [9.9991e-01, 8.7945e-05],
        [9.3459e-10, 1.0000e+00],
        [3.3792e-10, 1.0000e+00],
        [2.8512e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.3252e-07, 1.0000e+00],
        [9.9651e-01, 3.4936e-03],
        [9.9796e-01, 2.0369e-03],
        [2.1832e-08, 1.0000e+00],
        [2.1154e-09, 1.0000e+00],
        [5.6519e-11, 1.0000e+00],
        [3.9405e-07, 1.0000e+00],
        [2.8635e-08, 1.0000e+00],
        [1.3035e-08, 1.0000e+00],
        [9.9983e-01, 1.7316e-04],
        [9.9984e-01, 1.5669e-04],
        [1.7571e-08, 1.0000e+00],
        [1.1992e-10, 1.0000e+00],
        [1.1561e-06, 1.0000e+00],
        [4.0291e-05, 9.9996e-01],
        [8.8871e-01, 1.1129e-01],
        [1.6927e-04, 9.9983e-01],
        [9.9963e-01, 3.6801e-04],
        [9.9955e-01, 4.5459e-04],
        [1.0429e-05, 9.9999e-01],
        [8.2581e-09, 1.0000e+00],
        [9.8723e-01, 1.2766e-02],
        [9.9987e-01, 1.3313e-04],
        [4.1502e-10, 1.0000e+00],
        [1.1364e-07, 1.0000e+00],
        [4.0819e-09, 1.0000e+00],
        [2.9855e-05, 9.9997e-01],
        [9.9347e-01, 6.5317e-03],
        [1.9072e-10, 1.0000e+00],
        [1.8701e-06, 1.0000e+00],
        [1.5085e-06, 1.0000e+00],
        [1.3416e-06, 1.0000e+00],
        [9.9996e-01, 4.3358e-05],
        [9.9992e-01, 7.7740e-05],
        [9.9843e-01, 1.5690e-03],
        [2.0752e-05, 9.9998e-01],
        [1.2312e-05, 9.9999e-01],
        [1.9187e-09, 1.0000e+00],
        [2.1387e-05, 9.9998e-01],
        [9.9989e-01, 1.0654e-04],
        [9.9973e-01, 2.7443e-04],
        [2.4957e-10, 1.0000e+00],
        [3.7612e-07, 1.0000e+00],
        [9.9992e-01, 7.5281e-05],
        [2.3062e-08, 1.0000e+00],
        [1.9497e-07, 1.0000e+00],
        [2.4687e-08, 1.0000e+00],
        [1.2056e-05, 9.9999e-01],
        [3.2245e-08, 1.0000e+00],
        [9.9744e-01, 2.5613e-03],
        [5.2223e-07, 1.0000e+00],
        [6.0384e-07, 1.0000e+00],
        [5.2066e-10, 1.0000e+00],
        [9.9186e-01, 8.1438e-03],
        [6.5940e-10, 1.0000e+00],
        [9.8646e-01, 1.3542e-02],
        [4.4589e-03, 9.9554e-01],
        [5.3978e-07, 1.0000e+00],
        [5.3609e-07, 1.0000e+00],
        [9.9978e-01, 2.1538e-04],
        [5.4901e-08, 1.0000e+00],
        [6.7252e-07, 1.0000e+00],
        [1.4008e-07, 1.0000e+00],
        [3.2944e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.6162e-05, 9.9996e-01],
        [9.9992e-01, 7.5629e-05],
        [2.8575e-08, 1.0000e+00],
        [9.9989e-01, 1.1400e-04],
        [9.9982e-01, 1.8150e-04],
        [2.7292e-08, 1.0000e+00],
        [1.0756e-06, 1.0000e+00],
        [2.7340e-05, 9.9997e-01],
        [3.0097e-10, 1.0000e+00],
        [4.8113e-05, 9.9995e-01],
        [5.7465e-07, 1.0000e+00],
        [6.5365e-07, 1.0000e+00],
        [9.9977e-01, 2.2775e-04],
        [8.4306e-01, 1.5694e-01],
        [5.6485e-09, 1.0000e+00],
        [9.9749e-01, 2.5109e-03],
        [9.3403e-01, 6.5966e-02],
        [4.7371e-08, 1.0000e+00],
        [9.9605e-01, 3.9454e-03],
        [1.2113e-05, 9.9999e-01],
        [8.7903e-01, 1.2097e-01],
        [1.9782e-07, 1.0000e+00],
        [3.3462e-08, 1.0000e+00],
        [3.1468e-07, 1.0000e+00],
        [1.7169e-08, 1.0000e+00],
        [4.9131e-11, 1.0000e+00],
        [1.2299e-06, 1.0000e+00],
        [2.0114e-08, 1.0000e+00],
        [2.1327e-06, 1.0000e+00],
        [6.6497e-04, 9.9933e-01],
        [3.5925e-11, 1.0000e+00],
        [7.0308e-11, 1.0000e+00],
        [8.4713e-06, 9.9999e-01],
        [8.3643e-08, 1.0000e+00],
        [5.3130e-09, 1.0000e+00],
        [6.0764e-07, 1.0000e+00],
        [3.8981e-07, 1.0000e+00],
        [4.6147e-07, 1.0000e+00],
        [8.4677e-02, 9.1532e-01],
        [9.9697e-11, 1.0000e+00],
        [1.9940e-08, 1.0000e+00],
        [8.3916e-09, 1.0000e+00],
        [2.7835e-04, 9.9972e-01],
        [1.4134e-06, 1.0000e+00],
        [1.3852e-07, 1.0000e+00],
        [5.0933e-04, 9.9949e-01],
        [9.9994e-01, 6.1825e-05],
        [5.5680e-11, 1.0000e+00],
        [5.2843e-06, 9.9999e-01],
        [9.9989e-01, 1.1064e-04],
        [9.9617e-09, 1.0000e+00],
        [8.2438e-09, 1.0000e+00],
        [3.4263e-05, 9.9997e-01],
        [9.9956e-01, 4.4040e-04],
        [9.9991e-01, 9.3007e-05],
        [2.1247e-04, 9.9979e-01],
        [1.8042e-06, 1.0000e+00],
        [9.9985e-01, 1.5117e-04],
        [2.0812e-06, 1.0000e+00],
        [4.9790e-09, 1.0000e+00],
        [1.0616e-01, 8.9384e-01],
        [6.8581e-01, 3.1419e-01],
        [1.5766e-05, 9.9998e-01],
        [9.6355e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.0718e-09, 1.0000e+00],
        [1.2816e-05, 9.9999e-01],
        [6.9387e-07, 1.0000e+00],
        [4.3040e-05, 9.9996e-01],
        [2.9935e-11, 1.0000e+00],
        [9.9998e-01, 1.6871e-05],
        [3.4322e-09, 1.0000e+00],
        [1.9032e-02, 9.8097e-01],
        [9.9860e-01, 1.4031e-03],
        [9.9997e-01, 2.9319e-05],
        [1.1048e-06, 1.0000e+00],
        [5.2777e-08, 1.0000e+00],
        [4.5093e-08, 1.0000e+00],
        [8.9582e-03, 9.9104e-01],
        [9.9991e-01, 9.1324e-05],
        [1.7695e-07, 1.0000e+00],
        [5.1279e-08, 1.0000e+00],
        [1.6578e-07, 1.0000e+00],
        [1.4187e-07, 1.0000e+00],
        [4.2952e-08, 1.0000e+00],
        [1.4039e-06, 1.0000e+00],
        [3.2865e-08, 1.0000e+00],
        [3.7014e-09, 1.0000e+00],
        [1.4259e-05, 9.9999e-01],
        [6.3974e-07, 1.0000e+00],
        [1.2267e-08, 1.0000e+00],
        [1.3440e-07, 1.0000e+00],
        [7.7094e-10, 1.0000e+00],
        [3.5383e-05, 9.9996e-01],
        [1.1764e-04, 9.9988e-01],
        [4.7606e-03, 9.9524e-01],
        [4.5104e-08, 1.0000e+00],
        [2.9696e-06, 1.0000e+00],
        [2.1776e-05, 9.9998e-01],
        [1.5143e-08, 1.0000e+00],
        [1.2724e-08, 1.0000e+00],
        [4.7214e-10, 1.0000e+00],
        [2.3207e-07, 1.0000e+00],
        [9.9990e-01, 9.8859e-05],
        [1.5333e-08, 1.0000e+00],
        [6.5931e-08, 1.0000e+00],
        [3.5458e-07, 1.0000e+00],
        [9.9262e-01, 7.3785e-03],
        [1.1359e-05, 9.9999e-01],
        [1.0891e-02, 9.8911e-01],
        [9.9995e-01, 5.4525e-05],
        [4.4543e-11, 1.0000e+00],
        [1.5716e-08, 1.0000e+00],
        [9.9819e-01, 1.8127e-03],
        [1.0265e-06, 1.0000e+00],
        [2.8385e-08, 1.0000e+00],
        [1.5580e-03, 9.9844e-01],
        [8.8807e-05, 9.9991e-01],
        [9.9985e-01, 1.5256e-04],
        [9.6220e-01, 3.7803e-02],
        [5.3936e-10, 1.0000e+00],
        [8.6248e-07, 1.0000e+00],
        [1.7226e-05, 9.9998e-01],
        [9.9694e-10, 1.0000e+00],
        [9.8791e-01, 1.2088e-02],
        [6.7839e-01, 3.2161e-01],
        [1.3802e-06, 1.0000e+00],
        [4.1982e-11, 1.0000e+00],
        [2.1896e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.1267e-01, 3.8733e-01],
        [4.0737e-05, 9.9996e-01],
        [6.7991e-12, 1.0000e+00],
        [1.6594e-10, 1.0000e+00],
        [6.6499e-07, 1.0000e+00],
        [3.8329e-03, 9.9617e-01],
        [7.8793e-10, 1.0000e+00],
        [3.5877e-05, 9.9996e-01],
        [1.3859e-06, 1.0000e+00],
        [8.8205e-06, 9.9999e-01],
        [9.9974e-01, 2.6427e-04],
        [2.0653e-09, 1.0000e+00],
        [9.9993e-01, 6.7950e-05],
        [3.2546e-07, 1.0000e+00],
        [3.7771e-06, 1.0000e+00],
        [1.2376e-05, 9.9999e-01],
        [4.5281e-08, 1.0000e+00],
        [9.9993e-01, 7.3220e-05],
        [5.8798e-08, 1.0000e+00],
        [3.7989e-06, 1.0000e+00],
        [6.1538e-08, 1.0000e+00],
        [8.1948e-07, 1.0000e+00],
        [1.1839e-07, 1.0000e+00],
        [3.6039e-05, 9.9996e-01],
        [9.9816e-01, 1.8360e-03],
        [7.1031e-07, 1.0000e+00],
        [1.8348e-01, 8.1652e-01],
        [1.0790e-07, 1.0000e+00],
        [7.8135e-08, 1.0000e+00],
        [1.4991e-03, 9.9850e-01],
        [1.8317e-10, 1.0000e+00],
        [9.2099e-01, 7.9014e-02],
        [2.6435e-02, 9.7357e-01],
        [6.3120e-08, 1.0000e+00],
        [8.7745e-06, 9.9999e-01],
        [2.2547e-08, 1.0000e+00],
        [1.7642e-05, 9.9998e-01],
        [4.2840e-07, 1.0000e+00],
        [6.7256e-07, 1.0000e+00],
        [4.8935e-13, 1.0000e+00],
        [3.4461e-05, 9.9997e-01],
        [1.3606e-07, 1.0000e+00],
        [1.3238e-06, 1.0000e+00],
        [1.4858e-04, 9.9985e-01],
        [9.9986e-01, 1.3670e-04],
        [7.5555e-07, 1.0000e+00],
        [6.9163e-10, 1.0000e+00],
        [9.6441e-06, 9.9999e-01],
        [1.5219e-06, 1.0000e+00],
        [4.6364e-04, 9.9954e-01],
        [1.2076e-08, 1.0000e+00],
        [7.8700e-07, 1.0000e+00],
        [7.2032e-08, 1.0000e+00],
        [1.7250e-07, 1.0000e+00],
        [1.9392e-06, 1.0000e+00],
        [1.5438e-06, 1.0000e+00],
        [9.9962e-01, 3.8146e-04],
        [8.1874e-09, 1.0000e+00],
        [1.4014e-05, 9.9999e-01],
        [5.9815e-08, 1.0000e+00],
        [9.9970e-01, 2.9605e-04],
        [2.5864e-02, 9.7414e-01],
        [1.7572e-07, 1.0000e+00],
        [9.9977e-01, 2.3444e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9990e-01, 1.0067e-04],
        [1.0622e-02, 9.8938e-01],
        [1.5345e-09, 1.0000e+00],
        [1.3448e-02, 9.8655e-01],
        [3.8863e-10, 1.0000e+00],
        [7.1474e-06, 9.9999e-01],
        [1.7700e-08, 1.0000e+00],
        [6.8162e-05, 9.9993e-01],
        [2.6793e-08, 1.0000e+00],
        [9.9246e-01, 7.5389e-03],
        [2.8187e-06, 1.0000e+00],
        [5.0615e-05, 9.9995e-01],
        [1.3028e-04, 9.9987e-01],
        [3.1213e-01, 6.8787e-01],
        [1.0000e+00, 1.4039e-06],
        [1.3131e-07, 1.0000e+00],
        [9.9984e-01, 1.5906e-04],
        [1.0671e-05, 9.9999e-01],
        [2.3378e-10, 1.0000e+00],
        [7.0501e-06, 9.9999e-01],
        [8.8382e-08, 1.0000e+00],
        [2.2194e-09, 1.0000e+00],
        [9.3914e-01, 6.0863e-02],
        [5.2960e-07, 1.0000e+00],
        [7.0389e-08, 1.0000e+00],
        [9.9996e-01, 4.3248e-05],
        [1.3757e-05, 9.9999e-01],
        [8.3621e-08, 1.0000e+00],
        [3.1156e-07, 1.0000e+00],
        [2.5706e-04, 9.9974e-01],
        [9.9996e-01, 3.7211e-05],
        [1.9612e-05, 9.9998e-01],
        [9.9982e-01, 1.8001e-04],
        [1.1282e-08, 1.0000e+00],
        [7.2925e-09, 1.0000e+00],
        [9.9966e-01, 3.3753e-04],
        [5.8996e-08, 1.0000e+00],
        [5.1522e-08, 1.0000e+00],
        [1.1636e-08, 1.0000e+00],
        [1.8945e-02, 9.8105e-01],
        [1.5971e-09, 1.0000e+00],
        [9.9954e-01, 4.6304e-04],
        [2.2320e-08, 1.0000e+00],
        [5.1570e-10, 1.0000e+00],
        [5.8859e-06, 9.9999e-01],
        [6.8655e-06, 9.9999e-01],
        [1.0000e+00, 7.2988e-08],
        [4.8009e-10, 1.0000e+00],
        [3.2636e-06, 1.0000e+00],
        [9.9990e-01, 1.0254e-04],
        [2.4008e-09, 1.0000e+00],
        [6.5395e-03, 9.9346e-01],
        [1.2874e-01, 8.7126e-01],
        [2.1507e-07, 1.0000e+00],
        [1.2256e-08, 1.0000e+00],
        [4.2028e-01, 5.7972e-01],
        [3.7629e-08, 1.0000e+00],
        [4.6869e-04, 9.9953e-01],
        [8.3840e-07, 1.0000e+00],
        [2.6337e-05, 9.9997e-01],
        [9.9949e-01, 5.1119e-04],
        [9.9993e-01, 7.2995e-05],
        [9.9820e-01, 1.7974e-03],
        [9.1292e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.3572e-08, 1.0000e+00],
        [9.9879e-01, 1.2103e-03],
        [1.1771e-03, 9.9882e-01],
        [2.2057e-06, 1.0000e+00],
        [2.4939e-07, 1.0000e+00],
        [2.0919e-07, 1.0000e+00],
        [9.4352e-07, 1.0000e+00],
        [7.7371e-09, 1.0000e+00],
        [1.1145e-07, 1.0000e+00],
        [1.1852e-04, 9.9988e-01],
        [9.4179e-07, 1.0000e+00],
        [1.9436e-10, 1.0000e+00],
        [9.9992e-01, 7.9938e-05],
        [1.7594e-07, 1.0000e+00],
        [7.7846e-08, 1.0000e+00],
        [1.5225e-03, 9.9848e-01],
        [9.9960e-01, 3.9922e-04],
        [1.2728e-04, 9.9987e-01],
        [2.6037e-05, 9.9997e-01],
        [7.1486e-03, 9.9285e-01],
        [9.9996e-01, 3.9331e-05],
        [9.3388e-03, 9.9066e-01],
        [6.6566e-01, 3.3434e-01],
        [7.2841e-04, 9.9927e-01],
        [9.1694e-07, 1.0000e+00],
        [1.5483e-05, 9.9998e-01],
        [1.8572e-01, 8.1428e-01],
        [1.9034e-04, 9.9981e-01],
        [1.3783e-08, 1.0000e+00],
        [9.9999e-01, 1.0761e-05],
        [2.5515e-03, 9.9745e-01],
        [3.4493e-07, 1.0000e+00],
        [7.6054e-09, 1.0000e+00],
        [2.4050e-09, 1.0000e+00],
        [5.6963e-01, 4.3037e-01],
        [2.2878e-04, 9.9977e-01],
        [1.5261e-08, 1.0000e+00],
        [1.2043e-08, 1.0000e+00],
        [1.2513e-04, 9.9987e-01],
        [1.1737e-02, 9.8826e-01],
        [9.9985e-01, 1.4600e-04],
        [5.0584e-05, 9.9995e-01],
        [9.9996e-01, 4.2900e-05],
        [4.8133e-03, 9.9519e-01],
        [9.4697e-04, 9.9905e-01],
        [1.9077e-08, 1.0000e+00],
        [9.9991e-01, 9.3432e-05],
        [9.9970e-01, 3.0094e-04],
        [2.0051e-05, 9.9998e-01],
        [1.7993e-07, 1.0000e+00],
        [4.6138e-06, 1.0000e+00],
        [3.6023e-03, 9.9640e-01],
        [1.5454e-03, 9.9845e-01],
        [8.7041e-08, 1.0000e+00],
        [1.2975e-05, 9.9999e-01],
        [9.9967e-01, 3.3080e-04],
        [2.2956e-05, 9.9998e-01],
        [1.7682e-08, 1.0000e+00],
        [1.3105e-05, 9.9999e-01],
        [1.2189e-08, 1.0000e+00],
        [1.0995e-06, 1.0000e+00],
        [2.1667e-06, 1.0000e+00],
        [5.2598e-07, 1.0000e+00],
        [9.9971e-01, 2.9292e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9538e-01, 4.6243e-03],
        [1.5201e-06, 1.0000e+00],
        [2.3257e-08, 1.0000e+00],
        [9.9993e-01, 7.4409e-05],
        [2.0154e-07, 1.0000e+00],
        [1.3850e-07, 1.0000e+00],
        [9.9932e-01, 6.8063e-04],
        [2.1373e-06, 1.0000e+00],
        [9.9984e-01, 1.6213e-04],
        [9.9994e-01, 6.1633e-05],
        [9.4921e-06, 9.9999e-01],
        [9.9993e-01, 7.4397e-05],
        [1.3429e-08, 1.0000e+00],
        [6.7878e-05, 9.9993e-01],
        [2.1476e-10, 1.0000e+00],
        [6.9596e-11, 1.0000e+00],
        [4.4481e-05, 9.9996e-01],
        [5.7419e-06, 9.9999e-01],
        [1.5943e-06, 1.0000e+00],
        [2.4168e-06, 1.0000e+00],
        [9.9971e-01, 2.9322e-04],
        [9.9979e-01, 2.1014e-04],
        [9.9977e-01, 2.2641e-04],
        [2.1684e-11, 1.0000e+00],
        [9.1069e-01, 8.9312e-02],
        [3.0786e-04, 9.9969e-01],
        [2.2975e-04, 9.9977e-01],
        [5.3910e-09, 1.0000e+00],
        [9.9990e-01, 9.8091e-05],
        [4.5023e-03, 9.9550e-01],
        [2.0717e-06, 1.0000e+00],
        [2.1058e-01, 7.8942e-01],
        [2.0174e-05, 9.9998e-01],
        [2.5758e-09, 1.0000e+00],
        [1.2408e-05, 9.9999e-01],
        [9.9967e-01, 3.2575e-04],
        [2.1349e-03, 9.9787e-01],
        [5.2085e-09, 1.0000e+00],
        [5.2683e-05, 9.9995e-01],
        [9.9935e-01, 6.4987e-04],
        [6.5255e-05, 9.9993e-01],
        [6.6885e-05, 9.9993e-01],
        [2.0967e-07, 1.0000e+00],
        [9.9718e-01, 2.8234e-03],
        [6.8239e-06, 9.9999e-01],
        [6.6916e-04, 9.9933e-01],
        [3.4798e-04, 9.9965e-01],
        [2.6295e-09, 1.0000e+00],
        [9.9991e-01, 9.2441e-05],
        [1.3631e-03, 9.9864e-01],
        [3.1804e-03, 9.9682e-01],
        [9.9991e-01, 8.9208e-05],
        [9.9939e-01, 6.0593e-04],
        [1.5649e-07, 1.0000e+00],
        [7.5210e-06, 9.9999e-01],
        [6.0525e-07, 1.0000e+00],
        [1.6375e-02, 9.8362e-01],
        [1.1749e-10, 1.0000e+00],
        [3.4976e-03, 9.9650e-01],
        [9.9630e-01, 3.6953e-03],
        [4.8226e-07, 1.0000e+00],
        [6.9218e-05, 9.9993e-01],
        [8.6297e-06, 9.9999e-01],
        [7.1556e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.3742e-03, 9.9463e-01],
        [5.4245e-08, 1.0000e+00],
        [4.0166e-06, 1.0000e+00],
        [9.9986e-01, 1.4201e-04],
        [6.8092e-10, 1.0000e+00],
        [9.9879e-01, 1.2128e-03],
        [9.9586e-07, 1.0000e+00],
        [1.4396e-04, 9.9986e-01],
        [4.1383e-08, 1.0000e+00],
        [9.7033e-08, 1.0000e+00],
        [9.9980e-01, 2.0333e-04],
        [1.1062e-06, 1.0000e+00],
        [3.6246e-04, 9.9964e-01],
        [7.5955e-04, 9.9924e-01],
        [1.3159e-12, 1.0000e+00],
        [2.1910e-08, 1.0000e+00],
        [8.0516e-06, 9.9999e-01],
        [9.9903e-01, 9.6562e-04],
        [2.8575e-05, 9.9997e-01],
        [6.5483e-07, 1.0000e+00],
        [9.9244e-01, 7.5606e-03],
        [5.5747e-13, 1.0000e+00],
        [2.2936e-03, 9.9771e-01],
        [7.8275e-04, 9.9922e-01],
        [1.0954e-07, 1.0000e+00],
        [5.5179e-01, 4.4821e-01],
        [9.7244e-12, 1.0000e+00],
        [9.9971e-01, 2.8505e-04],
        [9.9950e-01, 4.9904e-04],
        [1.3461e-08, 1.0000e+00],
        [2.3432e-06, 1.0000e+00],
        [1.0512e-10, 1.0000e+00],
        [8.2666e-05, 9.9992e-01],
        [2.1910e-08, 1.0000e+00],
        [3.2953e-03, 9.9670e-01],
        [9.2746e-03, 9.9073e-01],
        [2.3622e-09, 1.0000e+00],
        [1.3724e-04, 9.9986e-01],
        [4.0762e-10, 1.0000e+00],
        [1.4771e-07, 1.0000e+00],
        [3.3992e-04, 9.9966e-01],
        [1.9363e-05, 9.9998e-01],
        [7.1851e-10, 1.0000e+00],
        [3.6828e-06, 1.0000e+00],
        [3.9663e-04, 9.9960e-01],
        [9.9981e-01, 1.9414e-04],
        [1.1584e-03, 9.9884e-01],
        [5.6361e-05, 9.9994e-01],
        [4.8354e-06, 1.0000e+00],
        [2.8037e-06, 1.0000e+00],
        [1.1618e-07, 1.0000e+00],
        [1.6328e-04, 9.9984e-01],
        [9.3003e-03, 9.9070e-01],
        [5.8752e-04, 9.9941e-01],
        [9.9972e-01, 2.8353e-04],
        [9.9959e-01, 4.0776e-04],
        [5.8854e-01, 4.1146e-01],
        [4.0527e-06, 1.0000e+00],
        [9.9988e-01, 1.2386e-04],
        [5.1883e-05, 9.9995e-01],
        [1.7892e-04, 9.9982e-01],
        [6.9225e-07, 1.0000e+00],
        [9.7391e-07, 1.0000e+00],
        [7.5160e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5958e-03, 9.9840e-01],
        [9.9992e-01, 7.7019e-05],
        [4.9344e-03, 9.9507e-01],
        [1.9377e-03, 9.9806e-01],
        [1.0950e-08, 1.0000e+00],
        [1.0824e-09, 1.0000e+00],
        [2.1971e-05, 9.9998e-01],
        [9.9966e-01, 3.4185e-04],
        [9.9999e-01, 9.9177e-06],
        [8.4798e-01, 1.5202e-01],
        [6.5353e-01, 3.4647e-01],
        [6.4044e-01, 3.5956e-01],
        [9.9994e-01, 5.9952e-05],
        [1.2455e-05, 9.9999e-01],
        [1.1430e-04, 9.9989e-01],
        [9.9989e-01, 1.1489e-04],
        [9.9987e-01, 1.3228e-04],
        [1.0675e-06, 1.0000e+00],
        [1.1561e-04, 9.9988e-01],
        [3.0994e-02, 9.6901e-01],
        [9.4209e-09, 1.0000e+00],
        [1.1939e-05, 9.9999e-01],
        [2.6560e-12, 1.0000e+00],
        [3.6795e-01, 6.3205e-01],
        [1.1093e-06, 1.0000e+00],
        [1.0610e-01, 8.9390e-01],
        [5.4548e-07, 1.0000e+00],
        [1.4358e-06, 1.0000e+00],
        [1.7870e-07, 1.0000e+00],
        [2.9601e-05, 9.9997e-01],
        [1.1957e-01, 8.8043e-01],
        [9.9788e-06, 9.9999e-01],
        [7.5835e-09, 1.0000e+00],
        [1.7768e-04, 9.9982e-01],
        [5.4232e-04, 9.9946e-01],
        [3.1635e-06, 1.0000e+00],
        [2.5474e-07, 1.0000e+00],
        [8.0682e-01, 1.9318e-01],
        [9.9982e-01, 1.8167e-04],
        [1.4243e-04, 9.9986e-01],
        [9.7998e-05, 9.9990e-01],
        [9.9983e-01, 1.6517e-04],
        [5.8251e-09, 1.0000e+00],
        [1.6248e-03, 9.9838e-01],
        [9.9986e-01, 1.4148e-04],
        [5.4107e-04, 9.9946e-01],
        [1.0345e-04, 9.9990e-01],
        [1.7416e-04, 9.9983e-01],
        [9.8895e-01, 1.1052e-02],
        [5.9754e-10, 1.0000e+00],
        [1.0438e-05, 9.9999e-01],
        [9.9971e-01, 2.8551e-04],
        [9.4850e-06, 9.9999e-01],
        [8.9638e-01, 1.0362e-01],
        [3.2465e-09, 1.0000e+00],
        [2.8367e-05, 9.9997e-01],
        [1.3117e-03, 9.9869e-01],
        [5.0385e-11, 1.0000e+00],
        [8.5107e-04, 9.9915e-01],
        [4.3107e-04, 9.9957e-01],
        [7.0822e-05, 9.9993e-01],
        [2.2271e-04, 9.9978e-01],
        [2.3326e-07, 1.0000e+00],
        [3.7407e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.6174e-08, 1.0000e+00],
        [2.1016e-06, 1.0000e+00],
        [9.9937e-01, 6.2780e-04],
        [4.7616e-08, 1.0000e+00],
        [2.5333e-03, 9.9747e-01],
        [3.2419e-06, 1.0000e+00],
        [3.7822e-08, 1.0000e+00],
        [1.1092e-05, 9.9999e-01],
        [7.6482e-06, 9.9999e-01],
        [5.8194e-07, 1.0000e+00],
        [2.0002e-04, 9.9980e-01],
        [1.9388e-04, 9.9981e-01],
        [9.9961e-01, 3.9259e-04],
        [7.3520e-04, 9.9926e-01],
        [9.9362e-01, 6.3832e-03],
        [9.9998e-01, 2.4768e-05],
        [4.2032e-09, 1.0000e+00],
        [9.9904e-01, 9.5688e-04],
        [9.9871e-01, 1.2880e-03],
        [5.6324e-03, 9.9437e-01],
        [3.4731e-07, 1.0000e+00],
        [1.1863e-05, 9.9999e-01],
        [2.6414e-09, 1.0000e+00],
        [2.0900e-08, 1.0000e+00],
        [2.4017e-05, 9.9998e-01],
        [9.9999e-01, 1.3585e-05],
        [1.4929e-06, 1.0000e+00],
        [2.1993e-13, 1.0000e+00],
        [9.0000e-08, 1.0000e+00],
        [2.4194e-07, 1.0000e+00],
        [9.9993e-01, 6.7466e-05],
        [7.3024e-08, 1.0000e+00],
        [9.9989e-01, 1.1241e-04],
        [9.9869e-01, 1.3142e-03],
        [1.9093e-05, 9.9998e-01],
        [1.3532e-07, 1.0000e+00],
        [9.9819e-01, 1.8106e-03],
        [1.4557e-02, 9.8544e-01],
        [1.8060e-06, 1.0000e+00],
        [1.3763e-07, 1.0000e+00],
        [4.3880e-06, 1.0000e+00],
        [7.1843e-03, 9.9282e-01],
        [6.0665e-02, 9.3933e-01],
        [2.5497e-04, 9.9975e-01],
        [1.0363e-04, 9.9990e-01],
        [1.0338e-09, 1.0000e+00],
        [9.9991e-01, 8.7075e-05],
        [4.9100e-07, 1.0000e+00],
        [9.9975e-01, 2.4993e-04],
        [1.8277e-09, 1.0000e+00],
        [1.1798e-07, 1.0000e+00],
        [1.7598e-09, 1.0000e+00],
        [9.8176e-01, 1.8240e-02],
        [7.2451e-05, 9.9993e-01],
        [1.1613e-05, 9.9999e-01],
        [6.2041e-05, 9.9994e-01],
        [9.9990e-01, 9.6717e-05],
        [9.9990e-01, 1.0164e-04],
        [1.0491e-04, 9.9990e-01],
        [4.1212e-06, 1.0000e+00],
        [9.6032e-01, 3.9682e-02],
        [2.6840e-07, 1.0000e+00],
        [9.9993e-01, 7.2418e-05],
        [2.4816e-04, 9.9975e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3707e-08, 1.0000e+00],
        [9.4256e-10, 1.0000e+00],
        [4.6925e-07, 1.0000e+00],
        [5.2061e-07, 1.0000e+00],
        [2.0502e-03, 9.9795e-01],
        [4.0623e-04, 9.9959e-01],
        [4.4089e-04, 9.9956e-01],
        [9.9987e-01, 1.2936e-04],
        [3.5498e-08, 1.0000e+00],
        [4.5818e-04, 9.9954e-01],
        [9.9989e-01, 1.1059e-04],
        [6.5665e-02, 9.3433e-01],
        [4.3880e-07, 1.0000e+00],
        [7.1736e-06, 9.9999e-01],
        [9.6163e-04, 9.9904e-01],
        [9.9952e-01, 4.7558e-04],
        [2.0489e-06, 1.0000e+00],
        [9.9987e-01, 1.2586e-04],
        [9.9046e-07, 1.0000e+00],
        [6.2817e-08, 1.0000e+00],
        [4.8615e-08, 1.0000e+00],
        [9.9988e-01, 1.1739e-04],
        [4.5288e-07, 1.0000e+00],
        [9.9979e-01, 2.1261e-04],
        [1.8433e-05, 9.9998e-01],
        [9.9941e-01, 5.9092e-04],
        [5.6508e-06, 9.9999e-01],
        [9.9954e-01, 4.6326e-04],
        [4.9054e-09, 1.0000e+00],
        [1.3269e-05, 9.9999e-01],
        [1.8121e-07, 1.0000e+00],
        [3.5046e-11, 1.0000e+00],
        [9.9376e-01, 6.2398e-03],
        [1.4121e-07, 1.0000e+00],
        [9.9167e-01, 8.3262e-03],
        [9.2443e-05, 9.9991e-01],
        [5.4785e-11, 1.0000e+00],
        [2.4964e-04, 9.9975e-01],
        [5.6949e-07, 1.0000e+00],
        [3.1339e-09, 1.0000e+00],
        [5.2792e-05, 9.9995e-01],
        [1.4366e-03, 9.9856e-01],
        [9.0398e-03, 9.9096e-01],
        [9.9988e-01, 1.1811e-04],
        [1.8897e-09, 1.0000e+00],
        [2.2154e-04, 9.9978e-01],
        [9.9416e-01, 5.8440e-03],
        [2.9457e-08, 1.0000e+00],
        [8.5886e-06, 9.9999e-01],
        [9.5976e-01, 4.0235e-02],
        [3.5415e-05, 9.9996e-01],
        [2.1383e-04, 9.9979e-01],
        [9.9991e-01, 9.1820e-05],
        [8.9188e-01, 1.0812e-01],
        [5.0807e-05, 9.9995e-01],
        [1.0006e-07, 1.0000e+00],
        [9.9994e-01, 5.9527e-05],
        [7.6556e-07, 1.0000e+00],
        [4.0136e-06, 1.0000e+00],
        [2.1074e-04, 9.9979e-01],
        [2.3668e-05, 9.9998e-01],
        [5.3048e-05, 9.9995e-01],
        [2.7212e-04, 9.9973e-01],
        [1.7088e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5482e-06, 1.0000e+00],
        [5.5921e-07, 1.0000e+00],
        [5.2148e-10, 1.0000e+00],
        [6.0977e-01, 3.9023e-01],
        [9.9994e-01, 5.7400e-05],
        [2.7812e-06, 1.0000e+00],
        [9.9996e-01, 4.0458e-05],
        [9.9999e-01, 1.2531e-05],
        [4.3883e-06, 1.0000e+00],
        [9.9999e-01, 7.0221e-06],
        [9.9974e-01, 2.6129e-04],
        [9.6985e-01, 3.0148e-02],
        [8.8173e-03, 9.9118e-01],
        [7.4257e-09, 1.0000e+00],
        [4.3928e-02, 9.5607e-01],
        [1.6330e-05, 9.9998e-01],
        [2.1088e-09, 1.0000e+00],
        [9.9888e-01, 1.1198e-03],
        [9.8206e-01, 1.7941e-02],
        [7.7619e-10, 1.0000e+00],
        [9.6958e-01, 3.0420e-02],
        [8.9567e-06, 9.9999e-01],
        [2.0678e-09, 1.0000e+00],
        [9.9938e-01, 6.1616e-04],
        [3.0418e-07, 1.0000e+00],
        [9.9991e-01, 9.1165e-05],
        [2.1195e-04, 9.9979e-01],
        [1.9102e-01, 8.0898e-01],
        [3.4510e-04, 9.9965e-01],
        [1.8398e-03, 9.9816e-01],
        [9.9718e-01, 2.8153e-03],
        [2.1752e-04, 9.9978e-01],
        [9.9751e-01, 2.4926e-03],
        [9.9949e-01, 5.0725e-04],
        [9.9988e-01, 1.2431e-04],
        [1.3489e-08, 1.0000e+00],
        [8.3515e-01, 1.6485e-01],
        [1.3841e-06, 1.0000e+00],
        [2.5286e-06, 1.0000e+00],
        [9.9986e-01, 1.4460e-04],
        [3.7457e-01, 6.2543e-01],
        [9.9956e-01, 4.4264e-04],
        [6.7900e-09, 1.0000e+00],
        [8.5974e-07, 1.0000e+00],
        [9.9988e-01, 1.1834e-04],
        [1.1855e-06, 1.0000e+00],
        [8.0539e-01, 1.9461e-01],
        [1.1641e-10, 1.0000e+00],
        [5.4269e-08, 1.0000e+00],
        [9.9993e-01, 6.6396e-05],
        [5.3817e-10, 1.0000e+00],
        [3.4489e-11, 1.0000e+00],
        [2.0057e-01, 7.9943e-01],
        [1.7409e-04, 9.9983e-01],
        [2.7010e-07, 1.0000e+00],
        [7.7374e-06, 9.9999e-01],
        [2.1646e-05, 9.9998e-01],
        [1.4742e-03, 9.9853e-01],
        [2.1794e-08, 1.0000e+00],
        [6.1929e-05, 9.9994e-01],
        [2.4997e-06, 1.0000e+00],
        [1.5923e-04, 9.9984e-01],
        [1.3069e-05, 9.9999e-01],
        [1.2108e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.1322e-07, 1.0000e+00],
        [7.1384e-06, 9.9999e-01],
        [2.5031e-03, 9.9750e-01],
        [8.1618e-08, 1.0000e+00],
        [2.4596e-01, 7.5404e-01],
        [9.9923e-01, 7.6592e-04],
        [1.5847e-03, 9.9842e-01],
        [5.3522e-04, 9.9946e-01],
        [9.9453e-01, 5.4748e-03],
        [1.2387e-13, 1.0000e+00],
        [6.7181e-03, 9.9328e-01],
        [2.0048e-07, 1.0000e+00],
        [9.9412e-01, 5.8784e-03],
        [4.1310e-08, 1.0000e+00],
        [1.4737e-05, 9.9999e-01],
        [3.8437e-04, 9.9962e-01],
        [2.2017e-04, 9.9978e-01],
        [3.3933e-08, 1.0000e+00],
        [2.3742e-03, 9.9763e-01],
        [4.4002e-09, 1.0000e+00],
        [3.0842e-08, 1.0000e+00],
        [9.9961e-01, 3.9100e-04],
        [9.9985e-01, 1.4682e-04],
        [6.1520e-04, 9.9938e-01],
        [8.5592e-01, 1.4408e-01],
        [9.9997e-01, 2.9636e-05],
        [2.3663e-06, 1.0000e+00],
        [4.5256e-02, 9.5474e-01],
        [3.0346e-05, 9.9997e-01],
        [6.7343e-05, 9.9993e-01],
        [1.7418e-05, 9.9998e-01],
        [1.0478e-07, 1.0000e+00],
        [9.9981e-01, 1.9420e-04],
        [9.9994e-01, 5.8510e-05],
        [1.5705e-05, 9.9998e-01],
        [3.8990e-05, 9.9996e-01],
        [7.7645e-06, 9.9999e-01],
        [1.3531e-05, 9.9999e-01],
        [1.0527e-09, 1.0000e+00],
        [1.2115e-08, 1.0000e+00],
        [5.3155e-09, 1.0000e+00],
        [8.1678e-05, 9.9992e-01],
        [9.9488e-01, 5.1151e-03],
        [2.4205e-06, 1.0000e+00],
        [7.4193e-06, 9.9999e-01],
        [4.7079e-10, 1.0000e+00],
        [1.6597e-05, 9.9998e-01],
        [8.9734e-01, 1.0266e-01],
        [9.9993e-01, 6.9599e-05],
        [2.1667e-05, 9.9998e-01],
        [4.0235e-09, 1.0000e+00],
        [3.8424e-05, 9.9996e-01],
        [1.0793e-06, 1.0000e+00],
        [6.3570e-08, 1.0000e+00],
        [5.6216e-03, 9.9438e-01],
        [7.3761e-06, 9.9999e-01],
        [1.7911e-06, 1.0000e+00],
        [2.1108e-05, 9.9998e-01],
        [9.7659e-07, 1.0000e+00],
        [6.1924e-07, 1.0000e+00],
        [3.0226e-10, 1.0000e+00],
        [3.1182e-04, 9.9969e-01],
        [1.2576e-06, 1.0000e+00],
        [2.3506e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.7865e-07, 1.0000e+00],
        [5.7839e-06, 9.9999e-01],
        [1.7739e-02, 9.8226e-01],
        [9.9984e-01, 1.6382e-04],
        [1.0274e-07, 1.0000e+00],
        [1.2877e-06, 1.0000e+00],
        [7.1210e-05, 9.9993e-01],
        [4.6401e-05, 9.9995e-01],
        [9.9938e-01, 6.2453e-04],
        [3.1760e-03, 9.9682e-01],
        [1.0138e-10, 1.0000e+00],
        [1.1748e-02, 9.8825e-01],
        [9.9742e-01, 2.5837e-03],
        [1.7130e-07, 1.0000e+00],
        [9.9959e-01, 4.0747e-04],
        [9.9977e-01, 2.2559e-04],
        [6.7492e-13, 1.0000e+00],
        [5.0478e-06, 9.9999e-01],
        [3.7695e-05, 9.9996e-01],
        [6.8226e-04, 9.9932e-01],
        [1.1655e-03, 9.9883e-01],
        [3.8489e-05, 9.9996e-01],
        [1.6787e-06, 1.0000e+00],
        [5.2912e-04, 9.9947e-01],
        [5.8613e-09, 1.0000e+00],
        [4.9238e-02, 9.5076e-01],
        [3.0114e-07, 1.0000e+00],
        [9.9927e-01, 7.2923e-04],
        [9.9979e-01, 2.1179e-04],
        [5.9027e-04, 9.9941e-01],
        [9.9724e-01, 2.7623e-03],
        [6.9119e-06, 9.9999e-01],
        [9.9982e-01, 1.7783e-04],
        [9.9996e-01, 4.4721e-05],
        [9.9997e-01, 2.5675e-05],
        [1.0780e-02, 9.8922e-01],
        [3.1002e-04, 9.9969e-01],
        [1.1714e-08, 1.0000e+00],
        [2.7519e-13, 1.0000e+00],
        [1.1555e-10, 1.0000e+00],
        [1.0899e-02, 9.8910e-01],
        [8.3570e-03, 9.9164e-01],
        [7.8041e-04, 9.9922e-01],
        [9.9988e-01, 1.2459e-04],
        [4.6525e-06, 1.0000e+00],
        [3.2218e-04, 9.9968e-01],
        [8.1675e-10, 1.0000e+00],
        [1.7267e-07, 1.0000e+00],
        [3.8570e-06, 1.0000e+00],
        [3.2872e-06, 1.0000e+00],
        [9.9985e-01, 1.4696e-04],
        [4.4206e-05, 9.9996e-01],
        [8.3636e-03, 9.9164e-01],
        [2.5568e-01, 7.4432e-01],
        [1.2035e-04, 9.9988e-01],
        [1.1376e-03, 9.9886e-01],
        [9.8241e-01, 1.7591e-02],
        [8.6559e-01, 1.3441e-01],
        [4.9177e-04, 9.9951e-01],
        [4.8481e-05, 9.9995e-01],
        [4.8577e-08, 1.0000e+00],
        [2.0173e-04, 9.9980e-01],
        [9.7047e-01, 2.9532e-02],
        [4.7533e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.0168e-06, 9.9999e-01],
        [2.9271e-06, 1.0000e+00],
        [4.3450e-06, 1.0000e+00],
        [5.4208e-05, 9.9995e-01],
        [9.9994e-01, 5.8328e-05],
        [9.9992e-01, 8.1858e-05],
        [2.6526e-07, 1.0000e+00],
        [9.9988e-01, 1.1679e-04],
        [2.3294e-04, 9.9977e-01],
        [9.9998e-01, 2.0574e-05],
        [9.9980e-01, 2.0176e-04],
        [5.8384e-05, 9.9994e-01],
        [1.6347e-07, 1.0000e+00],
        [5.1382e-07, 1.0000e+00],
        [9.9992e-01, 7.9364e-05],
        [4.1101e-04, 9.9959e-01],
        [3.0770e-01, 6.9230e-01],
        [1.1188e-07, 1.0000e+00],
        [3.6400e-04, 9.9964e-01],
        [1.5765e-04, 9.9984e-01],
        [4.3015e-08, 1.0000e+00],
        [2.1697e-07, 1.0000e+00],
        [9.9948e-01, 5.1986e-04],
        [3.3498e-04, 9.9967e-01],
        [4.2495e-06, 1.0000e+00],
        [9.3718e-09, 1.0000e+00],
        [8.9417e-03, 9.9106e-01],
        [9.9885e-01, 1.1482e-03],
        [1.3191e-07, 1.0000e+00],
        [4.7549e-08, 1.0000e+00],
        [9.9995e-01, 5.3809e-05],
        [2.8518e-08, 1.0000e+00],
        [2.0105e-07, 1.0000e+00],
        [9.9923e-01, 7.6886e-04],
        [2.6099e-09, 1.0000e+00],
        [2.0522e-05, 9.9998e-01],
        [1.8721e-09, 1.0000e+00],
        [3.5071e-04, 9.9965e-01],
        [9.2099e-08, 1.0000e+00],
        [4.9857e-08, 1.0000e+00],
        [1.6682e-07, 1.0000e+00],
        [1.1410e-04, 9.9989e-01],
        [5.9082e-05, 9.9994e-01],
        [6.5324e-10, 1.0000e+00],
        [3.0714e-07, 1.0000e+00],
        [6.0102e-07, 1.0000e+00],
        [9.9985e-01, 1.4942e-04],
        [1.3105e-04, 9.9987e-01],
        [9.9996e-01, 4.4678e-05],
        [2.1353e-05, 9.9998e-01],
        [1.6480e-08, 1.0000e+00],
        [9.6062e-01, 3.9377e-02],
        [3.2145e-07, 1.0000e+00],
        [1.1730e-04, 9.9988e-01],
        [2.4836e-06, 1.0000e+00],
        [2.7449e-09, 1.0000e+00],
        [2.6099e-09, 1.0000e+00],
        [9.3749e-04, 9.9906e-01],
        [2.7606e-07, 1.0000e+00],
        [9.9997e-01, 3.0329e-05],
        [2.0617e-10, 1.0000e+00],
        [1.2344e-05, 9.9999e-01],
        [6.0112e-07, 1.0000e+00],
        [1.5607e-02, 9.8439e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.4583e-03, 9.9654e-01],
        [4.2515e-02, 9.5749e-01],
        [9.8268e-05, 9.9990e-01],
        [9.6312e-01, 3.6882e-02],
        [4.0834e-06, 1.0000e+00],
        [1.5935e-08, 1.0000e+00],
        [9.9966e-01, 3.3619e-04],
        [1.0179e-01, 8.9821e-01],
        [7.2339e-10, 1.0000e+00],
        [2.6275e-04, 9.9974e-01],
        [9.9995e-01, 5.2362e-05],
        [9.9975e-01, 2.4583e-04],
        [9.9980e-01, 1.9728e-04],
        [1.2478e-04, 9.9988e-01],
        [5.7090e-02, 9.4291e-01],
        [5.9358e-09, 1.0000e+00],
        [4.8667e-08, 1.0000e+00],
        [1.0498e-04, 9.9989e-01],
        [9.7981e-01, 2.0190e-02],
        [1.5551e-06, 1.0000e+00],
        [1.5754e-04, 9.9984e-01],
        [2.3613e-09, 1.0000e+00],
        [1.0322e-04, 9.9990e-01],
        [1.6898e-06, 1.0000e+00],
        [2.5474e-08, 1.0000e+00],
        [5.3761e-05, 9.9995e-01],
        [2.7378e-03, 9.9726e-01],
        [9.9678e-01, 3.2202e-03],
        [3.9894e-07, 1.0000e+00],
        [4.5098e-08, 1.0000e+00],
        [9.9995e-01, 5.1031e-05],
        [1.6771e-06, 1.0000e+00],
        [9.9994e-01, 6.1561e-05],
        [1.0013e-08, 1.0000e+00],
        [4.3430e-08, 1.0000e+00],
        [6.7755e-07, 1.0000e+00],
        [9.9984e-01, 1.5670e-04],
        [9.9990e-01, 9.8998e-05],
        [3.3501e-09, 1.0000e+00],
        [3.3671e-05, 9.9997e-01],
        [8.8648e-04, 9.9911e-01],
        [9.4857e-06, 9.9999e-01],
        [2.5010e-08, 1.0000e+00],
        [5.1385e-07, 1.0000e+00],
        [3.4647e-04, 9.9965e-01],
        [9.8748e-01, 1.2523e-02],
        [5.1715e-04, 9.9948e-01],
        [3.8698e-07, 1.0000e+00],
        [6.2855e-05, 9.9994e-01],
        [9.9999e-01, 6.9986e-06],
        [1.4041e-01, 8.5959e-01],
        [8.7817e-08, 1.0000e+00],
        [2.3507e-04, 9.9976e-01],
        [1.9906e-03, 9.9801e-01],
        [3.0862e-06, 1.0000e+00],
        [1.0513e-05, 9.9999e-01],
        [8.9650e-10, 1.0000e+00],
        [9.9970e-01, 3.0221e-04],
        [2.8503e-04, 9.9971e-01],
        [1.4451e-04, 9.9986e-01],
        [1.5525e-08, 1.0000e+00],
        [2.5326e-04, 9.9975e-01],
        [3.1558e-05, 9.9997e-01],
        [3.9941e-01, 6.0059e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.8803e-05, 9.9997e-01],
        [4.0618e-07, 1.0000e+00],
        [9.9998e-01, 2.2560e-05],
        [5.7933e-07, 1.0000e+00],
        [2.5432e-12, 1.0000e+00],
        [9.9984e-01, 1.6103e-04],
        [2.7182e-07, 1.0000e+00],
        [1.4217e-01, 8.5783e-01],
        [5.3296e-08, 1.0000e+00],
        [5.8552e-09, 1.0000e+00],
        [3.7078e-05, 9.9996e-01],
        [5.2496e-09, 1.0000e+00],
        [1.4386e-03, 9.9856e-01],
        [6.7016e-03, 9.9330e-01],
        [7.9860e-01, 2.0140e-01],
        [2.4593e-06, 1.0000e+00],
        [8.1723e-08, 1.0000e+00],
        [9.3776e-05, 9.9991e-01],
        [1.1817e-06, 1.0000e+00],
        [4.0795e-05, 9.9996e-01],
        [1.9940e-06, 1.0000e+00],
        [9.9990e-01, 9.6038e-05],
        [9.9977e-01, 2.3182e-04],
        [5.7171e-04, 9.9943e-01],
        [2.0906e-09, 1.0000e+00],
        [1.3217e-09, 1.0000e+00],
        [4.7019e-07, 1.0000e+00],
        [8.2363e-05, 9.9992e-01],
        [4.1952e-04, 9.9958e-01],
        [3.1285e-07, 1.0000e+00],
        [2.9438e-06, 1.0000e+00],
        [3.5543e-07, 1.0000e+00],
        [3.3063e-01, 6.6937e-01],
        [9.4773e-10, 1.0000e+00],
        [2.7049e-04, 9.9973e-01],
        [9.9984e-01, 1.5564e-04],
        [7.6979e-06, 9.9999e-01],
        [5.3039e-03, 9.9470e-01],
        [1.2292e-08, 1.0000e+00],
        [6.1149e-09, 1.0000e+00],
        [7.8236e-10, 1.0000e+00],
        [9.9998e-01, 1.7645e-05],
        [9.6947e-05, 9.9990e-01],
        [1.3463e-09, 1.0000e+00],
        [1.5096e-03, 9.9849e-01],
        [9.5552e-01, 4.4478e-02],
        [8.5522e-01, 1.4478e-01],
        [9.7576e-12, 1.0000e+00],
        [7.9230e-05, 9.9992e-01],
        [4.9278e-06, 1.0000e+00],
        [8.9736e-06, 9.9999e-01],
        [2.6291e-06, 1.0000e+00],
        [9.9992e-01, 8.1127e-05],
        [1.3833e-01, 8.6167e-01],
        [9.9456e-01, 5.4430e-03],
        [1.2920e-07, 1.0000e+00],
        [2.6504e-08, 1.0000e+00],
        [1.7624e-04, 9.9982e-01],
        [1.1232e-06, 1.0000e+00],
        [1.1346e-08, 1.0000e+00],
        [8.1560e-07, 1.0000e+00],
        [3.4909e-06, 1.0000e+00],
        [9.9981e-01, 1.8828e-04],
        [9.9828e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9874e-01, 1.2572e-03],
        [3.3303e-09, 1.0000e+00],
        [1.8242e-05, 9.9998e-01],
        [2.7900e-05, 9.9997e-01],
        [1.2962e-08, 1.0000e+00],
        [2.3908e-10, 1.0000e+00],
        [3.4832e-10, 1.0000e+00],
        [1.3946e-03, 9.9861e-01],
        [2.9668e-06, 1.0000e+00],
        [7.2583e-10, 1.0000e+00],
        [1.2461e-05, 9.9999e-01],
        [6.1487e-09, 1.0000e+00],
        [5.3472e-05, 9.9995e-01],
        [3.0606e-05, 9.9997e-01],
        [3.4476e-02, 9.6552e-01],
        [8.0979e-08, 1.0000e+00],
        [2.6180e-11, 1.0000e+00],
        [9.9994e-01, 5.9090e-05],
        [1.8774e-04, 9.9981e-01],
        [1.9752e-04, 9.9980e-01],
        [7.9256e-07, 1.0000e+00],
        [9.9983e-01, 1.6904e-04],
        [8.2389e-05, 9.9992e-01],
        [4.3435e-10, 1.0000e+00],
        [9.9998e-01, 2.1065e-05],
        [6.1169e-10, 1.0000e+00],
        [8.7319e-01, 1.2681e-01],
        [1.1630e-07, 1.0000e+00],
        [5.3406e-10, 1.0000e+00],
        [8.2524e-08, 1.0000e+00],
        [1.2606e-02, 9.8739e-01],
        [9.9992e-01, 8.4852e-05],
        [1.5498e-09, 1.0000e+00],
        [4.1149e-09, 1.0000e+00],
        [2.3175e-09, 1.0000e+00],
        [2.7295e-10, 1.0000e+00],
        [1.3567e-05, 9.9999e-01],
        [9.8937e-05, 9.9990e-01],
        [2.2063e-05, 9.9998e-01],
        [7.7085e-06, 9.9999e-01],
        [1.6698e-07, 1.0000e+00],
        [7.4923e-06, 9.9999e-01],
        [2.3343e-06, 1.0000e+00],
        [5.2497e-06, 9.9999e-01],
        [1.2177e-02, 9.8782e-01],
        [3.4713e-06, 1.0000e+00],
        [9.9934e-01, 6.5879e-04],
        [9.9992e-01, 7.9653e-05],
        [3.8635e-05, 9.9996e-01],
        [9.9994e-01, 6.2176e-05],
        [2.8793e-03, 9.9712e-01],
        [1.3464e-05, 9.9999e-01],
        [2.2364e-07, 1.0000e+00],
        [1.7423e-10, 1.0000e+00],
        [1.7229e-07, 1.0000e+00],
        [5.5563e-02, 9.4444e-01],
        [8.4290e-10, 1.0000e+00],
        [2.1129e-05, 9.9998e-01],
        [1.4623e-07, 1.0000e+00],
        [9.9990e-01, 9.7651e-05],
        [3.4883e-01, 6.5117e-01],
        [9.9860e-01, 1.4011e-03],
        [5.9895e-07, 1.0000e+00],
        [1.0303e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9479e-01, 5.2148e-03],
        [4.8243e-12, 1.0000e+00],
        [8.4211e-10, 1.0000e+00],
        [2.8912e-07, 1.0000e+00],
        [6.4438e-04, 9.9936e-01],
        [3.4648e-05, 9.9997e-01],
        [1.2785e-04, 9.9987e-01],
        [1.5138e-07, 1.0000e+00],
        [9.9993e-01, 7.2683e-05],
        [9.9995e-01, 5.0858e-05],
        [7.2279e-07, 1.0000e+00],
        [1.4675e-05, 9.9999e-01],
        [1.2374e-05, 9.9999e-01],
        [9.9998e-01, 2.4825e-05],
        [3.1840e-07, 1.0000e+00],
        [1.4718e-06, 1.0000e+00],
        [9.9998e-01, 2.4398e-05],
        [1.7255e-02, 9.8274e-01],
        [4.8835e-08, 1.0000e+00],
        [8.1761e-09, 1.0000e+00],
        [2.5430e-06, 1.0000e+00],
        [3.3982e-07, 1.0000e+00],
        [4.8243e-12, 1.0000e+00],
        [5.6048e-06, 9.9999e-01],
        [2.3188e-03, 9.9768e-01],
        [6.0248e-07, 1.0000e+00],
        [2.0585e-07, 1.0000e+00],
        [9.9983e-01, 1.6586e-04],
        [4.9494e-09, 1.0000e+00],
        [3.4152e-07, 1.0000e+00],
        [9.9064e-08, 1.0000e+00],
        [2.8456e-01, 7.1544e-01],
        [1.7789e-07, 1.0000e+00],
        [4.8879e-08, 1.0000e+00],
        [3.3446e-02, 9.6655e-01],
        [5.4585e-08, 1.0000e+00],
        [2.3647e-07, 1.0000e+00],
        [3.6521e-05, 9.9996e-01],
        [2.2143e-06, 1.0000e+00],
        [8.8488e-10, 1.0000e+00],
        [1.5704e-02, 9.8430e-01],
        [9.9996e-01, 3.6365e-05],
        [4.8681e-04, 9.9951e-01],
        [1.7402e-02, 9.8260e-01],
        [6.1630e-04, 9.9938e-01],
        [1.2018e-03, 9.9880e-01],
        [2.2145e-10, 1.0000e+00],
        [3.9238e-06, 1.0000e+00],
        [3.2831e-06, 1.0000e+00],
        [4.5273e-04, 9.9955e-01],
        [2.6310e-08, 1.0000e+00],
        [9.9987e-01, 1.3150e-04],
        [9.9971e-01, 2.9498e-04],
        [3.6387e-07, 1.0000e+00],
        [1.9119e-08, 1.0000e+00],
        [1.2667e-07, 1.0000e+00],
        [4.1184e-05, 9.9996e-01],
        [9.9914e-01, 8.5680e-04],
        [3.1107e-01, 6.8893e-01],
        [6.0346e-10, 1.0000e+00],
        [6.9239e-07, 1.0000e+00],
        [9.2180e-08, 1.0000e+00],
        [1.6128e-04, 9.9984e-01],
        [9.3839e-05, 9.9991e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9044e-03, 9.9810e-01],
        [1.0011e-06, 1.0000e+00],
        [1.1807e-07, 1.0000e+00],
        [9.9972e-01, 2.8257e-04],
        [5.7946e-07, 1.0000e+00],
        [3.6222e-05, 9.9996e-01],
        [3.8231e-10, 1.0000e+00],
        [4.3726e-10, 1.0000e+00],
        [2.0076e-09, 1.0000e+00],
        [6.9084e-02, 9.3092e-01],
        [1.4265e-09, 1.0000e+00],
        [9.9994e-01, 5.5768e-05],
        [9.9169e-10, 1.0000e+00],
        [9.9993e-01, 6.5052e-05],
        [9.9995e-01, 4.7567e-05],
        [2.9519e-04, 9.9970e-01],
        [9.9993e-01, 7.1038e-05],
        [7.6885e-04, 9.9923e-01],
        [1.3375e-10, 1.0000e+00],
        [2.8064e-06, 1.0000e+00],
        [2.2788e-05, 9.9998e-01],
        [5.6448e-07, 1.0000e+00],
        [9.9992e-01, 7.8753e-05],
        [8.3699e-07, 1.0000e+00],
        [9.9998e-01, 1.5253e-05],
        [9.9850e-01, 1.4963e-03],
        [1.0821e-02, 9.8918e-01],
        [9.9993e-01, 7.1486e-05],
        [4.1646e-07, 1.0000e+00],
        [1.0000e+00, 1.0100e-06],
        [9.9983e-01, 1.7445e-04],
        [1.3023e-03, 9.9870e-01],
        [4.4289e-07, 1.0000e+00],
        [2.9321e-04, 9.9971e-01],
        [4.4210e-08, 1.0000e+00],
        [9.9991e-01, 8.6338e-05],
        [9.9987e-01, 1.3111e-04],
        [1.2305e-06, 1.0000e+00],
        [2.6425e-09, 1.0000e+00],
        [4.3798e-01, 5.6202e-01],
        [9.9927e-01, 7.2587e-04],
        [1.6394e-02, 9.8361e-01],
        [4.5151e-01, 5.4849e-01],
        [9.9989e-01, 1.0501e-04],
        [7.7772e-04, 9.9922e-01],
        [4.7665e-06, 1.0000e+00],
        [1.6041e-05, 9.9998e-01],
        [1.4007e-09, 1.0000e+00],
        [9.4590e-07, 1.0000e+00],
        [9.8912e-01, 1.0878e-02],
        [7.7315e-07, 1.0000e+00],
        [1.3011e-08, 1.0000e+00],
        [8.4927e-01, 1.5073e-01],
        [4.7864e-09, 1.0000e+00],
        [9.9990e-01, 9.7519e-05],
        [4.9996e-06, 9.9999e-01],
        [3.2932e-02, 9.6707e-01],
        [9.4464e-08, 1.0000e+00],
        [5.9830e-08, 1.0000e+00],
        [6.8695e-06, 9.9999e-01],
        [1.7459e-09, 1.0000e+00],
        [6.2875e-06, 9.9999e-01],
        [8.3569e-10, 1.0000e+00],
        [1.3974e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9976e-01, 2.4278e-04],
        [7.4066e-07, 1.0000e+00],
        [2.4953e-04, 9.9975e-01],
        [2.3533e-06, 1.0000e+00],
        [6.9858e-05, 9.9993e-01],
        [1.1218e-06, 1.0000e+00],
        [3.7129e-07, 1.0000e+00],
        [2.7180e-05, 9.9997e-01],
        [9.9751e-01, 2.4938e-03],
        [1.3807e-05, 9.9999e-01],
        [9.9997e-01, 2.5606e-05],
        [1.2741e-08, 1.0000e+00],
        [9.9892e-01, 1.0797e-03],
        [9.9942e-01, 5.7563e-04],
        [3.1121e-07, 1.0000e+00],
        [5.8487e-08, 1.0000e+00],
        [1.2739e-11, 1.0000e+00],
        [9.9989e-01, 1.1278e-04],
        [9.9980e-01, 2.0191e-04],
        [2.4283e-03, 9.9757e-01],
        [9.9972e-01, 2.8191e-04],
        [9.9997e-01, 2.8547e-05],
        [5.4292e-06, 9.9999e-01],
        [5.2305e-10, 1.0000e+00],
        [9.9563e-01, 4.3720e-03],
        [1.4665e-11, 1.0000e+00],
        [1.7937e-09, 1.0000e+00],
        [6.7309e-06, 9.9999e-01],
        [3.5958e-07, 1.0000e+00],
        [9.1282e-06, 9.9999e-01],
        [9.9570e-01, 4.2951e-03],
        [8.2240e-09, 1.0000e+00],
        [7.1994e-02, 9.2801e-01],
        [1.0532e-04, 9.9989e-01],
        [1.0898e-08, 1.0000e+00],
        [2.3006e-11, 1.0000e+00],
        [4.5808e-01, 5.4192e-01],
        [8.7832e-01, 1.2167e-01],
        [9.9903e-01, 9.7211e-04],
        [8.7794e-10, 1.0000e+00],
        [2.8139e-04, 9.9972e-01],
        [6.4122e-06, 9.9999e-01],
        [4.0622e-09, 1.0000e+00],
        [9.8845e-01, 1.1550e-02],
        [5.5451e-03, 9.9445e-01],
        [1.0675e-08, 1.0000e+00],
        [9.9256e-01, 7.4401e-03],
        [1.0092e-07, 1.0000e+00],
        [6.2678e-08, 1.0000e+00],
        [9.7925e-05, 9.9990e-01],
        [1.1625e-08, 1.0000e+00],
        [8.7782e-05, 9.9991e-01],
        [1.0826e-08, 1.0000e+00],
        [9.9399e-01, 6.0108e-03],
        [9.9980e-01, 1.9765e-04],
        [1.0792e-04, 9.9989e-01],
        [2.0520e-08, 1.0000e+00],
        [9.9992e-01, 8.1616e-05],
        [1.8349e-11, 1.0000e+00],
        [2.2051e-08, 1.0000e+00],
        [2.4078e-06, 1.0000e+00],
        [7.5271e-09, 1.0000e+00],
        [1.5064e-09, 1.0000e+00],
        [9.9981e-01, 1.8629e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9873e-10, 1.0000e+00],
        [9.9996e-01, 3.8502e-05],
        [7.0995e-12, 1.0000e+00],
        [1.0226e-09, 1.0000e+00],
        [3.3425e-06, 1.0000e+00],
        [1.3939e-04, 9.9986e-01],
        [9.9996e-01, 4.1262e-05],
        [9.9969e-01, 3.0646e-04],
        [2.1695e-08, 1.0000e+00],
        [1.4683e-04, 9.9985e-01],
        [3.2860e-08, 1.0000e+00],
        [9.9960e-01, 3.9532e-04],
        [1.3206e-11, 1.0000e+00],
        [7.2334e-10, 1.0000e+00],
        [9.9994e-01, 5.6808e-05],
        [1.3733e-05, 9.9999e-01],
        [1.3341e-08, 1.0000e+00],
        [2.7874e-05, 9.9997e-01],
        [7.2977e-08, 1.0000e+00],
        [7.4085e-07, 1.0000e+00],
        [2.7677e-06, 1.0000e+00],
        [2.8959e-07, 1.0000e+00],
        [1.0614e-08, 1.0000e+00],
        [3.4682e-07, 1.0000e+00],
        [7.3337e-05, 9.9993e-01],
        [9.9975e-01, 2.5129e-04],
        [8.2116e-02, 9.1788e-01],
        [2.7809e-04, 9.9972e-01],
        [6.6069e-07, 1.0000e+00],
        [5.9438e-07, 1.0000e+00],
        [6.3172e-07, 1.0000e+00],
        [9.9944e-01, 5.6037e-04],
        [1.9001e-09, 1.0000e+00],
        [9.9998e-01, 1.6020e-05],
        [1.6362e-04, 9.9984e-01],
        [9.9279e-01, 7.2132e-03],
        [2.3912e-07, 1.0000e+00],
        [2.3800e-06, 1.0000e+00],
        [1.1881e-07, 1.0000e+00],
        [1.8725e-08, 1.0000e+00],
        [5.2406e-06, 9.9999e-01],
        [1.8701e-07, 1.0000e+00],
        [5.9523e-09, 1.0000e+00],
        [2.1039e-07, 1.0000e+00],
        [9.9983e-01, 1.6608e-04],
        [5.9537e-07, 1.0000e+00],
        [6.5169e-11, 1.0000e+00],
        [9.7731e-11, 1.0000e+00],
        [9.9989e-01, 1.0978e-04],
        [4.5588e-02, 9.5441e-01],
        [7.2544e-12, 1.0000e+00],
        [9.9988e-01, 1.2089e-04],
        [9.9926e-01, 7.4503e-04],
        [9.9986e-01, 1.3687e-04],
        [7.8377e-09, 1.0000e+00],
        [3.8602e-04, 9.9961e-01],
        [1.9876e-07, 1.0000e+00],
        [1.0819e-06, 1.0000e+00],
        [2.3524e-10, 1.0000e+00],
        [3.4527e-02, 9.6547e-01],
        [4.4456e-05, 9.9996e-01],
        [1.2095e-06, 1.0000e+00],
        [7.9013e-04, 9.9921e-01],
        [7.5986e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7060e-08, 1.0000e+00],
        [4.9975e-04, 9.9950e-01],
        [2.7544e-08, 1.0000e+00],
        [6.4448e-04, 9.9936e-01],
        [9.8517e-01, 1.4827e-02],
        [9.9971e-01, 2.8672e-04],
        [2.7106e-06, 1.0000e+00],
        [2.3072e-06, 1.0000e+00],
        [5.8764e-06, 9.9999e-01],
        [2.0540e-04, 9.9979e-01],
        [1.3064e-04, 9.9987e-01],
        [1.3537e-07, 1.0000e+00],
        [9.9989e-01, 1.1171e-04],
        [3.0111e-03, 9.9699e-01],
        [9.9982e-01, 1.7505e-04],
        [1.1688e-07, 1.0000e+00],
        [9.8255e-01, 1.7450e-02],
        [9.9006e-01, 9.9396e-03],
        [1.2323e-06, 1.0000e+00],
        [9.9994e-01, 6.2395e-05],
        [8.8530e-09, 1.0000e+00],
        [6.6035e-07, 1.0000e+00],
        [7.1379e-11, 1.0000e+00],
        [1.3351e-03, 9.9866e-01],
        [3.8103e-08, 1.0000e+00],
        [1.0403e-08, 1.0000e+00],
        [6.6037e-08, 1.0000e+00],
        [9.9986e-01, 1.3670e-04],
        [9.9885e-01, 1.1482e-03],
        [3.6112e-04, 9.9964e-01],
        [8.6969e-06, 9.9999e-01],
        [9.9886e-01, 1.1428e-03],
        [9.9987e-01, 1.2587e-04],
        [9.9991e-01, 8.5514e-05],
        [1.0736e-04, 9.9989e-01],
        [5.6527e-03, 9.9435e-01],
        [9.9985e-01, 1.4541e-04],
        [5.3809e-10, 1.0000e+00],
        [6.4028e-06, 9.9999e-01],
        [2.4163e-12, 1.0000e+00],
        [1.8032e-08, 1.0000e+00],
        [1.4596e-01, 8.5404e-01],
        [9.9934e-01, 6.6264e-04],
        [1.5829e-03, 9.9842e-01],
        [1.8718e-06, 1.0000e+00],
        [3.3555e-04, 9.9966e-01],
        [9.9984e-01, 1.6410e-04],
        [8.9799e-07, 1.0000e+00],
        [1.3047e-07, 1.0000e+00],
        [1.4071e-08, 1.0000e+00],
        [9.9990e-01, 1.0168e-04],
        [3.2371e-11, 1.0000e+00],
        [1.9571e-07, 1.0000e+00],
        [1.8388e-11, 1.0000e+00],
        [2.8298e-09, 1.0000e+00],
        [3.6684e-08, 1.0000e+00],
        [1.5332e-06, 1.0000e+00],
        [9.9954e-01, 4.6409e-04],
        [1.1377e-07, 1.0000e+00],
        [9.9989e-01, 1.0837e-04],
        [7.8193e-10, 1.0000e+00],
        [3.3639e-04, 9.9966e-01],
        [9.9750e-09, 1.0000e+00],
        [2.1265e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.8275e-07, 1.0000e+00],
        [1.2461e-03, 9.9875e-01],
        [7.3563e-06, 9.9999e-01],
        [1.3620e-07, 1.0000e+00],
        [2.9982e-08, 1.0000e+00],
        [9.9994e-01, 5.6948e-05],
        [1.4818e-04, 9.9985e-01],
        [4.5401e-04, 9.9955e-01],
        [9.6660e-11, 1.0000e+00],
        [1.8602e-03, 9.9814e-01],
        [3.0940e-08, 1.0000e+00],
        [5.3568e-07, 1.0000e+00],
        [1.2299e-07, 1.0000e+00],
        [2.3405e-02, 9.7659e-01],
        [8.2739e-11, 1.0000e+00],
        [9.9865e-01, 1.3502e-03],
        [2.0751e-08, 1.0000e+00],
        [2.1337e-04, 9.9979e-01],
        [9.9899e-01, 1.0090e-03],
        [9.9986e-01, 1.3828e-04],
        [6.6054e-03, 9.9339e-01],
        [9.8259e-01, 1.7406e-02],
        [9.9850e-01, 1.4976e-03],
        [9.5578e-08, 1.0000e+00],
        [9.8901e-01, 1.0988e-02],
        [2.9551e-08, 1.0000e+00],
        [3.1277e-05, 9.9997e-01],
        [9.9985e-01, 1.4597e-04],
        [1.9421e-08, 1.0000e+00],
        [1.8949e-03, 9.9811e-01],
        [9.9968e-01, 3.2368e-04],
        [4.6065e-07, 1.0000e+00],
        [9.9993e-01, 6.5560e-05],
        [2.1425e-04, 9.9979e-01],
        [1.2124e-09, 1.0000e+00],
        [2.0261e-09, 1.0000e+00],
        [3.0338e-07, 1.0000e+00],
        [2.1184e-12, 1.0000e+00],
        [6.7589e-07, 1.0000e+00],
        [9.2862e-08, 1.0000e+00],
        [4.4883e-06, 1.0000e+00],
        [1.2488e-05, 9.9999e-01],
        [6.1798e-08, 1.0000e+00],
        [3.9731e-04, 9.9960e-01],
        [5.9074e-05, 9.9994e-01],
        [1.1242e-08, 1.0000e+00],
        [1.4560e-08, 1.0000e+00],
        [2.6510e-06, 1.0000e+00],
        [2.1490e-08, 1.0000e+00],
        [9.2617e-07, 1.0000e+00],
        [8.8556e-04, 9.9911e-01],
        [9.9999e-01, 1.3213e-05],
        [2.6215e-01, 7.3785e-01],
        [9.9984e-01, 1.6159e-04],
        [8.7439e-06, 9.9999e-01],
        [4.4340e-06, 1.0000e+00],
        [6.6120e-06, 9.9999e-01],
        [5.8683e-05, 9.9994e-01],
        [1.5142e-08, 1.0000e+00],
        [1.5177e-07, 1.0000e+00],
        [9.9742e-01, 2.5775e-03],
        [7.6593e-07, 1.0000e+00],
        [4.0561e-06, 1.0000e+00],
        [5.4936e-12, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6454e-06, 1.0000e+00],
        [1.5931e-08, 1.0000e+00],
        [2.1071e-05, 9.9998e-01],
        [6.4020e-05, 9.9994e-01],
        [4.3665e-06, 1.0000e+00],
        [1.2316e-10, 1.0000e+00],
        [5.3341e-07, 1.0000e+00],
        [9.9992e-01, 8.3538e-05],
        [6.1097e-10, 1.0000e+00],
        [2.6943e-07, 1.0000e+00],
        [2.7169e-09, 1.0000e+00],
        [2.9398e-04, 9.9971e-01],
        [9.9995e-01, 4.5160e-05],
        [5.6331e-01, 4.3669e-01],
        [4.7175e-08, 1.0000e+00],
        [9.9995e-01, 5.4343e-05],
        [3.3135e-08, 1.0000e+00],
        [2.6063e-09, 1.0000e+00],
        [1.6257e-05, 9.9998e-01],
        [4.1993e-07, 1.0000e+00],
        [9.9718e-01, 2.8178e-03],
        [4.5417e-09, 1.0000e+00],
        [5.5311e-04, 9.9945e-01],
        [7.5173e-06, 9.9999e-01],
        [2.9485e-05, 9.9997e-01],
        [6.3624e-05, 9.9994e-01],
        [4.8619e-03, 9.9514e-01],
        [9.4161e-04, 9.9906e-01],
        [9.9984e-01, 1.6203e-04],
        [1.4000e-07, 1.0000e+00],
        [2.7373e-08, 1.0000e+00],
        [9.9264e-01, 7.3588e-03],
        [9.3533e-01, 6.4672e-02],
        [7.2490e-02, 9.2751e-01],
        [6.4695e-06, 9.9999e-01],
        [9.9993e-01, 7.4290e-05],
        [6.6004e-05, 9.9993e-01],
        [9.9997e-01, 2.9348e-05],
        [2.4740e-06, 1.0000e+00],
        [6.9471e-08, 1.0000e+00],
        [9.3705e-01, 6.2948e-02],
        [1.0084e-09, 1.0000e+00],
        [5.9825e-08, 1.0000e+00],
        [7.7438e-10, 1.0000e+00],
        [5.3552e-08, 1.0000e+00],
        [1.8080e-07, 1.0000e+00],
        [1.0375e-06, 1.0000e+00],
        [9.0241e-07, 1.0000e+00],
        [5.9388e-07, 1.0000e+00],
        [1.2865e-08, 1.0000e+00],
        [1.8382e-03, 9.9816e-01],
        [4.7930e-07, 1.0000e+00],
        [3.9212e-05, 9.9996e-01],
        [9.9985e-01, 1.5423e-04],
        [1.6243e-03, 9.9838e-01],
        [7.6828e-04, 9.9923e-01],
        [1.6652e-05, 9.9998e-01],
        [1.1503e-04, 9.9988e-01],
        [8.2581e-08, 1.0000e+00],
        [6.9856e-09, 1.0000e+00],
        [9.9980e-01, 2.0494e-04],
        [8.6293e-08, 1.0000e+00],
        [9.3039e-01, 6.9606e-02],
        [7.0130e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0000e+00, 1.0725e-07],
        [9.9989e-01, 1.0947e-04],
        [1.7095e-05, 9.9998e-01],
        [9.7536e-05, 9.9990e-01],
        [1.0697e-09, 1.0000e+00],
        [1.0056e-07, 1.0000e+00],
        [1.8054e-08, 1.0000e+00],
        [5.7253e-01, 4.2747e-01],
        [1.0000e+00, 1.0761e-06],
        [9.5996e-01, 4.0036e-02],
        [9.9689e-01, 3.1116e-03],
        [2.4079e-05, 9.9998e-01],
        [3.2579e-07, 1.0000e+00],
        [1.1852e-09, 1.0000e+00],
        [1.0789e-06, 1.0000e+00],
        [9.9980e-01, 1.9560e-04],
        [4.6526e-06, 1.0000e+00],
        [9.9977e-01, 2.2848e-04],
        [9.9991e-01, 8.7495e-05],
        [3.8487e-08, 1.0000e+00],
        [9.9973e-01, 2.7085e-04],
        [1.7178e-06, 1.0000e+00],
        [9.9965e-01, 3.4960e-04],
        [1.0162e-06, 1.0000e+00],
        [2.8094e-07, 1.0000e+00],
        [1.0899e-04, 9.9989e-01],
        [7.1877e-09, 1.0000e+00],
        [1.2421e-06, 1.0000e+00],
        [1.0000e+00, 4.9068e-07],
        [9.2857e-08, 1.0000e+00],
        [5.1229e-02, 9.4877e-01],
        [3.9333e-09, 1.0000e+00],
        [1.4163e-03, 9.9858e-01],
        [7.0574e-08, 1.0000e+00],
        [3.1835e-07, 1.0000e+00],
        [3.6543e-04, 9.9963e-01],
        [1.6656e-06, 1.0000e+00],
        [2.1795e-05, 9.9998e-01],
        [1.3427e-07, 1.0000e+00],
        [1.9108e-06, 1.0000e+00],
        [2.8888e-06, 1.0000e+00],
        [1.6393e-07, 1.0000e+00],
        [8.2641e-07, 1.0000e+00],
        [5.9891e-02, 9.4011e-01],
        [6.4104e-08, 1.0000e+00],
        [3.4895e-08, 1.0000e+00],
        [3.7313e-08, 1.0000e+00],
        [1.1033e-10, 1.0000e+00],
        [7.2321e-04, 9.9928e-01],
        [9.9993e-01, 6.5208e-05],
        [3.8301e-06, 1.0000e+00],
        [2.7142e-08, 1.0000e+00],
        [9.8309e-01, 1.6912e-02],
        [1.7457e-02, 9.8254e-01],
        [9.9461e-01, 5.3857e-03],
        [1.8845e-07, 1.0000e+00],
        [1.1117e-07, 1.0000e+00],
        [1.3168e-09, 1.0000e+00],
        [9.9678e-01, 3.2216e-03],
        [2.3244e-05, 9.9998e-01],
        [1.2801e-03, 9.9872e-01],
        [2.0823e-04, 9.9979e-01],
        [3.7501e-07, 1.0000e+00],
        [4.6926e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.2329e-09, 1.0000e+00],
        [9.9948e-01, 5.1914e-04],
        [2.8610e-09, 1.0000e+00],
        [4.9716e-10, 1.0000e+00],
        [9.9992e-01, 7.6810e-05],
        [5.7433e-03, 9.9426e-01],
        [6.2500e-06, 9.9999e-01],
        [7.2841e-01, 2.7159e-01],
        [2.2000e-05, 9.9998e-01],
        [9.9980e-01, 1.9889e-04],
        [9.9985e-01, 1.5014e-04],
        [1.2845e-09, 1.0000e+00],
        [4.1877e-06, 1.0000e+00],
        [1.6029e-10, 1.0000e+00],
        [8.6054e-01, 1.3946e-01],
        [1.2623e-10, 1.0000e+00],
        [1.7605e-06, 1.0000e+00],
        [9.9920e-01, 8.0270e-04],
        [1.2739e-05, 9.9999e-01],
        [4.0973e-05, 9.9996e-01],
        [2.6072e-06, 1.0000e+00],
        [4.3960e-08, 1.0000e+00],
        [1.5214e-08, 1.0000e+00],
        [7.3024e-09, 1.0000e+00],
        [2.0284e-08, 1.0000e+00],
        [5.0861e-05, 9.9995e-01],
        [4.8512e-02, 9.5149e-01],
        [2.0836e-05, 9.9998e-01],
        [1.1042e-01, 8.8958e-01],
        [1.4931e-07, 1.0000e+00],
        [9.9998e-01, 2.1851e-05],
        [2.4635e-09, 1.0000e+00],
        [9.9938e-01, 6.2044e-04],
        [7.2487e-01, 2.7513e-01],
        [8.5550e-10, 1.0000e+00],
        [6.1841e-04, 9.9938e-01],
        [5.2977e-10, 1.0000e+00],
        [1.0532e-04, 9.9989e-01],
        [2.3718e-05, 9.9998e-01],
        [7.4355e-05, 9.9993e-01],
        [7.0040e-01, 2.9960e-01],
        [1.0000e+00, 2.7379e-06],
        [9.8513e-01, 1.4872e-02],
        [9.9665e-01, 3.3508e-03],
        [9.9988e-01, 1.2048e-04],
        [1.3710e-03, 9.9863e-01],
        [1.4523e-06, 1.0000e+00],
        [7.5804e-07, 1.0000e+00],
        [9.9696e-01, 3.0413e-03],
        [3.2219e-08, 1.0000e+00],
        [9.9360e-01, 6.3968e-03],
        [1.6666e-05, 9.9998e-01],
        [9.7493e-01, 2.5070e-02],
        [1.2642e-05, 9.9999e-01],
        [4.0523e-05, 9.9996e-01],
        [4.9437e-09, 1.0000e+00],
        [1.0522e-06, 1.0000e+00],
        [6.7500e-01, 3.2500e-01],
        [4.2798e-08, 1.0000e+00],
        [1.9601e-08, 1.0000e+00],
        [9.9988e-01, 1.2401e-04],
        [7.6783e-11, 1.0000e+00],
        [8.4605e-08, 1.0000e+00],
        [1.6461e-10, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0709e-06, 1.0000e+00],
        [2.0438e-10, 1.0000e+00],
        [2.2155e-04, 9.9978e-01],
        [2.7319e-05, 9.9997e-01],
        [7.2597e-05, 9.9993e-01],
        [1.1508e-06, 1.0000e+00],
        [2.7290e-06, 1.0000e+00],
        [1.0134e-09, 1.0000e+00],
        [9.9982e-01, 1.7922e-04],
        [8.5652e-07, 1.0000e+00],
        [6.8666e-05, 9.9993e-01],
        [6.5102e-07, 1.0000e+00],
        [1.8033e-02, 9.8197e-01],
        [2.1986e-06, 1.0000e+00],
        [1.1535e-01, 8.8465e-01],
        [1.7743e-04, 9.9982e-01],
        [7.7046e-06, 9.9999e-01],
        [4.7420e-11, 1.0000e+00],
        [3.1521e-07, 1.0000e+00],
        [9.9972e-01, 2.8267e-04],
        [1.3657e-06, 1.0000e+00],
        [4.0394e-08, 1.0000e+00],
        [1.6282e-10, 1.0000e+00],
        [3.2134e-08, 1.0000e+00],
        [4.7434e-07, 1.0000e+00],
        [3.7155e-06, 1.0000e+00],
        [7.0443e-06, 9.9999e-01],
        [2.7752e-02, 9.7225e-01],
        [2.0539e-09, 1.0000e+00],
        [2.5710e-07, 1.0000e+00],
        [5.7253e-03, 9.9427e-01],
        [2.1786e-10, 1.0000e+00],
        [3.9932e-04, 9.9960e-01],
        [8.6185e-11, 1.0000e+00],
        [1.0348e-01, 8.9652e-01],
        [4.0420e-08, 1.0000e+00],
        [2.6317e-01, 7.3683e-01],
        [5.0389e-05, 9.9995e-01],
        [1.0688e-04, 9.9989e-01],
        [8.0991e-01, 1.9009e-01],
        [2.1390e-04, 9.9979e-01],
        [8.1545e-08, 1.0000e+00],
        [2.5304e-02, 9.7470e-01],
        [2.5612e-06, 1.0000e+00],
        [9.8312e-06, 9.9999e-01],
        [5.8091e-05, 9.9994e-01],
        [1.0540e-05, 9.9999e-01],
        [2.8846e-04, 9.9971e-01],
        [3.2489e-08, 1.0000e+00],
        [1.0515e-05, 9.9999e-01],
        [5.3606e-02, 9.4639e-01],
        [4.8493e-10, 1.0000e+00],
        [9.9987e-01, 1.2646e-04],
        [2.2566e-10, 1.0000e+00],
        [1.8995e-04, 9.9981e-01],
        [1.3415e-11, 1.0000e+00],
        [9.9950e-01, 4.9589e-04],
        [2.0870e-06, 1.0000e+00],
        [1.9056e-06, 1.0000e+00],
        [9.9985e-01, 1.4962e-04],
        [1.6116e-06, 1.0000e+00],
        [4.3287e-06, 1.0000e+00],
        [2.4943e-02, 9.7506e-01],
        [9.9990e-01, 1.0095e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[4.2590e-04, 9.9957e-01],
        [2.2193e-06, 1.0000e+00],
        [1.8228e-06, 1.0000e+00],
        [9.9995e-01, 5.1789e-05],
        [6.0431e-05, 9.9994e-01],
        [1.4518e-07, 1.0000e+00],
        [9.2888e-01, 7.1120e-02],
        [2.5877e-06, 1.0000e+00],
        [1.1598e-08, 1.0000e+00],
        [9.9994e-01, 6.2973e-05],
        [2.3157e-04, 9.9977e-01],
        [9.9976e-01, 2.4235e-04],
        [9.8109e-01, 1.8907e-02],
        [9.4826e-07, 1.0000e+00],
        [1.3224e-07, 1.0000e+00],
        [2.3101e-03, 9.9769e-01],
        [4.6454e-08, 1.0000e+00],
        [4.7513e-08, 1.0000e+00],
        [9.9995e-01, 5.0057e-05],
        [1.7394e-07, 1.0000e+00],
        [9.9965e-01, 3.5329e-04],
        [1.9134e-08, 1.0000e+00],
        [3.6103e-10, 1.0000e+00],
        [1.3630e-01, 8.6370e-01],
        [9.9994e-01, 6.4595e-05],
        [9.9842e-01, 1.5831e-03],
        [6.5721e-05, 9.9993e-01],
        [9.9845e-01, 1.5454e-03],
        [4.2946e-07, 1.0000e+00],
        [9.9986e-01, 1.4449e-04],
        [1.0000e+00, 1.5313e-06],
        [2.0642e-08, 1.0000e+00],
        [3.2186e-02, 9.6781e-01],
        [1.7176e-02, 9.8282e-01],
        [1.0677e-03, 9.9893e-01],
        [1.5754e-04, 9.9984e-01],
        [1.7607e-05, 9.9998e-01],
        [2.9806e-10, 1.0000e+00],
        [4.9927e-08, 1.0000e+00],
        [5.6948e-07, 1.0000e+00],
        [1.7461e-04, 9.9983e-01],
        [9.9838e-01, 1.6199e-03],
        [4.7790e-07, 1.0000e+00],
        [6.5444e-06, 9.9999e-01],
        [9.9987e-01, 1.3161e-04],
        [2.3151e-05, 9.9998e-01],
        [3.9174e-09, 1.0000e+00],
        [3.6182e-07, 1.0000e+00],
        [4.1995e-09, 1.0000e+00],
        [8.6283e-05, 9.9991e-01],
        [1.4630e-06, 1.0000e+00],
        [9.0265e-04, 9.9910e-01],
        [1.3559e-04, 9.9986e-01],
        [4.8066e-08, 1.0000e+00],
        [9.8359e-01, 1.6407e-02],
        [7.4531e-05, 9.9993e-01],
        [9.9968e-01, 3.2082e-04],
        [2.4881e-06, 1.0000e+00],
        [1.1345e-07, 1.0000e+00],
        [1.1717e-08, 1.0000e+00],
        [9.9995e-01, 4.6902e-05],
        [1.0744e-01, 8.9256e-01],
        [9.9754e-01, 2.4587e-03],
        [3.7367e-17, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7274e-06, 1.0000e+00],
        [9.9985e-01, 1.4815e-04],
        [1.3166e-06, 1.0000e+00],
        [9.9942e-01, 5.8294e-04],
        [3.6040e-08, 1.0000e+00],
        [1.6725e-04, 9.9983e-01],
        [1.1677e-09, 1.0000e+00],
        [4.9668e-09, 1.0000e+00],
        [1.2115e-06, 1.0000e+00],
        [9.9988e-01, 1.2237e-04],
        [9.9987e-01, 1.3243e-04],
        [5.4813e-12, 1.0000e+00],
        [1.7762e-08, 1.0000e+00],
        [9.9997e-01, 3.4044e-05],
        [9.9974e-01, 2.6255e-04],
        [1.4903e-08, 1.0000e+00],
        [9.9974e-01, 2.5563e-04],
        [9.9986e-01, 1.4022e-04],
        [2.5971e-11, 1.0000e+00],
        [5.6784e-06, 9.9999e-01],
        [2.9691e-10, 1.0000e+00],
        [7.3361e-08, 1.0000e+00],
        [2.5208e-07, 1.0000e+00],
        [9.5578e-01, 4.4216e-02],
        [9.9994e-01, 5.5286e-05],
        [6.8124e-06, 9.9999e-01],
        [9.9907e-01, 9.3427e-04],
        [4.2860e-07, 1.0000e+00],
        [1.2302e-06, 1.0000e+00],
        [2.4214e-11, 1.0000e+00],
        [9.9987e-01, 1.2637e-04],
        [1.8077e-05, 9.9998e-01],
        [6.3961e-09, 1.0000e+00],
        [9.9989e-01, 1.0751e-04],
        [5.2667e-06, 9.9999e-01],
        [1.8380e-04, 9.9982e-01],
        [1.5650e-07, 1.0000e+00],
        [6.6241e-07, 1.0000e+00],
        [1.1907e-08, 1.0000e+00],
        [1.0034e-09, 1.0000e+00],
        [1.3756e-06, 1.0000e+00],
        [3.9772e-09, 1.0000e+00],
        [1.1003e-09, 1.0000e+00],
        [9.9883e-01, 1.1703e-03],
        [9.9994e-01, 5.6018e-05],
        [1.7435e-03, 9.9826e-01],
        [1.8721e-06, 1.0000e+00],
        [2.0964e-06, 1.0000e+00],
        [9.8962e-01, 1.0379e-02],
        [2.4208e-05, 9.9998e-01],
        [1.6221e-08, 1.0000e+00],
        [9.9959e-01, 4.1142e-04],
        [3.3447e-07, 1.0000e+00],
        [5.1781e-06, 9.9999e-01],
        [5.2215e-07, 1.0000e+00],
        [7.2012e-06, 9.9999e-01],
        [9.9988e-01, 1.1776e-04],
        [1.0299e-02, 9.8970e-01],
        [1.0000e-06, 1.0000e+00],
        [7.3357e-05, 9.9993e-01],
        [9.9953e-01, 4.7283e-04],
        [1.2282e-05, 9.9999e-01],
        [5.8487e-03, 9.9415e-01],
        [1.3383e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5502e-10, 1.0000e+00],
        [5.3592e-06, 9.9999e-01],
        [1.0203e-05, 9.9999e-01],
        [9.9980e-01, 1.9945e-04],
        [4.9647e-09, 1.0000e+00],
        [1.3273e-05, 9.9999e-01],
        [6.5967e-06, 9.9999e-01],
        [5.3085e-07, 1.0000e+00],
        [7.1413e-06, 9.9999e-01],
        [4.1362e-08, 1.0000e+00],
        [4.8645e-12, 1.0000e+00],
        [1.1309e-06, 1.0000e+00],
        [9.9999e-01, 9.5677e-06],
        [1.1318e-04, 9.9989e-01],
        [9.9076e-06, 9.9999e-01],
        [1.4914e-05, 9.9999e-01],
        [1.5042e-12, 1.0000e+00],
        [9.9050e-10, 1.0000e+00],
        [1.0264e-08, 1.0000e+00],
        [7.6108e-04, 9.9924e-01],
        [1.6606e-04, 9.9983e-01],
        [6.2158e-09, 1.0000e+00],
        [1.4908e-04, 9.9985e-01],
        [9.9983e-01, 1.7427e-04],
        [9.9982e-01, 1.7562e-04],
        [2.4376e-08, 1.0000e+00],
        [7.6812e-08, 1.0000e+00],
        [6.2148e-06, 9.9999e-01],
        [1.7696e-06, 1.0000e+00],
        [1.4621e-01, 8.5379e-01],
        [3.7035e-07, 1.0000e+00],
        [6.3407e-05, 9.9994e-01],
        [8.0820e-07, 1.0000e+00],
        [1.5890e-08, 1.0000e+00],
        [9.5250e-01, 4.7496e-02],
        [3.1964e-10, 1.0000e+00],
        [1.8476e-06, 1.0000e+00],
        [7.5807e-05, 9.9992e-01],
        [8.8233e-10, 1.0000e+00],
        [4.1235e-03, 9.9588e-01],
        [9.9982e-01, 1.8125e-04],
        [5.5321e-05, 9.9994e-01],
        [9.9996e-01, 3.8800e-05],
        [9.9999e-01, 9.0812e-06],
        [2.3798e-11, 1.0000e+00],
        [6.0190e-10, 1.0000e+00],
        [2.2616e-08, 1.0000e+00],
        [4.5685e-10, 1.0000e+00],
        [1.0035e-05, 9.9999e-01],
        [8.7127e-07, 1.0000e+00],
        [3.2640e-07, 1.0000e+00],
        [9.9972e-01, 2.8105e-04],
        [2.9751e-03, 9.9702e-01],
        [5.1005e-09, 1.0000e+00],
        [9.9974e-01, 2.5910e-04],
        [2.1714e-09, 1.0000e+00],
        [1.6563e-08, 1.0000e+00],
        [1.9050e-06, 1.0000e+00],
        [3.0625e-06, 1.0000e+00],
        [1.0152e-06, 1.0000e+00],
        [6.2508e-09, 1.0000e+00],
        [9.0889e-07, 1.0000e+00],
        [4.2855e-01, 5.7145e-01],
        [4.6702e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.6224e-10, 1.0000e+00],
        [4.1254e-06, 1.0000e+00],
        [6.7396e-09, 1.0000e+00],
        [3.0174e-06, 1.0000e+00],
        [2.9908e-05, 9.9997e-01],
        [2.5030e-09, 1.0000e+00],
        [2.3790e-07, 1.0000e+00],
        [4.4450e-05, 9.9996e-01],
        [2.0973e-07, 1.0000e+00],
        [9.9972e-01, 2.7678e-04],
        [2.1059e-05, 9.9998e-01],
        [1.0000e+00, 2.5361e-07],
        [5.6436e-05, 9.9994e-01],
        [9.9999e-01, 1.1875e-05],
        [8.7421e-06, 9.9999e-01],
        [1.0133e-08, 1.0000e+00],
        [2.8439e-08, 1.0000e+00],
        [4.6434e-08, 1.0000e+00],
        [3.8434e-04, 9.9962e-01],
        [9.9677e-01, 3.2288e-03],
        [4.1794e-08, 1.0000e+00],
        [4.8423e-10, 1.0000e+00],
        [9.5194e-09, 1.0000e+00],
        [2.6334e-06, 1.0000e+00],
        [9.9992e-01, 8.4164e-05],
        [9.8562e-01, 1.4378e-02],
        [1.4504e-08, 1.0000e+00],
        [6.3463e-04, 9.9937e-01],
        [5.4022e-08, 1.0000e+00],
        [1.3503e-08, 1.0000e+00],
        [1.1820e-06, 1.0000e+00],
        [1.8577e-06, 1.0000e+00],
        [5.1772e-07, 1.0000e+00],
        [7.3915e-05, 9.9993e-01],
        [2.1697e-06, 1.0000e+00],
        [4.2719e-07, 1.0000e+00],
        [4.8985e-07, 1.0000e+00],
        [5.4305e-04, 9.9946e-01],
        [8.9591e-04, 9.9910e-01],
        [5.2629e-06, 9.9999e-01],
        [1.4987e-05, 9.9998e-01],
        [1.5436e-08, 1.0000e+00],
        [5.0895e-12, 1.0000e+00],
        [7.3131e-06, 9.9999e-01],
        [6.8307e-07, 1.0000e+00],
        [9.9987e-01, 1.2632e-04],
        [3.9014e-10, 1.0000e+00],
        [1.0842e-05, 9.9999e-01],
        [4.7262e-04, 9.9953e-01],
        [3.6921e-06, 1.0000e+00],
        [4.3152e-04, 9.9957e-01],
        [1.3561e-05, 9.9999e-01],
        [9.9972e-01, 2.7644e-04],
        [1.6667e-08, 1.0000e+00],
        [9.9217e-01, 7.8337e-03],
        [2.9857e-07, 1.0000e+00],
        [6.3164e-06, 9.9999e-01],
        [1.3060e-10, 1.0000e+00],
        [3.4223e-02, 9.6578e-01],
        [1.9308e-09, 1.0000e+00],
        [3.2516e-03, 9.9675e-01],
        [8.7753e-10, 1.0000e+00],
        [1.0879e-05, 9.9999e-01],
        [7.4948e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3806e-04, 9.9976e-01],
        [1.0181e-06, 1.0000e+00],
        [9.9997e-01, 3.2485e-05],
        [1.8403e-05, 9.9998e-01],
        [2.3228e-07, 1.0000e+00],
        [3.0328e-05, 9.9997e-01],
        [3.8987e-09, 1.0000e+00],
        [3.9558e-09, 1.0000e+00],
        [8.3616e-08, 1.0000e+00],
        [1.3406e-06, 1.0000e+00],
        [5.5665e-07, 1.0000e+00],
        [5.6242e-07, 1.0000e+00],
        [5.6125e-06, 9.9999e-01],
        [6.3469e-09, 1.0000e+00],
        [8.4248e-10, 1.0000e+00],
        [3.3207e-05, 9.9997e-01],
        [2.6621e-15, 1.0000e+00],
        [8.8565e-01, 1.1435e-01],
        [1.4808e-10, 1.0000e+00],
        [1.3855e-08, 1.0000e+00],
        [9.9823e-01, 1.7685e-03],
        [1.1643e-08, 1.0000e+00],
        [1.4197e-01, 8.5803e-01],
        [4.7173e-04, 9.9953e-01],
        [2.2987e-07, 1.0000e+00],
        [8.0033e-05, 9.9992e-01],
        [1.2961e-07, 1.0000e+00],
        [9.9988e-01, 1.2188e-04],
        [7.4723e-04, 9.9925e-01],
        [1.3987e-06, 1.0000e+00],
        [3.9880e-03, 9.9601e-01],
        [9.9959e-01, 4.0583e-04],
        [1.3813e-09, 1.0000e+00],
        [4.5597e-05, 9.9995e-01],
        [5.8853e-07, 1.0000e+00],
        [1.0199e-05, 9.9999e-01],
        [9.9976e-01, 2.4292e-04],
        [4.0335e-03, 9.9597e-01],
        [4.4410e-09, 1.0000e+00],
        [9.9963e-01, 3.6596e-04],
        [1.1049e-07, 1.0000e+00],
        [9.9644e-01, 3.5646e-03],
        [2.4511e-09, 1.0000e+00],
        [8.9202e-07, 1.0000e+00],
        [1.0128e-06, 1.0000e+00],
        [4.8031e-09, 1.0000e+00],
        [1.2654e-05, 9.9999e-01],
        [9.9991e-01, 8.7494e-05],
        [9.9194e-08, 1.0000e+00],
        [5.4192e-11, 1.0000e+00],
        [4.5245e-07, 1.0000e+00],
        [3.2906e-11, 1.0000e+00],
        [1.3550e-06, 1.0000e+00],
        [5.1114e-08, 1.0000e+00],
        [1.7868e-06, 1.0000e+00],
        [2.2808e-05, 9.9998e-01],
        [5.7321e-01, 4.2679e-01],
        [9.9988e-01, 1.1520e-04],
        [1.0700e-07, 1.0000e+00],
        [5.8529e-06, 9.9999e-01],
        [1.8069e-06, 1.0000e+00],
        [9.9895e-01, 1.0534e-03],
        [9.9988e-01, 1.2018e-04],
        [9.9985e-01, 1.4854e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2707e-10, 1.0000e+00],
        [2.1430e-03, 9.9786e-01],
        [9.9994e-01, 6.2885e-05],
        [8.2648e-09, 1.0000e+00],
        [9.9979e-01, 2.0695e-04],
        [8.1174e-07, 1.0000e+00],
        [6.5861e-03, 9.9341e-01],
        [6.0636e-08, 1.0000e+00],
        [2.5471e-09, 1.0000e+00],
        [9.4190e-01, 5.8097e-02],
        [9.9984e-01, 1.6027e-04],
        [4.6639e-06, 1.0000e+00],
        [9.9987e-01, 1.2584e-04],
        [6.3295e-09, 1.0000e+00],
        [1.1919e-09, 1.0000e+00],
        [1.8428e-05, 9.9998e-01],
        [4.6332e-10, 1.0000e+00],
        [9.9445e-01, 5.5500e-03],
        [9.9980e-01, 1.9879e-04],
        [9.4526e-09, 1.0000e+00],
        [6.6690e-05, 9.9993e-01],
        [1.0347e-04, 9.9990e-01],
        [1.3122e-06, 1.0000e+00],
        [5.0480e-06, 9.9999e-01],
        [1.4232e-06, 1.0000e+00],
        [9.9995e-01, 4.8866e-05],
        [2.1334e-04, 9.9979e-01],
        [9.9571e-01, 4.2895e-03],
        [1.0771e-06, 1.0000e+00],
        [1.6371e-06, 1.0000e+00],
        [3.4536e-09, 1.0000e+00],
        [3.3231e-04, 9.9967e-01],
        [2.6531e-05, 9.9997e-01],
        [1.2301e-05, 9.9999e-01],
        [7.7061e-11, 1.0000e+00],
        [1.1418e-08, 1.0000e+00],
        [3.3779e-09, 1.0000e+00],
        [9.9964e-01, 3.6099e-04],
        [9.9996e-01, 4.4243e-05],
        [1.4723e-09, 1.0000e+00],
        [3.0141e-07, 1.0000e+00],
        [9.9997e-01, 3.3076e-05],
        [9.9986e-01, 1.4376e-04],
        [1.8140e-07, 1.0000e+00],
        [5.1173e-11, 1.0000e+00],
        [2.2744e-06, 1.0000e+00],
        [1.3266e-09, 1.0000e+00],
        [2.4839e-09, 1.0000e+00],
        [8.7853e-12, 1.0000e+00],
        [5.6886e-07, 1.0000e+00],
        [1.0257e-01, 8.9743e-01],
        [2.5524e-11, 1.0000e+00],
        [1.8195e-04, 9.9982e-01],
        [3.5653e-07, 1.0000e+00],
        [4.5068e-03, 9.9549e-01],
        [9.5049e-05, 9.9990e-01],
        [9.8479e-01, 1.5209e-02],
        [3.8917e-05, 9.9996e-01],
        [9.9976e-01, 2.3998e-04],
        [6.2030e-09, 1.0000e+00],
        [3.6278e-05, 9.9996e-01],
        [2.1066e-03, 9.9789e-01],
        [9.9974e-01, 2.6439e-04],
        [4.2938e-03, 9.9571e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9083e-01, 9.1736e-03],
        [9.6245e-09, 1.0000e+00],
        [2.4976e-09, 1.0000e+00],
        [3.8363e-03, 9.9616e-01],
        [1.2907e-07, 1.0000e+00],
        [9.9991e-01, 8.7900e-05],
        [1.7272e-06, 1.0000e+00],
        [9.0421e-01, 9.5794e-02],
        [3.0130e-04, 9.9970e-01],
        [9.9988e-01, 1.2057e-04],
        [4.3705e-07, 1.0000e+00],
        [3.0694e-11, 1.0000e+00],
        [9.9983e-01, 1.7502e-04],
        [9.9924e-01, 7.6160e-04],
        [4.9432e-06, 1.0000e+00],
        [1.6602e-03, 9.9834e-01],
        [2.7213e-05, 9.9997e-01],
        [9.9991e-01, 9.1444e-05],
        [9.3923e-01, 6.0775e-02],
        [5.2175e-09, 1.0000e+00],
        [1.5800e-09, 1.0000e+00],
        [6.8970e-01, 3.1030e-01],
        [2.8469e-09, 1.0000e+00],
        [4.2555e-11, 1.0000e+00],
        [9.9899e-01, 1.0053e-03],
        [9.6107e-01, 3.8928e-02],
        [8.6984e-10, 1.0000e+00],
        [1.8627e-07, 1.0000e+00],
        [1.5568e-06, 1.0000e+00],
        [6.0483e-08, 1.0000e+00],
        [9.9999e-01, 9.5200e-06],
        [9.7922e-12, 1.0000e+00],
        [1.9825e-07, 1.0000e+00],
        [1.7515e-05, 9.9998e-01],
        [1.6053e-10, 1.0000e+00],
        [2.2381e-08, 1.0000e+00],
        [1.5850e-08, 1.0000e+00],
        [2.6454e-03, 9.9735e-01],
        [6.2772e-06, 9.9999e-01],
        [2.7823e-07, 1.0000e+00],
        [8.8882e-10, 1.0000e+00],
        [1.4621e-05, 9.9999e-01],
        [9.8686e-01, 1.3142e-02],
        [1.2138e-08, 1.0000e+00],
        [1.5043e-04, 9.9985e-01],
        [9.5839e-10, 1.0000e+00],
        [5.5109e-11, 1.0000e+00],
        [5.7937e-07, 1.0000e+00],
        [1.1474e-08, 1.0000e+00],
        [6.9690e-08, 1.0000e+00],
        [2.8554e-04, 9.9971e-01],
        [5.5415e-04, 9.9945e-01],
        [9.8694e-04, 9.9901e-01],
        [1.4812e-08, 1.0000e+00],
        [3.6294e-12, 1.0000e+00],
        [1.1416e-03, 9.9886e-01],
        [8.4008e-08, 1.0000e+00],
        [1.7881e-07, 1.0000e+00],
        [2.5059e-11, 1.0000e+00],
        [2.4454e-02, 9.7555e-01],
        [1.2339e-04, 9.9988e-01],
        [1.6797e-06, 1.0000e+00],
        [9.9983e-01, 1.6766e-04],
        [8.7224e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9775e-01, 2.2493e-03],
        [9.0031e-08, 1.0000e+00],
        [2.4004e-05, 9.9998e-01],
        [1.9939e-04, 9.9980e-01],
        [8.0334e-05, 9.9992e-01],
        [4.3898e-11, 1.0000e+00],
        [4.2393e-09, 1.0000e+00],
        [2.8130e-03, 9.9719e-01],
        [1.1650e-11, 1.0000e+00],
        [2.9991e-06, 1.0000e+00],
        [7.4719e-08, 1.0000e+00],
        [9.9955e-01, 4.4767e-04],
        [1.0000e+00, 2.3378e-07],
        [1.6246e-08, 1.0000e+00],
        [9.5159e-07, 1.0000e+00],
        [1.0000e+00, 3.8672e-07],
        [5.1675e-03, 9.9483e-01],
        [1.5017e-12, 1.0000e+00],
        [6.1222e-04, 9.9939e-01],
        [1.7667e-10, 1.0000e+00],
        [9.8357e-01, 1.6432e-02],
        [9.9998e-01, 1.6362e-05],
        [1.5283e-07, 1.0000e+00],
        [3.2740e-10, 1.0000e+00],
        [2.8001e-08, 1.0000e+00],
        [5.2589e-03, 9.9474e-01],
        [2.7762e-03, 9.9722e-01],
        [3.9209e-08, 1.0000e+00],
        [8.7144e-09, 1.0000e+00],
        [9.9857e-01, 1.4315e-03],
        [2.5730e-09, 1.0000e+00],
        [7.0020e-05, 9.9993e-01],
        [7.0922e-01, 2.9078e-01],
        [6.0654e-10, 1.0000e+00],
        [5.5809e-08, 1.0000e+00],
        [6.2168e-01, 3.7832e-01],
        [9.2414e-07, 1.0000e+00],
        [6.8416e-06, 9.9999e-01],
        [1.2965e-07, 1.0000e+00],
        [9.9990e-01, 9.8890e-05],
        [9.9965e-01, 3.5183e-04],
        [5.1592e-11, 1.0000e+00],
        [3.9192e-11, 1.0000e+00],
        [9.9997e-01, 3.2558e-05],
        [9.9986e-01, 1.3542e-04],
        [8.7321e-05, 9.9991e-01],
        [3.7677e-07, 1.0000e+00],
        [9.9984e-01, 1.6130e-04],
        [9.0283e-11, 1.0000e+00],
        [3.6201e-09, 1.0000e+00],
        [9.9993e-01, 7.0630e-05],
        [2.0127e-09, 1.0000e+00],
        [9.9992e-01, 8.2397e-05],
        [9.9934e-01, 6.6478e-04],
        [9.9996e-01, 4.4893e-05],
        [1.7622e-06, 1.0000e+00],
        [7.4157e-06, 9.9999e-01],
        [4.7847e-07, 1.0000e+00],
        [9.9979e-01, 2.0763e-04],
        [9.9986e-01, 1.4191e-04],
        [9.9955e-01, 4.5462e-04],
        [4.4965e-06, 1.0000e+00],
        [4.5266e-08, 1.0000e+00],
        [3.3787e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9983e-01, 1.7105e-04],
        [1.3383e-08, 1.0000e+00],
        [3.3868e-08, 1.0000e+00],
        [1.0483e-04, 9.9990e-01],
        [9.9897e-01, 1.0286e-03],
        [9.9860e-01, 1.4047e-03],
        [1.7015e-06, 1.0000e+00],
        [4.7237e-08, 1.0000e+00],
        [8.1620e-04, 9.9918e-01],
        [1.8827e-07, 1.0000e+00],
        [9.9935e-01, 6.4830e-04],
        [3.7202e-12, 1.0000e+00],
        [9.5375e-01, 4.6251e-02],
        [3.3872e-01, 6.6128e-01],
        [1.7853e-05, 9.9998e-01],
        [4.3043e-08, 1.0000e+00],
        [9.9995e-01, 5.0854e-05],
        [2.3314e-07, 1.0000e+00],
        [1.5023e-07, 1.0000e+00],
        [9.9828e-01, 1.7181e-03],
        [9.9941e-01, 5.9248e-04],
        [9.8373e-07, 1.0000e+00],
        [1.7878e-09, 1.0000e+00],
        [2.3017e-05, 9.9998e-01],
        [2.9933e-06, 1.0000e+00],
        [9.9983e-01, 1.7266e-04],
        [3.6479e-05, 9.9996e-01],
        [8.7158e-07, 1.0000e+00],
        [4.9892e-10, 1.0000e+00],
        [3.7577e-08, 1.0000e+00],
        [1.6734e-07, 1.0000e+00],
        [9.2653e-01, 7.3465e-02],
        [1.8389e-05, 9.9998e-01],
        [2.5269e-05, 9.9997e-01],
        [2.9624e-05, 9.9997e-01],
        [9.9987e-01, 1.3235e-04],
        [5.4440e-09, 1.0000e+00],
        [7.4673e-05, 9.9993e-01],
        [9.8589e-01, 1.4107e-02],
        [1.7704e-04, 9.9982e-01],
        [9.9966e-01, 3.3821e-04],
        [5.1935e-05, 9.9995e-01],
        [2.8011e-04, 9.9972e-01],
        [3.3662e-03, 9.9663e-01],
        [1.6022e-03, 9.9840e-01],
        [1.7340e-07, 1.0000e+00],
        [4.8135e-04, 9.9952e-01],
        [9.9981e-01, 1.8656e-04],
        [2.6983e-10, 1.0000e+00],
        [1.4973e-06, 1.0000e+00],
        [1.0793e-07, 1.0000e+00],
        [1.0836e-08, 1.0000e+00],
        [5.0058e-10, 1.0000e+00],
        [2.3861e-05, 9.9998e-01],
        [1.4626e-08, 1.0000e+00],
        [1.4286e-10, 1.0000e+00],
        [1.1270e-03, 9.9887e-01],
        [2.9757e-04, 9.9970e-01],
        [1.2139e-08, 1.0000e+00],
        [2.1189e-09, 1.0000e+00],
        [1.4460e-08, 1.0000e+00],
        [5.8480e-01, 4.1520e-01],
        [1.3909e-06, 1.0000e+00],
        [8.6751e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.6714e-08, 1.0000e+00],
        [9.9979e-01, 2.1310e-04],
        [9.9996e-01, 3.6572e-05],
        [1.7974e-07, 1.0000e+00],
        [1.5719e-04, 9.9984e-01],
        [1.0503e-08, 1.0000e+00],
        [8.6211e-07, 1.0000e+00],
        [8.5198e-07, 1.0000e+00],
        [9.9974e-01, 2.6456e-04],
        [9.9991e-01, 9.4042e-05],
        [3.6888e-04, 9.9963e-01],
        [3.0160e-09, 1.0000e+00],
        [9.3201e-07, 1.0000e+00],
        [9.9974e-01, 2.6345e-04],
        [9.9986e-01, 1.4342e-04],
        [2.8480e-05, 9.9997e-01],
        [4.1625e-07, 1.0000e+00],
        [1.6322e-08, 1.0000e+00],
        [1.5853e-03, 9.9841e-01],
        [2.5470e-05, 9.9997e-01],
        [6.8977e-01, 3.1023e-01],
        [8.8020e-01, 1.1980e-01],
        [3.4431e-08, 1.0000e+00],
        [1.8721e-03, 9.9813e-01],
        [9.9398e-01, 6.0185e-03],
        [6.2503e-06, 9.9999e-01],
        [1.0655e-07, 1.0000e+00],
        [2.3466e-02, 9.7653e-01],
        [9.6566e-01, 3.4336e-02],
        [1.7828e-07, 1.0000e+00],
        [2.9375e-07, 1.0000e+00],
        [1.3128e-05, 9.9999e-01],
        [2.9685e-05, 9.9997e-01],
        [1.3798e-06, 1.0000e+00],
        [2.6503e-03, 9.9735e-01],
        [5.7574e-08, 1.0000e+00],
        [2.7366e-06, 1.0000e+00],
        [2.8592e-10, 1.0000e+00],
        [5.8842e-07, 1.0000e+00],
        [9.9991e-01, 8.9027e-05],
        [3.3627e-12, 1.0000e+00],
        [2.5908e-06, 1.0000e+00],
        [2.6983e-08, 1.0000e+00],
        [9.2398e-06, 9.9999e-01],
        [9.9982e-01, 1.8136e-04],
        [1.1131e-07, 1.0000e+00],
        [2.4718e-04, 9.9975e-01],
        [8.0945e-01, 1.9055e-01],
        [9.9952e-01, 4.7981e-04],
        [9.9288e-01, 7.1202e-03],
        [1.7172e-07, 1.0000e+00],
        [4.6346e-02, 9.5365e-01],
        [9.9955e-01, 4.5311e-04],
        [5.4776e-07, 1.0000e+00],
        [4.8740e-11, 1.0000e+00],
        [9.2495e-05, 9.9991e-01],
        [1.8855e-07, 1.0000e+00],
        [2.9267e-03, 9.9707e-01],
        [9.9995e-01, 4.5148e-05],
        [9.9807e-01, 1.9340e-03],
        [4.3232e-03, 9.9568e-01],
        [1.1969e-06, 1.0000e+00],
        [7.3217e-03, 9.9268e-01],
        [9.9976e-01, 2.3571e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.4711e-03, 9.9653e-01],
        [1.0443e-06, 1.0000e+00],
        [1.7988e-05, 9.9998e-01],
        [2.5893e-06, 1.0000e+00],
        [1.8416e-06, 1.0000e+00],
        [6.1944e-09, 1.0000e+00],
        [1.1137e-06, 1.0000e+00],
        [9.9972e-01, 2.7858e-04],
        [9.9995e-01, 4.8287e-05],
        [9.9997e-01, 3.0221e-05],
        [9.5533e-08, 1.0000e+00],
        [2.4338e-09, 1.0000e+00],
        [9.5687e-06, 9.9999e-01],
        [1.7683e-07, 1.0000e+00],
        [9.9507e-07, 1.0000e+00],
        [9.9991e-01, 8.8257e-05],
        [1.0422e-06, 1.0000e+00],
        [1.9270e-09, 1.0000e+00],
        [9.9971e-01, 2.8547e-04],
        [9.4479e-07, 1.0000e+00],
        [1.1947e-08, 1.0000e+00],
        [1.1071e-08, 1.0000e+00],
        [4.9035e-06, 1.0000e+00],
        [1.2609e-04, 9.9987e-01],
        [6.2462e-06, 9.9999e-01],
        [5.4485e-07, 1.0000e+00],
        [3.8722e-04, 9.9961e-01],
        [1.7560e-07, 1.0000e+00],
        [9.9869e-01, 1.3101e-03],
        [3.1637e-11, 1.0000e+00],
        [2.0591e-04, 9.9979e-01],
        [3.6864e-02, 9.6314e-01],
        [8.1296e-05, 9.9992e-01],
        [8.2568e-06, 9.9999e-01],
        [9.9989e-01, 1.1259e-04],
        [2.6205e-06, 1.0000e+00],
        [1.6749e-08, 1.0000e+00],
        [3.3338e-08, 1.0000e+00],
        [7.5311e-08, 1.0000e+00],
        [2.1096e-06, 1.0000e+00],
        [4.9711e-04, 9.9950e-01],
        [9.7312e-10, 1.0000e+00],
        [7.3623e-10, 1.0000e+00],
        [7.1930e-08, 1.0000e+00],
        [9.7625e-01, 2.3753e-02],
        [6.2086e-01, 3.7914e-01],
        [2.7351e-05, 9.9997e-01],
        [4.5840e-03, 9.9542e-01],
        [2.6757e-11, 1.0000e+00],
        [1.0610e-05, 9.9999e-01],
        [4.1183e-04, 9.9959e-01],
        [9.9988e-01, 1.1966e-04],
        [9.9980e-01, 1.9945e-04],
        [2.2262e-07, 1.0000e+00],
        [1.0623e-08, 1.0000e+00],
        [9.5784e-01, 4.2161e-02],
        [3.5712e-03, 9.9643e-01],
        [4.7276e-11, 1.0000e+00],
        [7.2474e-07, 1.0000e+00],
        [1.3858e-07, 1.0000e+00],
        [1.6207e-07, 1.0000e+00],
        [1.4519e-04, 9.9985e-01],
        [9.4925e-08, 1.0000e+00],
        [9.9926e-01, 7.4293e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[4.7792e-07, 1.0000e+00],
        [9.9989e-01, 1.0893e-04],
        [8.5286e-07, 1.0000e+00],
        [4.6625e-03, 9.9534e-01],
        [5.5534e-10, 1.0000e+00],
        [4.8453e-06, 1.0000e+00],
        [9.9981e-01, 1.8743e-04],
        [6.4464e-07, 1.0000e+00],
        [9.9524e-01, 4.7632e-03],
        [8.7272e-09, 1.0000e+00],
        [8.7449e-05, 9.9991e-01],
        [1.1592e-08, 1.0000e+00],
        [2.3434e-07, 1.0000e+00],
        [3.0852e-07, 1.0000e+00],
        [1.2836e-06, 1.0000e+00],
        [2.6674e-08, 1.0000e+00],
        [2.4878e-08, 1.0000e+00],
        [9.0581e-01, 9.4192e-02],
        [3.7911e-08, 1.0000e+00],
        [5.4597e-05, 9.9995e-01],
        [9.9938e-01, 6.1759e-04],
        [1.2938e-05, 9.9999e-01],
        [2.6660e-07, 1.0000e+00],
        [9.9993e-01, 7.4931e-05],
        [9.2484e-09, 1.0000e+00],
        [1.6235e-04, 9.9984e-01],
        [3.2023e-05, 9.9997e-01],
        [2.6864e-08, 1.0000e+00],
        [3.3978e-05, 9.9997e-01],
        [7.2759e-07, 1.0000e+00],
        [2.2838e-04, 9.9977e-01],
        [2.4839e-07, 1.0000e+00],
        [3.6932e-03, 9.9631e-01],
        [5.4382e-06, 9.9999e-01],
        [9.9993e-01, 6.7570e-05],
        [9.9993e-01, 7.3988e-05],
        [8.8298e-07, 1.0000e+00],
        [3.8215e-07, 1.0000e+00],
        [1.1580e-06, 1.0000e+00],
        [3.6562e-11, 1.0000e+00],
        [1.6473e-06, 1.0000e+00],
        [3.5814e-10, 1.0000e+00],
        [9.9755e-01, 2.4490e-03],
        [9.9730e-01, 2.6994e-03],
        [3.6529e-08, 1.0000e+00],
        [8.1514e-07, 1.0000e+00],
        [9.9831e-01, 1.6924e-03],
        [1.2971e-01, 8.7029e-01],
        [4.6935e-06, 1.0000e+00],
        [4.7587e-06, 1.0000e+00],
        [9.9986e-01, 1.3614e-04],
        [1.8310e-10, 1.0000e+00],
        [1.6467e-11, 1.0000e+00],
        [1.6742e-04, 9.9983e-01],
        [1.0796e-09, 1.0000e+00],
        [1.7704e-07, 1.0000e+00],
        [1.1860e-03, 9.9881e-01],
        [8.2812e-05, 9.9992e-01],
        [3.0737e-06, 1.0000e+00],
        [1.5332e-08, 1.0000e+00],
        [4.1610e-04, 9.9958e-01],
        [4.4814e-08, 1.0000e+00],
        [1.1827e-11, 1.0000e+00],
        [6.5029e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1078e-04, 9.9989e-01],
        [9.8398e-01, 1.6020e-02],
        [1.4339e-08, 1.0000e+00],
        [1.7605e-08, 1.0000e+00],
        [1.3766e-06, 1.0000e+00],
        [2.1307e-05, 9.9998e-01],
        [9.9968e-01, 3.2131e-04],
        [8.9388e-01, 1.0612e-01],
        [8.2345e-10, 1.0000e+00],
        [7.4330e-08, 1.0000e+00],
        [3.3652e-05, 9.9997e-01],
        [5.5869e-07, 1.0000e+00],
        [8.3994e-11, 1.0000e+00],
        [3.1407e-06, 1.0000e+00],
        [2.7604e-07, 1.0000e+00],
        [1.1637e-10, 1.0000e+00],
        [5.5344e-01, 4.4656e-01],
        [9.9993e-01, 7.0453e-05],
        [1.7654e-07, 1.0000e+00],
        [2.9487e-10, 1.0000e+00],
        [4.8744e-11, 1.0000e+00],
        [1.4697e-08, 1.0000e+00],
        [1.3203e-06, 1.0000e+00],
        [2.4613e-07, 1.0000e+00],
        [2.1167e-09, 1.0000e+00],
        [1.8761e-05, 9.9998e-01],
        [1.2078e-02, 9.8792e-01],
        [2.7019e-05, 9.9997e-01],
        [9.9998e-01, 1.6322e-05],
        [1.0702e-05, 9.9999e-01],
        [1.2582e-04, 9.9987e-01],
        [9.9987e-01, 1.3457e-04],
        [1.0000e+00, 5.3149e-07],
        [2.1490e-09, 1.0000e+00],
        [4.1432e-06, 1.0000e+00],
        [1.8016e-08, 1.0000e+00],
        [1.1404e-07, 1.0000e+00],
        [3.2265e-02, 9.6774e-01],
        [1.5052e-07, 1.0000e+00],
        [5.5394e-01, 4.4606e-01],
        [1.8711e-07, 1.0000e+00],
        [5.2206e-06, 9.9999e-01],
        [9.9962e-01, 3.7744e-04],
        [1.8211e-08, 1.0000e+00],
        [4.6572e-08, 1.0000e+00],
        [1.9319e-06, 1.0000e+00],
        [2.7087e-03, 9.9729e-01],
        [9.9963e-01, 3.6688e-04],
        [9.9893e-01, 1.0749e-03],
        [9.8739e-01, 1.2609e-02],
        [2.9254e-07, 1.0000e+00],
        [5.2311e-11, 1.0000e+00],
        [1.2943e-06, 1.0000e+00],
        [1.7069e-05, 9.9998e-01],
        [4.2810e-06, 1.0000e+00],
        [9.9986e-01, 1.4347e-04],
        [9.9973e-01, 2.6801e-04],
        [9.9983e-01, 1.7290e-04],
        [9.9983e-01, 1.7193e-04],
        [4.5992e-06, 1.0000e+00],
        [6.0006e-06, 9.9999e-01],
        [2.4656e-04, 9.9975e-01],
        [4.8733e-05, 9.9995e-01],
        [8.3574e-01, 1.6426e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.2246e-05, 9.9994e-01],
        [1.3473e-07, 1.0000e+00],
        [2.5851e-07, 1.0000e+00],
        [9.9981e-01, 1.8683e-04],
        [3.6813e-07, 1.0000e+00],
        [1.4936e-07, 1.0000e+00],
        [9.9999e-01, 1.2526e-05],
        [9.9998e-01, 2.3808e-05],
        [9.6588e-04, 9.9903e-01],
        [1.2874e-05, 9.9999e-01],
        [7.1422e-04, 9.9929e-01],
        [3.0118e-02, 9.6988e-01],
        [2.5129e-07, 1.0000e+00],
        [9.9988e-01, 1.2390e-04],
        [7.5956e-03, 9.9240e-01],
        [8.9436e-06, 9.9999e-01],
        [1.2928e-06, 1.0000e+00],
        [2.1590e-04, 9.9978e-01],
        [1.0016e-07, 1.0000e+00],
        [9.9959e-01, 4.0806e-04],
        [1.5014e-04, 9.9985e-01],
        [8.2375e-06, 9.9999e-01],
        [1.9633e-06, 1.0000e+00],
        [5.4151e-06, 9.9999e-01],
        [1.0121e-05, 9.9999e-01],
        [9.9735e-01, 2.6466e-03],
        [5.8878e-06, 9.9999e-01],
        [9.9971e-01, 2.8912e-04],
        [9.9997e-01, 3.4134e-05],
        [2.7417e-07, 1.0000e+00],
        [2.6273e-03, 9.9737e-01],
        [1.6174e-06, 1.0000e+00],
        [9.9993e-01, 6.5670e-05],
        [9.9991e-01, 8.8266e-05],
        [6.6904e-09, 1.0000e+00],
        [1.7595e-01, 8.2405e-01],
        [2.5974e-09, 1.0000e+00],
        [1.6774e-04, 9.9983e-01],
        [3.4932e-05, 9.9997e-01],
        [2.4274e-06, 1.0000e+00],
        [5.5611e-10, 1.0000e+00],
        [1.5345e-06, 1.0000e+00],
        [1.6603e-05, 9.9998e-01],
        [9.9861e-01, 1.3879e-03],
        [9.9947e-01, 5.3072e-04],
        [9.9977e-01, 2.2704e-04],
        [4.6401e-11, 1.0000e+00],
        [9.9983e-01, 1.6530e-04],
        [1.6902e-10, 1.0000e+00],
        [9.9997e-01, 3.1568e-05],
        [5.3569e-08, 1.0000e+00],
        [7.0609e-07, 1.0000e+00],
        [3.9190e-06, 1.0000e+00],
        [9.9440e-01, 5.5990e-03],
        [1.3689e-06, 1.0000e+00],
        [5.2960e-03, 9.9470e-01],
        [1.9571e-05, 9.9998e-01],
        [9.3090e-13, 1.0000e+00],
        [8.2633e-08, 1.0000e+00],
        [9.9984e-01, 1.6373e-04],
        [9.9972e-01, 2.7513e-04],
        [1.5108e-07, 1.0000e+00],
        [1.1345e-03, 9.9887e-01],
        [9.9981e-01, 1.8818e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9292e-01, 7.0834e-03],
        [9.4503e-01, 5.4967e-02],
        [2.1074e-14, 1.0000e+00],
        [9.9963e-01, 3.6707e-04],
        [9.9630e-01, 3.6985e-03],
        [3.0887e-02, 9.6911e-01],
        [1.7800e-09, 1.0000e+00],
        [9.1847e-09, 1.0000e+00],
        [9.9993e-01, 7.0785e-05],
        [9.6946e-01, 3.0538e-02],
        [3.8398e-04, 9.9962e-01],
        [1.2891e-03, 9.9871e-01],
        [4.9630e-04, 9.9950e-01],
        [3.6531e-09, 1.0000e+00],
        [1.3412e-09, 1.0000e+00],
        [3.3030e-06, 1.0000e+00],
        [4.5438e-09, 1.0000e+00],
        [2.1381e-11, 1.0000e+00],
        [1.3579e-06, 1.0000e+00],
        [1.3267e-03, 9.9867e-01],
        [9.9872e-01, 1.2758e-03],
        [9.9182e-01, 8.1767e-03],
        [1.5367e-02, 9.8463e-01],
        [2.6063e-01, 7.3937e-01],
        [4.5644e-04, 9.9954e-01],
        [1.0154e-08, 1.0000e+00],
        [9.9994e-01, 6.1521e-05],
        [1.9361e-08, 1.0000e+00],
        [3.6507e-06, 1.0000e+00],
        [1.4044e-10, 1.0000e+00],
        [9.9994e-01, 6.1003e-05],
        [3.0531e-04, 9.9969e-01],
        [7.1728e-06, 9.9999e-01],
        [3.8314e-09, 1.0000e+00],
        [8.0643e-03, 9.9194e-01],
        [5.4869e-06, 9.9999e-01],
        [9.9973e-01, 2.6969e-04],
        [2.2156e-02, 9.7784e-01],
        [5.4871e-06, 9.9999e-01],
        [2.0412e-06, 1.0000e+00],
        [3.4768e-09, 1.0000e+00],
        [1.6359e-08, 1.0000e+00],
        [4.4833e-05, 9.9996e-01],
        [1.0190e-10, 1.0000e+00],
        [3.7655e-06, 1.0000e+00],
        [2.2568e-05, 9.9998e-01],
        [2.7948e-06, 1.0000e+00],
        [2.6961e-08, 1.0000e+00],
        [1.5324e-05, 9.9998e-01],
        [1.5203e-06, 1.0000e+00],
        [1.4607e-04, 9.9985e-01],
        [8.1813e-12, 1.0000e+00],
        [9.9984e-01, 1.5772e-04],
        [9.6748e-01, 3.2525e-02],
        [3.5781e-09, 1.0000e+00],
        [1.1401e-04, 9.9989e-01],
        [4.7902e-09, 1.0000e+00],
        [9.9970e-01, 3.0128e-04],
        [1.3476e-07, 1.0000e+00],
        [3.2208e-05, 9.9997e-01],
        [7.6653e-04, 9.9923e-01],
        [9.9992e-01, 8.2549e-05],
        [6.8142e-03, 9.9319e-01],
        [6.5779e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.8024e-06, 9.9999e-01],
        [9.9997e-01, 3.3581e-05],
        [7.5087e-06, 9.9999e-01],
        [4.3153e-05, 9.9996e-01],
        [6.5810e-03, 9.9342e-01],
        [4.6248e-06, 1.0000e+00],
        [9.9985e-01, 1.4741e-04],
        [2.5887e-07, 1.0000e+00],
        [9.9986e-01, 1.3883e-04],
        [8.4361e-05, 9.9992e-01],
        [4.2979e-04, 9.9957e-01],
        [8.2823e-03, 9.9172e-01],
        [3.7399e-07, 1.0000e+00],
        [9.9992e-01, 8.1505e-05],
        [9.9372e-01, 6.2796e-03],
        [1.7804e-08, 1.0000e+00],
        [1.5670e-10, 1.0000e+00],
        [8.7413e-06, 9.9999e-01],
        [4.4560e-06, 1.0000e+00],
        [6.3379e-01, 3.6621e-01],
        [9.9986e-01, 1.3694e-04],
        [2.5271e-11, 1.0000e+00],
        [9.9967e-01, 3.3186e-04],
        [9.8403e-01, 1.5975e-02],
        [2.5150e-04, 9.9975e-01],
        [9.9930e-01, 7.0111e-04],
        [1.2039e-04, 9.9988e-01],
        [8.5160e-06, 9.9999e-01],
        [8.4173e-08, 1.0000e+00],
        [5.1532e-08, 1.0000e+00],
        [3.6921e-07, 1.0000e+00],
        [2.2583e-06, 1.0000e+00],
        [5.5707e-09, 1.0000e+00],
        [6.2461e-08, 1.0000e+00],
        [9.9996e-01, 3.9973e-05],
        [6.3641e-05, 9.9994e-01],
        [1.0697e-10, 1.0000e+00],
        [5.6307e-08, 1.0000e+00],
        [4.4476e-05, 9.9996e-01],
        [2.3963e-03, 9.9760e-01],
        [9.9986e-01, 1.3946e-04],
        [9.9821e-01, 1.7869e-03],
        [2.3869e-09, 1.0000e+00],
        [1.0948e-05, 9.9999e-01],
        [9.9657e-01, 3.4321e-03],
        [9.9981e-01, 1.8738e-04],
        [2.8197e-04, 9.9972e-01],
        [1.7115e-07, 1.0000e+00],
        [7.0518e-09, 1.0000e+00],
        [1.1872e-07, 1.0000e+00],
        [2.3805e-05, 9.9998e-01],
        [1.0000e+00, 4.1928e-06],
        [1.1064e-06, 1.0000e+00],
        [8.8033e-06, 9.9999e-01],
        [5.5190e-04, 9.9945e-01],
        [4.8871e-02, 9.5113e-01],
        [9.9994e-01, 5.8814e-05],
        [2.9516e-07, 1.0000e+00],
        [1.1110e-08, 1.0000e+00],
        [3.2608e-02, 9.6739e-01],
        [1.8422e-03, 9.9816e-01],
        [1.9482e-02, 9.8052e-01],
        [5.2926e-04, 9.9947e-01],
        [2.0998e-03, 9.9790e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7578e-06, 1.0000e+00],
        [3.0540e-09, 1.0000e+00],
        [9.9883e-01, 1.1668e-03],
        [5.5610e-05, 9.9994e-01],
        [5.9394e-05, 9.9994e-01],
        [8.5139e-10, 1.0000e+00],
        [9.9978e-01, 2.1542e-04],
        [1.3873e-07, 1.0000e+00],
        [7.1069e-08, 1.0000e+00],
        [9.9986e-01, 1.3785e-04],
        [9.9991e-01, 9.4227e-05],
        [1.8107e-08, 1.0000e+00],
        [6.1094e-09, 1.0000e+00],
        [5.4444e-08, 1.0000e+00],
        [9.9994e-01, 5.8951e-05],
        [5.4504e-09, 1.0000e+00],
        [4.1044e-03, 9.9590e-01],
        [5.4783e-03, 9.9452e-01],
        [5.8828e-05, 9.9994e-01],
        [9.9995e-01, 4.8013e-05],
        [9.9872e-01, 1.2781e-03],
        [1.6437e-02, 9.8356e-01],
        [2.7567e-08, 1.0000e+00],
        [7.7579e-04, 9.9922e-01],
        [1.0266e-06, 1.0000e+00],
        [7.0240e-12, 1.0000e+00],
        [1.6517e-03, 9.9835e-01],
        [9.9996e-01, 4.2539e-05],
        [6.9673e-06, 9.9999e-01],
        [1.7769e-05, 9.9998e-01],
        [2.4830e-03, 9.9752e-01],
        [9.9991e-01, 9.4227e-05],
        [3.5376e-04, 9.9965e-01],
        [9.4136e-01, 5.8636e-02],
        [1.0115e-08, 1.0000e+00],
        [9.9993e-01, 6.5046e-05],
        [5.2121e-04, 9.9948e-01],
        [9.9987e-01, 1.3097e-04],
        [3.6314e-07, 1.0000e+00],
        [3.7516e-08, 1.0000e+00],
        [3.4602e-07, 1.0000e+00],
        [7.5163e-07, 1.0000e+00],
        [9.9404e-01, 5.9551e-03],
        [7.3969e-06, 9.9999e-01],
        [1.4935e-09, 1.0000e+00],
        [1.7145e-07, 1.0000e+00],
        [2.9208e-05, 9.9997e-01],
        [2.9991e-07, 1.0000e+00],
        [4.6904e-07, 1.0000e+00],
        [9.5462e-01, 4.5382e-02],
        [9.7510e-08, 1.0000e+00],
        [3.9547e-09, 1.0000e+00],
        [1.7232e-08, 1.0000e+00],
        [9.9986e-01, 1.3907e-04],
        [9.9090e-01, 9.1029e-03],
        [9.3186e-07, 1.0000e+00],
        [1.5909e-05, 9.9998e-01],
        [1.1335e-02, 9.8866e-01],
        [9.9993e-01, 6.6344e-05],
        [3.1283e-06, 1.0000e+00],
        [9.9556e-01, 4.4394e-03],
        [9.6214e-01, 3.7863e-02],
        [1.3651e-01, 8.6349e-01],
        [3.4188e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3311e-02, 9.8669e-01],
        [4.7898e-01, 5.2102e-01],
        [8.7457e-04, 9.9913e-01],
        [2.2244e-06, 1.0000e+00],
        [7.9073e-07, 1.0000e+00],
        [1.3393e-07, 1.0000e+00],
        [9.9885e-01, 1.1512e-03],
        [1.4962e-04, 9.9985e-01],
        [2.5233e-10, 1.0000e+00],
        [4.8286e-07, 1.0000e+00],
        [1.0075e-06, 1.0000e+00],
        [4.9073e-09, 1.0000e+00],
        [9.9997e-01, 3.4773e-05],
        [1.5304e-02, 9.8470e-01],
        [9.9949e-01, 5.0829e-04],
        [9.9987e-01, 1.2904e-04],
        [9.9981e-01, 1.8703e-04],
        [1.7425e-05, 9.9998e-01],
        [2.0219e-07, 1.0000e+00],
        [9.9986e-01, 1.4308e-04],
        [1.4298e-08, 1.0000e+00],
        [6.9171e-05, 9.9993e-01],
        [4.7112e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.6607e-08, 1.0000e+00],
        [2.7458e-05, 9.9997e-01],
        [7.4892e-02, 9.2511e-01],
        [9.9981e-01, 1.8782e-04],
        [9.6564e-01, 3.4357e-02],
        [4.8862e-04, 9.9951e-01],
        [3.4127e-04, 9.9966e-01],
        [1.5057e-07, 1.0000e+00],
        [3.3622e-08, 1.0000e+00],
        [5.7537e-05, 9.9994e-01],
        [1.3956e-07, 1.0000e+00],
        [1.1135e-02, 9.8887e-01],
        [9.9300e-07, 1.0000e+00],
        [9.5336e-06, 9.9999e-01],
        [1.6929e-06, 1.0000e+00],
        [2.0896e-08, 1.0000e+00],
        [3.4791e-04, 9.9965e-01],
        [2.2115e-05, 9.9998e-01],
        [9.9987e-01, 1.2765e-04],
        [3.3255e-06, 1.0000e+00],
        [9.4529e-04, 9.9905e-01],
        [4.9006e-08, 1.0000e+00],
        [5.5611e-03, 9.9444e-01],
        [2.0180e-06, 1.0000e+00],
        [4.0258e-08, 1.0000e+00],
        [9.9786e-01, 2.1434e-03],
        [3.7323e-02, 9.6268e-01],
        [1.1676e-09, 1.0000e+00],
        [9.9988e-01, 1.2010e-04],
        [3.7017e-01, 6.2983e-01],
        [4.7056e-06, 1.0000e+00],
        [9.9973e-01, 2.7018e-04],
        [3.0218e-05, 9.9997e-01],
        [1.2247e-06, 1.0000e+00],
        [9.9990e-01, 1.0028e-04],
        [1.1602e-10, 1.0000e+00],
        [7.6532e-05, 9.9992e-01],
        [3.6066e-08, 1.0000e+00],
        [2.1350e-10, 1.0000e+00],
        [1.7681e-01, 8.2319e-01],
        [2.1204e-07, 1.0000e+00],
        [5.4085e-05, 9.9995e-01],
        [8.5735e-07, 1.0000e+00],
        [9.9264e-01, 7.3596e-03],
        [1.2043e-04, 9.9988e-01],
        [4.3246e-05, 9.9996e-01],
        [1.9629e-06, 1.0000e+00],
        [9.9986e-01, 1.3828e-04],
        [3.5239e-01, 6.4761e-01],
        [1.8525e-06, 1.0000e+00],
        [1.3513e-06, 1.0000e+00],
        [9.9940e-01, 6.0202e-04],
        [3.9861e-07, 1.0000e+00],
        [1.1921e-07, 1.0000e+00],
        [1.5522e-05, 9.9998e-01],
        [8.7577e-05, 9.9991e-01],
        [2.3985e-07, 1.0000e+00],
        [2.4715e-03, 9.9753e-01],
        [3.1008e-05, 9.9997e-01],
        [2.7849e-03, 9.9722e-01],
        [1.0256e-04, 9.9990e-01],
        [1.2396e-04, 9.9988e-01],
        [2.4559e-02, 9.7544e-01],
        [2.2586e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1033e-08, 1.0000e+00],
        [1.5955e-08, 1.0000e+00],
        [4.3483e-02, 9.5652e-01],
        [2.8559e-06, 1.0000e+00],
        [2.3855e-07, 1.0000e+00],
        [1.8021e-10, 1.0000e+00],
        [9.8194e-04, 9.9902e-01],
        [3.9074e-08, 1.0000e+00],
        [1.1224e-04, 9.9989e-01],
        [3.2527e-07, 1.0000e+00],
        [8.5260e-04, 9.9915e-01],
        [4.0224e-08, 1.0000e+00],
        [9.9999e-01, 1.1924e-05],
        [4.3865e-06, 1.0000e+00],
        [9.9863e-01, 1.3651e-03],
        [9.9948e-01, 5.2126e-04],
        [2.0482e-09, 1.0000e+00],
        [2.4482e-06, 1.0000e+00],
        [9.9965e-01, 3.4671e-04],
        [4.2015e-06, 1.0000e+00],
        [1.2996e-08, 1.0000e+00],
        [2.0118e-07, 1.0000e+00],
        [1.1723e-01, 8.8277e-01],
        [9.9942e-01, 5.7731e-04],
        [9.9990e-01, 1.0083e-04],
        [2.5737e-08, 1.0000e+00],
        [3.8951e-08, 1.0000e+00],
        [3.2329e-07, 1.0000e+00],
        [9.9986e-01, 1.4105e-04],
        [2.5417e-13, 1.0000e+00],
        [8.7497e-04, 9.9913e-01],
        [6.8095e-06, 9.9999e-01],
        [7.1180e-04, 9.9929e-01],
        [7.1471e-05, 9.9993e-01],
        [3.3032e-09, 1.0000e+00],
        [2.5466e-04, 9.9975e-01],
        [3.7027e-04, 9.9963e-01],
        [9.9168e-01, 8.3183e-03],
        [1.1644e-03, 9.9884e-01],
        [2.0890e-02, 9.7911e-01],
        [7.5995e-11, 1.0000e+00],
        [4.4200e-09, 1.0000e+00],
        [3.2199e-08, 1.0000e+00],
        [9.9996e-01, 4.3253e-05],
        [9.5420e-01, 4.5796e-02],
        [2.3404e-05, 9.9998e-01],
        [1.1460e-07, 1.0000e+00],
        [2.4121e-05, 9.9998e-01],
        [7.0189e-08, 1.0000e+00],
        [9.9990e-01, 1.0366e-04],
        [9.9988e-01, 1.2490e-04],
        [1.2738e-06, 1.0000e+00],
        [7.8061e-04, 9.9922e-01],
        [4.2319e-04, 9.9958e-01],
        [1.6924e-05, 9.9998e-01],
        [1.2759e-10, 1.0000e+00],
        [1.1673e-06, 1.0000e+00],
        [9.9987e-01, 1.3196e-04],
        [9.9993e-01, 6.9352e-05],
        [2.1182e-06, 1.0000e+00],
        [9.9999e-01, 1.0202e-05],
        [8.6862e-09, 1.0000e+00],
        [9.9986e-01, 1.3705e-04],
        [2.7456e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9934e-01, 6.5991e-04],
        [1.6329e-08, 1.0000e+00],
        [9.9984e-01, 1.6130e-04],
        [9.9594e-01, 4.0638e-03],
        [1.9940e-10, 1.0000e+00],
        [1.1532e-07, 1.0000e+00],
        [4.6439e-05, 9.9995e-01],
        [1.5777e-07, 1.0000e+00],
        [9.4721e-11, 1.0000e+00],
        [2.2237e-08, 1.0000e+00],
        [5.7145e-07, 1.0000e+00],
        [7.6717e-06, 9.9999e-01],
        [1.7379e-06, 1.0000e+00],
        [8.2096e-07, 1.0000e+00],
        [2.6439e-05, 9.9997e-01],
        [1.0309e-05, 9.9999e-01],
        [6.8005e-08, 1.0000e+00],
        [1.5914e-04, 9.9984e-01],
        [9.9995e-01, 4.7910e-05],
        [4.4816e-07, 1.0000e+00],
        [6.6299e-10, 1.0000e+00],
        [3.0882e-08, 1.0000e+00],
        [3.3797e-07, 1.0000e+00],
        [1.3750e-06, 1.0000e+00],
        [9.9940e-01, 6.0269e-04],
        [8.4641e-08, 1.0000e+00],
        [8.9778e-08, 1.0000e+00],
        [1.4440e-04, 9.9986e-01],
        [4.9823e-04, 9.9950e-01],
        [3.0999e-03, 9.9690e-01],
        [9.3752e-08, 1.0000e+00],
        [2.0014e-04, 9.9980e-01],
        [9.3595e-01, 6.4054e-02],
        [2.2857e-04, 9.9977e-01],
        [1.1762e-03, 9.9882e-01],
        [7.8861e-12, 1.0000e+00],
        [3.8168e-04, 9.9962e-01],
        [9.9834e-01, 1.6615e-03],
        [1.2454e-03, 9.9875e-01],
        [9.9943e-01, 5.6870e-04],
        [1.6329e-08, 1.0000e+00],
        [9.9973e-01, 2.7026e-04],
        [5.9577e-02, 9.4042e-01],
        [1.4094e-02, 9.8591e-01],
        [4.4602e-11, 1.0000e+00],
        [2.7330e-06, 1.0000e+00],
        [1.0318e-06, 1.0000e+00],
        [1.1493e-05, 9.9999e-01],
        [8.4970e-05, 9.9992e-01],
        [2.2990e-06, 1.0000e+00],
        [1.8263e-07, 1.0000e+00],
        [4.2103e-07, 1.0000e+00],
        [3.9380e-11, 1.0000e+00],
        [1.0146e-09, 1.0000e+00],
        [2.5178e-04, 9.9975e-01],
        [9.9993e-01, 6.9683e-05],
        [8.8709e-06, 9.9999e-01],
        [9.9946e-01, 5.4422e-04],
        [3.5303e-07, 1.0000e+00],
        [8.9391e-01, 1.0609e-01],
        [1.4744e-04, 9.9985e-01],
        [9.9951e-01, 4.9020e-04],
        [3.2010e-03, 9.9680e-01],
        [4.3809e-04, 9.9956e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.2541e-12, 1.0000e+00],
        [2.3253e-05, 9.9998e-01],
        [6.7043e-04, 9.9933e-01],
        [1.0228e-08, 1.0000e+00],
        [3.3181e-07, 1.0000e+00],
        [1.6146e-10, 1.0000e+00],
        [9.9994e-01, 5.8898e-05],
        [1.4292e-09, 1.0000e+00],
        [9.9346e-01, 6.5406e-03],
        [9.9990e-01, 1.0240e-04],
        [9.9998e-01, 1.7172e-05],
        [5.1021e-05, 9.9995e-01],
        [2.6633e-03, 9.9734e-01],
        [1.9965e-06, 1.0000e+00],
        [9.9999e-01, 8.2843e-06],
        [1.0684e-08, 1.0000e+00],
        [9.9976e-01, 2.4160e-04],
        [2.4738e-08, 1.0000e+00],
        [6.3654e-07, 1.0000e+00],
        [9.9991e-01, 8.8465e-05],
        [8.4206e-06, 9.9999e-01],
        [3.8119e-05, 9.9996e-01],
        [3.2201e-06, 1.0000e+00],
        [7.2785e-04, 9.9927e-01],
        [9.5462e-07, 1.0000e+00],
        [4.1611e-04, 9.9958e-01],
        [1.4799e-07, 1.0000e+00],
        [9.9991e-01, 8.9271e-05],
        [6.0305e-08, 1.0000e+00],
        [4.3378e-10, 1.0000e+00],
        [7.9828e-09, 1.0000e+00],
        [1.3886e-07, 1.0000e+00],
        [1.2111e-07, 1.0000e+00],
        [9.9992e-01, 8.0538e-05],
        [8.9643e-07, 1.0000e+00],
        [9.2296e-01, 7.7041e-02],
        [5.0031e-09, 1.0000e+00],
        [9.9891e-01, 1.0908e-03],
        [1.3737e-08, 1.0000e+00],
        [1.5344e-06, 1.0000e+00],
        [2.5828e-10, 1.0000e+00],
        [4.1691e-07, 1.0000e+00],
        [1.9703e-10, 1.0000e+00],
        [3.2939e-07, 1.0000e+00],
        [9.9981e-01, 1.8523e-04],
        [4.9898e-07, 1.0000e+00],
        [1.0246e-06, 1.0000e+00],
        [9.9999e-01, 1.3069e-05],
        [8.5491e-03, 9.9145e-01],
        [1.2474e-02, 9.8753e-01],
        [6.8991e-08, 1.0000e+00],
        [6.7063e-07, 1.0000e+00],
        [2.7325e-08, 1.0000e+00],
        [2.0010e-06, 1.0000e+00],
        [9.9978e-01, 2.2345e-04],
        [1.9924e-05, 9.9998e-01],
        [4.0740e-06, 1.0000e+00],
        [3.4342e-06, 1.0000e+00],
        [9.1755e-01, 8.2445e-02],
        [9.9992e-01, 7.7696e-05],
        [9.7035e-08, 1.0000e+00],
        [1.4628e-10, 1.0000e+00],
        [1.7359e-04, 9.9983e-01],
        [4.2038e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9975e-01, 2.5463e-04],
        [9.9988e-01, 1.2445e-04],
        [2.0028e-06, 1.0000e+00],
        [4.3272e-12, 1.0000e+00],
        [8.9578e-04, 9.9910e-01],
        [9.1043e-07, 1.0000e+00],
        [4.3140e-07, 1.0000e+00],
        [3.5240e-08, 1.0000e+00],
        [2.0649e-04, 9.9979e-01],
        [4.2171e-04, 9.9958e-01],
        [1.6544e-04, 9.9983e-01],
        [9.9995e-01, 5.4109e-05],
        [1.9454e-05, 9.9998e-01],
        [2.2106e-07, 1.0000e+00],
        [4.0575e-03, 9.9594e-01],
        [9.2096e-05, 9.9991e-01],
        [8.1926e-02, 9.1807e-01],
        [2.5036e-08, 1.0000e+00],
        [2.5138e-06, 1.0000e+00],
        [1.7776e-07, 1.0000e+00],
        [1.2920e-04, 9.9987e-01],
        [3.5250e-07, 1.0000e+00],
        [2.2006e-08, 1.0000e+00],
        [8.9484e-08, 1.0000e+00],
        [1.8792e-10, 1.0000e+00],
        [2.3193e-09, 1.0000e+00],
        [1.3836e-04, 9.9986e-01],
        [2.1124e-06, 1.0000e+00],
        [1.8397e-05, 9.9998e-01],
        [2.8138e-05, 9.9997e-01],
        [5.3056e-06, 9.9999e-01],
        [2.3399e-09, 1.0000e+00],
        [7.8180e-07, 1.0000e+00],
        [1.5691e-04, 9.9984e-01],
        [7.9988e-08, 1.0000e+00],
        [1.9353e-05, 9.9998e-01],
        [9.9989e-01, 1.1040e-04],
        [1.4486e-09, 1.0000e+00],
        [8.3002e-07, 1.0000e+00],
        [4.5737e-05, 9.9995e-01],
        [3.1406e-06, 1.0000e+00],
        [7.6482e-05, 9.9992e-01],
        [5.9842e-10, 1.0000e+00],
        [9.9993e-01, 7.0498e-05],
        [2.2128e-05, 9.9998e-01],
        [1.6942e-06, 1.0000e+00],
        [2.1065e-04, 9.9979e-01],
        [1.7675e-06, 1.0000e+00],
        [7.3615e-05, 9.9993e-01],
        [5.7132e-11, 1.0000e+00],
        [9.9981e-01, 1.9459e-04],
        [4.3282e-08, 1.0000e+00],
        [9.9973e-01, 2.6849e-04],
        [4.2225e-08, 1.0000e+00],
        [9.9992e-01, 8.2519e-05],
        [4.0969e-04, 9.9959e-01],
        [2.3866e-08, 1.0000e+00],
        [1.0958e-06, 1.0000e+00],
        [2.7230e-04, 9.9973e-01],
        [4.9058e-08, 1.0000e+00],
        [1.5562e-07, 1.0000e+00],
        [9.9953e-01, 4.6988e-04],
        [3.1745e-10, 1.0000e+00],
        [9.5213e-01, 4.7868e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2658e-06, 1.0000e+00],
        [9.9984e-01, 1.5831e-04],
        [1.5500e-08, 1.0000e+00],
        [9.9990e-01, 9.8263e-05],
        [2.8867e-10, 1.0000e+00],
        [1.3000e-07, 1.0000e+00],
        [9.9997e-01, 3.2392e-05],
        [1.3902e-11, 1.0000e+00],
        [2.3209e-03, 9.9768e-01],
        [2.8270e-08, 1.0000e+00],
        [9.5728e-05, 9.9990e-01],
        [6.1855e-07, 1.0000e+00],
        [1.0076e-07, 1.0000e+00],
        [2.0397e-05, 9.9998e-01],
        [9.1360e-07, 1.0000e+00],
        [9.9994e-01, 6.1037e-05],
        [9.9981e-01, 1.9089e-04],
        [9.9005e-01, 9.9507e-03],
        [7.9828e-05, 9.9992e-01],
        [5.9001e-09, 1.0000e+00],
        [9.5399e-01, 4.6010e-02],
        [2.9369e-06, 1.0000e+00],
        [9.9979e-01, 2.0853e-04],
        [1.3044e-03, 9.9870e-01],
        [9.9995e-01, 4.7221e-05],
        [9.3538e-05, 9.9991e-01],
        [6.6440e-08, 1.0000e+00],
        [7.5481e-10, 1.0000e+00],
        [2.2682e-01, 7.7318e-01],
        [1.5511e-11, 1.0000e+00],
        [1.8207e-03, 9.9818e-01],
        [1.4032e-09, 1.0000e+00],
        [6.6267e-11, 1.0000e+00],
        [6.2279e-10, 1.0000e+00],
        [3.5082e-06, 1.0000e+00],
        [9.9994e-01, 6.4267e-05],
        [1.6990e-03, 9.9830e-01],
        [2.0497e-07, 1.0000e+00],
        [1.4174e-07, 1.0000e+00],
        [6.2403e-06, 9.9999e-01],
        [4.5700e-07, 1.0000e+00],
        [8.6473e-09, 1.0000e+00],
        [1.3943e-08, 1.0000e+00],
        [3.8094e-04, 9.9962e-01],
        [1.7983e-04, 9.9982e-01],
        [3.6427e-09, 1.0000e+00],
        [9.9972e-01, 2.8470e-04],
        [5.5737e-09, 1.0000e+00],
        [8.1758e-07, 1.0000e+00],
        [9.9998e-01, 2.4403e-05],
        [4.4050e-09, 1.0000e+00],
        [2.5943e-08, 1.0000e+00],
        [1.3241e-04, 9.9987e-01],
        [9.9905e-01, 9.4782e-04],
        [7.0303e-05, 9.9993e-01],
        [4.4667e-05, 9.9996e-01],
        [2.6208e-06, 1.0000e+00],
        [1.6891e-06, 1.0000e+00],
        [4.8956e-01, 5.1044e-01],
        [5.3276e-01, 4.6724e-01],
        [1.2015e-03, 9.9880e-01],
        [1.1305e-05, 9.9999e-01],
        [8.4403e-07, 1.0000e+00],
        [8.4710e-05, 9.9992e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0949e-07, 1.0000e+00],
        [2.0011e-07, 1.0000e+00],
        [1.8265e-08, 1.0000e+00],
        [2.7337e-10, 1.0000e+00],
        [9.9993e-01, 6.8305e-05],
        [2.6363e-06, 1.0000e+00],
        [4.8296e-05, 9.9995e-01],
        [2.0416e-09, 1.0000e+00],
        [2.5151e-10, 1.0000e+00],
        [9.4749e-01, 5.2507e-02],
        [8.5534e-01, 1.4466e-01],
        [2.9225e-05, 9.9997e-01],
        [9.9771e-01, 2.2886e-03],
        [9.5711e-08, 1.0000e+00],
        [9.0667e-09, 1.0000e+00],
        [9.9994e-01, 5.8841e-05],
        [1.3428e-04, 9.9987e-01],
        [9.9979e-01, 2.1230e-04],
        [9.9935e-01, 6.5000e-04],
        [1.6336e-04, 9.9984e-01],
        [1.2010e-05, 9.9999e-01],
        [1.8293e-10, 1.0000e+00],
        [1.6870e-09, 1.0000e+00],
        [1.5170e-06, 1.0000e+00],
        [4.6126e-01, 5.3874e-01],
        [9.9958e-01, 4.1903e-04],
        [9.6195e-01, 3.8052e-02],
        [6.4367e-07, 1.0000e+00],
        [1.1783e-07, 1.0000e+00],
        [9.8651e-04, 9.9901e-01],
        [4.5444e-07, 1.0000e+00],
        [9.3613e-05, 9.9991e-01],
        [9.9947e-01, 5.2504e-04],
        [1.5393e-07, 1.0000e+00],
        [6.8584e-10, 1.0000e+00],
        [6.6653e-06, 9.9999e-01],
        [3.1244e-07, 1.0000e+00],
        [7.4587e-06, 9.9999e-01],
        [4.6103e-08, 1.0000e+00],
        [9.9933e-01, 6.6873e-04],
        [2.4974e-04, 9.9975e-01],
        [5.4252e-10, 1.0000e+00],
        [9.9994e-01, 6.2393e-05],
        [5.3217e-10, 1.0000e+00],
        [8.9991e-14, 1.0000e+00],
        [4.9673e-07, 1.0000e+00],
        [9.9996e-01, 4.1042e-05],
        [9.9949e-01, 5.1001e-04],
        [1.7164e-09, 1.0000e+00],
        [4.1230e-06, 1.0000e+00],
        [1.9166e-08, 1.0000e+00],
        [1.1075e-05, 9.9999e-01],
        [3.3426e-06, 1.0000e+00],
        [6.5776e-07, 1.0000e+00],
        [1.0028e-07, 1.0000e+00],
        [9.9990e-01, 9.6265e-05],
        [1.9475e-09, 1.0000e+00],
        [3.4977e-10, 1.0000e+00],
        [9.9647e-01, 3.5298e-03],
        [8.9534e-07, 1.0000e+00],
        [1.0692e-04, 9.9989e-01],
        [3.6979e-09, 1.0000e+00],
        [1.0689e-10, 1.0000e+00],
        [1.0116e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9962e-01, 3.7604e-04],
        [3.4435e-07, 1.0000e+00],
        [4.1191e-10, 1.0000e+00],
        [9.9996e-01, 4.4189e-05],
        [9.2690e-01, 7.3104e-02],
        [7.8037e-06, 9.9999e-01],
        [3.3429e-01, 6.6571e-01],
        [9.9990e-01, 9.8587e-05],
        [1.1764e-04, 9.9988e-01],
        [5.2520e-03, 9.9475e-01],
        [2.4343e-09, 1.0000e+00],
        [1.7848e-06, 1.0000e+00],
        [2.8616e-07, 1.0000e+00],
        [1.7818e-04, 9.9982e-01],
        [1.1192e-08, 1.0000e+00],
        [3.3596e-04, 9.9966e-01],
        [9.9970e-01, 2.9918e-04],
        [9.9998e-01, 1.7991e-05],
        [2.5039e-09, 1.0000e+00],
        [1.2737e-03, 9.9873e-01],
        [9.9949e-01, 5.1116e-04],
        [2.7413e-07, 1.0000e+00],
        [1.7224e-02, 9.8278e-01],
        [1.5703e-09, 1.0000e+00],
        [6.2052e-07, 1.0000e+00],
        [5.9555e-07, 1.0000e+00],
        [1.2722e-11, 1.0000e+00],
        [9.9215e-01, 7.8513e-03],
        [1.2068e-06, 1.0000e+00],
        [9.2474e-08, 1.0000e+00],
        [3.3023e-07, 1.0000e+00],
        [4.0031e-05, 9.9996e-01],
        [6.5214e-10, 1.0000e+00],
        [9.2421e-05, 9.9991e-01],
        [1.5954e-07, 1.0000e+00],
        [1.6764e-08, 1.0000e+00],
        [1.7288e-06, 1.0000e+00],
        [1.6393e-03, 9.9836e-01],
        [6.7924e-09, 1.0000e+00],
        [4.1837e-08, 1.0000e+00],
        [3.9569e-01, 6.0431e-01],
        [6.9959e-07, 1.0000e+00],
        [5.2622e-10, 1.0000e+00],
        [6.5942e-04, 9.9934e-01],
        [1.8446e-08, 1.0000e+00],
        [1.4576e-09, 1.0000e+00],
        [3.5757e-03, 9.9642e-01],
        [3.2937e-10, 1.0000e+00],
        [2.3514e-08, 1.0000e+00],
        [1.9580e-03, 9.9804e-01],
        [5.9842e-06, 9.9999e-01],
        [1.4939e-04, 9.9985e-01],
        [1.7157e-05, 9.9998e-01],
        [1.2540e-06, 1.0000e+00],
        [6.9202e-08, 1.0000e+00],
        [1.0963e-07, 1.0000e+00],
        [5.9887e-10, 1.0000e+00],
        [2.1158e-07, 1.0000e+00],
        [1.4836e-08, 1.0000e+00],
        [9.9990e-01, 9.4980e-05],
        [6.5256e-13, 1.0000e+00],
        [6.2771e-04, 9.9937e-01],
        [5.3021e-04, 9.9947e-01],
        [1.2786e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.8442e-07, 1.0000e+00],
        [9.7678e-08, 1.0000e+00],
        [5.4110e-02, 9.4589e-01],
        [1.1382e-09, 1.0000e+00],
        [9.1051e-05, 9.9991e-01],
        [9.9996e-01, 3.5928e-05],
        [8.9210e-05, 9.9991e-01],
        [1.2698e-05, 9.9999e-01],
        [9.9984e-01, 1.5919e-04],
        [9.9637e-01, 3.6328e-03],
        [1.6091e-06, 1.0000e+00],
        [2.9023e-02, 9.7098e-01],
        [3.1725e-11, 1.0000e+00],
        [6.4512e-08, 1.0000e+00],
        [9.9996e-01, 4.4845e-05],
        [1.6394e-04, 9.9984e-01],
        [2.5308e-06, 1.0000e+00],
        [2.8130e-06, 1.0000e+00],
        [9.9994e-01, 6.1602e-05],
        [4.1382e-09, 1.0000e+00],
        [2.1133e-06, 1.0000e+00],
        [1.1799e-10, 1.0000e+00],
        [2.2292e-09, 1.0000e+00],
        [9.1698e-08, 1.0000e+00],
        [1.7259e-09, 1.0000e+00],
        [1.1341e-03, 9.9887e-01],
        [6.6941e-06, 9.9999e-01],
        [2.0354e-05, 9.9998e-01],
        [1.5736e-05, 9.9998e-01],
        [1.3195e-04, 9.9987e-01],
        [1.2356e-06, 1.0000e+00],
        [9.9819e-01, 1.8122e-03],
        [4.1573e-11, 1.0000e+00],
        [2.3249e-06, 1.0000e+00],
        [9.9974e-01, 2.6267e-04],
        [7.0346e-11, 1.0000e+00],
        [5.3478e-01, 4.6522e-01],
        [3.1788e-01, 6.8212e-01],
        [1.3110e-05, 9.9999e-01],
        [3.7205e-09, 1.0000e+00],
        [9.9992e-01, 8.0776e-05],
        [3.0560e-06, 1.0000e+00],
        [7.2554e-10, 1.0000e+00],
        [6.6099e-07, 1.0000e+00],
        [6.8305e-11, 1.0000e+00],
        [1.7376e-04, 9.9983e-01],
        [9.9843e-01, 1.5744e-03],
        [1.2284e-09, 1.0000e+00],
        [8.5330e-09, 1.0000e+00],
        [6.8761e-07, 1.0000e+00],
        [1.1652e-07, 1.0000e+00],
        [1.3957e-11, 1.0000e+00],
        [2.0694e-05, 9.9998e-01],
        [8.9382e-06, 9.9999e-01],
        [5.0107e-08, 1.0000e+00],
        [1.7992e-08, 1.0000e+00],
        [1.1250e-02, 9.8875e-01],
        [2.1748e-08, 1.0000e+00],
        [3.7524e-10, 1.0000e+00],
        [1.6642e-05, 9.9998e-01],
        [9.9976e-01, 2.4407e-04],
        [1.4015e-09, 1.0000e+00],
        [2.6291e-07, 1.0000e+00],
        [6.6689e-04, 9.9933e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9985e-01, 1.5133e-04],
        [7.8579e-06, 9.9999e-01],
        [5.9037e-11, 1.0000e+00],
        [9.8624e-01, 1.3759e-02],
        [9.9997e-01, 2.5001e-05],
        [2.5377e-10, 1.0000e+00],
        [9.1975e-08, 1.0000e+00],
        [9.9998e-01, 1.5625e-05],
        [5.8735e-01, 4.1265e-01],
        [1.6782e-13, 1.0000e+00],
        [1.0814e-06, 1.0000e+00],
        [4.8853e-06, 1.0000e+00],
        [6.6473e-07, 1.0000e+00],
        [9.9993e-01, 7.4684e-05],
        [7.5683e-08, 1.0000e+00],
        [4.2844e-08, 1.0000e+00],
        [3.8341e-10, 1.0000e+00],
        [8.9459e-07, 1.0000e+00],
        [4.5044e-04, 9.9955e-01],
        [5.9208e-10, 1.0000e+00],
        [6.5693e-07, 1.0000e+00],
        [8.8093e-07, 1.0000e+00],
        [2.1641e-05, 9.9998e-01],
        [9.9986e-01, 1.3936e-04],
        [9.9458e-01, 5.4208e-03],
        [6.7628e-08, 1.0000e+00],
        [9.4129e-05, 9.9991e-01],
        [1.8513e-04, 9.9981e-01],
        [5.2368e-06, 9.9999e-01],
        [1.0000e+00, 3.5232e-06],
        [9.9991e-01, 8.6114e-05],
        [8.9896e-08, 1.0000e+00],
        [6.6183e-07, 1.0000e+00],
        [1.0000e+00, 3.3475e-06],
        [5.6409e-15, 1.0000e+00],
        [9.9866e-01, 1.3425e-03],
        [4.0963e-06, 1.0000e+00],
        [9.9938e-01, 6.2340e-04],
        [5.2761e-08, 1.0000e+00],
        [7.5010e-06, 9.9999e-01],
        [2.2853e-09, 1.0000e+00],
        [1.8294e-08, 1.0000e+00],
        [9.9996e-01, 4.3313e-05],
        [1.5724e-04, 9.9984e-01],
        [7.0014e-10, 1.0000e+00],
        [2.7421e-05, 9.9997e-01],
        [1.7513e-11, 1.0000e+00],
        [8.8550e-07, 1.0000e+00],
        [7.4256e-09, 1.0000e+00],
        [1.9157e-03, 9.9808e-01],
        [2.0461e-04, 9.9980e-01],
        [9.9986e-01, 1.3934e-04],
        [1.1755e-11, 1.0000e+00],
        [9.9974e-01, 2.5636e-04],
        [3.3076e-09, 1.0000e+00],
        [4.4134e-06, 1.0000e+00],
        [1.2908e-07, 1.0000e+00],
        [2.9192e-08, 1.0000e+00],
        [5.5476e-10, 1.0000e+00],
        [4.0008e-04, 9.9960e-01],
        [6.8383e-07, 1.0000e+00],
        [2.4011e-10, 1.0000e+00],
        [5.2751e-01, 4.7249e-01],
        [9.6051e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9985e-01, 1.5385e-04],
        [2.4588e-03, 9.9754e-01],
        [9.9976e-01, 2.4428e-04],
        [1.8098e-04, 9.9982e-01],
        [5.6440e-07, 1.0000e+00],
        [9.9944e-01, 5.6250e-04],
        [7.7537e-05, 9.9992e-01],
        [6.3593e-08, 1.0000e+00],
        [8.3879e-06, 9.9999e-01],
        [1.3595e-05, 9.9999e-01],
        [1.8815e-07, 1.0000e+00],
        [9.9982e-01, 1.7530e-04],
        [5.8847e-05, 9.9994e-01],
        [1.2903e-08, 1.0000e+00],
        [9.9977e-01, 2.2582e-04],
        [1.0277e-07, 1.0000e+00],
        [1.8224e-04, 9.9982e-01],
        [9.7407e-01, 2.5932e-02],
        [2.1554e-07, 1.0000e+00],
        [5.7220e-05, 9.9994e-01],
        [3.0231e-07, 1.0000e+00],
        [9.8048e-01, 1.9517e-02],
        [1.4824e-02, 9.8518e-01],
        [4.4697e-04, 9.9955e-01],
        [1.8828e-06, 1.0000e+00],
        [4.6175e-08, 1.0000e+00],
        [1.0780e-04, 9.9989e-01],
        [9.9983e-01, 1.7215e-04],
        [1.8940e-08, 1.0000e+00],
        [7.3570e-09, 1.0000e+00],
        [1.2735e-08, 1.0000e+00],
        [9.9996e-01, 3.8993e-05],
        [4.3339e-06, 1.0000e+00],
        [9.9965e-01, 3.5470e-04],
        [9.9990e-01, 9.8594e-05],
        [5.2476e-13, 1.0000e+00],
        [9.9987e-01, 1.3411e-04],
        [9.9776e-01, 2.2378e-03],
        [9.9966e-01, 3.4316e-04],
        [1.8283e-08, 1.0000e+00],
        [2.4187e-06, 1.0000e+00],
        [3.5446e-05, 9.9996e-01],
        [2.2193e-08, 1.0000e+00],
        [4.6834e-04, 9.9953e-01],
        [9.9800e-01, 2.0022e-03],
        [7.0896e-05, 9.9993e-01],
        [3.1921e-11, 1.0000e+00],
        [3.5092e-07, 1.0000e+00],
        [9.9985e-01, 1.5150e-04],
        [3.5898e-03, 9.9641e-01],
        [1.7726e-07, 1.0000e+00],
        [9.9819e-01, 1.8065e-03],
        [9.9985e-01, 1.5067e-04],
        [1.0508e-05, 9.9999e-01],
        [1.2718e-05, 9.9999e-01],
        [3.2237e-02, 9.6776e-01],
        [4.2315e-08, 1.0000e+00],
        [1.4199e-05, 9.9999e-01],
        [2.1044e-07, 1.0000e+00],
        [3.0504e-08, 1.0000e+00],
        [2.5274e-02, 9.7473e-01],
        [6.2884e-07, 1.0000e+00],
        [8.6204e-01, 1.3796e-01],
        [5.4256e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0000e+00, 5.9222e-07],
        [6.8038e-07, 1.0000e+00],
        [1.1777e-04, 9.9988e-01],
        [3.5067e-05, 9.9996e-01],
        [1.9258e-03, 9.9807e-01],
        [8.1242e-10, 1.0000e+00],
        [9.9950e-01, 4.9852e-04],
        [4.9354e-10, 1.0000e+00],
        [1.4067e-06, 1.0000e+00],
        [6.0573e-04, 9.9939e-01],
        [1.2779e-04, 9.9987e-01],
        [2.6776e-06, 1.0000e+00],
        [9.9978e-01, 2.1809e-04],
        [9.9952e-01, 4.7666e-04],
        [2.4747e-04, 9.9975e-01],
        [9.9993e-01, 7.3084e-05],
        [9.9954e-01, 4.5510e-04],
        [1.5332e-12, 1.0000e+00],
        [1.2614e-04, 9.9987e-01],
        [7.2675e-05, 9.9993e-01],
        [9.9934e-01, 6.6175e-04],
        [1.3183e-05, 9.9999e-01],
        [3.6714e-01, 6.3286e-01],
        [9.9989e-01, 1.1365e-04],
        [7.7786e-06, 9.9999e-01],
        [7.9123e-11, 1.0000e+00],
        [2.9309e-08, 1.0000e+00],
        [9.9998e-01, 2.1436e-05],
        [9.9976e-01, 2.3897e-04],
        [4.5428e-05, 9.9995e-01],
        [1.1655e-06, 1.0000e+00],
        [7.8609e-10, 1.0000e+00],
        [3.9570e-02, 9.6043e-01],
        [4.3463e-08, 1.0000e+00],
        [9.9995e-01, 4.6757e-05],
        [3.8281e-03, 9.9617e-01],
        [9.9988e-01, 1.1960e-04],
        [1.1857e-06, 1.0000e+00],
        [9.9980e-01, 1.9723e-04],
        [9.9977e-01, 2.3017e-04],
        [2.7749e-12, 1.0000e+00],
        [3.9533e-06, 1.0000e+00],
        [1.5940e-06, 1.0000e+00],
        [1.0192e-11, 1.0000e+00],
        [1.1241e-10, 1.0000e+00],
        [3.6536e-05, 9.9996e-01],
        [4.7807e-05, 9.9995e-01],
        [1.3311e-02, 9.8669e-01],
        [1.8316e-03, 9.9817e-01],
        [8.2041e-05, 9.9992e-01],
        [9.9983e-01, 1.6632e-04],
        [7.6723e-09, 1.0000e+00],
        [9.9969e-01, 3.1343e-04],
        [5.0845e-04, 9.9949e-01],
        [9.9504e-01, 4.9559e-03],
        [9.0464e-04, 9.9910e-01],
        [1.9708e-07, 1.0000e+00],
        [1.1297e-07, 1.0000e+00],
        [9.9972e-01, 2.7654e-04],
        [3.9073e-06, 1.0000e+00],
        [4.1544e-06, 1.0000e+00],
        [1.5491e-05, 9.9998e-01],
        [9.9996e-01, 4.4410e-05],
        [9.9992e-01, 7.5232e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7989e-08, 1.0000e+00],
        [4.2147e-08, 1.0000e+00],
        [9.9977e-01, 2.2643e-04],
        [2.2369e-11, 1.0000e+00],
        [9.9959e-01, 4.1461e-04],
        [1.1632e-11, 1.0000e+00],
        [9.9959e-01, 4.0517e-04],
        [7.0523e-05, 9.9993e-01],
        [1.5068e-03, 9.9849e-01],
        [1.2373e-05, 9.9999e-01],
        [9.9994e-01, 6.4185e-05],
        [3.2331e-07, 1.0000e+00],
        [2.3984e-04, 9.9976e-01],
        [2.9113e-10, 1.0000e+00],
        [2.0549e-09, 1.0000e+00],
        [1.5376e-06, 1.0000e+00],
        [2.2244e-04, 9.9978e-01],
        [8.5853e-06, 9.9999e-01],
        [9.9943e-01, 5.6919e-04],
        [1.2467e-04, 9.9988e-01],
        [6.3728e-08, 1.0000e+00],
        [5.5323e-09, 1.0000e+00],
        [3.3643e-06, 1.0000e+00],
        [3.7063e-05, 9.9996e-01],
        [6.4533e-08, 1.0000e+00],
        [2.5017e-06, 1.0000e+00],
        [2.0377e-05, 9.9998e-01],
        [1.5285e-06, 1.0000e+00],
        [1.5619e-09, 1.0000e+00],
        [1.7958e-06, 1.0000e+00],
        [9.9995e-01, 4.5683e-05],
        [1.8831e-09, 1.0000e+00],
        [9.9997e-01, 2.7731e-05],
        [9.9982e-01, 1.8088e-04],
        [6.1458e-08, 1.0000e+00],
        [5.3552e-04, 9.9946e-01],
        [1.2571e-06, 1.0000e+00],
        [2.9620e-07, 1.0000e+00],
        [1.2483e-06, 1.0000e+00],
        [9.9834e-01, 1.6565e-03],
        [2.5933e-07, 1.0000e+00],
        [3.0576e-06, 1.0000e+00],
        [1.0710e-07, 1.0000e+00],
        [9.9984e-01, 1.5802e-04],
        [7.6862e-04, 9.9923e-01],
        [1.4230e-05, 9.9999e-01],
        [2.9222e-09, 1.0000e+00],
        [5.8525e-02, 9.4148e-01],
        [1.1719e-07, 1.0000e+00],
        [9.1877e-01, 8.1232e-02],
        [4.4091e-02, 9.5591e-01],
        [2.7053e-10, 1.0000e+00],
        [9.9993e-01, 7.2358e-05],
        [8.7822e-08, 1.0000e+00],
        [1.7583e-09, 1.0000e+00],
        [7.0562e-13, 1.0000e+00],
        [1.1120e-05, 9.9999e-01],
        [1.3924e-06, 1.0000e+00],
        [2.4528e-06, 1.0000e+00],
        [2.3048e-08, 1.0000e+00],
        [3.9754e-07, 1.0000e+00],
        [5.6254e-04, 9.9944e-01],
        [9.7442e-03, 9.9026e-01],
        [9.9945e-01, 5.5253e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[4.3471e-09, 1.0000e+00],
        [1.3056e-04, 9.9987e-01],
        [5.0280e-05, 9.9995e-01],
        [8.9785e-09, 1.0000e+00],
        [1.2251e-07, 1.0000e+00],
        [1.1381e-05, 9.9999e-01],
        [8.5046e-01, 1.4954e-01],
        [1.6634e-03, 9.9834e-01],
        [9.9979e-01, 2.0623e-04],
        [1.9236e-11, 1.0000e+00],
        [4.0453e-09, 1.0000e+00],
        [9.9887e-01, 1.1345e-03],
        [1.5094e-07, 1.0000e+00],
        [2.8184e-06, 1.0000e+00],
        [4.5699e-08, 1.0000e+00],
        [5.4664e-06, 9.9999e-01],
        [2.4228e-07, 1.0000e+00],
        [6.7771e-01, 3.2229e-01],
        [9.9972e-01, 2.7914e-04],
        [9.9994e-01, 5.7699e-05],
        [1.1653e-08, 1.0000e+00],
        [3.2828e-07, 1.0000e+00],
        [3.8425e-06, 1.0000e+00],
        [1.3946e-06, 1.0000e+00],
        [7.5836e-07, 1.0000e+00],
        [3.8131e-05, 9.9996e-01],
        [1.1562e-05, 9.9999e-01],
        [1.7647e-09, 1.0000e+00],
        [7.7196e-10, 1.0000e+00],
        [1.8768e-07, 1.0000e+00],
        [1.0959e-06, 1.0000e+00],
        [3.1621e-07, 1.0000e+00],
        [1.9259e-09, 1.0000e+00],
        [8.3762e-08, 1.0000e+00],
        [1.1466e-07, 1.0000e+00],
        [1.6074e-05, 9.9998e-01],
        [1.5390e-04, 9.9985e-01],
        [4.1356e-05, 9.9996e-01],
        [1.9414e-06, 1.0000e+00],
        [1.8829e-08, 1.0000e+00],
        [9.9751e-01, 2.4894e-03],
        [2.4366e-07, 1.0000e+00],
        [2.0605e-07, 1.0000e+00],
        [9.9991e-01, 8.5239e-05],
        [1.1321e-06, 1.0000e+00],
        [1.3190e-05, 9.9999e-01],
        [1.9146e-09, 1.0000e+00],
        [9.9977e-01, 2.3024e-04],
        [9.9947e-01, 5.3079e-04],
        [5.1468e-06, 9.9999e-01],
        [9.9976e-01, 2.3974e-04],
        [3.3681e-06, 1.0000e+00],
        [1.3390e-06, 1.0000e+00],
        [9.9927e-01, 7.2950e-04],
        [2.9215e-05, 9.9997e-01],
        [9.9982e-01, 1.8109e-04],
        [2.5212e-06, 1.0000e+00],
        [5.6957e-04, 9.9943e-01],
        [5.0249e-11, 1.0000e+00],
        [9.9985e-01, 1.5235e-04],
        [9.9447e-01, 5.5304e-03],
        [9.9976e-01, 2.4289e-04],
        [9.9983e-01, 1.7121e-04],
        [7.9013e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9834e-01, 1.6585e-03],
        [4.9148e-09, 1.0000e+00],
        [2.3539e-08, 1.0000e+00],
        [2.9554e-09, 1.0000e+00],
        [9.9991e-01, 8.8994e-05],
        [9.9991e-01, 9.4565e-05],
        [1.5885e-07, 1.0000e+00],
        [1.5974e-07, 1.0000e+00],
        [9.9972e-01, 2.8423e-04],
        [9.9999e-01, 1.3761e-05],
        [9.9944e-01, 5.6408e-04],
        [9.4034e-10, 1.0000e+00],
        [9.5157e-08, 1.0000e+00],
        [7.4699e-01, 2.5301e-01],
        [4.2827e-06, 1.0000e+00],
        [9.6278e-11, 1.0000e+00],
        [5.7229e-08, 1.0000e+00],
        [8.0872e-07, 1.0000e+00],
        [9.9992e-01, 8.4798e-05],
        [9.9993e-01, 7.0500e-05],
        [7.6173e-10, 1.0000e+00],
        [1.1901e-11, 1.0000e+00],
        [4.1947e-07, 1.0000e+00],
        [1.7160e-04, 9.9983e-01],
        [3.2637e-06, 1.0000e+00],
        [1.0762e-06, 1.0000e+00],
        [1.3780e-08, 1.0000e+00],
        [1.3494e-06, 1.0000e+00],
        [9.9950e-01, 5.0280e-04],
        [1.2141e-04, 9.9988e-01],
        [4.2746e-07, 1.0000e+00],
        [9.9997e-01, 2.8047e-05],
        [9.9975e-01, 2.4794e-04],
        [9.7937e-08, 1.0000e+00],
        [9.9973e-01, 2.6642e-04],
        [4.7813e-06, 1.0000e+00],
        [9.4318e-01, 5.6816e-02],
        [9.9993e-01, 7.4932e-05],
        [1.4496e-09, 1.0000e+00],
        [9.9992e-01, 7.5755e-05],
        [9.9985e-01, 1.5161e-04],
        [6.1499e-07, 1.0000e+00],
        [3.0605e-05, 9.9997e-01],
        [4.1115e-10, 1.0000e+00],
        [2.3912e-10, 1.0000e+00],
        [9.9863e-01, 1.3677e-03],
        [1.9535e-07, 1.0000e+00],
        [1.4578e-04, 9.9985e-01],
        [9.9983e-01, 1.6738e-04],
        [9.9981e-01, 1.9115e-04],
        [5.4777e-07, 1.0000e+00],
        [9.9994e-01, 6.2303e-05],
        [1.9401e-05, 9.9998e-01],
        [9.9983e-01, 1.7052e-04],
        [1.4065e-04, 9.9986e-01],
        [8.5459e-05, 9.9991e-01],
        [9.9982e-01, 1.7911e-04],
        [5.4496e-11, 1.0000e+00],
        [7.1843e-06, 9.9999e-01],
        [9.4936e-01, 5.0638e-02],
        [9.8997e-01, 1.0029e-02],
        [1.0866e-06, 1.0000e+00],
        [9.9960e-01, 3.9644e-04],
        [6.1792e-03, 9.9382e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0984e-08, 1.0000e+00],
        [9.9997e-01, 2.7330e-05],
        [6.0801e-07, 1.0000e+00],
        [5.1844e-06, 9.9999e-01],
        [9.9997e-01, 2.5237e-05],
        [9.9996e-01, 3.9693e-05],
        [1.8213e-10, 1.0000e+00],
        [2.1665e-07, 1.0000e+00],
        [1.5027e-04, 9.9985e-01],
        [2.0544e-11, 1.0000e+00],
        [5.2102e-12, 1.0000e+00],
        [4.1017e-07, 1.0000e+00],
        [9.9997e-01, 2.6237e-05],
        [1.2551e-06, 1.0000e+00],
        [7.4806e-03, 9.9252e-01],
        [1.5466e-06, 1.0000e+00],
        [1.7315e-08, 1.0000e+00],
        [9.5900e-01, 4.0998e-02],
        [1.2274e-09, 1.0000e+00],
        [5.7188e-08, 1.0000e+00],
        [8.2683e-07, 1.0000e+00],
        [1.8617e-04, 9.9981e-01],
        [9.9827e-01, 1.7324e-03],
        [7.7828e-08, 1.0000e+00],
        [9.9986e-01, 1.4122e-04],
        [1.8718e-09, 1.0000e+00],
        [4.2314e-03, 9.9577e-01],
        [1.5989e-05, 9.9998e-01],
        [1.8147e-03, 9.9819e-01],
        [2.8705e-04, 9.9971e-01],
        [1.1019e-06, 1.0000e+00],
        [5.7566e-09, 1.0000e+00],
        [9.9517e-01, 4.8292e-03],
        [8.3606e-08, 1.0000e+00],
        [7.8869e-04, 9.9921e-01],
        [3.1685e-06, 1.0000e+00],
        [3.5053e-05, 9.9996e-01],
        [9.9995e-01, 4.7265e-05],
        [1.0658e-11, 1.0000e+00],
        [5.4149e-06, 9.9999e-01],
        [9.9996e-01, 4.2835e-05],
        [6.0360e-15, 1.0000e+00],
        [9.9997e-01, 3.4898e-05],
        [4.7699e-02, 9.5230e-01],
        [9.9992e-01, 7.5869e-05],
        [2.3367e-04, 9.9977e-01],
        [9.9995e-01, 4.5597e-05],
        [2.1967e-03, 9.9780e-01],
        [6.2452e-04, 9.9938e-01],
        [5.1777e-02, 9.4822e-01],
        [5.1816e-08, 1.0000e+00],
        [5.3072e-07, 1.0000e+00],
        [9.9961e-01, 3.9142e-04],
        [7.4376e-03, 9.9256e-01],
        [1.4178e-08, 1.0000e+00],
        [5.1941e-08, 1.0000e+00],
        [9.9909e-01, 9.0827e-04],
        [1.1186e-09, 1.0000e+00],
        [2.7485e-07, 1.0000e+00],
        [1.4959e-05, 9.9999e-01],
        [2.1784e-06, 1.0000e+00],
        [3.1029e-05, 9.9997e-01],
        [4.2447e-04, 9.9958e-01],
        [9.9895e-01, 1.0474e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.6098e-07, 1.0000e+00],
        [9.9982e-01, 1.7616e-04],
        [9.9251e-10, 1.0000e+00],
        [9.8559e-01, 1.4413e-02],
        [9.9967e-01, 3.3447e-04],
        [1.1040e-10, 1.0000e+00],
        [1.2037e-10, 1.0000e+00],
        [7.7319e-01, 2.2681e-01],
        [3.4643e-06, 1.0000e+00],
        [5.6775e-05, 9.9994e-01],
        [5.7120e-09, 1.0000e+00],
        [9.9975e-01, 2.4808e-04],
        [4.7818e-11, 1.0000e+00],
        [5.8625e-02, 9.4137e-01],
        [1.4814e-04, 9.9985e-01],
        [6.3176e-10, 1.0000e+00],
        [5.6183e-08, 1.0000e+00],
        [1.7072e-08, 1.0000e+00],
        [2.5890e-08, 1.0000e+00],
        [5.0980e-06, 9.9999e-01],
        [9.9987e-01, 1.3435e-04],
        [1.0797e-09, 1.0000e+00],
        [8.9736e-08, 1.0000e+00],
        [3.3052e-05, 9.9997e-01],
        [1.2160e-04, 9.9988e-01],
        [7.0206e-07, 1.0000e+00],
        [7.5557e-05, 9.9992e-01],
        [2.7749e-09, 1.0000e+00],
        [3.1913e-05, 9.9997e-01],
        [4.2006e-07, 1.0000e+00],
        [8.2154e-01, 1.7846e-01],
        [3.6454e-06, 1.0000e+00],
        [9.9665e-01, 3.3525e-03],
        [1.6677e-03, 9.9833e-01],
        [1.8553e-05, 9.9998e-01],
        [2.7483e-08, 1.0000e+00],
        [1.0352e-09, 1.0000e+00],
        [7.4965e-11, 1.0000e+00],
        [5.3855e-07, 1.0000e+00],
        [7.9693e-01, 2.0307e-01],
        [9.8793e-01, 1.2067e-02],
        [1.5554e-05, 9.9998e-01],
        [1.0000e+00, 1.1871e-06],
        [7.3017e-01, 2.6983e-01],
        [6.4548e-06, 9.9999e-01],
        [1.1694e-05, 9.9999e-01],
        [9.9987e-01, 1.3477e-04],
        [3.3170e-05, 9.9997e-01],
        [4.3762e-07, 1.0000e+00],
        [7.9932e-07, 1.0000e+00],
        [3.3853e-05, 9.9997e-01],
        [9.9985e-01, 1.5135e-04],
        [3.4328e-08, 1.0000e+00],
        [2.5870e-08, 1.0000e+00],
        [3.6669e-04, 9.9963e-01],
        [4.2922e-03, 9.9571e-01],
        [9.0797e-05, 9.9991e-01],
        [3.8415e-04, 9.9962e-01],
        [8.3574e-01, 1.6426e-01],
        [1.6829e-05, 9.9998e-01],
        [1.2199e-07, 1.0000e+00],
        [2.9047e-08, 1.0000e+00],
        [6.8872e-07, 1.0000e+00],
        [5.7604e-01, 4.2395e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4546e-05, 9.9999e-01],
        [1.6324e-04, 9.9984e-01],
        [3.3511e-03, 9.9665e-01],
        [3.8562e-08, 1.0000e+00],
        [9.5319e-09, 1.0000e+00],
        [1.0620e-07, 1.0000e+00],
        [9.6370e-08, 1.0000e+00],
        [2.8986e-09, 1.0000e+00],
        [3.2050e-04, 9.9968e-01],
        [2.4643e-09, 1.0000e+00],
        [9.9992e-01, 8.1777e-05],
        [1.5424e-11, 1.0000e+00],
        [1.6062e-05, 9.9998e-01],
        [7.3337e-08, 1.0000e+00],
        [9.9981e-01, 1.8710e-04],
        [1.2002e-07, 1.0000e+00],
        [2.9181e-06, 1.0000e+00],
        [4.5551e-06, 1.0000e+00],
        [1.5289e-04, 9.9985e-01],
        [6.7761e-05, 9.9993e-01],
        [9.9997e-01, 3.3496e-05],
        [7.7550e-03, 9.9225e-01],
        [1.9752e-09, 1.0000e+00],
        [1.1207e-03, 9.9888e-01],
        [5.0764e-08, 1.0000e+00],
        [4.4302e-06, 1.0000e+00],
        [9.8221e-06, 9.9999e-01],
        [6.6103e-12, 1.0000e+00],
        [9.9991e-01, 8.9005e-05],
        [9.9909e-01, 9.1300e-04],
        [5.6379e-02, 9.4362e-01],
        [4.9316e-08, 1.0000e+00],
        [9.9989e-01, 1.0693e-04],
        [2.2007e-05, 9.9998e-01],
        [3.9796e-08, 1.0000e+00],
        [1.7233e-04, 9.9983e-01],
        [9.9997e-01, 2.9603e-05],
        [9.9974e-01, 2.6317e-04],
        [4.8461e-09, 1.0000e+00],
        [3.5003e-06, 1.0000e+00],
        [6.1158e-06, 9.9999e-01],
        [9.9994e-01, 6.4776e-05],
        [1.9782e-09, 1.0000e+00],
        [4.4115e-06, 1.0000e+00],
        [9.8709e-01, 1.2909e-02],
        [1.7972e-05, 9.9998e-01],
        [1.4261e-08, 1.0000e+00],
        [6.1853e-03, 9.9381e-01],
        [4.7726e-05, 9.9995e-01],
        [1.1491e-06, 1.0000e+00],
        [1.0708e-05, 9.9999e-01],
        [9.9206e-01, 7.9412e-03],
        [1.4388e-02, 9.8561e-01],
        [1.2764e-07, 1.0000e+00],
        [3.4360e-01, 6.5640e-01],
        [3.1916e-07, 1.0000e+00],
        [1.7172e-07, 1.0000e+00],
        [1.2732e-04, 9.9987e-01],
        [9.9983e-01, 1.6634e-04],
        [1.0625e-05, 9.9999e-01],
        [3.5727e-06, 1.0000e+00],
        [3.0533e-02, 9.6947e-01],
        [9.9991e-01, 8.7053e-05],
        [2.8233e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7181e-06, 1.0000e+00],
        [9.8472e-01, 1.5277e-02],
        [7.1185e-03, 9.9288e-01],
        [7.0835e-07, 1.0000e+00],
        [9.7729e-01, 2.2713e-02],
        [8.7341e-05, 9.9991e-01],
        [4.7553e-09, 1.0000e+00],
        [2.5068e-07, 1.0000e+00],
        [9.9970e-01, 2.9752e-04],
        [8.1266e-04, 9.9919e-01],
        [7.4934e-03, 9.9251e-01],
        [1.6303e-11, 1.0000e+00],
        [3.9604e-05, 9.9996e-01],
        [1.8656e-06, 1.0000e+00],
        [9.9967e-01, 3.2714e-04],
        [7.6484e-06, 9.9999e-01],
        [2.8281e-07, 1.0000e+00],
        [9.9996e-01, 4.4393e-05],
        [5.7992e-10, 1.0000e+00],
        [1.5017e-07, 1.0000e+00],
        [8.7045e-05, 9.9991e-01],
        [9.9999e-01, 1.2466e-05],
        [9.9819e-01, 1.8079e-03],
        [9.9992e-01, 8.0810e-05],
        [1.0431e-03, 9.9896e-01],
        [9.9965e-01, 3.4635e-04],
        [1.1522e-05, 9.9999e-01],
        [2.6404e-07, 1.0000e+00],
        [7.6775e-08, 1.0000e+00],
        [7.4404e-09, 1.0000e+00],
        [9.9979e-01, 2.0995e-04],
        [1.8855e-07, 1.0000e+00],
        [1.7631e-06, 1.0000e+00],
        [6.4153e-05, 9.9994e-01],
        [2.1147e-06, 1.0000e+00],
        [8.0734e-04, 9.9919e-01],
        [6.8818e-07, 1.0000e+00],
        [9.9995e-01, 5.2735e-05],
        [1.0621e-05, 9.9999e-01],
        [2.4875e-04, 9.9975e-01],
        [2.0007e-06, 1.0000e+00],
        [9.9996e-01, 3.9188e-05],
        [6.6939e-07, 1.0000e+00],
        [9.9984e-01, 1.6083e-04],
        [4.4412e-07, 1.0000e+00],
        [9.9988e-01, 1.1703e-04],
        [3.3067e-06, 1.0000e+00],
        [1.0653e-06, 1.0000e+00],
        [3.0282e-07, 1.0000e+00],
        [6.6676e-09, 1.0000e+00],
        [9.9957e-01, 4.3379e-04],
        [8.0417e-04, 9.9920e-01],
        [6.2926e-07, 1.0000e+00],
        [4.7907e-08, 1.0000e+00],
        [3.8146e-06, 1.0000e+00],
        [4.8410e-07, 1.0000e+00],
        [3.1107e-06, 1.0000e+00],
        [3.7270e-06, 1.0000e+00],
        [6.0920e-08, 1.0000e+00],
        [2.2186e-06, 1.0000e+00],
        [1.2528e-08, 1.0000e+00],
        [4.7336e-08, 1.0000e+00],
        [1.4518e-03, 9.9855e-01],
        [9.9988e-01, 1.1763e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[4.7415e-06, 1.0000e+00],
        [3.7459e-05, 9.9996e-01],
        [8.8149e-04, 9.9912e-01],
        [5.3048e-09, 1.0000e+00],
        [1.3625e-05, 9.9999e-01],
        [9.9969e-01, 3.0929e-04],
        [9.9996e-01, 4.3005e-05],
        [1.1018e-06, 1.0000e+00],
        [2.9085e-03, 9.9709e-01],
        [9.9932e-01, 6.8234e-04],
        [5.1074e-06, 9.9999e-01],
        [1.6391e-08, 1.0000e+00],
        [8.5450e-09, 1.0000e+00],
        [3.1351e-06, 1.0000e+00],
        [1.4453e-04, 9.9986e-01],
        [4.3430e-02, 9.5657e-01],
        [1.8963e-08, 1.0000e+00],
        [9.7917e-04, 9.9902e-01],
        [1.8599e-05, 9.9998e-01],
        [5.4843e-08, 1.0000e+00],
        [1.3501e-07, 1.0000e+00],
        [3.8179e-06, 1.0000e+00],
        [2.5580e-04, 9.9974e-01],
        [3.2902e-05, 9.9997e-01],
        [6.3673e-08, 1.0000e+00],
        [1.0000e+00, 2.6421e-06],
        [5.1749e-06, 9.9999e-01],
        [9.9988e-01, 1.2059e-04],
        [9.9981e-01, 1.9111e-04],
        [7.0851e-09, 1.0000e+00],
        [3.8180e-08, 1.0000e+00],
        [8.6785e-01, 1.3215e-01],
        [2.6110e-04, 9.9974e-01],
        [9.7922e-04, 9.9902e-01],
        [1.2834e-09, 1.0000e+00],
        [2.3975e-07, 1.0000e+00],
        [6.3117e-01, 3.6883e-01],
        [1.0918e-02, 9.8908e-01],
        [1.3223e-05, 9.9999e-01],
        [9.9992e-01, 7.9706e-05],
        [9.9753e-01, 2.4701e-03],
        [9.9982e-01, 1.8044e-04],
        [9.6739e-01, 3.2608e-02],
        [9.9995e-01, 4.9610e-05],
        [9.9991e-01, 9.0323e-05],
        [6.4512e-08, 1.0000e+00],
        [9.9998e-01, 2.1066e-05],
        [1.7612e-06, 1.0000e+00],
        [9.9716e-01, 2.8381e-03],
        [3.5742e-06, 1.0000e+00],
        [3.2902e-05, 9.9997e-01],
        [1.9403e-08, 1.0000e+00],
        [6.7647e-07, 1.0000e+00],
        [6.0775e-06, 9.9999e-01],
        [7.0851e-09, 1.0000e+00],
        [4.0489e-03, 9.9595e-01],
        [9.9989e-01, 1.0938e-04],
        [2.0369e-05, 9.9998e-01],
        [1.8170e-06, 1.0000e+00],
        [9.9581e-01, 4.1920e-03],
        [2.6845e-05, 9.9997e-01],
        [2.3904e-10, 1.0000e+00],
        [1.6049e-04, 9.9984e-01],
        [9.9979e-01, 2.1001e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9914e-01, 8.6126e-04],
        [5.2874e-10, 1.0000e+00],
        [5.3797e-05, 9.9995e-01],
        [9.9993e-01, 7.2457e-05],
        [7.1093e-10, 1.0000e+00],
        [3.5345e-06, 1.0000e+00],
        [9.9978e-01, 2.1834e-04],
        [2.2551e-08, 1.0000e+00],
        [7.0783e-08, 1.0000e+00],
        [9.9416e-01, 5.8368e-03],
        [2.8123e-04, 9.9972e-01],
        [1.9083e-05, 9.9998e-01],
        [1.3643e-07, 1.0000e+00],
        [9.9198e-05, 9.9990e-01],
        [2.6732e-09, 1.0000e+00],
        [3.2765e-05, 9.9997e-01],
        [9.9944e-01, 5.5780e-04],
        [4.9885e-09, 1.0000e+00],
        [9.2871e-10, 1.0000e+00],
        [3.2719e-05, 9.9997e-01],
        [6.7586e-07, 1.0000e+00],
        [2.0117e-02, 9.7988e-01],
        [9.9962e-01, 3.8152e-04],
        [1.8091e-02, 9.8191e-01],
        [6.2966e-07, 1.0000e+00],
        [8.3138e-02, 9.1686e-01],
        [3.4463e-09, 1.0000e+00],
        [1.2568e-02, 9.8743e-01],
        [7.8874e-09, 1.0000e+00],
        [2.0093e-08, 1.0000e+00],
        [9.9647e-01, 3.5328e-03],
        [9.4970e-09, 1.0000e+00],
        [9.9982e-01, 1.8207e-04],
        [2.2888e-06, 1.0000e+00],
        [7.1326e-05, 9.9993e-01],
        [1.5654e-05, 9.9998e-01],
        [9.9990e-01, 9.5859e-05],
        [1.0000e+00, 3.0329e-07],
        [4.8830e-03, 9.9512e-01],
        [2.1316e-09, 1.0000e+00],
        [3.3291e-10, 1.0000e+00],
        [2.4954e-05, 9.9998e-01],
        [7.7914e-10, 1.0000e+00],
        [4.8198e-06, 1.0000e+00],
        [4.3094e-06, 1.0000e+00],
        [1.6996e-06, 1.0000e+00],
        [4.6987e-06, 1.0000e+00],
        [1.8834e-07, 1.0000e+00],
        [6.8677e-07, 1.0000e+00],
        [1.2458e-06, 1.0000e+00],
        [1.2437e-04, 9.9988e-01],
        [2.0003e-04, 9.9980e-01],
        [2.2879e-06, 1.0000e+00],
        [1.7274e-08, 1.0000e+00],
        [1.0000e+00, 9.2590e-07],
        [6.0384e-08, 1.0000e+00],
        [9.9832e-01, 1.6756e-03],
        [6.2130e-08, 1.0000e+00],
        [3.7617e-01, 6.2383e-01],
        [1.2069e-05, 9.9999e-01],
        [9.9976e-01, 2.3992e-04],
        [7.9647e-06, 9.9999e-01],
        [7.9551e-06, 9.9999e-01],
        [2.1415e-03, 9.9786e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7113e-05, 9.9998e-01],
        [6.6116e-08, 1.0000e+00],
        [2.3238e-01, 7.6762e-01],
        [1.3939e-06, 1.0000e+00],
        [9.8413e-01, 1.5865e-02],
        [3.1447e-13, 1.0000e+00],
        [9.9917e-01, 8.3202e-04],
        [6.9838e-04, 9.9930e-01],
        [4.2988e-05, 9.9996e-01],
        [9.9989e-01, 1.1125e-04],
        [9.9990e-01, 9.6511e-05],
        [7.3267e-08, 1.0000e+00],
        [5.2894e-10, 1.0000e+00],
        [1.8720e-09, 1.0000e+00],
        [1.2204e-07, 1.0000e+00],
        [1.4242e-07, 1.0000e+00],
        [1.6197e-03, 9.9838e-01],
        [7.5076e-06, 9.9999e-01],
        [9.9817e-01, 1.8320e-03],
        [1.0034e-01, 8.9966e-01],
        [9.9986e-01, 1.4426e-04],
        [3.2640e-06, 1.0000e+00],
        [9.9932e-01, 6.7571e-04],
        [8.4246e-10, 1.0000e+00],
        [9.0570e-01, 9.4297e-02],
        [1.7165e-05, 9.9998e-01],
        [9.9988e-01, 1.1740e-04],
        [3.2967e-04, 9.9967e-01],
        [9.9907e-01, 9.3121e-04],
        [3.4085e-02, 9.6591e-01],
        [9.9995e-01, 4.8759e-05],
        [5.4997e-05, 9.9995e-01],
        [1.9224e-04, 9.9981e-01],
        [9.9918e-01, 8.2302e-04],
        [1.6005e-04, 9.9984e-01],
        [6.2461e-05, 9.9994e-01],
        [2.6178e-05, 9.9997e-01],
        [2.0558e-03, 9.9794e-01],
        [6.2841e-09, 1.0000e+00],
        [5.7334e-07, 1.0000e+00],
        [9.7379e-01, 2.6205e-02],
        [3.1922e-08, 1.0000e+00],
        [3.8254e-03, 9.9617e-01],
        [7.2413e-02, 9.2759e-01],
        [9.9931e-01, 6.9361e-04],
        [7.0596e-07, 1.0000e+00],
        [1.4551e-08, 1.0000e+00],
        [1.0364e-05, 9.9999e-01],
        [6.7727e-07, 1.0000e+00],
        [9.9568e-01, 4.3215e-03],
        [6.1212e-06, 9.9999e-01],
        [5.7682e-08, 1.0000e+00],
        [5.7823e-06, 9.9999e-01],
        [3.5141e-06, 1.0000e+00],
        [5.3000e-04, 9.9947e-01],
        [3.8551e-01, 6.1449e-01],
        [3.1768e-10, 1.0000e+00],
        [1.1972e-08, 1.0000e+00],
        [1.1304e-05, 9.9999e-01],
        [1.3735e-10, 1.0000e+00],
        [9.9991e-01, 9.2667e-05],
        [1.1852e-05, 9.9999e-01],
        [8.7507e-07, 1.0000e+00],
        [3.7930e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.9956e-09, 1.0000e+00],
        [1.8036e-06, 1.0000e+00],
        [9.9956e-01, 4.4335e-04],
        [1.1247e-04, 9.9989e-01],
        [6.8332e-07, 1.0000e+00],
        [1.3802e-05, 9.9999e-01],
        [2.3104e-05, 9.9998e-01],
        [9.9960e-01, 3.9602e-04],
        [9.9876e-01, 1.2410e-03],
        [1.5520e-04, 9.9984e-01],
        [1.4432e-05, 9.9999e-01],
        [1.2007e-05, 9.9999e-01],
        [1.4649e-04, 9.9985e-01],
        [1.0200e-04, 9.9990e-01],
        [1.6235e-05, 9.9998e-01],
        [1.0267e-10, 1.0000e+00],
        [9.9978e-01, 2.2196e-04],
        [5.3405e-10, 1.0000e+00],
        [4.6334e-06, 1.0000e+00],
        [9.9998e-01, 1.9954e-05],
        [5.7423e-09, 1.0000e+00],
        [1.9912e-08, 1.0000e+00],
        [5.4067e-09, 1.0000e+00],
        [9.9991e-01, 8.8565e-05],
        [1.4986e-03, 9.9850e-01],
        [2.3774e-04, 9.9976e-01],
        [1.3470e-04, 9.9987e-01],
        [3.7763e-05, 9.9996e-01],
        [7.2146e-03, 9.9279e-01],
        [6.8332e-07, 1.0000e+00],
        [5.5218e-04, 9.9945e-01],
        [9.9910e-01, 8.9953e-04],
        [9.9996e-01, 4.3348e-05],
        [9.9890e-01, 1.0981e-03],
        [7.7254e-07, 1.0000e+00],
        [3.4072e-06, 1.0000e+00],
        [1.5542e-06, 1.0000e+00],
        [4.9824e-04, 9.9950e-01],
        [9.0681e-05, 9.9991e-01],
        [2.9237e-03, 9.9708e-01],
        [8.2542e-06, 9.9999e-01],
        [9.9993e-01, 6.9527e-05],
        [9.9991e-01, 8.8856e-05],
        [9.9986e-01, 1.4321e-04],
        [1.6216e-09, 1.0000e+00],
        [9.8479e-01, 1.5206e-02],
        [6.5653e-06, 9.9999e-01],
        [1.6944e-06, 1.0000e+00],
        [1.0000e+00, 4.7865e-06],
        [6.5430e-06, 9.9999e-01],
        [3.2550e-08, 1.0000e+00],
        [9.9992e-01, 7.6336e-05],
        [1.0369e-07, 1.0000e+00],
        [8.2278e-05, 9.9992e-01],
        [1.7934e-08, 1.0000e+00],
        [3.9006e-05, 9.9996e-01],
        [3.2246e-03, 9.9678e-01],
        [1.4150e-08, 1.0000e+00],
        [5.0796e-03, 9.9492e-01],
        [8.1265e-09, 1.0000e+00],
        [7.5075e-05, 9.9992e-01],
        [1.1069e-04, 9.9989e-01],
        [2.6222e-02, 9.7378e-01],
        [1.4275e-04, 9.9986e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6903e-04, 9.9983e-01],
        [1.2749e-08, 1.0000e+00],
        [6.3622e-07, 1.0000e+00],
        [8.6875e-10, 1.0000e+00],
        [4.8172e-06, 1.0000e+00],
        [9.9980e-01, 2.0460e-04],
        [1.4047e-05, 9.9999e-01],
        [2.0020e-06, 1.0000e+00],
        [1.7594e-04, 9.9982e-01],
        [7.4603e-09, 1.0000e+00],
        [1.1643e-07, 1.0000e+00],
        [3.2195e-05, 9.9997e-01],
        [7.2589e-07, 1.0000e+00],
        [2.2211e-08, 1.0000e+00],
        [2.5787e-06, 1.0000e+00],
        [9.8043e-01, 1.9571e-02],
        [2.2168e-06, 1.0000e+00],
        [3.5883e-04, 9.9964e-01],
        [1.0286e-06, 1.0000e+00],
        [6.5110e-07, 1.0000e+00],
        [8.8240e-10, 1.0000e+00],
        [1.7054e-11, 1.0000e+00],
        [9.9994e-01, 6.3471e-05],
        [9.9988e-01, 1.2438e-04],
        [4.4286e-01, 5.5714e-01],
        [4.1729e-04, 9.9958e-01],
        [9.9987e-01, 1.2806e-04],
        [1.6053e-08, 1.0000e+00],
        [1.4765e-01, 8.5235e-01],
        [5.9617e-05, 9.9994e-01],
        [2.9474e-05, 9.9997e-01],
        [2.2418e-04, 9.9978e-01],
        [2.9026e-03, 9.9710e-01],
        [1.1076e-06, 1.0000e+00],
        [8.4432e-05, 9.9992e-01],
        [2.2924e-04, 9.9977e-01],
        [1.2316e-09, 1.0000e+00],
        [6.6426e-12, 1.0000e+00],
        [9.9998e-01, 2.4730e-05],
        [9.9998e-01, 2.0845e-05],
        [9.9949e-01, 5.0517e-04],
        [9.9989e-01, 1.0945e-04],
        [5.8856e-02, 9.4114e-01],
        [3.4278e-08, 1.0000e+00],
        [1.4932e-06, 1.0000e+00],
        [3.5851e-01, 6.4149e-01],
        [9.9995e-01, 5.4681e-05],
        [9.6563e-06, 9.9999e-01],
        [2.7831e-06, 1.0000e+00],
        [8.4130e-07, 1.0000e+00],
        [2.6419e-06, 1.0000e+00],
        [2.4533e-06, 1.0000e+00],
        [5.3179e-01, 4.6821e-01],
        [9.3284e-08, 1.0000e+00],
        [1.0465e-02, 9.8953e-01],
        [6.8606e-08, 1.0000e+00],
        [9.9991e-01, 9.3689e-05],
        [3.8697e-05, 9.9996e-01],
        [9.9985e-01, 1.4577e-04],
        [9.9993e-01, 6.6083e-05],
        [2.1353e-06, 1.0000e+00],
        [2.5755e-04, 9.9974e-01],
        [2.2699e-13, 1.0000e+00],
        [9.9991e-01, 9.0388e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9998e-08, 1.0000e+00],
        [4.1800e-06, 1.0000e+00],
        [7.2360e-08, 1.0000e+00],
        [5.4916e-07, 1.0000e+00],
        [2.1208e-06, 1.0000e+00],
        [9.9983e-01, 1.6503e-04],
        [6.4483e-03, 9.9355e-01],
        [4.8801e-08, 1.0000e+00],
        [4.3055e-05, 9.9996e-01],
        [8.3544e-08, 1.0000e+00],
        [7.7595e-07, 1.0000e+00],
        [9.0456e-08, 1.0000e+00],
        [7.6277e-06, 9.9999e-01],
        [9.9873e-01, 1.2696e-03],
        [2.5591e-04, 9.9974e-01],
        [1.1428e-01, 8.8572e-01],
        [3.0299e-02, 9.6970e-01],
        [2.2410e-04, 9.9978e-01],
        [6.1800e-08, 1.0000e+00],
        [1.6520e-05, 9.9998e-01],
        [9.9987e-01, 1.3366e-04],
        [3.4211e-09, 1.0000e+00],
        [1.1300e-07, 1.0000e+00],
        [9.9995e-01, 5.4372e-05],
        [9.9995e-01, 4.9699e-05],
        [2.6834e-07, 1.0000e+00],
        [4.1982e-04, 9.9958e-01],
        [5.8200e-09, 1.0000e+00],
        [1.7328e-05, 9.9998e-01],
        [7.4721e-05, 9.9993e-01],
        [4.4153e-05, 9.9996e-01],
        [2.6402e-04, 9.9974e-01],
        [7.3063e-09, 1.0000e+00],
        [3.0455e-01, 6.9545e-01],
        [9.9993e-01, 7.0583e-05],
        [8.7113e-07, 1.0000e+00],
        [1.7298e-06, 1.0000e+00],
        [1.8271e-05, 9.9998e-01],
        [2.9271e-07, 1.0000e+00],
        [1.3606e-06, 1.0000e+00],
        [1.0466e-01, 8.9534e-01],
        [6.0025e-06, 9.9999e-01],
        [6.1549e-05, 9.9994e-01],
        [2.3298e-06, 1.0000e+00],
        [9.9727e-01, 2.7348e-03],
        [1.3845e-05, 9.9999e-01],
        [8.6003e-07, 1.0000e+00],
        [4.5170e-08, 1.0000e+00],
        [9.5889e-05, 9.9990e-01],
        [5.0608e-02, 9.4939e-01],
        [2.6983e-09, 1.0000e+00],
        [9.9850e-05, 9.9990e-01],
        [2.4660e-05, 9.9998e-01],
        [4.1493e-04, 9.9959e-01],
        [3.8466e-04, 9.9962e-01],
        [2.6443e-04, 9.9974e-01],
        [1.0000e+00, 1.4771e-06],
        [3.5386e-13, 1.0000e+00],
        [9.9995e-01, 5.0884e-05],
        [1.5957e-03, 9.9840e-01],
        [5.5602e-09, 1.0000e+00],
        [1.3845e-05, 9.9999e-01],
        [2.4295e-07, 1.0000e+00],
        [2.9112e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9987e-01, 1.2633e-04],
        [9.9983e-01, 1.6509e-04],
        [1.4628e-04, 9.9985e-01],
        [9.8351e-05, 9.9990e-01],
        [1.0364e-09, 1.0000e+00],
        [9.9998e-01, 1.9420e-05],
        [7.8962e-07, 1.0000e+00],
        [2.3648e-08, 1.0000e+00],
        [7.9602e-08, 1.0000e+00],
        [1.8057e-05, 9.9998e-01],
        [9.9821e-01, 1.7945e-03],
        [9.9968e-01, 3.1764e-04],
        [1.1919e-05, 9.9999e-01],
        [1.0598e-05, 9.9999e-01],
        [1.4363e-03, 9.9856e-01],
        [1.8684e-06, 1.0000e+00],
        [3.3869e-07, 1.0000e+00],
        [3.6101e-04, 9.9964e-01],
        [9.9998e-01, 2.1300e-05],
        [1.8575e-09, 1.0000e+00],
        [6.5055e-06, 9.9999e-01],
        [7.6342e-06, 9.9999e-01],
        [1.2173e-04, 9.9988e-01],
        [2.1205e-07, 1.0000e+00],
        [1.2484e-04, 9.9988e-01],
        [9.9999e-01, 6.5753e-06],
        [9.9968e-01, 3.1645e-04],
        [4.2267e-08, 1.0000e+00],
        [4.1858e-05, 9.9996e-01],
        [6.6338e-04, 9.9934e-01],
        [6.3453e-05, 9.9994e-01],
        [1.9009e-08, 1.0000e+00],
        [2.0974e-07, 1.0000e+00],
        [9.6300e-09, 1.0000e+00],
        [2.6380e-06, 1.0000e+00],
        [6.5057e-08, 1.0000e+00],
        [9.9951e-01, 4.8629e-04],
        [1.2082e-06, 1.0000e+00],
        [2.2234e-07, 1.0000e+00],
        [4.3061e-05, 9.9996e-01],
        [1.0314e-08, 1.0000e+00],
        [3.2940e-07, 1.0000e+00],
        [2.6840e-02, 9.7316e-01],
        [9.2994e-05, 9.9991e-01],
        [6.4655e-09, 1.0000e+00],
        [1.0534e-01, 8.9466e-01],
        [9.9990e-01, 1.0182e-04],
        [9.9986e-01, 1.3949e-04],
        [9.3975e-01, 6.0247e-02],
        [9.9994e-01, 6.2267e-05],
        [2.6517e-09, 1.0000e+00],
        [9.9961e-01, 3.8925e-04],
        [4.6214e-07, 1.0000e+00],
        [9.9925e-01, 7.4977e-04],
        [1.6780e-07, 1.0000e+00],
        [1.4976e-09, 1.0000e+00],
        [9.3974e-06, 9.9999e-01],
        [9.9997e-01, 2.9798e-05],
        [2.8602e-09, 1.0000e+00],
        [1.6495e-07, 1.0000e+00],
        [2.0410e-06, 1.0000e+00],
        [5.6103e-07, 1.0000e+00],
        [6.9982e-06, 9.9999e-01],
        [9.9995e-01, 5.4349e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[8.5288e-04, 9.9915e-01],
        [1.4488e-10, 1.0000e+00],
        [1.2430e-06, 1.0000e+00],
        [2.2949e-06, 1.0000e+00],
        [1.4549e-11, 1.0000e+00],
        [1.2099e-05, 9.9999e-01],
        [4.7584e-03, 9.9524e-01],
        [9.9969e-01, 3.0755e-04],
        [9.1658e-07, 1.0000e+00],
        [8.4886e-06, 9.9999e-01],
        [1.3669e-01, 8.6331e-01],
        [3.2907e-06, 1.0000e+00],
        [2.3690e-07, 1.0000e+00],
        [5.6196e-04, 9.9944e-01],
        [1.4346e-05, 9.9999e-01],
        [3.6710e-10, 1.0000e+00],
        [2.4845e-05, 9.9998e-01],
        [1.0143e-03, 9.9899e-01],
        [9.9984e-01, 1.5803e-04],
        [9.3440e-07, 1.0000e+00],
        [9.9994e-01, 5.9352e-05],
        [4.2873e-09, 1.0000e+00],
        [6.4026e-03, 9.9360e-01],
        [1.0567e-05, 9.9999e-01],
        [8.4716e-06, 9.9999e-01],
        [4.4486e-08, 1.0000e+00],
        [4.0402e-05, 9.9996e-01],
        [2.5490e-06, 1.0000e+00],
        [7.8542e-06, 9.9999e-01],
        [2.3954e-06, 1.0000e+00],
        [9.9994e-01, 6.0210e-05],
        [9.9994e-01, 5.8601e-05],
        [9.9996e-01, 4.1974e-05],
        [5.1210e-04, 9.9949e-01],
        [3.0123e-09, 1.0000e+00],
        [3.1833e-10, 1.0000e+00],
        [9.9992e-01, 7.6349e-05],
        [1.4195e-06, 1.0000e+00],
        [8.1080e-05, 9.9992e-01],
        [9.8801e-01, 1.1994e-02],
        [9.9979e-01, 2.1385e-04],
        [9.9864e-01, 1.3617e-03],
        [4.1787e-09, 1.0000e+00],
        [9.9779e-01, 2.2145e-03],
        [7.1490e-07, 1.0000e+00],
        [1.8236e-05, 9.9998e-01],
        [9.9995e-01, 4.9110e-05],
        [1.2049e-07, 1.0000e+00],
        [8.2933e-06, 9.9999e-01],
        [1.1566e-05, 9.9999e-01],
        [4.9901e-04, 9.9950e-01],
        [1.0000e+00, 2.8533e-06],
        [2.0466e-05, 9.9998e-01],
        [5.4762e-05, 9.9995e-01],
        [3.5326e-08, 1.0000e+00],
        [9.9990e-01, 1.0070e-04],
        [9.9991e-01, 9.4429e-05],
        [7.6353e-03, 9.9236e-01],
        [8.7067e-07, 1.0000e+00],
        [9.9985e-01, 1.5181e-04],
        [2.0393e-06, 1.0000e+00],
        [9.3678e-01, 6.3217e-02],
        [9.1538e-08, 1.0000e+00],
        [1.6635e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1141e-01, 8.8859e-01],
        [5.1171e-02, 9.4883e-01],
        [6.9062e-08, 1.0000e+00],
        [5.0216e-10, 1.0000e+00],
        [9.9999e-01, 1.0759e-05],
        [9.9995e-01, 5.3230e-05],
        [6.3189e-04, 9.9937e-01],
        [1.0000e+00, 2.1120e-07],
        [1.9809e-03, 9.9802e-01],
        [1.5090e-05, 9.9998e-01],
        [5.2980e-08, 1.0000e+00],
        [9.9983e-01, 1.7412e-04],
        [1.2853e-04, 9.9987e-01],
        [3.8162e-07, 1.0000e+00],
        [9.9991e-01, 9.0145e-05],
        [7.2982e-05, 9.9993e-01],
        [2.5612e-06, 1.0000e+00],
        [5.9037e-08, 1.0000e+00],
        [2.6921e-08, 1.0000e+00],
        [3.9492e-02, 9.6051e-01],
        [5.5344e-05, 9.9994e-01],
        [1.5737e-10, 1.0000e+00],
        [7.9188e-09, 1.0000e+00],
        [4.1666e-09, 1.0000e+00],
        [9.3029e-01, 6.9714e-02],
        [1.6959e-04, 9.9983e-01],
        [2.1632e-04, 9.9978e-01],
        [2.3193e-05, 9.9998e-01],
        [1.3448e-09, 1.0000e+00],
        [9.9341e-03, 9.9007e-01],
        [8.2324e-04, 9.9918e-01],
        [3.4277e-08, 1.0000e+00],
        [1.6248e-05, 9.9998e-01],
        [5.2645e-04, 9.9947e-01],
        [9.9996e-01, 3.6253e-05],
        [1.0263e-03, 9.9897e-01],
        [1.9921e-03, 9.9801e-01],
        [1.8784e-11, 1.0000e+00],
        [4.4351e-09, 1.0000e+00],
        [8.9759e-09, 1.0000e+00],
        [1.0411e-04, 9.9990e-01],
        [1.9879e-07, 1.0000e+00],
        [2.1405e-05, 9.9998e-01],
        [1.8213e-05, 9.9998e-01],
        [1.1028e-01, 8.8972e-01],
        [5.4880e-10, 1.0000e+00],
        [3.6313e-04, 9.9964e-01],
        [8.1859e-12, 1.0000e+00],
        [2.9824e-10, 1.0000e+00],
        [9.9953e-01, 4.6621e-04],
        [9.9985e-01, 1.4850e-04],
        [4.8202e-07, 1.0000e+00],
        [4.1903e-07, 1.0000e+00],
        [4.2397e-07, 1.0000e+00],
        [1.7456e-04, 9.9983e-01],
        [9.9997e-01, 2.5801e-05],
        [9.9917e-01, 8.2904e-04],
        [7.5150e-09, 1.0000e+00],
        [2.7966e-09, 1.0000e+00],
        [9.9709e-01, 2.9116e-03],
        [8.3394e-01, 1.6606e-01],
        [1.0824e-02, 9.8918e-01],
        [6.5646e-08, 1.0000e+00],
        [1.2228e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.6655e-03, 9.9733e-01],
        [9.9967e-01, 3.3251e-04],
        [9.9996e-01, 4.2770e-05],
        [9.9997e-01, 2.7552e-05],
        [9.9952e-01, 4.8287e-04],
        [4.9046e-09, 1.0000e+00],
        [8.9472e-07, 1.0000e+00],
        [2.0525e-01, 7.9475e-01],
        [3.8095e-04, 9.9962e-01],
        [9.9986e-01, 1.4310e-04],
        [9.6484e-01, 3.5162e-02],
        [8.8697e-06, 9.9999e-01],
        [1.4534e-07, 1.0000e+00],
        [9.7551e-01, 2.4490e-02],
        [1.2850e-08, 1.0000e+00],
        [1.1220e-06, 1.0000e+00],
        [7.1324e-05, 9.9993e-01],
        [2.7610e-08, 1.0000e+00],
        [9.2540e-08, 1.0000e+00],
        [2.7409e-02, 9.7259e-01],
        [8.6443e-02, 9.1356e-01],
        [7.5860e-11, 1.0000e+00],
        [4.6012e-06, 1.0000e+00],
        [2.5153e-07, 1.0000e+00],
        [7.0081e-01, 2.9919e-01],
        [9.9995e-01, 4.8450e-05],
        [5.4257e-08, 1.0000e+00],
        [5.1612e-04, 9.9948e-01],
        [5.7818e-03, 9.9422e-01],
        [2.8406e-07, 1.0000e+00],
        [3.3087e-07, 1.0000e+00],
        [5.5911e-06, 9.9999e-01],
        [3.1841e-02, 9.6816e-01],
        [2.5283e-08, 1.0000e+00],
        [4.2974e-06, 1.0000e+00],
        [7.8781e-05, 9.9992e-01],
        [7.9854e-07, 1.0000e+00],
        [6.0339e-10, 1.0000e+00],
        [6.4190e-04, 9.9936e-01],
        [2.1402e-06, 1.0000e+00],
        [6.5471e-06, 9.9999e-01],
        [1.0981e-09, 1.0000e+00],
        [9.9999e-01, 8.1143e-06],
        [2.3421e-03, 9.9766e-01],
        [3.6472e-06, 1.0000e+00],
        [3.1619e-08, 1.0000e+00],
        [1.2814e-09, 1.0000e+00],
        [2.3210e-09, 1.0000e+00],
        [9.9987e-01, 1.3324e-04],
        [1.4718e-06, 1.0000e+00],
        [4.6356e-01, 5.3644e-01],
        [7.7305e-11, 1.0000e+00],
        [1.3040e-07, 1.0000e+00],
        [9.9996e-01, 4.0956e-05],
        [2.3300e-08, 1.0000e+00],
        [9.9999e-01, 5.9751e-06],
        [9.9921e-01, 7.9048e-04],
        [7.5096e-10, 1.0000e+00],
        [9.9967e-01, 3.2833e-04],
        [4.4330e-06, 1.0000e+00],
        [9.9999e-01, 1.2253e-05],
        [3.8095e-04, 9.9962e-01],
        [6.9769e-08, 1.0000e+00],
        [8.7478e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9995e-01, 4.8082e-05],
        [9.9940e-01, 6.0467e-04],
        [3.9225e-04, 9.9961e-01],
        [9.9983e-01, 1.7203e-04],
        [5.3979e-07, 1.0000e+00],
        [3.7669e-03, 9.9623e-01],
        [1.0000e+00, 3.4840e-07],
        [1.3936e-07, 1.0000e+00],
        [1.2601e-09, 1.0000e+00],
        [1.8205e-04, 9.9982e-01],
        [2.6921e-03, 9.9731e-01],
        [2.7861e-03, 9.9721e-01],
        [3.0104e-07, 1.0000e+00],
        [1.3135e-07, 1.0000e+00],
        [5.7230e-04, 9.9943e-01],
        [6.4192e-08, 1.0000e+00],
        [9.9982e-01, 1.7606e-04],
        [9.9409e-01, 5.9111e-03],
        [3.9398e-05, 9.9996e-01],
        [6.0673e-08, 1.0000e+00],
        [9.9986e-01, 1.3704e-04],
        [1.0000e+00, 1.3667e-07],
        [6.1802e-07, 1.0000e+00],
        [7.4308e-08, 1.0000e+00],
        [9.9968e-01, 3.1703e-04],
        [1.2593e-05, 9.9999e-01],
        [3.1543e-08, 1.0000e+00],
        [5.3211e-06, 9.9999e-01],
        [9.9752e-01, 2.4830e-03],
        [1.3624e-09, 1.0000e+00],
        [3.5785e-11, 1.0000e+00],
        [1.0848e-02, 9.8915e-01],
        [9.9989e-01, 1.0772e-04],
        [1.2778e-09, 1.0000e+00],
        [7.6255e-10, 1.0000e+00],
        [9.9993e-01, 7.1794e-05],
        [8.1893e-08, 1.0000e+00],
        [6.3903e-07, 1.0000e+00],
        [9.9274e-01, 7.2551e-03],
        [9.9992e-01, 8.2410e-05],
        [2.2510e-07, 1.0000e+00],
        [7.0504e-09, 1.0000e+00],
        [8.7990e-08, 1.0000e+00],
        [9.9742e-01, 2.5783e-03],
        [2.6808e-07, 1.0000e+00],
        [1.0346e-06, 1.0000e+00],
        [6.7895e-09, 1.0000e+00],
        [1.5251e-02, 9.8475e-01],
        [4.7267e-04, 9.9953e-01],
        [9.9995e-01, 4.6351e-05],
        [1.6846e-05, 9.9998e-01],
        [9.9719e-01, 2.8072e-03],
        [1.1789e-05, 9.9999e-01],
        [9.9996e-01, 3.5018e-05],
        [1.3287e-05, 9.9999e-01],
        [7.0274e-06, 9.9999e-01],
        [1.3270e-06, 1.0000e+00],
        [7.6557e-05, 9.9992e-01],
        [1.0607e-07, 1.0000e+00],
        [9.9961e-01, 3.9138e-04],
        [8.7544e-01, 1.2456e-01],
        [1.1945e-04, 9.9988e-01],
        [9.9983e-01, 1.7373e-04],
        [2.7549e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5485e-09, 1.0000e+00],
        [9.9993e-01, 7.1255e-05],
        [2.5171e-02, 9.7483e-01],
        [8.4202e-08, 1.0000e+00],
        [8.9240e-08, 1.0000e+00],
        [9.9994e-01, 5.8064e-05],
        [1.2218e-07, 1.0000e+00],
        [9.9999e-01, 1.1890e-05],
        [9.9970e-01, 2.9806e-04],
        [1.4674e-05, 9.9999e-01],
        [9.9633e-01, 3.6673e-03],
        [9.9604e-01, 3.9595e-03],
        [1.7092e-06, 1.0000e+00],
        [1.5684e-07, 1.0000e+00],
        [4.5737e-03, 9.9543e-01],
        [9.9946e-01, 5.4019e-04],
        [5.6714e-04, 9.9943e-01],
        [1.0885e-09, 1.0000e+00],
        [7.5071e-05, 9.9992e-01],
        [1.4720e-03, 9.9853e-01],
        [9.9563e-01, 4.3741e-03],
        [5.5697e-08, 1.0000e+00],
        [2.6708e-09, 1.0000e+00],
        [1.3141e-09, 1.0000e+00],
        [1.4031e-01, 8.5969e-01],
        [2.4449e-04, 9.9976e-01],
        [5.2710e-08, 1.0000e+00],
        [8.1730e-01, 1.8270e-01],
        [1.1951e-06, 1.0000e+00],
        [6.5903e-01, 3.4097e-01],
        [1.2034e-08, 1.0000e+00],
        [5.1431e-04, 9.9949e-01],
        [2.5288e-09, 1.0000e+00],
        [1.8881e-06, 1.0000e+00],
        [2.0253e-06, 1.0000e+00],
        [2.0725e-05, 9.9998e-01],
        [2.7688e-06, 1.0000e+00],
        [2.5076e-11, 1.0000e+00],
        [8.2772e-04, 9.9917e-01],
        [3.5033e-07, 1.0000e+00],
        [1.3555e-10, 1.0000e+00],
        [1.5270e-08, 1.0000e+00],
        [9.9992e-01, 8.1723e-05],
        [2.1559e-08, 1.0000e+00],
        [1.3167e-07, 1.0000e+00],
        [1.2965e-09, 1.0000e+00],
        [1.0316e-04, 9.9990e-01],
        [4.9190e-08, 1.0000e+00],
        [5.7679e-06, 9.9999e-01],
        [1.2213e-10, 1.0000e+00],
        [1.2558e-09, 1.0000e+00],
        [7.4639e-06, 9.9999e-01],
        [3.0291e-09, 1.0000e+00],
        [1.8458e-05, 9.9998e-01],
        [9.9290e-01, 7.0963e-03],
        [9.9988e-01, 1.2265e-04],
        [5.1235e-09, 1.0000e+00],
        [3.1492e-08, 1.0000e+00],
        [3.1664e-04, 9.9968e-01],
        [1.3967e-05, 9.9999e-01],
        [1.3029e-10, 1.0000e+00],
        [8.2591e-04, 9.9917e-01],
        [8.0670e-01, 1.9330e-01],
        [2.3071e-04, 9.9977e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3044e-10, 1.0000e+00],
        [9.3444e-11, 1.0000e+00],
        [3.5308e-07, 1.0000e+00],
        [4.4792e-05, 9.9996e-01],
        [2.7719e-05, 9.9997e-01],
        [2.8738e-06, 1.0000e+00],
        [1.7019e-04, 9.9983e-01],
        [9.9626e-07, 1.0000e+00],
        [4.6197e-07, 1.0000e+00],
        [3.9583e-05, 9.9996e-01],
        [3.1442e-02, 9.6856e-01],
        [1.0519e-02, 9.8948e-01],
        [6.6319e-10, 1.0000e+00],
        [3.7739e-03, 9.9623e-01],
        [9.9990e-01, 1.0417e-04],
        [1.4351e-03, 9.9856e-01],
        [1.6973e-04, 9.9983e-01],
        [1.0730e-06, 1.0000e+00],
        [9.8816e-01, 1.1840e-02],
        [2.0559e-07, 1.0000e+00],
        [9.9946e-01, 5.4430e-04],
        [4.1429e-08, 1.0000e+00],
        [2.1147e-02, 9.7885e-01],
        [5.9570e-07, 1.0000e+00],
        [1.6505e-07, 1.0000e+00],
        [3.9366e-06, 1.0000e+00],
        [9.9980e-01, 1.9916e-04],
        [9.9999e-01, 1.3685e-05],
        [8.7012e-07, 1.0000e+00],
        [1.2539e-04, 9.9987e-01],
        [9.9999e-01, 1.1588e-05],
        [1.0370e-04, 9.9990e-01],
        [3.9021e-07, 1.0000e+00],
        [1.2959e-05, 9.9999e-01],
        [9.0578e-09, 1.0000e+00],
        [3.5454e-03, 9.9645e-01],
        [2.9955e-04, 9.9970e-01],
        [1.6231e-03, 9.9838e-01],
        [1.4029e-03, 9.9860e-01],
        [1.6114e-07, 1.0000e+00],
        [3.4885e-01, 6.5115e-01],
        [1.1995e-06, 1.0000e+00],
        [9.9996e-01, 3.8171e-05],
        [9.4656e-01, 5.3438e-02],
        [9.3913e-07, 1.0000e+00],
        [6.4456e-02, 9.3554e-01],
        [2.0445e-03, 9.9796e-01],
        [6.6879e-08, 1.0000e+00],
        [5.2777e-05, 9.9995e-01],
        [3.0728e-07, 1.0000e+00],
        [9.9965e-01, 3.4662e-04],
        [9.9951e-01, 4.9298e-04],
        [5.3351e-08, 1.0000e+00],
        [4.9198e-10, 1.0000e+00],
        [4.3770e-11, 1.0000e+00],
        [7.5612e-04, 9.9924e-01],
        [1.7084e-05, 9.9998e-01],
        [1.8717e-06, 1.0000e+00],
        [9.6837e-01, 3.1633e-02],
        [6.3495e-08, 1.0000e+00],
        [9.9990e-01, 9.7616e-05],
        [3.7732e-09, 1.0000e+00],
        [1.8008e-05, 9.9998e-01],
        [1.3865e-02, 9.8613e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5693e-09, 1.0000e+00],
        [2.5589e-05, 9.9997e-01],
        [2.4009e-05, 9.9998e-01],
        [4.3466e-06, 1.0000e+00],
        [5.5769e-01, 4.4231e-01],
        [4.2377e-07, 1.0000e+00],
        [2.6508e-06, 1.0000e+00],
        [3.0738e-01, 6.9262e-01],
        [2.2986e-06, 1.0000e+00],
        [1.4449e-10, 1.0000e+00],
        [2.0886e-04, 9.9979e-01],
        [8.9462e-05, 9.9991e-01],
        [9.9997e-01, 3.0359e-05],
        [1.0196e-06, 1.0000e+00],
        [1.6489e-05, 9.9998e-01],
        [1.2886e-06, 1.0000e+00],
        [6.3828e-05, 9.9994e-01],
        [5.8065e-07, 1.0000e+00],
        [3.7130e-09, 1.0000e+00],
        [2.0431e-04, 9.9980e-01],
        [2.2791e-10, 1.0000e+00],
        [9.9983e-01, 1.7284e-04],
        [9.8977e-01, 1.0231e-02],
        [9.9975e-01, 2.4941e-04],
        [1.2865e-10, 1.0000e+00],
        [7.8400e-09, 1.0000e+00],
        [1.2117e-07, 1.0000e+00],
        [2.7812e-05, 9.9997e-01],
        [1.7626e-10, 1.0000e+00],
        [2.6466e-05, 9.9997e-01],
        [4.4848e-08, 1.0000e+00],
        [3.1879e-09, 1.0000e+00],
        [3.0323e-08, 1.0000e+00],
        [9.8692e-07, 1.0000e+00],
        [9.9990e-01, 1.0049e-04],
        [4.3163e-05, 9.9996e-01],
        [1.4459e-09, 1.0000e+00],
        [1.1427e-11, 1.0000e+00],
        [1.1104e-07, 1.0000e+00],
        [3.8787e-06, 1.0000e+00],
        [1.7711e-08, 1.0000e+00],
        [2.8646e-05, 9.9997e-01],
        [9.9966e-01, 3.4426e-04],
        [9.9991e-01, 9.1927e-05],
        [9.9967e-01, 3.3368e-04],
        [9.9808e-01, 1.9216e-03],
        [9.3277e-05, 9.9991e-01],
        [2.0440e-11, 1.0000e+00],
        [1.3375e-07, 1.0000e+00],
        [1.1370e-09, 1.0000e+00],
        [5.4675e-05, 9.9995e-01],
        [1.0302e-10, 1.0000e+00],
        [4.4299e-10, 1.0000e+00],
        [1.3180e-07, 1.0000e+00],
        [9.9987e-01, 1.2565e-04],
        [9.9903e-01, 9.6674e-04],
        [4.9330e-10, 1.0000e+00],
        [2.8604e-06, 1.0000e+00],
        [3.2537e-07, 1.0000e+00],
        [3.3346e-09, 1.0000e+00],
        [1.4136e-08, 1.0000e+00],
        [7.2437e-10, 1.0000e+00],
        [1.5010e-10, 1.0000e+00],
        [9.8957e-01, 1.0433e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.6384e-07, 1.0000e+00],
        [9.9816e-01, 1.8352e-03],
        [9.9973e-01, 2.6812e-04],
        [9.5996e-07, 1.0000e+00],
        [7.1268e-07, 1.0000e+00],
        [9.9990e-01, 9.9658e-05],
        [9.9996e-01, 3.8852e-05],
        [9.9621e-01, 3.7884e-03],
        [3.8179e-12, 1.0000e+00],
        [9.9991e-01, 8.7536e-05],
        [9.9983e-01, 1.7283e-04],
        [1.3443e-06, 1.0000e+00],
        [9.9987e-01, 1.3222e-04],
        [9.9963e-01, 3.6664e-04],
        [7.1777e-08, 1.0000e+00],
        [7.8647e-04, 9.9921e-01],
        [4.0972e-09, 1.0000e+00],
        [9.9991e-01, 9.2155e-05],
        [5.3625e-05, 9.9995e-01],
        [9.9453e-01, 5.4721e-03],
        [4.4406e-10, 1.0000e+00],
        [9.9984e-01, 1.6481e-04],
        [5.5548e-04, 9.9944e-01],
        [5.4546e-10, 1.0000e+00],
        [9.7626e-10, 1.0000e+00],
        [2.2208e-07, 1.0000e+00],
        [5.8572e-07, 1.0000e+00],
        [2.0056e-08, 1.0000e+00],
        [7.9212e-07, 1.0000e+00],
        [3.2561e-04, 9.9967e-01],
        [5.1188e-05, 9.9995e-01],
        [3.2803e-08, 1.0000e+00],
        [9.9993e-01, 7.2269e-05],
        [5.0804e-06, 9.9999e-01],
        [1.5655e-06, 1.0000e+00],
        [9.9954e-01, 4.6285e-04],
        [8.2851e-01, 1.7149e-01],
        [2.4755e-10, 1.0000e+00],
        [9.9413e-01, 5.8656e-03],
        [8.0103e-07, 1.0000e+00],
        [4.9966e-02, 9.5003e-01],
        [1.1196e-01, 8.8804e-01],
        [4.6573e-05, 9.9995e-01],
        [1.4795e-06, 1.0000e+00],
        [9.9253e-01, 7.4674e-03],
        [9.9964e-01, 3.5606e-04],
        [1.2574e-10, 1.0000e+00],
        [9.9991e-01, 9.1214e-05],
        [5.4829e-06, 9.9999e-01],
        [9.9902e-01, 9.7633e-04],
        [1.2528e-03, 9.9875e-01],
        [9.7336e-01, 2.6644e-02],
        [6.6814e-06, 9.9999e-01],
        [5.6474e-07, 1.0000e+00],
        [9.9959e-01, 4.0729e-04],
        [1.1553e-10, 1.0000e+00],
        [5.8733e-13, 1.0000e+00],
        [4.5928e-05, 9.9995e-01],
        [2.0372e-13, 1.0000e+00],
        [1.1425e-10, 1.0000e+00],
        [2.7875e-10, 1.0000e+00],
        [9.9799e-01, 2.0089e-03],
        [2.5928e-08, 1.0000e+00],
        [4.0402e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9947e-01, 5.3102e-04],
        [3.1175e-08, 1.0000e+00],
        [3.9473e-10, 1.0000e+00],
        [1.6892e-09, 1.0000e+00],
        [9.9847e-01, 1.5344e-03],
        [2.3181e-04, 9.9977e-01],
        [1.7244e-03, 9.9828e-01],
        [5.4498e-04, 9.9946e-01],
        [3.5765e-08, 1.0000e+00],
        [3.8822e-07, 1.0000e+00],
        [5.0969e-09, 1.0000e+00],
        [3.1028e-11, 1.0000e+00],
        [1.5623e-06, 1.0000e+00],
        [1.3547e-11, 1.0000e+00],
        [9.9973e-01, 2.7082e-04],
        [1.7315e-07, 1.0000e+00],
        [5.9890e-09, 1.0000e+00],
        [9.5946e-08, 1.0000e+00],
        [1.0035e-07, 1.0000e+00],
        [3.8837e-15, 1.0000e+00],
        [9.9980e-01, 2.0268e-04],
        [9.9587e-01, 4.1287e-03],
        [9.9999e-01, 1.0138e-05],
        [2.9638e-11, 1.0000e+00],
        [1.8187e-08, 1.0000e+00],
        [1.0257e-06, 1.0000e+00],
        [1.7123e-07, 1.0000e+00],
        [2.0122e-05, 9.9998e-01],
        [3.6781e-03, 9.9632e-01],
        [3.0291e-06, 1.0000e+00],
        [1.3203e-07, 1.0000e+00],
        [1.4769e-10, 1.0000e+00],
        [3.3943e-04, 9.9966e-01],
        [2.7931e-10, 1.0000e+00],
        [9.9908e-01, 9.1517e-04],
        [1.8005e-06, 1.0000e+00],
        [9.9974e-01, 2.6268e-04],
        [7.9118e-07, 1.0000e+00],
        [4.1732e-05, 9.9996e-01],
        [1.3597e-11, 1.0000e+00],
        [3.8642e-07, 1.0000e+00],
        [6.5806e-06, 9.9999e-01],
        [9.9984e-01, 1.6450e-04],
        [1.6539e-07, 1.0000e+00],
        [9.9979e-01, 2.0729e-04],
        [9.9981e-01, 1.8508e-04],
        [1.7715e-08, 1.0000e+00],
        [9.9970e-01, 3.0022e-04],
        [8.4850e-10, 1.0000e+00],
        [1.5613e-05, 9.9998e-01],
        [1.0995e-06, 1.0000e+00],
        [9.9982e-01, 1.7874e-04],
        [1.1411e-08, 1.0000e+00],
        [1.3749e-08, 1.0000e+00],
        [1.1807e-04, 9.9988e-01],
        [8.0915e-08, 1.0000e+00],
        [5.0874e-06, 9.9999e-01],
        [3.7655e-03, 9.9623e-01],
        [1.0642e-03, 9.9894e-01],
        [5.3642e-08, 1.0000e+00],
        [7.8043e-10, 1.0000e+00],
        [3.4895e-05, 9.9997e-01],
        [1.9053e-07, 1.0000e+00],
        [9.9282e-01, 7.1842e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[5.8722e-07, 1.0000e+00],
        [8.3512e-10, 1.0000e+00],
        [1.0065e-10, 1.0000e+00],
        [7.7078e-05, 9.9992e-01],
        [9.4544e-09, 1.0000e+00],
        [3.0144e-11, 1.0000e+00],
        [2.9740e-02, 9.7026e-01],
        [3.6518e-12, 1.0000e+00],
        [1.8471e-04, 9.9982e-01],
        [8.6443e-01, 1.3557e-01],
        [1.3736e-09, 1.0000e+00],
        [8.8892e-12, 1.0000e+00],
        [8.2587e-10, 1.0000e+00],
        [2.3933e-11, 1.0000e+00],
        [1.0971e-09, 1.0000e+00],
        [9.9059e-01, 9.4064e-03],
        [1.3234e-07, 1.0000e+00],
        [1.9912e-06, 1.0000e+00],
        [8.5261e-10, 1.0000e+00],
        [9.9990e-01, 1.0208e-04],
        [9.9969e-01, 3.1462e-04],
        [7.6699e-10, 1.0000e+00],
        [9.9947e-01, 5.2751e-04],
        [9.9939e-01, 6.1265e-04],
        [1.1006e-06, 1.0000e+00],
        [8.8525e-11, 1.0000e+00],
        [5.1181e-08, 1.0000e+00],
        [6.5562e-07, 1.0000e+00],
        [9.9947e-01, 5.2718e-04],
        [8.6457e-09, 1.0000e+00],
        [2.8901e-05, 9.9997e-01],
        [1.1895e-06, 1.0000e+00],
        [2.3588e-10, 1.0000e+00],
        [8.4736e-04, 9.9915e-01],
        [9.9995e-01, 5.2483e-05],
        [2.6616e-13, 1.0000e+00],
        [9.9999e-01, 1.1278e-05],
        [3.2192e-05, 9.9997e-01],
        [3.2968e-07, 1.0000e+00],
        [9.9982e-01, 1.7515e-04],
        [6.0428e-03, 9.9396e-01],
        [1.7208e-04, 9.9983e-01],
        [1.2720e-08, 1.0000e+00],
        [5.6341e-07, 1.0000e+00],
        [9.7325e-07, 1.0000e+00],
        [4.6827e-07, 1.0000e+00],
        [9.1884e-01, 8.1161e-02],
        [1.0284e-05, 9.9999e-01],
        [1.3141e-07, 1.0000e+00],
        [4.8585e-12, 1.0000e+00],
        [9.9950e-01, 4.9649e-04],
        [9.9983e-01, 1.6667e-04],
        [9.8863e-01, 1.1369e-02],
        [1.8477e-01, 8.1523e-01],
        [3.8984e-05, 9.9996e-01],
        [2.9172e-07, 1.0000e+00],
        [1.6377e-08, 1.0000e+00],
        [9.9960e-01, 3.9772e-04],
        [2.2524e-05, 9.9998e-01],
        [7.0152e-08, 1.0000e+00],
        [1.0542e-08, 1.0000e+00],
        [1.5665e-06, 1.0000e+00],
        [6.9796e-06, 9.9999e-01],
        [1.3736e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9079e-07, 1.0000e+00],
        [1.4267e-12, 1.0000e+00],
        [2.1350e-05, 9.9998e-01],
        [1.8751e-08, 1.0000e+00],
        [9.9784e-01, 2.1631e-03],
        [9.8685e-01, 1.3151e-02],
        [2.5393e-05, 9.9997e-01],
        [5.9232e-08, 1.0000e+00],
        [7.0186e-03, 9.9298e-01],
        [1.6315e-05, 9.9998e-01],
        [3.4165e-08, 1.0000e+00],
        [9.9879e-01, 1.2141e-03],
        [8.2796e-11, 1.0000e+00],
        [1.3381e-06, 1.0000e+00],
        [9.4347e-08, 1.0000e+00],
        [7.8493e-07, 1.0000e+00],
        [2.6654e-10, 1.0000e+00],
        [2.0131e-04, 9.9980e-01],
        [6.2944e-11, 1.0000e+00],
        [2.0025e-11, 1.0000e+00],
        [8.0312e-01, 1.9688e-01],
        [9.9998e-01, 2.2199e-05],
        [7.7043e-09, 1.0000e+00],
        [9.9982e-01, 1.7795e-04],
        [3.3552e-09, 1.0000e+00],
        [3.4591e-03, 9.9654e-01],
        [1.4632e-06, 1.0000e+00],
        [9.8741e-01, 1.2588e-02],
        [3.8614e-04, 9.9961e-01],
        [6.8242e-08, 1.0000e+00],
        [3.3297e-11, 1.0000e+00],
        [1.8726e-05, 9.9998e-01],
        [3.7270e-12, 1.0000e+00],
        [9.9620e-01, 3.7985e-03],
        [9.9967e-01, 3.2726e-04],
        [1.9524e-06, 1.0000e+00],
        [4.8172e-06, 1.0000e+00],
        [2.8403e-10, 1.0000e+00],
        [1.0531e-04, 9.9989e-01],
        [1.0191e-07, 1.0000e+00],
        [9.9958e-01, 4.1538e-04],
        [2.0630e-06, 1.0000e+00],
        [4.5925e-07, 1.0000e+00],
        [2.1740e-03, 9.9783e-01],
        [9.9951e-01, 4.8729e-04],
        [9.6075e-11, 1.0000e+00],
        [5.7683e-05, 9.9994e-01],
        [5.6962e-02, 9.4304e-01],
        [4.6511e-03, 9.9535e-01],
        [2.2683e-09, 1.0000e+00],
        [3.2828e-08, 1.0000e+00],
        [3.4548e-05, 9.9997e-01],
        [8.9315e-12, 1.0000e+00],
        [2.6627e-09, 1.0000e+00],
        [6.4347e-01, 3.5653e-01],
        [1.1568e-07, 1.0000e+00],
        [9.9839e-01, 1.6137e-03],
        [9.9985e-01, 1.5308e-04],
        [9.9850e-01, 1.5019e-03],
        [9.9000e-01, 1.0001e-02],
        [5.9499e-08, 1.0000e+00],
        [7.1587e-08, 1.0000e+00],
        [5.3171e-11, 1.0000e+00],
        [5.5001e-01, 4.4999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9996e-01, 4.1995e-05],
        [3.4623e-06, 1.0000e+00],
        [1.8735e-07, 1.0000e+00],
        [2.0347e-01, 7.9653e-01],
        [1.2253e-08, 1.0000e+00],
        [9.8974e-01, 1.0258e-02],
        [2.1399e-12, 1.0000e+00],
        [4.5423e-04, 9.9955e-01],
        [9.9700e-01, 3.0020e-03],
        [7.8240e-06, 9.9999e-01],
        [9.9989e-01, 1.0594e-04],
        [9.3269e-02, 9.0673e-01],
        [9.1343e-01, 8.6572e-02],
        [1.9760e-05, 9.9998e-01],
        [9.9981e-01, 1.9413e-04],
        [8.2359e-04, 9.9918e-01],
        [6.4787e-09, 1.0000e+00],
        [9.9922e-01, 7.8060e-04],
        [9.9922e-01, 7.7832e-04],
        [1.0696e-06, 1.0000e+00],
        [1.6003e-09, 1.0000e+00],
        [9.9980e-01, 1.9520e-04],
        [1.0000e+00, 4.3958e-06],
        [1.4159e-04, 9.9986e-01],
        [1.2936e-05, 9.9999e-01],
        [1.6325e-06, 1.0000e+00],
        [7.8352e-13, 1.0000e+00],
        [1.2434e-07, 1.0000e+00],
        [9.7563e-01, 2.4373e-02],
        [2.1989e-08, 1.0000e+00],
        [3.0727e-10, 1.0000e+00],
        [4.5209e-10, 1.0000e+00],
        [9.9871e-01, 1.2872e-03],
        [8.5860e-08, 1.0000e+00],
        [7.3446e-08, 1.0000e+00],
        [8.2040e-08, 1.0000e+00],
        [1.0000e+00, 2.8830e-07],
        [8.8222e-10, 1.0000e+00],
        [1.0205e-07, 1.0000e+00],
        [1.9156e-10, 1.0000e+00],
        [9.9973e-01, 2.7037e-04],
        [2.3364e-09, 1.0000e+00],
        [4.9094e-03, 9.9509e-01],
        [5.2287e-12, 1.0000e+00],
        [1.5875e-05, 9.9998e-01],
        [8.9286e-01, 1.0714e-01],
        [1.9190e-05, 9.9998e-01],
        [7.8083e-04, 9.9922e-01],
        [9.9990e-01, 9.8271e-05],
        [6.5381e-07, 1.0000e+00],
        [1.3220e-06, 1.0000e+00],
        [5.8251e-08, 1.0000e+00],
        [4.9321e-10, 1.0000e+00],
        [5.7458e-11, 1.0000e+00],
        [7.1500e-10, 1.0000e+00],
        [9.9968e-01, 3.1690e-04],
        [1.0000e+00, 3.8079e-06],
        [2.4832e-05, 9.9998e-01],
        [1.5083e-06, 1.0000e+00],
        [1.6813e-08, 1.0000e+00],
        [1.0903e-08, 1.0000e+00],
        [5.0680e-11, 1.0000e+00],
        [1.7651e-10, 1.0000e+00],
        [1.8196e-10, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0838e-05, 9.9997e-01],
        [1.1238e-08, 1.0000e+00],
        [2.3262e-03, 9.9767e-01],
        [9.9991e-01, 8.8083e-05],
        [2.2463e-08, 1.0000e+00],
        [9.9951e-01, 4.8795e-04],
        [7.8334e-01, 2.1666e-01],
        [1.8009e-03, 9.9820e-01],
        [1.9905e-10, 1.0000e+00],
        [4.8794e-01, 5.1206e-01],
        [2.7404e-11, 1.0000e+00],
        [2.0394e-10, 1.0000e+00],
        [8.6149e-09, 1.0000e+00],
        [9.8422e-01, 1.5779e-02],
        [3.2009e-04, 9.9968e-01],
        [1.8839e-10, 1.0000e+00],
        [1.0587e-10, 1.0000e+00],
        [1.0306e-09, 1.0000e+00],
        [9.9976e-01, 2.4174e-04],
        [2.1932e-06, 1.0000e+00],
        [4.3708e-09, 1.0000e+00],
        [4.0232e-09, 1.0000e+00],
        [5.2394e-09, 1.0000e+00],
        [1.3998e-08, 1.0000e+00],
        [8.3095e-04, 9.9917e-01],
        [9.1074e-08, 1.0000e+00],
        [9.9978e-01, 2.1604e-04],
        [9.7609e-01, 2.3915e-02],
        [9.9978e-01, 2.2320e-04],
        [1.4423e-11, 1.0000e+00],
        [1.1764e-03, 9.9882e-01],
        [6.5168e-07, 1.0000e+00],
        [8.8143e-07, 1.0000e+00],
        [9.9823e-01, 1.7735e-03],
        [9.9945e-01, 5.4832e-04],
        [2.5869e-08, 1.0000e+00],
        [1.2395e-07, 1.0000e+00],
        [8.4483e-05, 9.9992e-01],
        [9.9927e-01, 7.2643e-04],
        [4.3822e-10, 1.0000e+00],
        [2.9278e-10, 1.0000e+00],
        [2.1311e-06, 1.0000e+00],
        [9.9413e-01, 5.8688e-03],
        [5.3104e-08, 1.0000e+00],
        [9.9990e-01, 1.0016e-04],
        [1.1413e-09, 1.0000e+00],
        [6.2302e-06, 9.9999e-01],
        [3.0612e-10, 1.0000e+00],
        [2.5683e-02, 9.7432e-01],
        [4.0750e-06, 1.0000e+00],
        [9.9930e-01, 6.9524e-04],
        [2.6359e-07, 1.0000e+00],
        [1.7457e-05, 9.9998e-01],
        [2.0773e-02, 9.7923e-01],
        [7.6106e-04, 9.9924e-01],
        [3.6700e-06, 1.0000e+00],
        [6.9713e-07, 1.0000e+00],
        [5.7734e-07, 1.0000e+00],
        [2.4403e-08, 1.0000e+00],
        [1.0751e-07, 1.0000e+00],
        [9.5896e-01, 4.1042e-02],
        [4.7233e-09, 1.0000e+00],
        [6.9109e-07, 1.0000e+00],
        [4.3917e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.3982e-08, 1.0000e+00],
        [1.9054e-04, 9.9981e-01],
        [4.5098e-07, 1.0000e+00],
        [9.9933e-01, 6.6764e-04],
        [1.0147e-08, 1.0000e+00],
        [9.9983e-01, 1.7208e-04],
        [7.8417e-02, 9.2158e-01],
        [5.8152e-02, 9.4185e-01],
        [1.5031e-05, 9.9998e-01],
        [1.2849e-11, 1.0000e+00],
        [8.3579e-04, 9.9916e-01],
        [8.4669e-04, 9.9915e-01],
        [9.9985e-01, 1.5047e-04],
        [2.1755e-08, 1.0000e+00],
        [2.6188e-09, 1.0000e+00],
        [5.1099e-08, 1.0000e+00],
        [9.9984e-01, 1.5619e-04],
        [9.9979e-01, 2.1236e-04],
        [2.0184e-08, 1.0000e+00],
        [9.9969e-01, 3.1089e-04],
        [5.1357e-07, 1.0000e+00],
        [6.6223e-05, 9.9993e-01],
        [9.9934e-01, 6.6014e-04],
        [9.9983e-01, 1.7130e-04],
        [6.7116e-08, 1.0000e+00],
        [3.8277e-08, 1.0000e+00],
        [1.6046e-07, 1.0000e+00],
        [7.6140e-13, 1.0000e+00],
        [7.4677e-08, 1.0000e+00],
        [6.1376e-03, 9.9386e-01],
        [9.9987e-01, 1.3111e-04],
        [4.4439e-03, 9.9556e-01],
        [1.1667e-10, 1.0000e+00],
        [1.9753e-02, 9.8025e-01],
        [1.4415e-07, 1.0000e+00],
        [4.1101e-11, 1.0000e+00],
        [2.1597e-05, 9.9998e-01],
        [1.0913e-07, 1.0000e+00],
        [7.9417e-01, 2.0583e-01],
        [6.4174e-05, 9.9994e-01],
        [2.8085e-04, 9.9972e-01],
        [9.0630e-01, 9.3695e-02],
        [6.9288e-06, 9.9999e-01],
        [3.0332e-04, 9.9970e-01],
        [2.3026e-09, 1.0000e+00],
        [9.3857e-12, 1.0000e+00],
        [1.5704e-07, 1.0000e+00],
        [4.9457e-07, 1.0000e+00],
        [4.4125e-03, 9.9559e-01],
        [1.0000e+00, 4.2607e-06],
        [9.5899e-01, 4.1010e-02],
        [9.9973e-01, 2.6813e-04],
        [8.0272e-01, 1.9728e-01],
        [2.6899e-11, 1.0000e+00],
        [9.9987e-01, 1.2620e-04],
        [2.1605e-06, 1.0000e+00],
        [5.9707e-01, 4.0293e-01],
        [1.5604e-05, 9.9998e-01],
        [4.8819e-08, 1.0000e+00],
        [9.9995e-01, 4.5456e-05],
        [5.1464e-01, 4.8536e-01],
        [3.5778e-09, 1.0000e+00],
        [4.6220e-08, 1.0000e+00],
        [9.9926e-01, 7.4244e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0948e-08, 1.0000e+00],
        [2.4790e-05, 9.9998e-01],
        [1.7410e-07, 1.0000e+00],
        [9.9370e-01, 6.2979e-03],
        [9.9977e-01, 2.3492e-04],
        [8.1015e-07, 1.0000e+00],
        [7.4293e-11, 1.0000e+00],
        [1.5312e-01, 8.4688e-01],
        [6.7021e-05, 9.9993e-01],
        [1.3784e-07, 1.0000e+00],
        [6.9233e-07, 1.0000e+00],
        [1.5850e-05, 9.9998e-01],
        [9.9881e-01, 1.1911e-03],
        [3.0508e-06, 1.0000e+00],
        [4.3554e-05, 9.9996e-01],
        [9.9958e-01, 4.2094e-04],
        [9.9973e-01, 2.7253e-04],
        [1.3412e-07, 1.0000e+00],
        [1.8591e-06, 1.0000e+00],
        [1.6511e-04, 9.9983e-01],
        [9.9984e-01, 1.5533e-04],
        [1.0940e-05, 9.9999e-01],
        [7.2751e-07, 1.0000e+00],
        [2.4261e-04, 9.9976e-01],
        [8.3609e-10, 1.0000e+00],
        [7.6774e-05, 9.9992e-01],
        [5.1536e-10, 1.0000e+00],
        [9.3844e-06, 9.9999e-01],
        [7.2800e-03, 9.9272e-01],
        [5.3521e-10, 1.0000e+00],
        [2.9803e-06, 1.0000e+00],
        [4.2179e-10, 1.0000e+00],
        [9.9757e-01, 2.4262e-03],
        [1.8953e-08, 1.0000e+00],
        [7.2145e-03, 9.9279e-01],
        [9.9989e-01, 1.1242e-04],
        [5.5668e-03, 9.9443e-01],
        [3.4236e-06, 1.0000e+00],
        [2.8361e-10, 1.0000e+00],
        [7.1525e-08, 1.0000e+00],
        [9.9997e-01, 2.5523e-05],
        [3.9667e-05, 9.9996e-01],
        [6.6031e-01, 3.3969e-01],
        [3.4086e-08, 1.0000e+00],
        [9.3101e-01, 6.8987e-02],
        [7.0102e-11, 1.0000e+00],
        [1.1581e-05, 9.9999e-01],
        [9.9942e-01, 5.7676e-04],
        [2.7590e-09, 1.0000e+00],
        [6.8524e-08, 1.0000e+00],
        [1.6500e-01, 8.3500e-01],
        [9.9980e-01, 2.0277e-04],
        [9.1717e-09, 1.0000e+00],
        [9.9981e-01, 1.9226e-04],
        [8.2941e-03, 9.9171e-01],
        [3.8945e-09, 1.0000e+00],
        [9.9977e-01, 2.3492e-04],
        [9.9990e-01, 1.0054e-04],
        [9.9979e-01, 2.0963e-04],
        [3.3438e-11, 1.0000e+00],
        [9.9796e-01, 2.0352e-03],
        [9.9935e-01, 6.5085e-04],
        [3.6134e-11, 1.0000e+00],
        [9.7964e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5336e-06, 1.0000e+00],
        [1.1400e-06, 1.0000e+00],
        [1.5371e-09, 1.0000e+00],
        [1.0855e-11, 1.0000e+00],
        [2.6681e-06, 1.0000e+00],
        [5.1007e-10, 1.0000e+00],
        [5.7794e-03, 9.9422e-01],
        [9.9950e-01, 4.9746e-04],
        [1.1637e-07, 1.0000e+00],
        [2.6994e-11, 1.0000e+00],
        [9.6304e-01, 3.6963e-02],
        [1.8803e-06, 1.0000e+00],
        [3.0854e-05, 9.9997e-01],
        [5.2873e-02, 9.4713e-01],
        [6.7624e-05, 9.9993e-01],
        [5.0523e-05, 9.9995e-01],
        [9.9990e-01, 9.5174e-05],
        [1.1410e-07, 1.0000e+00],
        [3.6043e-10, 1.0000e+00],
        [5.3570e-11, 1.0000e+00],
        [5.4442e-07, 1.0000e+00],
        [9.9946e-01, 5.4284e-04],
        [5.4792e-07, 1.0000e+00],
        [3.0882e-05, 9.9997e-01],
        [2.5239e-07, 1.0000e+00],
        [1.8089e-02, 9.8191e-01],
        [1.2423e-06, 1.0000e+00],
        [4.2512e-08, 1.0000e+00],
        [4.3834e-10, 1.0000e+00],
        [6.9488e-07, 1.0000e+00],
        [1.0000e+00, 3.5062e-09],
        [1.4674e-07, 1.0000e+00],
        [1.1693e-08, 1.0000e+00],
        [1.3023e-04, 9.9987e-01],
        [9.9484e-01, 5.1577e-03],
        [5.7398e-01, 4.2602e-01],
        [9.9978e-01, 2.1971e-04],
        [9.9829e-01, 1.7103e-03],
        [1.4693e-02, 9.8531e-01],
        [4.7712e-06, 1.0000e+00],
        [7.0723e-08, 1.0000e+00],
        [2.6965e-04, 9.9973e-01],
        [6.7224e-07, 1.0000e+00],
        [9.9988e-01, 1.2116e-04],
        [6.7961e-12, 1.0000e+00],
        [5.2344e-11, 1.0000e+00],
        [9.6959e-04, 9.9903e-01],
        [1.1580e-08, 1.0000e+00],
        [1.1592e-06, 1.0000e+00],
        [1.2556e-04, 9.9987e-01],
        [9.9986e-01, 1.4092e-04],
        [4.6372e-06, 1.0000e+00],
        [9.9994e-01, 6.2403e-05],
        [1.4644e-09, 1.0000e+00],
        [9.8912e-01, 1.0881e-02],
        [3.7801e-07, 1.0000e+00],
        [6.9457e-06, 9.9999e-01],
        [8.0618e-08, 1.0000e+00],
        [5.3536e-05, 9.9995e-01],
        [1.0243e-05, 9.9999e-01],
        [9.9982e-01, 1.7961e-04],
        [9.9990e-01, 1.0107e-04],
        [9.8317e-01, 1.6825e-02],
        [9.9921e-01, 7.8777e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[7.9918e-03, 9.9201e-01],
        [1.2357e-04, 9.9988e-01],
        [6.3040e-07, 1.0000e+00],
        [4.8785e-06, 1.0000e+00],
        [6.9408e-08, 1.0000e+00],
        [1.8334e-08, 1.0000e+00],
        [9.9325e-01, 6.7484e-03],
        [9.9930e-01, 6.9796e-04],
        [5.4825e-04, 9.9945e-01],
        [1.0000e+00, 3.9702e-06],
        [1.8687e-08, 1.0000e+00],
        [1.5640e-05, 9.9998e-01],
        [8.9753e-05, 9.9991e-01],
        [6.3220e-08, 1.0000e+00],
        [9.9989e-01, 1.0652e-04],
        [9.9997e-01, 2.9475e-05],
        [3.1628e-07, 1.0000e+00],
        [9.9067e-05, 9.9990e-01],
        [1.7488e-08, 1.0000e+00],
        [1.7005e-11, 1.0000e+00],
        [3.0895e-10, 1.0000e+00],
        [2.3153e-06, 1.0000e+00],
        [1.0928e-08, 1.0000e+00],
        [9.0532e-06, 9.9999e-01],
        [8.8188e-04, 9.9912e-01],
        [4.1602e-10, 1.0000e+00],
        [6.8050e-08, 1.0000e+00],
        [9.9974e-01, 2.6429e-04],
        [2.4267e-05, 9.9998e-01],
        [1.2352e-06, 1.0000e+00],
        [9.9659e-01, 3.4112e-03],
        [4.8558e-07, 1.0000e+00],
        [9.9944e-01, 5.6066e-04],
        [1.6218e-08, 1.0000e+00],
        [9.9901e-01, 9.8900e-04],
        [9.9986e-01, 1.4027e-04],
        [1.1088e-04, 9.9989e-01],
        [3.9015e-05, 9.9996e-01],
        [9.9999e-01, 7.5582e-06],
        [1.1007e-03, 9.9890e-01],
        [9.9987e-01, 1.3468e-04],
        [3.7991e-10, 1.0000e+00],
        [8.1741e-07, 1.0000e+00],
        [2.8097e-08, 1.0000e+00],
        [1.7763e-07, 1.0000e+00],
        [1.3494e-05, 9.9999e-01],
        [2.9413e-08, 1.0000e+00],
        [7.9778e-01, 2.0222e-01],
        [8.3755e-08, 1.0000e+00],
        [3.6970e-03, 9.9630e-01],
        [1.6970e-09, 1.0000e+00],
        [7.9179e-10, 1.0000e+00],
        [2.5174e-06, 1.0000e+00],
        [1.0000e+00, 6.4690e-07],
        [1.6223e-05, 9.9998e-01],
        [8.8115e-05, 9.9991e-01],
        [2.2352e-06, 1.0000e+00],
        [5.3551e-06, 9.9999e-01],
        [1.9489e-06, 1.0000e+00],
        [2.1451e-10, 1.0000e+00],
        [9.9985e-01, 1.5447e-04],
        [1.2854e-07, 1.0000e+00],
        [2.3993e-02, 9.7601e-01],
        [9.9991e-01, 9.3064e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[8.2160e-07, 1.0000e+00],
        [2.7177e-03, 9.9728e-01],
        [9.1364e-05, 9.9991e-01],
        [9.9984e-01, 1.5675e-04],
        [6.7196e-05, 9.9993e-01],
        [2.3002e-10, 1.0000e+00],
        [9.9342e-01, 6.5757e-03],
        [3.4157e-13, 1.0000e+00],
        [1.8338e-06, 1.0000e+00],
        [5.7651e-03, 9.9423e-01],
        [9.9988e-01, 1.1969e-04],
        [9.8002e-01, 1.9978e-02],
        [4.5114e-08, 1.0000e+00],
        [4.2940e-06, 1.0000e+00],
        [1.6433e-09, 1.0000e+00],
        [9.9993e-01, 7.4991e-05],
        [9.9991e-01, 9.1805e-05],
        [6.5389e-09, 1.0000e+00],
        [6.4975e-07, 1.0000e+00],
        [1.7355e-05, 9.9998e-01],
        [9.6651e-09, 1.0000e+00],
        [1.5719e-01, 8.4281e-01],
        [1.1435e-07, 1.0000e+00],
        [1.1093e-01, 8.8907e-01],
        [1.9040e-04, 9.9981e-01],
        [9.9618e-09, 1.0000e+00],
        [2.9637e-08, 1.0000e+00],
        [9.9987e-01, 1.2762e-04],
        [7.5167e-01, 2.4833e-01],
        [3.7302e-06, 1.0000e+00],
        [6.4255e-09, 1.0000e+00],
        [3.3182e-01, 6.6818e-01],
        [9.0697e-06, 9.9999e-01],
        [4.2206e-08, 1.0000e+00],
        [1.4072e-04, 9.9986e-01],
        [1.1137e-01, 8.8863e-01],
        [9.9998e-01, 2.0409e-05],
        [4.6078e-07, 1.0000e+00],
        [5.3395e-07, 1.0000e+00],
        [7.6180e-06, 9.9999e-01],
        [9.9958e-01, 4.2443e-04],
        [1.1691e-04, 9.9988e-01],
        [5.6013e-08, 1.0000e+00],
        [4.2335e-04, 9.9958e-01],
        [9.5842e-01, 4.1581e-02],
        [2.6460e-01, 7.3540e-01],
        [9.9994e-01, 6.2715e-05],
        [7.1447e-05, 9.9993e-01],
        [9.9989e-01, 1.1241e-04],
        [2.7122e-05, 9.9997e-01],
        [9.9993e-01, 6.5368e-05],
        [1.9492e-07, 1.0000e+00],
        [7.1172e-08, 1.0000e+00],
        [9.9982e-01, 1.8026e-04],
        [4.6852e-06, 1.0000e+00],
        [1.1135e-08, 1.0000e+00],
        [1.4735e-08, 1.0000e+00],
        [8.9967e-01, 1.0033e-01],
        [2.5062e-08, 1.0000e+00],
        [2.2173e-07, 1.0000e+00],
        [5.0035e-05, 9.9995e-01],
        [9.9829e-01, 1.7123e-03],
        [2.4975e-08, 1.0000e+00],
        [4.5815e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.3136e-06, 9.9999e-01],
        [1.8990e-05, 9.9998e-01],
        [2.9745e-06, 1.0000e+00],
        [1.5854e-09, 1.0000e+00],
        [9.9975e-01, 2.4571e-04],
        [2.5744e-05, 9.9997e-01],
        [4.3357e-05, 9.9996e-01],
        [5.4811e-05, 9.9995e-01],
        [1.4333e-06, 1.0000e+00],
        [3.2929e-11, 1.0000e+00],
        [2.6430e-08, 1.0000e+00],
        [1.3965e-05, 9.9999e-01],
        [1.1131e-08, 1.0000e+00],
        [9.9996e-01, 3.6663e-05],
        [9.9994e-01, 5.7689e-05],
        [9.9918e-01, 8.2047e-04],
        [7.6009e-08, 1.0000e+00],
        [7.4803e-02, 9.2520e-01],
        [2.8060e-08, 1.0000e+00],
        [6.3322e-02, 9.3668e-01],
        [1.9347e-06, 1.0000e+00],
        [4.6410e-06, 1.0000e+00],
        [8.5901e-01, 1.4099e-01],
        [1.6669e-07, 1.0000e+00],
        [1.1317e-09, 1.0000e+00],
        [9.9995e-01, 4.6436e-05],
        [2.6276e-08, 1.0000e+00],
        [9.9997e-01, 3.4487e-05],
        [9.0158e-11, 1.0000e+00],
        [8.9917e-01, 1.0083e-01],
        [9.9987e-01, 1.3268e-04],
        [1.0416e-02, 9.8958e-01],
        [3.1138e-02, 9.6886e-01],
        [9.9649e-01, 3.5091e-03],
        [1.9004e-05, 9.9998e-01],
        [1.3185e-09, 1.0000e+00],
        [5.7792e-06, 9.9999e-01],
        [5.5911e-06, 9.9999e-01],
        [7.6143e-07, 1.0000e+00],
        [9.7666e-01, 2.3337e-02],
        [7.0934e-01, 2.9066e-01],
        [1.4412e-07, 1.0000e+00],
        [2.1025e-06, 1.0000e+00],
        [9.9967e-01, 3.2851e-04],
        [4.3044e-06, 1.0000e+00],
        [5.0824e-09, 1.0000e+00],
        [9.9992e-01, 7.7510e-05],
        [4.5522e-04, 9.9954e-01],
        [1.6348e-10, 1.0000e+00],
        [9.9967e-01, 3.3275e-04],
        [1.7332e-06, 1.0000e+00],
        [9.2608e-06, 9.9999e-01],
        [9.9879e-01, 1.2128e-03],
        [3.3069e-08, 1.0000e+00],
        [1.6213e-05, 9.9998e-01],
        [1.0936e-04, 9.9989e-01],
        [2.0044e-08, 1.0000e+00],
        [1.4276e-06, 1.0000e+00],
        [9.9976e-01, 2.4282e-04],
        [7.3339e-02, 9.2666e-01],
        [9.9998e-01, 1.9098e-05],
        [9.9417e-01, 5.8330e-03],
        [2.5584e-08, 1.0000e+00],
        [2.0060e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.8144e-08, 1.0000e+00],
        [8.5628e-03, 9.9144e-01],
        [5.7070e-05, 9.9994e-01],
        [9.9485e-01, 5.1520e-03],
        [1.3370e-07, 1.0000e+00],
        [9.9945e-01, 5.5245e-04],
        [1.6778e-07, 1.0000e+00],
        [7.9162e-07, 1.0000e+00],
        [1.5628e-07, 1.0000e+00],
        [2.6248e-02, 9.7375e-01],
        [1.6684e-06, 1.0000e+00],
        [4.1814e-07, 1.0000e+00],
        [3.1434e-02, 9.6857e-01],
        [9.9996e-01, 4.0905e-05],
        [2.5949e-06, 1.0000e+00],
        [1.8970e-09, 1.0000e+00],
        [1.6794e-04, 9.9983e-01],
        [9.9491e-01, 5.0938e-03],
        [1.1539e-08, 1.0000e+00],
        [1.3952e-03, 9.9860e-01],
        [2.8591e-10, 1.0000e+00],
        [4.0411e-06, 1.0000e+00],
        [3.1505e-07, 1.0000e+00],
        [3.9232e-07, 1.0000e+00],
        [2.7946e-07, 1.0000e+00],
        [4.1864e-09, 1.0000e+00],
        [2.9691e-07, 1.0000e+00],
        [3.1574e-04, 9.9968e-01],
        [2.8000e-04, 9.9972e-01],
        [1.6520e-09, 1.0000e+00],
        [9.9997e-01, 2.5304e-05],
        [1.3046e-07, 1.0000e+00],
        [9.9997e-01, 2.5641e-05],
        [9.9991e-01, 9.3409e-05],
        [3.3937e-07, 1.0000e+00],
        [4.0510e-09, 1.0000e+00],
        [5.5745e-08, 1.0000e+00],
        [4.2880e-02, 9.5712e-01],
        [5.2425e-06, 9.9999e-01],
        [9.9992e-01, 7.5868e-05],
        [6.9281e-03, 9.9307e-01],
        [2.3140e-07, 1.0000e+00],
        [7.0847e-04, 9.9929e-01],
        [9.9984e-01, 1.5944e-04],
        [7.7478e-05, 9.9992e-01],
        [1.9604e-09, 1.0000e+00],
        [8.1586e-05, 9.9992e-01],
        [4.7213e-05, 9.9995e-01],
        [6.0572e-09, 1.0000e+00],
        [1.3974e-03, 9.9860e-01],
        [8.4955e-06, 9.9999e-01],
        [9.9985e-01, 1.5035e-04],
        [1.0491e-05, 9.9999e-01],
        [1.9139e-03, 9.9809e-01],
        [7.7850e-06, 9.9999e-01],
        [2.8297e-06, 1.0000e+00],
        [9.9980e-01, 1.9675e-04],
        [8.6506e-06, 9.9999e-01],
        [1.0589e-05, 9.9999e-01],
        [1.1913e-08, 1.0000e+00],
        [1.3635e-07, 1.0000e+00],
        [9.9983e-01, 1.6838e-04],
        [4.3162e-11, 1.0000e+00],
        [1.5393e-02, 9.8461e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9966e-01, 3.4469e-04],
        [9.0567e-07, 1.0000e+00],
        [1.5052e-07, 1.0000e+00],
        [5.3941e-04, 9.9946e-01],
        [5.8484e-09, 1.0000e+00],
        [1.5375e-10, 1.0000e+00],
        [1.5377e-06, 1.0000e+00],
        [5.2118e-06, 9.9999e-01],
        [9.2494e-09, 1.0000e+00],
        [8.3713e-04, 9.9916e-01],
        [2.3955e-05, 9.9998e-01],
        [6.3866e-07, 1.0000e+00],
        [1.0943e-03, 9.9891e-01],
        [9.9994e-01, 6.4923e-05],
        [9.9301e-01, 6.9865e-03],
        [9.9994e-01, 6.3075e-05],
        [1.1912e-06, 1.0000e+00],
        [1.1157e-05, 9.9999e-01],
        [8.3751e-04, 9.9916e-01],
        [6.5409e-03, 9.9346e-01],
        [2.4513e-08, 1.0000e+00],
        [8.7644e-09, 1.0000e+00],
        [4.3017e-05, 9.9996e-01],
        [4.0804e-08, 1.0000e+00],
        [9.9992e-01, 7.7697e-05],
        [7.0872e-08, 1.0000e+00],
        [7.5537e-07, 1.0000e+00],
        [7.7266e-06, 9.9999e-01],
        [2.3939e-06, 1.0000e+00],
        [1.0887e-01, 8.9113e-01],
        [9.9994e-01, 6.4180e-05],
        [1.1013e-01, 8.8987e-01],
        [4.8459e-07, 1.0000e+00],
        [1.7084e-07, 1.0000e+00],
        [9.9985e-01, 1.4816e-04],
        [9.9938e-01, 6.2145e-04],
        [1.5377e-06, 1.0000e+00],
        [4.6572e-07, 1.0000e+00],
        [1.9277e-08, 1.0000e+00],
        [5.6908e-08, 1.0000e+00],
        [1.5206e-05, 9.9998e-01],
        [1.7561e-04, 9.9982e-01],
        [9.9979e-01, 2.0791e-04],
        [1.2214e-06, 1.0000e+00],
        [9.9965e-01, 3.4712e-04],
        [9.9999e-01, 6.7947e-06],
        [9.9991e-01, 8.5500e-05],
        [5.6546e-06, 9.9999e-01],
        [1.4703e-03, 9.9853e-01],
        [4.1897e-03, 9.9581e-01],
        [8.1165e-06, 9.9999e-01],
        [9.8128e-11, 1.0000e+00],
        [9.9991e-01, 8.8134e-05],
        [5.0783e-09, 1.0000e+00],
        [5.2630e-06, 9.9999e-01],
        [6.7912e-07, 1.0000e+00],
        [9.2433e-05, 9.9991e-01],
        [3.4691e-05, 9.9997e-01],
        [4.4725e-06, 1.0000e+00],
        [3.4895e-07, 1.0000e+00],
        [1.6674e-06, 1.0000e+00],
        [5.3837e-13, 1.0000e+00],
        [7.3547e-06, 9.9999e-01],
        [1.7621e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.8344e-08, 1.0000e+00],
        [4.5255e-05, 9.9995e-01],
        [4.5926e-11, 1.0000e+00],
        [2.8928e-06, 1.0000e+00],
        [5.3538e-02, 9.4646e-01],
        [9.9982e-01, 1.8384e-04],
        [1.1315e-03, 9.9887e-01],
        [9.9978e-01, 2.1947e-04],
        [9.9999e-01, 7.0576e-06],
        [3.4391e-01, 6.5609e-01],
        [6.1877e-06, 9.9999e-01],
        [9.9997e-01, 2.8153e-05],
        [3.1735e-06, 1.0000e+00],
        [9.9294e-01, 7.0609e-03],
        [1.9395e-02, 9.8061e-01],
        [9.9986e-01, 1.4144e-04],
        [3.8925e-02, 9.6107e-01],
        [4.1751e-10, 1.0000e+00],
        [7.1129e-02, 9.2887e-01],
        [3.3725e-08, 1.0000e+00],
        [1.8928e-11, 1.0000e+00],
        [1.4409e-05, 9.9999e-01],
        [3.6367e-09, 1.0000e+00],
        [9.9995e-01, 4.7515e-05],
        [8.3687e-09, 1.0000e+00],
        [5.9809e-03, 9.9402e-01],
        [8.0000e-06, 9.9999e-01],
        [9.9923e-01, 7.7189e-04],
        [2.1762e-06, 1.0000e+00],
        [1.7989e-05, 9.9998e-01],
        [7.7106e-11, 1.0000e+00],
        [9.9994e-01, 5.6412e-05],
        [8.0520e-06, 9.9999e-01],
        [1.0380e-01, 8.9620e-01],
        [1.9869e-09, 1.0000e+00],
        [1.8085e-04, 9.9982e-01],
        [5.3427e-08, 1.0000e+00],
        [1.0015e-07, 1.0000e+00],
        [9.5547e-01, 4.4529e-02],
        [5.1797e-05, 9.9995e-01],
        [2.6389e-09, 1.0000e+00],
        [6.4115e-06, 9.9999e-01],
        [2.4727e-06, 1.0000e+00],
        [2.2296e-10, 1.0000e+00],
        [3.4404e-06, 1.0000e+00],
        [8.7128e-02, 9.1287e-01],
        [1.9017e-02, 9.8098e-01],
        [1.1752e-07, 1.0000e+00],
        [9.9884e-01, 1.1551e-03],
        [9.9997e-01, 3.1935e-05],
        [9.8445e-01, 1.5554e-02],
        [9.9994e-01, 6.2803e-05],
        [6.1038e-05, 9.9994e-01],
        [2.7870e-04, 9.9972e-01],
        [3.7475e-03, 9.9625e-01],
        [6.3512e-05, 9.9994e-01],
        [9.9986e-01, 1.4256e-04],
        [1.3379e-06, 1.0000e+00],
        [2.4385e-08, 1.0000e+00],
        [5.6924e-01, 4.3076e-01],
        [7.4470e-05, 9.9993e-01],
        [5.0391e-10, 1.0000e+00],
        [5.0880e-08, 1.0000e+00],
        [1.7764e-04, 9.9982e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.3126e-03, 9.9569e-01],
        [9.9982e-01, 1.7505e-04],
        [1.3105e-04, 9.9987e-01],
        [3.1080e-06, 1.0000e+00],
        [9.9999e-01, 1.0490e-05],
        [3.7118e-06, 1.0000e+00],
        [1.7837e-06, 1.0000e+00],
        [2.1016e-09, 1.0000e+00],
        [7.3507e-03, 9.9265e-01],
        [1.0966e-05, 9.9999e-01],
        [9.9988e-01, 1.2121e-04],
        [3.7122e-06, 1.0000e+00],
        [4.5578e-06, 1.0000e+00],
        [9.4783e-05, 9.9991e-01],
        [1.5909e-04, 9.9984e-01],
        [1.5563e-05, 9.9998e-01],
        [8.5358e-01, 1.4642e-01],
        [2.6130e-09, 1.0000e+00],
        [4.9129e-07, 1.0000e+00],
        [2.1492e-02, 9.7851e-01],
        [4.2157e-07, 1.0000e+00],
        [3.6372e-03, 9.9636e-01],
        [1.7809e-07, 1.0000e+00],
        [1.8119e-03, 9.9819e-01],
        [1.4307e-05, 9.9999e-01],
        [9.9978e-01, 2.2306e-04],
        [9.9994e-01, 5.8549e-05],
        [4.7710e-11, 1.0000e+00],
        [1.1640e-06, 1.0000e+00],
        [2.1216e-03, 9.9788e-01],
        [1.5105e-04, 9.9985e-01],
        [9.9992e-01, 7.8422e-05],
        [9.3980e-09, 1.0000e+00],
        [2.4196e-07, 1.0000e+00],
        [2.2713e-02, 9.7729e-01],
        [1.6287e-07, 1.0000e+00],
        [5.8746e-05, 9.9994e-01],
        [4.3411e-04, 9.9957e-01],
        [4.0696e-06, 1.0000e+00],
        [6.2099e-09, 1.0000e+00],
        [1.2313e-07, 1.0000e+00],
        [4.5420e-05, 9.9995e-01],
        [1.5535e-04, 9.9984e-01],
        [1.0000e+00, 1.8965e-07],
        [3.7437e-05, 9.9996e-01],
        [3.5341e-07, 1.0000e+00],
        [9.9994e-01, 5.8805e-05],
        [1.9494e-09, 1.0000e+00],
        [1.8712e-03, 9.9813e-01],
        [1.5283e-05, 9.9998e-01],
        [1.6583e-07, 1.0000e+00],
        [3.4594e-05, 9.9997e-01],
        [9.9995e-01, 5.3745e-05],
        [1.9103e-05, 9.9998e-01],
        [1.6635e-04, 9.9983e-01],
        [2.0680e-04, 9.9979e-01],
        [6.5475e-11, 1.0000e+00],
        [5.6854e-09, 1.0000e+00],
        [3.5255e-09, 1.0000e+00],
        [1.3284e-04, 9.9987e-01],
        [9.8710e-01, 1.2904e-02],
        [2.8430e-05, 9.9997e-01],
        [9.5357e-01, 4.6428e-02],
        [2.7097e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.4815e-05, 9.9997e-01],
        [9.9997e-01, 2.6003e-05],
        [4.9290e-02, 9.5071e-01],
        [1.0093e-03, 9.9899e-01],
        [2.4996e-07, 1.0000e+00],
        [1.9300e-03, 9.9807e-01],
        [1.8217e-11, 1.0000e+00],
        [2.8356e-06, 1.0000e+00],
        [1.4632e-07, 1.0000e+00],
        [9.9991e-01, 8.5562e-05],
        [6.5050e-09, 1.0000e+00],
        [1.1246e-05, 9.9999e-01],
        [9.9868e-01, 1.3188e-03],
        [3.9646e-04, 9.9960e-01],
        [5.7584e-07, 1.0000e+00],
        [4.4402e-06, 1.0000e+00],
        [9.5453e-03, 9.9045e-01],
        [9.6266e-08, 1.0000e+00],
        [9.9998e-01, 2.4228e-05],
        [2.2719e-06, 1.0000e+00],
        [3.0790e-05, 9.9997e-01],
        [4.1740e-04, 9.9958e-01],
        [2.3018e-08, 1.0000e+00],
        [3.6789e-08, 1.0000e+00],
        [5.2620e-07, 1.0000e+00],
        [1.4497e-02, 9.8550e-01],
        [2.5963e-04, 9.9974e-01],
        [1.8303e-03, 9.9817e-01],
        [9.9991e-01, 8.9664e-05],
        [1.4821e-06, 1.0000e+00],
        [1.3037e-07, 1.0000e+00],
        [4.1884e-01, 5.8116e-01],
        [9.3317e-03, 9.9067e-01],
        [2.6140e-07, 1.0000e+00],
        [9.2214e-01, 7.7860e-02],
        [7.1186e-04, 9.9929e-01],
        [2.7305e-03, 9.9727e-01],
        [2.2119e-03, 9.9779e-01],
        [4.7310e-07, 1.0000e+00],
        [1.6398e-06, 1.0000e+00],
        [1.8538e-03, 9.9815e-01],
        [8.6308e-01, 1.3692e-01],
        [8.3689e-04, 9.9916e-01],
        [1.1766e-04, 9.9988e-01],
        [9.9987e-01, 1.3374e-04],
        [2.7749e-05, 9.9997e-01],
        [7.0995e-07, 1.0000e+00],
        [9.9989e-01, 1.1016e-04],
        [8.9696e-03, 9.9103e-01],
        [3.3198e-07, 1.0000e+00],
        [9.9974e-01, 2.5592e-04],
        [1.2088e-07, 1.0000e+00],
        [4.0830e-07, 1.0000e+00],
        [2.9144e-08, 1.0000e+00],
        [3.4585e-10, 1.0000e+00],
        [3.3906e-10, 1.0000e+00],
        [9.8312e-02, 9.0169e-01],
        [7.8780e-10, 1.0000e+00],
        [1.3263e-09, 1.0000e+00],
        [4.8650e-01, 5.1350e-01],
        [2.3344e-07, 1.0000e+00],
        [9.9992e-01, 8.3926e-05],
        [9.9999e-01, 9.1228e-06],
        [4.4402e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8740e-07, 1.0000e+00],
        [1.7481e-07, 1.0000e+00],
        [7.4271e-06, 9.9999e-01],
        [5.6436e-05, 9.9994e-01],
        [8.7352e-02, 9.1265e-01],
        [1.8426e-10, 1.0000e+00],
        [4.0260e-04, 9.9960e-01],
        [6.6040e-04, 9.9934e-01],
        [5.7938e-08, 1.0000e+00],
        [1.8864e-04, 9.9981e-01],
        [9.2599e-04, 9.9907e-01],
        [3.3229e-07, 1.0000e+00],
        [9.9990e-01, 1.0146e-04],
        [9.8627e-06, 9.9999e-01],
        [3.1027e-03, 9.9690e-01],
        [1.9754e-04, 9.9980e-01],
        [9.9987e-01, 1.2680e-04],
        [1.4610e-01, 8.5390e-01],
        [2.3515e-02, 9.7648e-01],
        [1.9860e-05, 9.9998e-01],
        [1.1080e-06, 1.0000e+00],
        [6.0308e-09, 1.0000e+00],
        [1.0228e-06, 1.0000e+00],
        [9.9993e-01, 6.7133e-05],
        [9.9984e-01, 1.6288e-04],
        [2.3040e-06, 1.0000e+00],
        [2.4569e-04, 9.9975e-01],
        [9.9990e-01, 1.0388e-04],
        [9.7813e-08, 1.0000e+00],
        [8.4578e-10, 1.0000e+00],
        [9.9995e-01, 5.2670e-05],
        [6.4455e-06, 9.9999e-01],
        [9.9742e-01, 2.5800e-03],
        [9.9997e-01, 3.2616e-05],
        [7.7519e-04, 9.9922e-01],
        [5.7264e-08, 1.0000e+00],
        [9.9994e-01, 5.6760e-05],
        [9.9952e-01, 4.7765e-04],
        [7.5668e-08, 1.0000e+00],
        [2.9706e-03, 9.9703e-01],
        [2.4657e-10, 1.0000e+00],
        [8.7094e-07, 1.0000e+00],
        [1.7522e-07, 1.0000e+00],
        [3.4637e-04, 9.9965e-01],
        [1.4092e-04, 9.9986e-01],
        [1.4576e-01, 8.5424e-01],
        [2.4652e-07, 1.0000e+00],
        [4.5884e-05, 9.9995e-01],
        [9.9993e-01, 7.4993e-05],
        [1.0000e+00, 1.8542e-06],
        [1.1077e-06, 1.0000e+00],
        [4.0059e-04, 9.9960e-01],
        [1.0612e-06, 1.0000e+00],
        [1.2054e-07, 1.0000e+00],
        [9.9996e-01, 4.3672e-05],
        [1.5230e-03, 9.9848e-01],
        [9.9996e-01, 4.0890e-05],
        [6.0372e-08, 1.0000e+00],
        [2.5938e-09, 1.0000e+00],
        [1.0524e-08, 1.0000e+00],
        [1.2835e-04, 9.9987e-01],
        [4.1209e-08, 1.0000e+00],
        [1.2031e-07, 1.0000e+00],
        [8.4959e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.0656e-05, 9.9992e-01],
        [1.8679e-10, 1.0000e+00],
        [9.9995e-01, 4.9327e-05],
        [2.7399e-10, 1.0000e+00],
        [7.9385e-03, 9.9206e-01],
        [9.9751e-01, 2.4862e-03],
        [2.6986e-09, 1.0000e+00],
        [1.2428e-08, 1.0000e+00],
        [6.9092e-07, 1.0000e+00],
        [9.9984e-01, 1.6138e-04],
        [2.1882e-04, 9.9978e-01],
        [2.3755e-04, 9.9976e-01],
        [3.0926e-06, 1.0000e+00],
        [1.0865e-05, 9.9999e-01],
        [2.3944e-07, 1.0000e+00],
        [5.7506e-05, 9.9994e-01],
        [1.0879e-05, 9.9999e-01],
        [9.9945e-01, 5.4509e-04],
        [1.0764e-03, 9.9892e-01],
        [2.3384e-06, 1.0000e+00],
        [2.1191e-05, 9.9998e-01],
        [1.9073e-08, 1.0000e+00],
        [9.9991e-01, 8.8404e-05],
        [9.9988e-01, 1.2024e-04],
        [4.0605e-04, 9.9959e-01],
        [8.4426e-07, 1.0000e+00],
        [1.7574e-06, 1.0000e+00],
        [1.1239e-08, 1.0000e+00],
        [1.3127e-04, 9.9987e-01],
        [3.3897e-08, 1.0000e+00],
        [2.0417e-05, 9.9998e-01],
        [1.3903e-06, 1.0000e+00],
        [9.9992e-01, 8.2756e-05],
        [3.2038e-06, 1.0000e+00],
        [1.1866e-02, 9.8813e-01],
        [2.6928e-01, 7.3072e-01],
        [4.0894e-07, 1.0000e+00],
        [9.9993e-01, 6.9684e-05],
        [8.7044e-03, 9.9130e-01],
        [1.0512e-07, 1.0000e+00],
        [4.5852e-06, 1.0000e+00],
        [2.2352e-01, 7.7648e-01],
        [3.3850e-08, 1.0000e+00],
        [9.9949e-01, 5.1169e-04],
        [5.1351e-09, 1.0000e+00],
        [2.3366e-07, 1.0000e+00],
        [2.6393e-04, 9.9974e-01],
        [7.7215e-01, 2.2785e-01],
        [9.9979e-01, 2.0949e-04],
        [3.9478e-07, 1.0000e+00],
        [9.9997e-01, 3.0136e-05],
        [1.5923e-10, 1.0000e+00],
        [2.7277e-03, 9.9727e-01],
        [8.2208e-06, 9.9999e-01],
        [4.7532e-06, 1.0000e+00],
        [3.3215e-02, 9.6678e-01],
        [1.6984e-10, 1.0000e+00],
        [9.9931e-01, 6.8712e-04],
        [3.9461e-10, 1.0000e+00],
        [1.7689e-09, 1.0000e+00],
        [9.9977e-01, 2.2746e-04],
        [1.0000e+00, 6.0728e-07],
        [3.2249e-05, 9.9997e-01],
        [1.4461e-03, 9.9855e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6264e-11, 1.0000e+00],
        [4.0108e-09, 1.0000e+00],
        [1.2007e-06, 1.0000e+00],
        [1.9531e-05, 9.9998e-01],
        [9.9985e-01, 1.4836e-04],
        [9.9984e-01, 1.6482e-04],
        [2.1614e-08, 1.0000e+00],
        [3.1566e-06, 1.0000e+00],
        [3.4874e-04, 9.9965e-01],
        [3.0183e-08, 1.0000e+00],
        [1.7436e-03, 9.9826e-01],
        [9.9986e-01, 1.4367e-04],
        [2.0065e-08, 1.0000e+00],
        [1.1795e-09, 1.0000e+00],
        [9.9773e-01, 2.2743e-03],
        [3.2865e-06, 1.0000e+00],
        [9.1573e-08, 1.0000e+00],
        [5.8439e-08, 1.0000e+00],
        [8.2858e-06, 9.9999e-01],
        [9.9997e-01, 3.4115e-05],
        [7.5482e-04, 9.9925e-01],
        [9.5757e-05, 9.9990e-01],
        [1.3692e-04, 9.9986e-01],
        [1.6019e-10, 1.0000e+00],
        [5.8637e-01, 4.1363e-01],
        [6.0076e-06, 9.9999e-01],
        [3.0952e-04, 9.9969e-01],
        [9.2457e-01, 7.5429e-02],
        [4.1736e-07, 1.0000e+00],
        [2.0251e-04, 9.9980e-01],
        [3.0046e-09, 1.0000e+00],
        [9.2457e-01, 7.5429e-02],
        [9.9946e-01, 5.4220e-04],
        [9.8709e-01, 1.2910e-02],
        [3.0620e-10, 1.0000e+00],
        [5.1335e-07, 1.0000e+00],
        [1.9057e-10, 1.0000e+00],
        [1.5430e-09, 1.0000e+00],
        [8.1615e-04, 9.9918e-01],
        [8.3662e-07, 1.0000e+00],
        [8.0931e-03, 9.9191e-01],
        [9.9957e-01, 4.3341e-04],
        [2.9682e-09, 1.0000e+00],
        [9.9992e-01, 7.5370e-05],
        [9.9996e-01, 4.1914e-05],
        [2.4369e-07, 1.0000e+00],
        [1.5015e-06, 1.0000e+00],
        [8.8367e-04, 9.9912e-01],
        [1.5329e-09, 1.0000e+00],
        [2.0252e-04, 9.9980e-01],
        [1.0285e-09, 1.0000e+00],
        [4.9853e-06, 9.9999e-01],
        [2.8361e-06, 1.0000e+00],
        [9.9999e-01, 5.9197e-06],
        [1.7889e-08, 1.0000e+00],
        [2.0681e-03, 9.9793e-01],
        [1.7314e-05, 9.9998e-01],
        [9.9995e-01, 5.4033e-05],
        [4.8225e-10, 1.0000e+00],
        [9.4847e-11, 1.0000e+00],
        [5.2227e-11, 1.0000e+00],
        [1.1362e-08, 1.0000e+00],
        [5.2930e-08, 1.0000e+00],
        [2.6176e-04, 9.9974e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[4.0635e-05, 9.9996e-01],
        [5.2003e-06, 9.9999e-01],
        [1.6722e-08, 1.0000e+00],
        [1.2722e-09, 1.0000e+00],
        [2.5018e-01, 7.4982e-01],
        [3.3307e-03, 9.9667e-01],
        [1.8485e-07, 1.0000e+00],
        [4.6386e-05, 9.9995e-01],
        [7.1272e-07, 1.0000e+00],
        [9.9987e-01, 1.3191e-04],
        [9.9981e-01, 1.9431e-04],
        [9.2604e-07, 1.0000e+00],
        [3.1165e-06, 1.0000e+00],
        [2.2465e-07, 1.0000e+00],
        [9.9983e-01, 1.6820e-04],
        [5.7359e-07, 1.0000e+00],
        [3.5946e-11, 1.0000e+00],
        [2.7847e-06, 1.0000e+00],
        [9.9988e-01, 1.1653e-04],
        [2.7215e-06, 1.0000e+00],
        [3.8480e-04, 9.9962e-01],
        [4.8579e-04, 9.9951e-01],
        [1.4791e-07, 1.0000e+00],
        [9.4821e-05, 9.9991e-01],
        [7.5444e-09, 1.0000e+00],
        [9.9874e-01, 1.2592e-03],
        [8.4999e-09, 1.0000e+00],
        [6.8963e-07, 1.0000e+00],
        [6.3980e-05, 9.9994e-01],
        [1.7258e-04, 9.9983e-01],
        [1.4654e-07, 1.0000e+00],
        [4.6442e-12, 1.0000e+00],
        [1.2929e-12, 1.0000e+00],
        [9.9977e-01, 2.2560e-04],
        [4.5251e-09, 1.0000e+00],
        [3.6452e-01, 6.3548e-01],
        [9.9993e-01, 7.0640e-05],
        [4.7473e-05, 9.9995e-01],
        [2.8936e-06, 1.0000e+00],
        [8.9845e-06, 9.9999e-01],
        [9.9989e-01, 1.1247e-04],
        [1.1275e-06, 1.0000e+00],
        [9.9864e-01, 1.3573e-03],
        [3.6347e-01, 6.3653e-01],
        [4.7920e-05, 9.9995e-01],
        [9.9993e-01, 7.4574e-05],
        [3.1516e-02, 9.6848e-01],
        [6.8227e-04, 9.9932e-01],
        [3.2550e-09, 1.0000e+00],
        [5.1877e-07, 1.0000e+00],
        [9.9981e-01, 1.8589e-04],
        [2.4043e-06, 1.0000e+00],
        [2.1025e-06, 1.0000e+00],
        [9.9985e-01, 1.5054e-04],
        [1.5990e-08, 1.0000e+00],
        [9.9627e-04, 9.9900e-01],
        [3.0746e-11, 1.0000e+00],
        [5.4641e-04, 9.9945e-01],
        [7.4557e-03, 9.9254e-01],
        [2.1013e-04, 9.9979e-01],
        [1.1357e-01, 8.8643e-01],
        [9.9984e-01, 1.5689e-04],
        [9.9914e-01, 8.5983e-04],
        [1.4511e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.6251e-08, 1.0000e+00],
        [2.8724e-04, 9.9971e-01],
        [2.7925e-11, 1.0000e+00],
        [1.2623e-08, 1.0000e+00],
        [3.1638e-08, 1.0000e+00],
        [9.9053e-01, 9.4655e-03],
        [5.0442e-04, 9.9950e-01],
        [8.5752e-06, 9.9999e-01],
        [1.5462e-09, 1.0000e+00],
        [6.3156e-01, 3.6844e-01],
        [5.3129e-10, 1.0000e+00],
        [2.2272e-09, 1.0000e+00],
        [9.9536e-01, 4.6444e-03],
        [1.3169e-08, 1.0000e+00],
        [1.8559e-06, 1.0000e+00],
        [1.6786e-06, 1.0000e+00],
        [3.8807e-07, 1.0000e+00],
        [1.2135e-11, 1.0000e+00],
        [7.8147e-07, 1.0000e+00],
        [9.9991e-01, 8.7853e-05],
        [9.8723e-01, 1.2772e-02],
        [9.9983e-01, 1.7503e-04],
        [9.9989e-01, 1.0631e-04],
        [1.2402e-05, 9.9999e-01],
        [6.4716e-07, 1.0000e+00],
        [9.9989e-01, 1.1119e-04],
        [1.0216e-11, 1.0000e+00],
        [2.3055e-06, 1.0000e+00],
        [5.5080e-03, 9.9449e-01],
        [9.9874e-01, 1.2642e-03],
        [1.4525e-08, 1.0000e+00],
        [9.9987e-01, 1.2552e-04],
        [1.2652e-09, 1.0000e+00],
        [1.0556e-10, 1.0000e+00],
        [9.9992e-01, 7.9826e-05],
        [2.8855e-09, 1.0000e+00],
        [2.2336e-06, 1.0000e+00],
        [5.9979e-08, 1.0000e+00],
        [9.9987e-01, 1.3445e-04],
        [1.0243e-06, 1.0000e+00],
        [9.9969e-01, 3.0544e-04],
        [3.7345e-08, 1.0000e+00],
        [8.8265e-10, 1.0000e+00],
        [2.8290e-08, 1.0000e+00],
        [9.9994e-01, 5.9940e-05],
        [6.8134e-07, 1.0000e+00],
        [2.3431e-08, 1.0000e+00],
        [8.0812e-10, 1.0000e+00],
        [1.5387e-05, 9.9998e-01],
        [3.4916e-10, 1.0000e+00],
        [2.3151e-05, 9.9998e-01],
        [9.9963e-01, 3.6596e-04],
        [8.7577e-03, 9.9124e-01],
        [1.2057e-03, 9.9879e-01],
        [1.8704e-05, 9.9998e-01],
        [9.9994e-01, 6.4227e-05],
        [1.0500e-07, 1.0000e+00],
        [7.2550e-09, 1.0000e+00],
        [9.9920e-01, 7.9690e-04],
        [9.9967e-01, 3.3407e-04],
        [4.2469e-05, 9.9996e-01],
        [2.8780e-09, 1.0000e+00],
        [8.7677e-12, 1.0000e+00],
        [8.5777e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3812e-06, 1.0000e+00],
        [1.2422e-05, 9.9999e-01],
        [2.0168e-09, 1.0000e+00],
        [2.1801e-03, 9.9782e-01],
        [9.9992e-01, 8.3608e-05],
        [1.0769e-02, 9.8923e-01],
        [8.4746e-01, 1.5254e-01],
        [1.3812e-09, 1.0000e+00],
        [4.0272e-16, 1.0000e+00],
        [3.9302e-07, 1.0000e+00],
        [9.9986e-01, 1.4019e-04],
        [2.6473e-06, 1.0000e+00],
        [9.9579e-01, 4.2132e-03],
        [6.6888e-07, 1.0000e+00],
        [4.6202e-07, 1.0000e+00],
        [1.4891e-04, 9.9985e-01],
        [1.5176e-05, 9.9998e-01],
        [3.3803e-08, 1.0000e+00],
        [1.2999e-07, 1.0000e+00],
        [3.7566e-01, 6.2434e-01],
        [1.0203e-08, 1.0000e+00],
        [2.2723e-04, 9.9977e-01],
        [4.7327e-05, 9.9995e-01],
        [1.5405e-06, 1.0000e+00],
        [1.0952e-06, 1.0000e+00],
        [9.9542e-01, 4.5801e-03],
        [9.7951e-01, 2.0487e-02],
        [3.1530e-11, 1.0000e+00],
        [3.1152e-06, 1.0000e+00],
        [9.7158e-01, 2.8419e-02],
        [1.0949e-10, 1.0000e+00],
        [9.9877e-01, 1.2268e-03],
        [8.2131e-09, 1.0000e+00],
        [9.9988e-01, 1.2256e-04],
        [8.3198e-07, 1.0000e+00],
        [5.8882e-09, 1.0000e+00],
        [1.1522e-08, 1.0000e+00],
        [3.4793e-08, 1.0000e+00],
        [8.9338e-11, 1.0000e+00],
        [9.9067e-01, 9.3318e-03],
        [2.4240e-08, 1.0000e+00],
        [2.2519e-08, 1.0000e+00],
        [9.6614e-09, 1.0000e+00],
        [1.3076e-08, 1.0000e+00],
        [7.7437e-09, 1.0000e+00],
        [3.3410e-05, 9.9997e-01],
        [4.5605e-09, 1.0000e+00],
        [4.7167e-08, 1.0000e+00],
        [2.2753e-08, 1.0000e+00],
        [1.3717e-05, 9.9999e-01],
        [9.9992e-01, 8.2861e-05],
        [1.2104e-10, 1.0000e+00],
        [3.8565e-08, 1.0000e+00],
        [6.2867e-10, 1.0000e+00],
        [3.4425e-04, 9.9966e-01],
        [1.2653e-04, 9.9987e-01],
        [9.3048e-05, 9.9991e-01],
        [3.3491e-02, 9.6651e-01],
        [3.3060e-06, 1.0000e+00],
        [6.7462e-03, 9.9325e-01],
        [9.9974e-01, 2.6476e-04],
        [9.3231e-08, 1.0000e+00],
        [3.3060e-06, 1.0000e+00],
        [5.8272e-01, 4.1728e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7031e-11, 1.0000e+00],
        [1.4805e-06, 1.0000e+00],
        [9.9624e-01, 3.7584e-03],
        [8.2402e-10, 1.0000e+00],
        [9.9840e-01, 1.5980e-03],
        [5.6157e-08, 1.0000e+00],
        [9.3123e-06, 9.9999e-01],
        [7.7336e-10, 1.0000e+00],
        [3.9537e-09, 1.0000e+00],
        [4.5355e-06, 1.0000e+00],
        [5.1783e-06, 9.9999e-01],
        [9.9974e-01, 2.6108e-04],
        [4.0596e-11, 1.0000e+00],
        [2.7814e-10, 1.0000e+00],
        [1.4397e-11, 1.0000e+00],
        [7.9507e-05, 9.9992e-01],
        [6.4480e-08, 1.0000e+00],
        [9.9626e-01, 3.7374e-03],
        [4.2543e-11, 1.0000e+00],
        [9.9873e-01, 1.2723e-03],
        [1.8574e-06, 1.0000e+00],
        [2.3491e-05, 9.9998e-01],
        [9.9401e-09, 1.0000e+00],
        [9.9992e-01, 8.0897e-05],
        [9.9994e-01, 6.4865e-05],
        [6.9217e-09, 1.0000e+00],
        [9.9988e-01, 1.2454e-04],
        [2.0860e-08, 1.0000e+00],
        [9.1408e-01, 8.5920e-02],
        [9.6503e-07, 1.0000e+00],
        [1.0279e-07, 1.0000e+00],
        [4.7998e-06, 1.0000e+00],
        [2.1895e-10, 1.0000e+00],
        [1.3648e-10, 1.0000e+00],
        [4.4370e-10, 1.0000e+00],
        [1.3375e-09, 1.0000e+00],
        [9.9999e-01, 1.4575e-05],
        [5.1939e-09, 1.0000e+00],
        [5.3795e-12, 1.0000e+00],
        [8.3994e-08, 1.0000e+00],
        [4.3478e-13, 1.0000e+00],
        [6.0414e-04, 9.9940e-01],
        [1.0440e-06, 1.0000e+00],
        [3.4996e-09, 1.0000e+00],
        [3.5764e-10, 1.0000e+00],
        [9.9995e-01, 5.3825e-05],
        [2.1035e-05, 9.9998e-01],
        [9.0143e-08, 1.0000e+00],
        [2.4334e-07, 1.0000e+00],
        [5.8216e-03, 9.9418e-01],
        [3.6819e-07, 1.0000e+00],
        [4.4517e-09, 1.0000e+00],
        [5.8645e-04, 9.9941e-01],
        [2.5061e-05, 9.9997e-01],
        [2.6325e-07, 1.0000e+00],
        [6.1074e-08, 1.0000e+00],
        [4.6186e-07, 1.0000e+00],
        [5.5959e-08, 1.0000e+00],
        [3.4967e-09, 1.0000e+00],
        [4.1980e-13, 1.0000e+00],
        [9.2532e-12, 1.0000e+00],
        [7.3909e-03, 9.9261e-01],
        [9.9989e-01, 1.0989e-04],
        [1.7991e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3602e-10, 1.0000e+00],
        [9.7956e-01, 2.0440e-02],
        [6.1099e-01, 3.8901e-01],
        [1.0752e-05, 9.9999e-01],
        [4.9268e-02, 9.5073e-01],
        [1.9477e-09, 1.0000e+00],
        [8.0627e-06, 9.9999e-01],
        [3.1265e-09, 1.0000e+00],
        [5.8690e-06, 9.9999e-01],
        [3.9442e-11, 1.0000e+00],
        [9.6421e-01, 3.5793e-02],
        [1.7630e-04, 9.9982e-01],
        [6.2055e-07, 1.0000e+00],
        [1.4197e-07, 1.0000e+00],
        [9.9989e-01, 1.0991e-04],
        [3.8551e-11, 1.0000e+00],
        [7.2142e-05, 9.9993e-01],
        [1.4019e-09, 1.0000e+00],
        [2.0363e-14, 1.0000e+00],
        [9.9988e-01, 1.1644e-04],
        [9.9967e-01, 3.3359e-04],
        [9.9925e-01, 7.4961e-04],
        [9.9991e-01, 8.8551e-05],
        [2.0772e-04, 9.9979e-01],
        [9.6797e-01, 3.2032e-02],
        [9.9924e-01, 7.5551e-04],
        [6.1257e-09, 1.0000e+00],
        [1.6670e-02, 9.8333e-01],
        [9.8364e-06, 9.9999e-01],
        [1.0052e-11, 1.0000e+00],
        [1.5118e-06, 1.0000e+00],
        [9.9896e-01, 1.0422e-03],
        [8.8868e-10, 1.0000e+00],
        [2.0435e-08, 1.0000e+00],
        [3.2582e-09, 1.0000e+00],
        [4.0441e-08, 1.0000e+00],
        [1.2246e-08, 1.0000e+00],
        [9.9925e-01, 7.5173e-04],
        [9.9974e-01, 2.6489e-04],
        [4.7549e-10, 1.0000e+00],
        [1.2413e-03, 9.9876e-01],
        [9.9976e-01, 2.4133e-04],
        [6.7341e-08, 1.0000e+00],
        [6.0805e-06, 9.9999e-01],
        [3.8172e-07, 1.0000e+00],
        [2.4029e-08, 1.0000e+00],
        [9.9896e-01, 1.0446e-03],
        [9.9375e-01, 6.2486e-03],
        [2.5474e-08, 1.0000e+00],
        [1.0825e-06, 1.0000e+00],
        [9.9205e-01, 7.9531e-03],
        [6.0277e-09, 1.0000e+00],
        [1.1516e-09, 1.0000e+00],
        [2.0199e-07, 1.0000e+00],
        [5.2071e-06, 9.9999e-01],
        [1.4416e-09, 1.0000e+00],
        [1.8571e-10, 1.0000e+00],
        [9.2198e-06, 9.9999e-01],
        [1.4930e-07, 1.0000e+00],
        [6.9997e-01, 3.0003e-01],
        [9.9966e-01, 3.4109e-04],
        [9.9980e-01, 2.0325e-04],
        [5.2480e-05, 9.9995e-01],
        [9.9946e-01, 5.3592e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2838e-07, 1.0000e+00],
        [8.7687e-09, 1.0000e+00],
        [1.1900e-13, 1.0000e+00],
        [1.0572e-01, 8.9428e-01],
        [1.1007e-11, 1.0000e+00],
        [9.9993e-01, 6.6680e-05],
        [4.0809e-08, 1.0000e+00],
        [2.2923e-01, 7.7077e-01],
        [1.4046e-08, 1.0000e+00],
        [9.9967e-01, 3.3034e-04],
        [3.0570e-07, 1.0000e+00],
        [1.6330e-09, 1.0000e+00],
        [3.7608e-07, 1.0000e+00],
        [1.4139e-09, 1.0000e+00],
        [7.2859e-07, 1.0000e+00],
        [7.1251e-05, 9.9993e-01],
        [6.0452e-08, 1.0000e+00],
        [4.0051e-01, 5.9949e-01],
        [5.7504e-08, 1.0000e+00],
        [4.4765e-09, 1.0000e+00],
        [3.9210e-06, 1.0000e+00],
        [2.1106e-08, 1.0000e+00],
        [9.9770e-01, 2.3021e-03],
        [2.5686e-06, 1.0000e+00],
        [5.6206e-10, 1.0000e+00],
        [6.7057e-06, 9.9999e-01],
        [9.9992e-01, 7.9731e-05],
        [3.7332e-06, 1.0000e+00],
        [2.4225e-08, 1.0000e+00],
        [1.5145e-07, 1.0000e+00],
        [7.0882e-01, 2.9118e-01],
        [7.8686e-07, 1.0000e+00],
        [2.3707e-09, 1.0000e+00],
        [2.5143e-06, 1.0000e+00],
        [6.4510e-08, 1.0000e+00],
        [7.0439e-10, 1.0000e+00],
        [4.8623e-06, 1.0000e+00],
        [4.8519e-07, 1.0000e+00],
        [9.9885e-01, 1.1519e-03],
        [4.5211e-07, 1.0000e+00],
        [2.9379e-08, 1.0000e+00],
        [4.1594e-11, 1.0000e+00],
        [9.9627e-01, 3.7345e-03],
        [9.9957e-01, 4.2584e-04],
        [1.1140e-04, 9.9989e-01],
        [9.9922e-01, 7.7733e-04],
        [2.7443e-08, 1.0000e+00],
        [9.9833e-01, 1.6660e-03],
        [2.9803e-07, 1.0000e+00],
        [9.9990e-01, 1.0133e-04],
        [2.0526e-09, 1.0000e+00],
        [1.0284e-08, 1.0000e+00],
        [6.0410e-06, 9.9999e-01],
        [9.9994e-01, 6.3981e-05],
        [9.9981e-01, 1.8712e-04],
        [1.4769e-01, 8.5231e-01],
        [4.2998e-07, 1.0000e+00],
        [2.6293e-09, 1.0000e+00],
        [2.8100e-10, 1.0000e+00],
        [5.2457e-10, 1.0000e+00],
        [8.7472e-02, 9.1253e-01],
        [6.8798e-05, 9.9993e-01],
        [7.7221e-07, 1.0000e+00],
        [9.9964e-01, 3.5868e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3754e-07, 1.0000e+00],
        [9.9677e-01, 3.2317e-03],
        [9.8852e-12, 1.0000e+00],
        [1.0289e-08, 1.0000e+00],
        [4.9912e-05, 9.9995e-01],
        [3.1005e-10, 1.0000e+00],
        [1.5085e-10, 1.0000e+00],
        [6.5105e-05, 9.9993e-01],
        [9.9981e-01, 1.9290e-04],
        [6.2655e-09, 1.0000e+00],
        [4.5961e-07, 1.0000e+00],
        [4.2427e-01, 5.7573e-01],
        [9.9971e-01, 2.8997e-04],
        [4.3454e-07, 1.0000e+00],
        [9.5706e-01, 4.2942e-02],
        [1.7171e-01, 8.2829e-01],
        [5.9915e-05, 9.9994e-01],
        [4.0132e-10, 1.0000e+00],
        [5.7335e-13, 1.0000e+00],
        [1.0456e-06, 1.0000e+00],
        [1.3877e-06, 1.0000e+00],
        [3.9744e-01, 6.0256e-01],
        [8.9345e-09, 1.0000e+00],
        [1.2235e-04, 9.9988e-01],
        [4.3892e-07, 1.0000e+00],
        [6.8634e-12, 1.0000e+00],
        [1.1031e-03, 9.9890e-01],
        [5.4511e-11, 1.0000e+00],
        [3.2486e-04, 9.9968e-01],
        [4.4637e-09, 1.0000e+00],
        [3.5668e-11, 1.0000e+00],
        [3.6904e-06, 1.0000e+00],
        [2.3588e-02, 9.7641e-01],
        [9.9990e-01, 9.5500e-05],
        [9.9987e-01, 1.2501e-04],
        [2.3236e-09, 1.0000e+00],
        [1.3815e-08, 1.0000e+00],
        [9.7872e-01, 2.1278e-02],
        [1.0883e-07, 1.0000e+00],
        [6.4402e-12, 1.0000e+00],
        [5.1879e-12, 1.0000e+00],
        [1.8578e-07, 1.0000e+00],
        [6.6927e-07, 1.0000e+00],
        [3.8851e-05, 9.9996e-01],
        [5.2316e-10, 1.0000e+00],
        [2.3138e-07, 1.0000e+00],
        [9.6291e-09, 1.0000e+00],
        [9.9776e-07, 1.0000e+00],
        [8.6680e-09, 1.0000e+00],
        [9.2405e-01, 7.5946e-02],
        [3.5458e-09, 1.0000e+00],
        [1.5149e-09, 1.0000e+00],
        [6.0909e-10, 1.0000e+00],
        [9.3717e-01, 6.2831e-02],
        [4.3774e-05, 9.9996e-01],
        [9.4059e-01, 5.9413e-02],
        [4.0701e-06, 1.0000e+00],
        [9.9978e-01, 2.1753e-04],
        [1.9787e-10, 1.0000e+00],
        [2.7385e-10, 1.0000e+00],
        [1.3493e-09, 1.0000e+00],
        [5.6983e-10, 1.0000e+00],
        [6.9163e-07, 1.0000e+00],
        [5.2485e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.1980e-09, 1.0000e+00],
        [1.3722e-09, 1.0000e+00],
        [1.0442e-07, 1.0000e+00],
        [1.0587e-10, 1.0000e+00],
        [1.9588e-08, 1.0000e+00],
        [1.4603e-06, 1.0000e+00],
        [2.0168e-10, 1.0000e+00],
        [1.3648e-08, 1.0000e+00],
        [8.5724e-07, 1.0000e+00],
        [3.3259e-09, 1.0000e+00],
        [3.9332e-05, 9.9996e-01],
        [1.1110e-10, 1.0000e+00],
        [9.5773e-06, 9.9999e-01],
        [4.7089e-11, 1.0000e+00],
        [9.9999e-01, 1.1201e-05],
        [1.2740e-07, 1.0000e+00],
        [1.6246e-11, 1.0000e+00],
        [1.0000e+00, 5.8513e-07],
        [2.1125e-07, 1.0000e+00],
        [1.5957e-05, 9.9998e-01],
        [1.7303e-02, 9.8270e-01],
        [9.2804e-05, 9.9991e-01],
        [6.7939e-04, 9.9932e-01],
        [2.4149e-12, 1.0000e+00],
        [1.2299e-08, 1.0000e+00],
        [4.6192e-05, 9.9995e-01],
        [6.5348e-06, 9.9999e-01],
        [4.7089e-11, 1.0000e+00],
        [3.1672e-08, 1.0000e+00],
        [1.0634e-04, 9.9989e-01],
        [2.5463e-09, 1.0000e+00],
        [2.6684e-04, 9.9973e-01],
        [4.0440e-09, 1.0000e+00],
        [5.2096e-09, 1.0000e+00],
        [2.1056e-04, 9.9979e-01],
        [6.5837e-06, 9.9999e-01],
        [9.9990e-01, 1.0174e-04],
        [3.4865e-05, 9.9997e-01],
        [9.1798e-05, 9.9991e-01],
        [1.0019e-08, 1.0000e+00],
        [3.7484e-09, 1.0000e+00],
        [8.3869e-11, 1.0000e+00],
        [1.4537e-06, 1.0000e+00],
        [2.1552e-06, 1.0000e+00],
        [9.9981e-01, 1.9484e-04],
        [9.9952e-01, 4.7723e-04],
        [1.0242e-06, 1.0000e+00],
        [6.9670e-08, 1.0000e+00],
        [3.6300e-12, 1.0000e+00],
        [9.8129e-01, 1.8706e-02],
        [9.9772e-01, 2.2818e-03],
        [4.1664e-09, 1.0000e+00],
        [2.6897e-09, 1.0000e+00],
        [9.8719e-01, 1.2811e-02],
        [2.2520e-08, 1.0000e+00],
        [9.9993e-01, 7.4840e-05],
        [2.5015e-10, 1.0000e+00],
        [9.9402e-01, 5.9825e-03],
        [2.4398e-05, 9.9998e-01],
        [3.8915e-09, 1.0000e+00],
        [3.6992e-06, 1.0000e+00],
        [6.8128e-01, 3.1872e-01],
        [2.4149e-12, 1.0000e+00],
        [9.9991e-01, 9.1588e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8868e-09, 1.0000e+00],
        [8.8859e-01, 1.1141e-01],
        [9.8603e-01, 1.3975e-02],
        [9.9934e-01, 6.6087e-04],
        [9.0420e-08, 1.0000e+00],
        [3.5738e-08, 1.0000e+00],
        [6.7124e-09, 1.0000e+00],
        [1.7611e-07, 1.0000e+00],
        [1.3647e-08, 1.0000e+00],
        [4.1065e-03, 9.9589e-01],
        [9.8753e-01, 1.2471e-02],
        [6.9505e-07, 1.0000e+00],
        [5.3635e-04, 9.9946e-01],
        [6.1397e-01, 3.8603e-01],
        [7.0388e-11, 1.0000e+00],
        [1.2316e-07, 1.0000e+00],
        [3.6065e-04, 9.9964e-01],
        [9.9921e-01, 7.8803e-04],
        [4.7781e-04, 9.9952e-01],
        [4.6333e-09, 1.0000e+00],
        [6.2166e-03, 9.9378e-01],
        [4.9559e-11, 1.0000e+00],
        [2.0310e-03, 9.9797e-01],
        [2.5996e-06, 1.0000e+00],
        [1.7588e-02, 9.8241e-01],
        [1.5937e-07, 1.0000e+00],
        [3.3464e-09, 1.0000e+00],
        [1.5328e-09, 1.0000e+00],
        [1.6607e-09, 1.0000e+00],
        [8.1942e-08, 1.0000e+00],
        [5.1570e-09, 1.0000e+00],
        [8.9661e-01, 1.0339e-01],
        [1.1365e-07, 1.0000e+00],
        [3.7406e-04, 9.9963e-01],
        [3.1145e-04, 9.9969e-01],
        [4.3061e-08, 1.0000e+00],
        [1.0196e-09, 1.0000e+00],
        [1.8941e-03, 9.9811e-01],
        [9.7926e-11, 1.0000e+00],
        [9.9989e-01, 1.0813e-04],
        [9.9985e-01, 1.5032e-04],
        [6.9058e-02, 9.3094e-01],
        [3.3446e-07, 1.0000e+00],
        [2.7302e-09, 1.0000e+00],
        [6.0048e-05, 9.9994e-01],
        [6.3481e-08, 1.0000e+00],
        [3.7526e-07, 1.0000e+00],
        [9.9979e-01, 2.1058e-04],
        [1.0000e+00, 5.6383e-07],
        [4.1629e-09, 1.0000e+00],
        [3.7117e-07, 1.0000e+00],
        [1.2302e-08, 1.0000e+00],
        [2.8168e-06, 1.0000e+00],
        [9.9979e-01, 2.0726e-04],
        [8.9211e-01, 1.0789e-01],
        [8.8222e-07, 1.0000e+00],
        [3.4616e-06, 1.0000e+00],
        [4.3613e-05, 9.9996e-01],
        [4.9718e-07, 1.0000e+00],
        [9.9998e-01, 1.8377e-05],
        [2.3083e-06, 1.0000e+00],
        [5.5560e-05, 9.9994e-01],
        [9.5396e-07, 1.0000e+00],
        [9.9992e-01, 7.6383e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9991e-01, 9.0205e-05],
        [1.4476e-07, 1.0000e+00],
        [3.2733e-04, 9.9967e-01],
        [7.8815e-05, 9.9992e-01],
        [2.8096e-02, 9.7190e-01],
        [1.1851e-09, 1.0000e+00],
        [5.7889e-07, 1.0000e+00],
        [2.6149e-12, 1.0000e+00],
        [9.9932e-01, 6.7563e-04],
        [5.4060e-08, 1.0000e+00],
        [2.4591e-05, 9.9998e-01],
        [3.5916e-06, 1.0000e+00],
        [4.0670e-08, 1.0000e+00],
        [9.9996e-01, 3.8671e-05],
        [8.6418e-01, 1.3582e-01],
        [6.2812e-06, 9.9999e-01],
        [9.9977e-01, 2.2789e-04],
        [2.3242e-07, 1.0000e+00],
        [2.3698e-08, 1.0000e+00],
        [6.1247e-06, 9.9999e-01],
        [6.9513e-01, 3.0487e-01],
        [2.9817e-02, 9.7018e-01],
        [4.1516e-05, 9.9996e-01],
        [3.9743e-08, 1.0000e+00],
        [6.9083e-06, 9.9999e-01],
        [5.7838e-06, 9.9999e-01],
        [6.1549e-09, 1.0000e+00],
        [5.9164e-01, 4.0836e-01],
        [1.8165e-04, 9.9982e-01],
        [2.8713e-09, 1.0000e+00],
        [9.7868e-01, 2.1324e-02],
        [9.9973e-01, 2.6717e-04],
        [4.0931e-08, 1.0000e+00],
        [1.0670e-04, 9.9989e-01],
        [7.9856e-04, 9.9920e-01],
        [6.3770e-04, 9.9936e-01],
        [9.1225e-03, 9.9088e-01],
        [3.6666e-08, 1.0000e+00],
        [9.9980e-01, 1.9528e-04],
        [9.9913e-01, 8.7379e-04],
        [4.1587e-09, 1.0000e+00],
        [9.9999e-01, 8.6118e-06],
        [1.0072e-06, 1.0000e+00],
        [9.3622e-05, 9.9991e-01],
        [5.4514e-09, 1.0000e+00],
        [4.5499e-05, 9.9995e-01],
        [2.2993e-07, 1.0000e+00],
        [1.3043e-07, 1.0000e+00],
        [5.5994e-12, 1.0000e+00],
        [9.2474e-10, 1.0000e+00],
        [2.2927e-03, 9.9771e-01],
        [1.4524e-07, 1.0000e+00],
        [2.8128e-10, 1.0000e+00],
        [7.1590e-06, 9.9999e-01],
        [3.5491e-07, 1.0000e+00],
        [9.9917e-01, 8.3425e-04],
        [1.0900e-03, 9.9891e-01],
        [9.9995e-01, 5.2291e-05],
        [1.3463e-05, 9.9999e-01],
        [9.9038e-01, 9.6247e-03],
        [1.6409e-07, 1.0000e+00],
        [1.2331e-11, 1.0000e+00],
        [9.9983e-01, 1.6548e-04],
        [9.0023e-05, 9.9991e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9988e-01, 1.2459e-04],
        [6.6623e-10, 1.0000e+00],
        [4.1723e-07, 1.0000e+00],
        [2.2703e-01, 7.7297e-01],
        [1.1253e-06, 1.0000e+00],
        [1.0862e-08, 1.0000e+00],
        [5.4814e-05, 9.9995e-01],
        [1.5731e-09, 1.0000e+00],
        [1.3796e-02, 9.8620e-01],
        [2.7386e-01, 7.2614e-01],
        [1.3784e-06, 1.0000e+00],
        [1.6010e-10, 1.0000e+00],
        [1.7310e-05, 9.9998e-01],
        [9.9972e-01, 2.8377e-04],
        [1.0264e-04, 9.9990e-01],
        [1.7349e-04, 9.9983e-01],
        [7.8797e-01, 2.1203e-01],
        [1.1707e-06, 1.0000e+00],
        [7.7844e-10, 1.0000e+00],
        [5.4956e-03, 9.9450e-01],
        [6.0855e-09, 1.0000e+00],
        [1.6709e-04, 9.9983e-01],
        [6.1332e-07, 1.0000e+00],
        [9.9993e-01, 7.3462e-05],
        [1.4973e-08, 1.0000e+00],
        [6.7253e-07, 1.0000e+00],
        [9.5121e-01, 4.8786e-02],
        [3.7209e-07, 1.0000e+00],
        [5.1869e-08, 1.0000e+00],
        [9.9990e-01, 1.0167e-04],
        [3.2860e-12, 1.0000e+00],
        [4.7937e-06, 1.0000e+00],
        [6.0067e-07, 1.0000e+00],
        [2.3243e-05, 9.9998e-01],
        [8.5438e-05, 9.9991e-01],
        [1.8342e-03, 9.9817e-01],
        [1.1445e-03, 9.9886e-01],
        [4.6880e-09, 1.0000e+00],
        [4.6069e-06, 1.0000e+00],
        [1.7249e-06, 1.0000e+00],
        [9.9990e-01, 1.0323e-04],
        [2.4580e-07, 1.0000e+00],
        [1.2041e-08, 1.0000e+00],
        [3.6550e-10, 1.0000e+00],
        [2.6453e-08, 1.0000e+00],
        [9.9995e-01, 4.6891e-05],
        [9.1753e-07, 1.0000e+00],
        [5.2507e-05, 9.9995e-01],
        [1.3250e-06, 1.0000e+00],
        [9.6836e-01, 3.1644e-02],
        [9.9989e-01, 1.1050e-04],
        [9.9990e-01, 9.5212e-05],
        [3.8638e-06, 1.0000e+00],
        [1.3808e-04, 9.9986e-01],
        [5.2987e-08, 1.0000e+00],
        [4.8703e-06, 1.0000e+00],
        [9.8504e-10, 1.0000e+00],
        [5.4928e-08, 1.0000e+00],
        [2.9412e-06, 1.0000e+00],
        [9.9324e-01, 6.7632e-03],
        [9.9980e-01, 1.9608e-04],
        [1.7689e-07, 1.0000e+00],
        [2.6806e-05, 9.9997e-01],
        [1.6000e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.4086e-06, 9.9999e-01],
        [3.2718e-04, 9.9967e-01],
        [9.9119e-01, 8.8084e-03],
        [3.6789e-07, 1.0000e+00],
        [9.9882e-01, 1.1848e-03],
        [9.9990e-01, 9.6149e-05],
        [1.4484e-07, 1.0000e+00],
        [6.7552e-06, 9.9999e-01],
        [7.9700e-05, 9.9992e-01],
        [1.4292e-03, 9.9857e-01],
        [9.9994e-01, 6.2219e-05],
        [4.7699e-04, 9.9952e-01],
        [9.9970e-01, 3.0282e-04],
        [3.2259e-04, 9.9968e-01],
        [4.4922e-01, 5.5078e-01],
        [1.0040e-05, 9.9999e-01],
        [4.0682e-10, 1.0000e+00],
        [1.6620e-02, 9.8338e-01],
        [2.4152e-08, 1.0000e+00],
        [2.1257e-08, 1.0000e+00],
        [4.0278e-03, 9.9597e-01],
        [1.5564e-05, 9.9998e-01],
        [9.6364e-01, 3.6358e-02],
        [5.1746e-01, 4.8254e-01],
        [3.8413e-04, 9.9962e-01],
        [9.9996e-01, 4.1535e-05],
        [1.6647e-05, 9.9998e-01],
        [2.6836e-06, 1.0000e+00],
        [9.9911e-01, 8.9192e-04],
        [4.4524e-04, 9.9955e-01],
        [1.5453e-07, 1.0000e+00],
        [2.2302e-04, 9.9978e-01],
        [7.1838e-10, 1.0000e+00],
        [1.9112e-05, 9.9998e-01],
        [9.9993e-01, 7.1822e-05],
        [1.1772e-06, 1.0000e+00],
        [1.5656e-04, 9.9984e-01],
        [8.7219e-09, 1.0000e+00],
        [2.1521e-08, 1.0000e+00],
        [1.2316e-07, 1.0000e+00],
        [2.4958e-11, 1.0000e+00],
        [9.9969e-01, 3.1099e-04],
        [6.2917e-10, 1.0000e+00],
        [9.9996e-01, 4.2039e-05],
        [9.9996e-01, 3.6255e-05],
        [8.7709e-01, 1.2291e-01],
        [9.3609e-09, 1.0000e+00],
        [7.1056e-11, 1.0000e+00],
        [9.9506e-01, 4.9419e-03],
        [7.4421e-01, 2.5579e-01],
        [9.9990e-01, 1.0165e-04],
        [1.5861e-07, 1.0000e+00],
        [5.1167e-06, 9.9999e-01],
        [1.9108e-07, 1.0000e+00],
        [8.1867e-08, 1.0000e+00],
        [2.2371e-08, 1.0000e+00],
        [1.5520e-06, 1.0000e+00],
        [4.5249e-07, 1.0000e+00],
        [1.2277e-09, 1.0000e+00],
        [9.9996e-01, 3.9315e-05],
        [1.5041e-06, 1.0000e+00],
        [8.5884e-05, 9.9991e-01],
        [1.3920e-01, 8.6080e-01],
        [8.3163e-01, 1.6837e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9994e-01, 6.2709e-05],
        [9.9955e-01, 4.5090e-04],
        [1.7492e-07, 1.0000e+00],
        [9.9036e-08, 1.0000e+00],
        [1.3248e-07, 1.0000e+00],
        [1.5227e-07, 1.0000e+00],
        [1.0537e-08, 1.0000e+00],
        [6.1718e-03, 9.9383e-01],
        [6.2524e-07, 1.0000e+00],
        [5.1449e-04, 9.9949e-01],
        [1.0313e-07, 1.0000e+00],
        [7.4114e-08, 1.0000e+00],
        [5.6150e-03, 9.9439e-01],
        [2.8673e-07, 1.0000e+00],
        [3.9594e-08, 1.0000e+00],
        [6.5263e-05, 9.9993e-01],
        [9.9997e-01, 2.5758e-05],
        [2.7727e-05, 9.9997e-01],
        [1.1597e-09, 1.0000e+00],
        [3.0596e-08, 1.0000e+00],
        [9.9996e-01, 4.1714e-05],
        [9.9993e-01, 7.2588e-05],
        [1.3212e-05, 9.9999e-01],
        [1.4798e-05, 9.9999e-01],
        [1.1432e-09, 1.0000e+00],
        [1.1059e-03, 9.9889e-01],
        [9.9977e-01, 2.2517e-04],
        [6.4106e-10, 1.0000e+00],
        [9.9978e-01, 2.2189e-04],
        [9.7945e-01, 2.0548e-02],
        [8.2684e-05, 9.9992e-01],
        [9.9989e-01, 1.1345e-04],
        [1.8973e-02, 9.8103e-01],
        [9.6301e-01, 3.6992e-02],
        [5.0993e-06, 9.9999e-01],
        [4.4169e-06, 1.0000e+00],
        [1.0264e-03, 9.9897e-01],
        [9.9984e-01, 1.5631e-04],
        [1.3211e-09, 1.0000e+00],
        [1.1210e-04, 9.9989e-01],
        [9.9978e-01, 2.2424e-04],
        [9.1235e-11, 1.0000e+00],
        [5.7178e-04, 9.9943e-01],
        [4.9115e-04, 9.9951e-01],
        [6.3188e-07, 1.0000e+00],
        [1.4942e-04, 9.9985e-01],
        [9.9201e-09, 1.0000e+00],
        [9.9995e-01, 5.0474e-05],
        [9.9999e-01, 1.0737e-05],
        [9.9993e-01, 7.0239e-05],
        [2.1987e-10, 1.0000e+00],
        [9.9993e-01, 6.9127e-05],
        [9.9867e-01, 1.3315e-03],
        [7.1909e-06, 9.9999e-01],
        [2.0414e-07, 1.0000e+00],
        [9.9877e-01, 1.2275e-03],
        [9.9999e-01, 1.1293e-05],
        [9.9991e-01, 9.2667e-05],
        [1.7244e-07, 1.0000e+00],
        [7.3926e-03, 9.9261e-01],
        [9.9997e-01, 2.9745e-05],
        [4.4179e-06, 1.0000e+00],
        [1.7907e-05, 9.9998e-01],
        [4.6569e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.7612e-09, 1.0000e+00],
        [7.2737e-07, 1.0000e+00],
        [1.2896e-11, 1.0000e+00],
        [9.9995e-01, 4.6612e-05],
        [3.2723e-04, 9.9967e-01],
        [9.9975e-01, 2.5079e-04],
        [8.3596e-07, 1.0000e+00],
        [9.9989e-01, 1.0539e-04],
        [1.6900e-08, 1.0000e+00],
        [6.9145e-06, 9.9999e-01],
        [1.8440e-03, 9.9816e-01],
        [7.7316e-05, 9.9992e-01],
        [4.4519e-09, 1.0000e+00],
        [5.5008e-08, 1.0000e+00],
        [3.1925e-09, 1.0000e+00],
        [5.4576e-03, 9.9454e-01],
        [9.9996e-01, 4.2378e-05],
        [6.6728e-06, 9.9999e-01],
        [9.9994e-01, 5.9525e-05],
        [3.7428e-11, 1.0000e+00],
        [3.7526e-07, 1.0000e+00],
        [9.8278e-01, 1.7225e-02],
        [8.7860e-03, 9.9121e-01],
        [2.7657e-01, 7.2343e-01],
        [5.7594e-08, 1.0000e+00],
        [2.2672e-04, 9.9977e-01],
        [9.9996e-01, 4.2391e-05],
        [9.9140e-01, 8.5980e-03],
        [1.6153e-04, 9.9984e-01],
        [5.8130e-03, 9.9419e-01],
        [1.2343e-08, 1.0000e+00],
        [9.9996e-01, 4.3457e-05],
        [3.7428e-10, 1.0000e+00],
        [1.6070e-05, 9.9998e-01],
        [9.7719e-05, 9.9990e-01],
        [1.0913e-04, 9.9989e-01],
        [2.2374e-05, 9.9998e-01],
        [6.0532e-10, 1.0000e+00],
        [1.1978e-05, 9.9999e-01],
        [7.5236e-07, 1.0000e+00],
        [5.0413e-04, 9.9950e-01],
        [6.0983e-03, 9.9390e-01],
        [5.0700e-08, 1.0000e+00],
        [1.4784e-04, 9.9985e-01],
        [7.2081e-01, 2.7919e-01],
        [1.6317e-08, 1.0000e+00],
        [3.8028e-01, 6.1972e-01],
        [3.3352e-04, 9.9967e-01],
        [8.1539e-05, 9.9992e-01],
        [8.1317e-02, 9.1868e-01],
        [2.6152e-04, 9.9974e-01],
        [6.1667e-05, 9.9994e-01],
        [9.6269e-01, 3.7313e-02],
        [1.3163e-06, 1.0000e+00],
        [4.7073e-04, 9.9953e-01],
        [6.5084e-04, 9.9935e-01],
        [6.8342e-04, 9.9932e-01],
        [8.0397e-07, 1.0000e+00],
        [9.9999e-01, 8.5397e-06],
        [5.6503e-04, 9.9944e-01],
        [3.1806e-10, 1.0000e+00],
        [1.2712e-04, 9.9987e-01],
        [1.6649e-06, 1.0000e+00],
        [1.2958e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.5411e-08, 1.0000e+00],
        [9.9999e-01, 6.8291e-06],
        [1.3466e-03, 9.9865e-01],
        [2.1389e-08, 1.0000e+00],
        [9.9998e-01, 1.7829e-05],
        [9.9986e-01, 1.4449e-04],
        [1.2708e-08, 1.0000e+00],
        [1.4759e-07, 1.0000e+00],
        [7.3097e-07, 1.0000e+00],
        [2.2427e-06, 1.0000e+00],
        [4.2834e-07, 1.0000e+00],
        [9.9981e-01, 1.9008e-04],
        [2.5247e-03, 9.9748e-01],
        [4.2570e-07, 1.0000e+00],
        [9.9990e-01, 9.5827e-05],
        [1.0341e-06, 1.0000e+00],
        [9.9891e-01, 1.0852e-03],
        [1.4460e-06, 1.0000e+00],
        [1.5898e-04, 9.9984e-01],
        [5.5790e-02, 9.4421e-01],
        [4.6845e-05, 9.9995e-01],
        [1.2283e-05, 9.9999e-01],
        [3.6865e-03, 9.9631e-01],
        [8.2176e-05, 9.9992e-01],
        [5.1840e-05, 9.9995e-01],
        [1.7417e-05, 9.9998e-01],
        [9.0122e-01, 9.8785e-02],
        [9.9986e-01, 1.4224e-04],
        [2.5020e-04, 9.9975e-01],
        [1.5052e-04, 9.9985e-01],
        [9.9992e-01, 8.3002e-05],
        [2.1018e-06, 1.0000e+00],
        [9.9978e-01, 2.2303e-04],
        [1.6242e-08, 1.0000e+00],
        [1.1722e-07, 1.0000e+00],
        [2.1442e-07, 1.0000e+00],
        [2.2147e-03, 9.9779e-01],
        [5.5940e-05, 9.9994e-01],
        [4.4892e-08, 1.0000e+00],
        [7.4214e-07, 1.0000e+00],
        [3.2085e-07, 1.0000e+00],
        [8.6152e-04, 9.9914e-01],
        [7.3214e-10, 1.0000e+00],
        [3.4458e-04, 9.9966e-01],
        [2.9335e-06, 1.0000e+00],
        [1.1557e-02, 9.8844e-01],
        [2.4609e-05, 9.9998e-01],
        [2.6388e-03, 9.9736e-01],
        [1.1866e-06, 1.0000e+00],
        [9.9989e-01, 1.1405e-04],
        [1.8488e-07, 1.0000e+00],
        [9.9994e-01, 6.1566e-05],
        [5.6298e-01, 4.3702e-01],
        [6.1166e-01, 3.8834e-01],
        [4.8058e-07, 1.0000e+00],
        [9.8256e-05, 9.9990e-01],
        [1.9544e-05, 9.9998e-01],
        [1.6421e-08, 1.0000e+00],
        [2.0091e-04, 9.9980e-01],
        [9.9996e-01, 4.3858e-05],
        [9.2265e-05, 9.9991e-01],
        [1.2212e-07, 1.0000e+00],
        [5.8881e-02, 9.4112e-01],
        [9.9940e-01, 5.9621e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[2.5008e-05, 9.9997e-01],
        [6.8852e-10, 1.0000e+00],
        [9.9996e-01, 4.2554e-05],
        [2.4312e-10, 1.0000e+00],
        [8.4553e-03, 9.9154e-01],
        [1.7916e-05, 9.9998e-01],
        [7.5515e-04, 9.9924e-01],
        [5.4596e-09, 1.0000e+00],
        [4.0754e-05, 9.9996e-01],
        [9.9987e-01, 1.3332e-04],
        [4.5523e-10, 1.0000e+00],
        [6.6554e-04, 9.9933e-01],
        [6.7875e-10, 1.0000e+00],
        [1.7822e-07, 1.0000e+00],
        [8.0474e-06, 9.9999e-01],
        [1.7028e-03, 9.9830e-01],
        [2.1465e-07, 1.0000e+00],
        [9.9989e-01, 1.0633e-04],
        [2.0999e-07, 1.0000e+00],
        [1.4624e-08, 1.0000e+00],
        [9.9979e-01, 2.0899e-04],
        [3.2431e-01, 6.7569e-01],
        [1.1261e-07, 1.0000e+00],
        [3.0859e-07, 1.0000e+00],
        [9.9995e-01, 4.6350e-05],
        [4.0263e-05, 9.9996e-01],
        [8.9623e-05, 9.9991e-01],
        [8.9740e-01, 1.0260e-01],
        [9.9739e-01, 2.6113e-03],
        [4.5949e-05, 9.9995e-01],
        [9.7354e-03, 9.9026e-01],
        [9.9873e-01, 1.2700e-03],
        [7.0194e-03, 9.9298e-01],
        [4.3429e-03, 9.9566e-01],
        [2.6792e-04, 9.9973e-01],
        [9.9997e-01, 3.3600e-05],
        [4.4964e-04, 9.9955e-01],
        [1.2278e-02, 9.8772e-01],
        [1.6216e-05, 9.9998e-01],
        [2.0706e-04, 9.9979e-01],
        [5.0223e-04, 9.9950e-01],
        [4.4082e-10, 1.0000e+00],
        [3.1589e-10, 1.0000e+00],
        [5.2460e-09, 1.0000e+00],
        [1.0592e-08, 1.0000e+00],
        [8.6881e-06, 9.9999e-01],
        [1.2084e-05, 9.9999e-01],
        [9.9999e-01, 6.9173e-06],
        [2.2488e-02, 9.7751e-01],
        [1.2086e-04, 9.9988e-01],
        [4.9061e-03, 9.9509e-01],
        [1.3063e-05, 9.9999e-01],
        [9.9700e-01, 3.0036e-03],
        [9.9998e-01, 1.9138e-05],
        [6.0959e-05, 9.9994e-01],
        [4.9441e-06, 1.0000e+00],
        [9.9990e-01, 9.6490e-05],
        [1.3424e-02, 9.8658e-01],
        [2.1885e-06, 1.0000e+00],
        [9.6308e-02, 9.0369e-01],
        [4.0680e-07, 1.0000e+00],
        [2.3380e-08, 1.0000e+00],
        [9.9988e-01, 1.1797e-04],
        [1.1869e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8407e-05, 9.9998e-01],
        [9.9991e-01, 8.5290e-05],
        [9.9930e-01, 7.0233e-04],
        [1.0749e-01, 8.9251e-01],
        [9.9457e-01, 5.4317e-03],
        [3.3969e-08, 1.0000e+00],
        [5.6128e-06, 9.9999e-01],
        [1.7588e-03, 9.9824e-01],
        [1.4893e-06, 1.0000e+00],
        [2.7050e-06, 1.0000e+00],
        [8.7188e-04, 9.9913e-01],
        [2.1783e-08, 1.0000e+00],
        [8.3002e-03, 9.9170e-01],
        [1.1337e-03, 9.9887e-01],
        [3.0880e-06, 1.0000e+00],
        [1.3949e-10, 1.0000e+00],
        [1.4874e-07, 1.0000e+00],
        [5.9222e-10, 1.0000e+00],
        [2.2063e-04, 9.9978e-01],
        [1.1003e-05, 9.9999e-01],
        [2.3378e-02, 9.7662e-01],
        [1.4504e-09, 1.0000e+00],
        [2.0955e-07, 1.0000e+00],
        [4.2272e-10, 1.0000e+00],
        [9.9996e-01, 4.1837e-05],
        [9.9009e-01, 9.9080e-03],
        [2.7651e-08, 1.0000e+00],
        [1.9308e-05, 9.9998e-01],
        [5.8075e-07, 1.0000e+00],
        [5.3033e-06, 9.9999e-01],
        [5.0814e-04, 9.9949e-01],
        [1.7035e-08, 1.0000e+00],
        [4.0392e-08, 1.0000e+00],
        [1.0102e-04, 9.9990e-01],
        [3.7729e-10, 1.0000e+00],
        [3.1486e-01, 6.8514e-01],
        [9.9990e-01, 9.9301e-05],
        [2.3150e-06, 1.0000e+00],
        [2.4690e-04, 9.9975e-01],
        [6.8844e-04, 9.9931e-01],
        [6.2205e-08, 1.0000e+00],
        [1.5329e-07, 1.0000e+00],
        [3.1207e-02, 9.6879e-01],
        [9.9991e-01, 9.2144e-05],
        [2.0430e-05, 9.9998e-01],
        [6.7423e-04, 9.9933e-01],
        [6.7043e-06, 9.9999e-01],
        [1.1102e-05, 9.9999e-01],
        [1.3564e-06, 1.0000e+00],
        [9.9993e-01, 6.6051e-05],
        [1.1502e-07, 1.0000e+00],
        [9.9978e-01, 2.1676e-04],
        [9.9979e-01, 2.0746e-04],
        [1.3564e-06, 1.0000e+00],
        [4.5253e-09, 1.0000e+00],
        [1.9782e-06, 1.0000e+00],
        [4.1692e-04, 9.9958e-01],
        [1.3451e-06, 1.0000e+00],
        [5.0668e-04, 9.9949e-01],
        [3.0269e-10, 1.0000e+00],
        [1.2327e-08, 1.0000e+00],
        [2.3840e-09, 1.0000e+00],
        [4.8822e-08, 1.0000e+00],
        [1.0854e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5955e-11, 1.0000e+00],
        [1.0342e-07, 1.0000e+00],
        [3.3018e-08, 1.0000e+00],
        [2.1668e-04, 9.9978e-01],
        [2.1416e-05, 9.9998e-01],
        [9.9990e-01, 9.6708e-05],
        [1.0006e-02, 9.8999e-01],
        [1.1295e-07, 1.0000e+00],
        [1.0974e-09, 1.0000e+00],
        [1.0342e-07, 1.0000e+00],
        [7.3027e-05, 9.9993e-01],
        [2.1043e-07, 1.0000e+00],
        [1.2950e-06, 1.0000e+00],
        [4.2633e-03, 9.9574e-01],
        [2.4454e-07, 1.0000e+00],
        [2.2310e-10, 1.0000e+00],
        [6.7242e-10, 1.0000e+00],
        [1.6758e-06, 1.0000e+00],
        [1.4347e-03, 9.9857e-01],
        [4.9384e-09, 1.0000e+00],
        [9.9773e-01, 2.2714e-03],
        [8.9009e-01, 1.0991e-01],
        [8.8242e-06, 9.9999e-01],
        [4.1244e-06, 1.0000e+00],
        [3.9428e-01, 6.0572e-01],
        [6.1574e-11, 1.0000e+00],
        [3.2898e-09, 1.0000e+00],
        [8.5456e-01, 1.4544e-01],
        [7.7378e-07, 1.0000e+00],
        [1.2021e-01, 8.7979e-01],
        [5.6378e-05, 9.9994e-01],
        [6.8335e-08, 1.0000e+00],
        [9.9988e-01, 1.2264e-04],
        [9.9999e-01, 6.2757e-06],
        [1.1567e-07, 1.0000e+00],
        [3.2880e-04, 9.9967e-01],
        [9.9946e-01, 5.4055e-04],
        [9.9997e-01, 2.8982e-05],
        [1.8804e-07, 1.0000e+00],
        [1.4956e-06, 1.0000e+00],
        [9.9996e-01, 3.5905e-05],
        [1.4545e-08, 1.0000e+00],
        [7.5726e-06, 9.9999e-01],
        [1.3556e-10, 1.0000e+00],
        [3.0923e-01, 6.9077e-01],
        [8.8121e-09, 1.0000e+00],
        [9.9991e-01, 8.7689e-05],
        [6.1211e-06, 9.9999e-01],
        [9.9991e-01, 9.3797e-05],
        [9.9995e-01, 4.6427e-05],
        [1.7730e-04, 9.9982e-01],
        [9.9997e-01, 3.4719e-05],
        [5.4204e-08, 1.0000e+00],
        [2.4737e-03, 9.9753e-01],
        [1.1595e-03, 9.9884e-01],
        [1.2775e-07, 1.0000e+00],
        [9.9999e-01, 8.2924e-06],
        [2.9523e-07, 1.0000e+00],
        [7.0232e-07, 1.0000e+00],
        [3.1998e-03, 9.9680e-01],
        [1.3082e-03, 9.9869e-01],
        [2.5171e-06, 1.0000e+00],
        [8.3420e-01, 1.6580e-01],
        [1.5831e-03, 9.9842e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.3638e-05, 9.9995e-01],
        [1.1345e-05, 9.9999e-01],
        [2.5979e-11, 1.0000e+00],
        [1.8730e-09, 1.0000e+00],
        [9.9997e-01, 3.3568e-05],
        [8.6592e-01, 1.3408e-01],
        [9.9988e-01, 1.1715e-04],
        [3.9704e-03, 9.9603e-01],
        [2.3682e-10, 1.0000e+00],
        [3.7389e-04, 9.9963e-01],
        [8.5565e-07, 1.0000e+00],
        [9.8474e-01, 1.5255e-02],
        [2.5335e-07, 1.0000e+00],
        [2.5175e-05, 9.9997e-01],
        [1.3739e-05, 9.9999e-01],
        [9.9997e-01, 3.2487e-05],
        [9.9952e-01, 4.7858e-04],
        [1.0829e-04, 9.9989e-01],
        [9.9567e-01, 4.3275e-03],
        [9.9987e-01, 1.2824e-04],
        [1.0507e-08, 1.0000e+00],
        [6.6824e-10, 1.0000e+00],
        [4.5377e-12, 1.0000e+00],
        [4.4783e-06, 1.0000e+00],
        [1.2392e-08, 1.0000e+00],
        [9.9996e-01, 4.1286e-05],
        [1.7490e-05, 9.9998e-01],
        [9.1102e-05, 9.9991e-01],
        [5.4320e-08, 1.0000e+00],
        [9.9981e-01, 1.9430e-04],
        [4.0694e-08, 1.0000e+00],
        [1.9212e-08, 1.0000e+00],
        [6.5330e-11, 1.0000e+00],
        [9.9983e-01, 1.6638e-04],
        [1.9573e-10, 1.0000e+00],
        [9.9991e-01, 8.9844e-05],
        [1.2434e-05, 9.9999e-01],
        [1.6540e-05, 9.9998e-01],
        [2.1180e-01, 7.8820e-01],
        [7.8117e-01, 2.1883e-01],
        [7.9957e-01, 2.0043e-01],
        [9.9739e-01, 2.6111e-03],
        [9.6452e-01, 3.5479e-02],
        [9.5971e-11, 1.0000e+00],
        [2.9396e-08, 1.0000e+00],
        [3.9104e-05, 9.9996e-01],
        [1.2224e-05, 9.9999e-01],
        [2.9837e-06, 1.0000e+00],
        [8.2126e-05, 9.9992e-01],
        [9.7002e-01, 2.9977e-02],
        [1.7614e-08, 1.0000e+00],
        [9.9974e-01, 2.6209e-04],
        [9.9988e-01, 1.2119e-04],
        [2.6876e-14, 1.0000e+00],
        [9.9768e-09, 1.0000e+00],
        [1.1138e-04, 9.9989e-01],
        [3.9191e-04, 9.9961e-01],
        [3.5861e-08, 1.0000e+00],
        [4.5316e-04, 9.9955e-01],
        [7.3795e-08, 1.0000e+00],
        [5.1618e-07, 1.0000e+00],
        [9.9965e-01, 3.5049e-04],
        [1.4377e-13, 1.0000e+00],
        [5.4949e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.7176e-11, 1.0000e+00],
        [5.9610e-08, 1.0000e+00],
        [3.8018e-09, 1.0000e+00],
        [5.0948e-07, 1.0000e+00],
        [9.9997e-01, 2.8622e-05],
        [4.6105e-13, 1.0000e+00],
        [2.7361e-04, 9.9973e-01],
        [9.9996e-01, 3.7277e-05],
        [3.5864e-10, 1.0000e+00],
        [3.3374e-03, 9.9666e-01],
        [2.3665e-02, 9.7634e-01],
        [6.6207e-10, 1.0000e+00],
        [4.8051e-07, 1.0000e+00],
        [9.9994e-01, 5.5214e-05],
        [9.9992e-01, 8.1254e-05],
        [7.2876e-06, 9.9999e-01],
        [7.1334e-08, 1.0000e+00],
        [4.4794e-08, 1.0000e+00],
        [3.7819e-07, 1.0000e+00],
        [9.9991e-01, 8.8365e-05],
        [3.9727e-02, 9.6027e-01],
        [2.1974e-03, 9.9780e-01],
        [4.4642e-11, 1.0000e+00],
        [2.2098e-08, 1.0000e+00],
        [6.6758e-05, 9.9993e-01],
        [9.9993e-01, 6.9818e-05],
        [9.9994e-01, 5.5632e-05],
        [9.9878e-01, 1.2235e-03],
        [9.9984e-01, 1.6010e-04],
        [2.6846e-07, 1.0000e+00],
        [1.0343e-05, 9.9999e-01],
        [2.8868e-07, 1.0000e+00],
        [5.6395e-10, 1.0000e+00],
        [6.0434e-09, 1.0000e+00],
        [1.4433e-05, 9.9999e-01],
        [8.1167e-01, 1.8833e-01],
        [5.8867e-08, 1.0000e+00],
        [2.0229e-04, 9.9980e-01],
        [3.7028e-07, 1.0000e+00],
        [4.5734e-06, 1.0000e+00],
        [9.9988e-01, 1.2416e-04],
        [3.4796e-08, 1.0000e+00],
        [9.9979e-01, 2.0735e-04],
        [2.3708e-08, 1.0000e+00],
        [9.4776e-09, 1.0000e+00],
        [9.9797e-01, 2.0318e-03],
        [1.5961e-07, 1.0000e+00],
        [9.9131e-01, 8.6908e-03],
        [6.8559e-18, 1.0000e+00],
        [4.3311e-08, 1.0000e+00],
        [1.3152e-06, 1.0000e+00],
        [3.2429e-07, 1.0000e+00],
        [7.9890e-01, 2.0110e-01],
        [7.4383e-03, 9.9256e-01],
        [4.1807e-09, 1.0000e+00],
        [2.5230e-06, 1.0000e+00],
        [3.8195e-10, 1.0000e+00],
        [7.9130e-06, 9.9999e-01],
        [5.8078e-08, 1.0000e+00],
        [2.1154e-10, 1.0000e+00],
        [4.6739e-10, 1.0000e+00],
        [6.0563e-07, 1.0000e+00],
        [6.8380e-08, 1.0000e+00],
        [9.9998e-01, 2.1671e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[6.2916e-07, 1.0000e+00],
        [1.8422e-05, 9.9998e-01],
        [9.7403e-01, 2.5968e-02],
        [1.0422e-07, 1.0000e+00],
        [9.9913e-01, 8.7353e-04],
        [3.3943e-10, 1.0000e+00],
        [9.9992e-01, 8.0791e-05],
        [2.6833e-04, 9.9973e-01],
        [3.0443e-07, 1.0000e+00],
        [2.9369e-10, 1.0000e+00],
        [1.0106e-02, 9.8989e-01],
        [9.9973e-01, 2.6852e-04],
        [1.1413e-07, 1.0000e+00],
        [9.9991e-01, 9.2674e-05],
        [2.6163e-04, 9.9974e-01],
        [2.7783e-09, 1.0000e+00],
        [1.2103e-10, 1.0000e+00],
        [9.9965e-01, 3.4519e-04],
        [1.1888e-03, 9.9881e-01],
        [1.3868e-04, 9.9986e-01],
        [4.7863e-09, 1.0000e+00],
        [2.4928e-07, 1.0000e+00],
        [1.1988e-05, 9.9999e-01],
        [7.8061e-11, 1.0000e+00],
        [6.2321e-05, 9.9994e-01],
        [9.9995e-01, 4.8344e-05],
        [1.4564e-05, 9.9999e-01],
        [1.1306e-07, 1.0000e+00],
        [9.6814e-01, 3.1859e-02],
        [7.8688e-10, 1.0000e+00],
        [7.1043e-07, 1.0000e+00],
        [2.1068e-04, 9.9979e-01],
        [9.9997e-01, 2.5256e-05],
        [3.1769e-09, 1.0000e+00],
        [3.9618e-06, 1.0000e+00],
        [9.1955e-11, 1.0000e+00],
        [4.6006e-06, 1.0000e+00],
        [4.2247e-09, 1.0000e+00],
        [9.6211e-03, 9.9038e-01],
        [9.9992e-01, 7.5759e-05],
        [3.5207e-07, 1.0000e+00],
        [1.2118e-09, 1.0000e+00],
        [5.4382e-06, 9.9999e-01],
        [1.1555e-03, 9.9884e-01],
        [9.6901e-04, 9.9903e-01],
        [4.2637e-10, 1.0000e+00],
        [1.7908e-11, 1.0000e+00],
        [9.6117e-01, 3.8826e-02],
        [3.8260e-06, 1.0000e+00],
        [2.0232e-08, 1.0000e+00],
        [1.1996e-06, 1.0000e+00],
        [2.7587e-03, 9.9724e-01],
        [9.9941e-01, 5.8833e-04],
        [1.3566e-06, 1.0000e+00],
        [1.3649e-11, 1.0000e+00],
        [5.0767e-01, 4.9233e-01],
        [3.7825e-05, 9.9996e-01],
        [1.5343e-03, 9.9847e-01],
        [7.2608e-09, 1.0000e+00],
        [1.6166e-11, 1.0000e+00],
        [4.8378e-05, 9.9995e-01],
        [1.6544e-10, 1.0000e+00],
        [9.1354e-13, 1.0000e+00],
        [1.9644e-10, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.0268e-12, 1.0000e+00],
        [9.9992e-01, 8.3652e-05],
        [5.8853e-02, 9.4115e-01],
        [9.5579e-01, 4.4215e-02],
        [1.3072e-07, 1.0000e+00],
        [1.7063e-08, 1.0000e+00],
        [4.6168e-09, 1.0000e+00],
        [6.9216e-03, 9.9308e-01],
        [1.9034e-06, 1.0000e+00],
        [4.0489e-05, 9.9996e-01],
        [5.3698e-08, 1.0000e+00],
        [1.4793e-11, 1.0000e+00],
        [6.6124e-09, 1.0000e+00],
        [3.7754e-11, 1.0000e+00],
        [5.0691e-09, 1.0000e+00],
        [3.6893e-06, 1.0000e+00],
        [9.5704e-01, 4.2964e-02],
        [2.6955e-11, 1.0000e+00],
        [1.1334e-06, 1.0000e+00],
        [2.8075e-09, 1.0000e+00],
        [1.1539e-04, 9.9988e-01],
        [9.6768e-01, 3.2321e-02],
        [5.7704e-05, 9.9994e-01],
        [9.8213e-01, 1.7874e-02],
        [9.9880e-01, 1.2027e-03],
        [3.1482e-09, 1.0000e+00],
        [9.9984e-01, 1.5996e-04],
        [1.4910e-09, 1.0000e+00],
        [4.9683e-05, 9.9995e-01],
        [9.9208e-01, 7.9242e-03],
        [7.7776e-06, 9.9999e-01],
        [8.5379e-09, 1.0000e+00],
        [9.9493e-01, 5.0698e-03],
        [9.9921e-01, 7.9182e-04],
        [5.3550e-09, 1.0000e+00],
        [9.9979e-01, 2.0699e-04],
        [9.9989e-01, 1.1107e-04],
        [7.2546e-06, 9.9999e-01],
        [3.8472e-11, 1.0000e+00],
        [5.2007e-07, 1.0000e+00],
        [3.1800e-10, 1.0000e+00],
        [3.1447e-05, 9.9997e-01],
        [4.6729e-08, 1.0000e+00],
        [1.4732e-06, 1.0000e+00],
        [9.2478e-01, 7.5215e-02],
        [2.6812e-08, 1.0000e+00],
        [2.9538e-05, 9.9997e-01],
        [1.8946e-09, 1.0000e+00],
        [9.9995e-01, 5.3663e-05],
        [3.1392e-07, 1.0000e+00],
        [1.6875e-12, 1.0000e+00],
        [1.0155e-08, 1.0000e+00],
        [2.0300e-10, 1.0000e+00],
        [1.2566e-07, 1.0000e+00],
        [1.7436e-10, 1.0000e+00],
        [7.7290e-06, 9.9999e-01],
        [5.9522e-08, 1.0000e+00],
        [2.8168e-07, 1.0000e+00],
        [9.9824e-01, 1.7635e-03],
        [4.2578e-06, 1.0000e+00],
        [3.8643e-05, 9.9996e-01],
        [9.9983e-01, 1.6521e-04],
        [1.3494e-10, 1.0000e+00],
        [9.6375e-13, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8974e-01, 1.0257e-02],
        [8.4938e-11, 1.0000e+00],
        [4.3766e-12, 1.0000e+00],
        [6.0931e-06, 9.9999e-01],
        [7.5370e-01, 2.4630e-01],
        [1.0331e-07, 1.0000e+00],
        [7.4058e-08, 1.0000e+00],
        [6.4269e-06, 9.9999e-01],
        [9.2684e-10, 1.0000e+00],
        [9.2813e-01, 7.1872e-02],
        [9.9999e-01, 8.2603e-06],
        [6.3517e-07, 1.0000e+00],
        [1.9394e-03, 9.9806e-01],
        [2.6549e-07, 1.0000e+00],
        [1.3129e-06, 1.0000e+00],
        [3.6452e-02, 9.6355e-01],
        [3.3300e-08, 1.0000e+00],
        [9.5221e-01, 4.7789e-02],
        [4.8095e-04, 9.9952e-01],
        [1.5316e-06, 1.0000e+00],
        [7.8625e-14, 1.0000e+00],
        [4.8041e-11, 1.0000e+00],
        [7.0898e-01, 2.9102e-01],
        [9.9993e-01, 6.9028e-05],
        [8.2762e-10, 1.0000e+00],
        [6.2699e-05, 9.9994e-01],
        [9.6722e-01, 3.2777e-02],
        [7.7765e-09, 1.0000e+00],
        [6.0296e-03, 9.9397e-01],
        [1.1357e-07, 1.0000e+00],
        [1.0775e-10, 1.0000e+00],
        [6.6244e-10, 1.0000e+00],
        [1.1790e-08, 1.0000e+00],
        [9.9738e-01, 2.6241e-03],
        [6.0182e-06, 9.9999e-01],
        [1.1822e-07, 1.0000e+00],
        [2.0881e-13, 1.0000e+00],
        [9.9958e-01, 4.2042e-04],
        [9.9983e-01, 1.6575e-04],
        [2.1497e-06, 1.0000e+00],
        [5.6282e-07, 1.0000e+00],
        [8.2166e-06, 9.9999e-01],
        [9.9954e-01, 4.6445e-04],
        [2.1089e-06, 1.0000e+00],
        [9.9984e-01, 1.5514e-04],
        [8.9507e-09, 1.0000e+00],
        [1.3377e-03, 9.9866e-01],
        [1.8247e-05, 9.9998e-01],
        [3.9322e-04, 9.9961e-01],
        [1.1311e-04, 9.9989e-01],
        [4.1269e-07, 1.0000e+00],
        [2.5038e-07, 1.0000e+00],
        [9.2341e-01, 7.6592e-02],
        [8.3467e-12, 1.0000e+00],
        [1.3732e-06, 1.0000e+00],
        [1.0063e-07, 1.0000e+00],
        [4.6388e-07, 1.0000e+00],
        [1.9195e-03, 9.9808e-01],
        [5.4629e-09, 1.0000e+00],
        [1.4875e-11, 1.0000e+00],
        [2.2085e-11, 1.0000e+00],
        [2.1735e-07, 1.0000e+00],
        [1.2414e-07, 1.0000e+00],
        [5.3163e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9989e-01, 1.0778e-04],
        [6.5173e-08, 1.0000e+00],
        [9.9996e-01, 3.7619e-05],
        [9.9928e-01, 7.2274e-04],
        [9.9992e-01, 7.9503e-05],
        [5.7092e-10, 1.0000e+00],
        [1.6207e-13, 1.0000e+00],
        [9.8253e-01, 1.7470e-02],
        [1.8166e-07, 1.0000e+00],
        [4.4744e-07, 1.0000e+00],
        [1.7901e-01, 8.2099e-01],
        [4.1779e-09, 1.0000e+00],
        [1.1812e-06, 1.0000e+00],
        [3.7606e-07, 1.0000e+00],
        [5.4973e-10, 1.0000e+00],
        [2.3401e-05, 9.9998e-01],
        [1.5409e-05, 9.9998e-01],
        [4.5019e-01, 5.4981e-01],
        [1.5409e-07, 1.0000e+00],
        [2.3313e-05, 9.9998e-01],
        [9.9986e-01, 1.3704e-04],
        [9.1946e-02, 9.0805e-01],
        [1.4830e-10, 1.0000e+00],
        [9.9848e-01, 1.5151e-03],
        [9.9988e-01, 1.2008e-04],
        [3.9644e-07, 1.0000e+00],
        [2.6788e-07, 1.0000e+00],
        [9.7093e-01, 2.9066e-02],
        [2.9281e-04, 9.9971e-01],
        [4.4472e-08, 1.0000e+00],
        [9.9968e-01, 3.2021e-04],
        [1.3618e-06, 1.0000e+00],
        [1.7038e-01, 8.2962e-01],
        [5.1454e-07, 1.0000e+00],
        [7.7675e-04, 9.9922e-01],
        [1.7782e-09, 1.0000e+00],
        [5.8070e-11, 1.0000e+00],
        [9.9975e-01, 2.4822e-04],
        [1.3956e-06, 1.0000e+00],
        [6.9523e-04, 9.9930e-01],
        [5.1390e-13, 1.0000e+00],
        [9.9984e-01, 1.6028e-04],
        [6.9157e-08, 1.0000e+00],
        [4.5185e-06, 1.0000e+00],
        [5.6803e-09, 1.0000e+00],
        [9.9996e-01, 4.4757e-05],
        [9.9993e-01, 6.8203e-05],
        [1.6703e-07, 1.0000e+00],
        [8.5938e-11, 1.0000e+00],
        [2.4316e-06, 1.0000e+00],
        [4.7374e-13, 1.0000e+00],
        [1.3715e-07, 1.0000e+00],
        [4.5766e-10, 1.0000e+00],
        [9.9940e-01, 6.0045e-04],
        [5.8474e-04, 9.9942e-01],
        [2.4536e-12, 1.0000e+00],
        [9.9974e-01, 2.5975e-04],
        [1.0000e+00, 1.1797e-07],
        [3.1859e-05, 9.9997e-01],
        [3.2808e-05, 9.9997e-01],
        [8.8873e-07, 1.0000e+00],
        [4.2161e-10, 1.0000e+00],
        [3.2515e-11, 1.0000e+00],
        [1.7773e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8211e-09, 1.0000e+00],
        [3.4667e-05, 9.9997e-01],
        [9.9886e-01, 1.1362e-03],
        [9.9997e-01, 3.2408e-05],
        [5.8152e-09, 1.0000e+00],
        [9.4482e-01, 5.5185e-02],
        [1.0425e-10, 1.0000e+00],
        [1.2643e-07, 1.0000e+00],
        [6.2292e-06, 9.9999e-01],
        [1.7904e-09, 1.0000e+00],
        [4.8838e-09, 1.0000e+00],
        [6.0928e-10, 1.0000e+00],
        [1.2576e-04, 9.9987e-01],
        [9.9995e-01, 4.6778e-05],
        [3.8479e-08, 1.0000e+00],
        [1.0679e-09, 1.0000e+00],
        [1.0037e-05, 9.9999e-01],
        [3.4336e-07, 1.0000e+00],
        [1.9716e-09, 1.0000e+00],
        [9.9987e-01, 1.2650e-04],
        [1.4320e-10, 1.0000e+00],
        [9.9977e-01, 2.2702e-04],
        [9.9963e-01, 3.6760e-04],
        [3.4291e-11, 1.0000e+00],
        [4.2717e-10, 1.0000e+00],
        [2.8627e-12, 1.0000e+00],
        [1.8788e-06, 1.0000e+00],
        [3.1037e-11, 1.0000e+00],
        [3.2198e-07, 1.0000e+00],
        [8.4468e-08, 1.0000e+00],
        [9.9985e-01, 1.4797e-04],
        [8.8438e-08, 1.0000e+00],
        [4.0928e-09, 1.0000e+00],
        [9.9996e-01, 4.1835e-05],
        [9.9997e-01, 2.5319e-05],
        [2.2397e-07, 1.0000e+00],
        [2.4816e-08, 1.0000e+00],
        [1.2694e-10, 1.0000e+00],
        [9.9985e-01, 1.4617e-04],
        [9.9854e-01, 1.4627e-03],
        [9.9830e-01, 1.7030e-03],
        [3.2733e-04, 9.9967e-01],
        [9.9934e-01, 6.5510e-04],
        [1.3033e-04, 9.9987e-01],
        [7.7997e-10, 1.0000e+00],
        [3.8141e-09, 1.0000e+00],
        [5.8926e-10, 1.0000e+00],
        [2.2531e-08, 1.0000e+00],
        [9.9996e-01, 4.2116e-05],
        [9.9986e-01, 1.4000e-04],
        [5.9891e-10, 1.0000e+00],
        [1.0036e-09, 1.0000e+00],
        [3.4976e-04, 9.9965e-01],
        [2.0558e-06, 1.0000e+00],
        [9.9994e-01, 6.0786e-05],
        [1.0882e-10, 1.0000e+00],
        [1.9062e-03, 9.9809e-01],
        [1.0000e+00, 1.1366e-07],
        [3.6448e-09, 1.0000e+00],
        [8.9301e-07, 1.0000e+00],
        [3.3523e-06, 1.0000e+00],
        [2.8102e-10, 1.0000e+00],
        [2.3088e-09, 1.0000e+00],
        [4.1904e-10, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.2020e-07, 1.0000e+00],
        [4.4813e-10, 1.0000e+00],
        [1.4349e-05, 9.9999e-01],
        [1.0663e-05, 9.9999e-01],
        [1.1193e-04, 9.9989e-01],
        [5.3716e-07, 1.0000e+00],
        [7.3416e-01, 2.6584e-01],
        [2.9996e-06, 1.0000e+00],
        [9.9991e-01, 8.5905e-05],
        [6.7567e-12, 1.0000e+00],
        [9.9981e-01, 1.8994e-04],
        [9.9185e-01, 8.1454e-03],
        [9.9962e-01, 3.7575e-04],
        [1.0000e+00, 1.3729e-06],
        [1.0473e-10, 1.0000e+00],
        [5.7868e-07, 1.0000e+00],
        [5.8846e-10, 1.0000e+00],
        [9.4948e-10, 1.0000e+00],
        [1.6916e-06, 1.0000e+00],
        [2.1531e-05, 9.9998e-01],
        [9.9976e-01, 2.3591e-04],
        [9.9979e-01, 2.0730e-04],
        [1.5837e-07, 1.0000e+00],
        [7.3364e-08, 1.0000e+00],
        [3.0570e-07, 1.0000e+00],
        [6.6132e-08, 1.0000e+00],
        [9.6791e-04, 9.9903e-01],
        [2.1128e-04, 9.9979e-01],
        [3.5385e-10, 1.0000e+00],
        [9.9987e-01, 1.2969e-04],
        [1.3760e-04, 9.9986e-01],
        [8.7922e-08, 1.0000e+00],
        [1.0990e-04, 9.9989e-01],
        [9.9968e-01, 3.1788e-04],
        [9.7485e-12, 1.0000e+00],
        [2.6268e-08, 1.0000e+00],
        [5.5696e-07, 1.0000e+00],
        [1.2390e-02, 9.8761e-01],
        [8.1975e-10, 1.0000e+00],
        [9.9960e-01, 4.0023e-04],
        [1.3674e-04, 9.9986e-01],
        [3.8134e-06, 1.0000e+00],
        [1.4817e-04, 9.9985e-01],
        [9.9901e-01, 9.9151e-04],
        [2.3082e-08, 1.0000e+00],
        [7.9112e-04, 9.9921e-01],
        [5.1691e-06, 9.9999e-01],
        [2.3748e-09, 1.0000e+00],
        [1.3075e-06, 1.0000e+00],
        [9.9767e-01, 2.3267e-03],
        [2.4262e-11, 1.0000e+00],
        [2.6123e-05, 9.9997e-01],
        [5.1736e-10, 1.0000e+00],
        [2.1415e-10, 1.0000e+00],
        [7.2400e-01, 2.7600e-01],
        [9.9757e-01, 2.4314e-03],
        [1.5286e-01, 8.4714e-01],
        [1.7622e-06, 1.0000e+00],
        [4.0680e-10, 1.0000e+00],
        [9.9907e-01, 9.2505e-04],
        [9.9991e-01, 9.0770e-05],
        [5.1164e-02, 9.4884e-01],
        [6.3557e-09, 1.0000e+00],
        [9.6465e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.1282e-06, 9.9999e-01],
        [9.9993e-01, 6.7203e-05],
        [1.9950e-08, 1.0000e+00],
        [3.0067e-07, 1.0000e+00],
        [3.3320e-09, 1.0000e+00],
        [1.0688e-08, 1.0000e+00],
        [1.5343e-02, 9.8466e-01],
        [9.9998e-01, 2.4969e-05],
        [4.2315e-10, 1.0000e+00],
        [9.9981e-01, 1.9437e-04],
        [4.2158e-07, 1.0000e+00],
        [6.5552e-12, 1.0000e+00],
        [6.2833e-07, 1.0000e+00],
        [1.9945e-07, 1.0000e+00],
        [4.6738e-08, 1.0000e+00],
        [1.4007e-02, 9.8599e-01],
        [9.7667e-01, 2.3331e-02],
        [7.3005e-01, 2.6995e-01],
        [3.2784e-06, 1.0000e+00],
        [9.9990e-01, 9.8265e-05],
        [2.0428e-10, 1.0000e+00],
        [9.7601e-01, 2.3991e-02],
        [9.9991e-01, 8.9229e-05],
        [9.2318e-09, 1.0000e+00],
        [6.0471e-05, 9.9994e-01],
        [3.7919e-06, 1.0000e+00],
        [1.3640e-08, 1.0000e+00],
        [1.5633e-06, 1.0000e+00],
        [9.9973e-01, 2.6598e-04],
        [9.9973e-01, 2.6598e-04],
        [3.1113e-10, 1.0000e+00],
        [6.8206e-05, 9.9993e-01],
        [7.3342e-06, 9.9999e-01],
        [2.2541e-07, 1.0000e+00],
        [7.3375e-10, 1.0000e+00],
        [4.6904e-06, 1.0000e+00],
        [1.9654e-10, 1.0000e+00],
        [6.3174e-10, 1.0000e+00],
        [2.1718e-06, 1.0000e+00],
        [8.7287e-05, 9.9991e-01],
        [6.2892e-11, 1.0000e+00],
        [4.3036e-05, 9.9996e-01],
        [9.9996e-01, 3.8062e-05],
        [3.2097e-07, 1.0000e+00],
        [9.9967e-01, 3.3264e-04],
        [6.9440e-09, 1.0000e+00],
        [2.6347e-08, 1.0000e+00],
        [9.3451e-01, 6.5493e-02],
        [8.9787e-01, 1.0213e-01],
        [1.9276e-06, 1.0000e+00],
        [3.7480e-06, 1.0000e+00],
        [1.2281e-08, 1.0000e+00],
        [7.9627e-05, 9.9992e-01],
        [1.4586e-10, 1.0000e+00],
        [7.2093e-08, 1.0000e+00],
        [1.6921e-05, 9.9998e-01],
        [9.7395e-11, 1.0000e+00],
        [1.5789e-03, 9.9842e-01],
        [9.3051e-06, 9.9999e-01],
        [9.9987e-01, 1.2966e-04],
        [1.7580e-09, 1.0000e+00],
        [9.8541e-01, 1.4592e-02],
        [2.2912e-06, 1.0000e+00],
        [6.4289e-05, 9.9994e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[6.3305e-09, 1.0000e+00],
        [2.1353e-11, 1.0000e+00],
        [2.1910e-03, 9.9781e-01],
        [2.2642e-03, 9.9774e-01],
        [9.9994e-01, 6.2428e-05],
        [9.9857e-01, 1.4293e-03],
        [1.7763e-02, 9.8224e-01],
        [9.9990e-01, 1.0464e-04],
        [2.6252e-04, 9.9974e-01],
        [9.9968e-01, 3.2195e-04],
        [2.2011e-10, 1.0000e+00],
        [6.9657e-01, 3.0343e-01],
        [9.9983e-01, 1.6727e-04],
        [2.5256e-03, 9.9747e-01],
        [8.2932e-05, 9.9992e-01],
        [4.6077e-07, 1.0000e+00],
        [1.0133e-11, 1.0000e+00],
        [6.9050e-08, 1.0000e+00],
        [9.7220e-01, 2.7803e-02],
        [1.4877e-06, 1.0000e+00],
        [6.4722e-05, 9.9994e-01],
        [1.8110e-05, 9.9998e-01],
        [2.5217e-04, 9.9975e-01],
        [8.9414e-01, 1.0586e-01],
        [4.0460e-07, 1.0000e+00],
        [1.5209e-08, 1.0000e+00],
        [5.1363e-11, 1.0000e+00],
        [6.2322e-07, 1.0000e+00],
        [4.4236e-05, 9.9996e-01],
        [1.5560e-01, 8.4440e-01],
        [6.7890e-04, 9.9932e-01],
        [1.3332e-09, 1.0000e+00],
        [9.9959e-01, 4.1013e-04],
        [9.9991e-01, 9.3979e-05],
        [1.1461e-06, 1.0000e+00],
        [1.1486e-06, 1.0000e+00],
        [5.7363e-09, 1.0000e+00],
        [5.0421e-07, 1.0000e+00],
        [1.1032e-12, 1.0000e+00],
        [1.0413e-05, 9.9999e-01],
        [1.3536e-01, 8.6464e-01],
        [9.9579e-01, 4.2122e-03],
        [1.2083e-10, 1.0000e+00],
        [9.8633e-01, 1.3675e-02],
        [9.9993e-01, 7.0509e-05],
        [9.9338e-01, 6.6170e-03],
        [9.9820e-01, 1.7985e-03],
        [9.9995e-01, 5.3207e-05],
        [3.3751e-04, 9.9966e-01],
        [9.9926e-01, 7.3508e-04],
        [1.3278e-05, 9.9999e-01],
        [2.8212e-09, 1.0000e+00],
        [7.8621e-10, 1.0000e+00],
        [9.7783e-01, 2.2166e-02],
        [7.6289e-07, 1.0000e+00],
        [1.5364e-03, 9.9846e-01],
        [4.3228e-06, 1.0000e+00],
        [7.7036e-10, 1.0000e+00],
        [9.9990e-01, 9.7978e-05],
        [3.1586e-10, 1.0000e+00],
        [1.1149e-04, 9.9989e-01],
        [1.7581e-10, 1.0000e+00],
        [9.9164e-01, 8.3630e-03],
        [2.7093e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5664e-05, 9.9998e-01],
        [3.8269e-12, 1.0000e+00],
        [1.0666e-04, 9.9989e-01],
        [3.8832e-07, 1.0000e+00],
        [9.9999e-01, 1.0988e-05],
        [1.7191e-04, 9.9983e-01],
        [9.9996e-01, 4.2463e-05],
        [2.4544e-06, 1.0000e+00],
        [2.1538e-04, 9.9978e-01],
        [1.3314e-08, 1.0000e+00],
        [2.6557e-09, 1.0000e+00],
        [1.1045e-06, 1.0000e+00],
        [9.9137e-01, 8.6258e-03],
        [9.9974e-01, 2.5630e-04],
        [4.9089e-02, 9.5091e-01],
        [1.1530e-08, 1.0000e+00],
        [9.6688e-09, 1.0000e+00],
        [1.2798e-05, 9.9999e-01],
        [9.9995e-01, 5.1946e-05],
        [2.5513e-08, 1.0000e+00],
        [3.1267e-09, 1.0000e+00],
        [9.3778e-01, 6.2216e-02],
        [7.1705e-03, 9.9283e-01],
        [4.5662e-08, 1.0000e+00],
        [4.9986e-04, 9.9950e-01],
        [9.9984e-01, 1.6469e-04],
        [6.6703e-08, 1.0000e+00],
        [1.4248e-04, 9.9986e-01],
        [2.9042e-09, 1.0000e+00],
        [9.3754e-04, 9.9906e-01],
        [2.1356e-07, 1.0000e+00],
        [4.3312e-11, 1.0000e+00],
        [4.7401e-11, 1.0000e+00],
        [4.4961e-05, 9.9996e-01],
        [6.9267e-05, 9.9993e-01],
        [3.1193e-06, 1.0000e+00],
        [9.9991e-01, 9.1346e-05],
        [9.9974e-01, 2.5527e-04],
        [5.9790e-09, 1.0000e+00],
        [8.5010e-06, 9.9999e-01],
        [8.1468e-03, 9.9185e-01],
        [3.8074e-05, 9.9996e-01],
        [1.3867e-08, 1.0000e+00],
        [5.6584e-07, 1.0000e+00],
        [2.1356e-07, 1.0000e+00],
        [6.2213e-06, 9.9999e-01],
        [9.9988e-01, 1.1776e-04],
        [7.4774e-08, 1.0000e+00],
        [3.6579e-05, 9.9996e-01],
        [9.9971e-01, 2.9088e-04],
        [9.9986e-01, 1.3896e-04],
        [1.9587e-05, 9.9998e-01],
        [1.1333e-03, 9.9887e-01],
        [2.0657e-07, 1.0000e+00],
        [9.6386e-09, 1.0000e+00],
        [2.2060e-07, 1.0000e+00],
        [1.3216e-06, 1.0000e+00],
        [1.6158e-05, 9.9998e-01],
        [2.9350e-08, 1.0000e+00],
        [1.1781e-02, 9.8822e-01],
        [1.1059e-06, 1.0000e+00],
        [1.2649e-04, 9.9987e-01],
        [2.7970e-11, 1.0000e+00],
        [9.9996e-01, 3.8088e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7288e-03, 9.9827e-01],
        [9.9992e-01, 7.5811e-05],
        [9.9995e-01, 4.9521e-05],
        [3.1661e-09, 1.0000e+00],
        [9.8383e-06, 9.9999e-01],
        [2.2802e-12, 1.0000e+00],
        [1.4332e-07, 1.0000e+00],
        [7.8082e-08, 1.0000e+00],
        [2.8482e-04, 9.9972e-01],
        [1.1194e-02, 9.8881e-01],
        [9.8474e-07, 1.0000e+00],
        [3.3671e-06, 1.0000e+00],
        [1.4948e-03, 9.9851e-01],
        [9.4604e-08, 1.0000e+00],
        [1.9690e-05, 9.9998e-01],
        [3.5349e-04, 9.9965e-01],
        [2.7939e-10, 1.0000e+00],
        [2.7136e-06, 1.0000e+00],
        [2.0382e-06, 1.0000e+00],
        [1.9286e-05, 9.9998e-01],
        [9.7426e-09, 1.0000e+00],
        [6.6156e-05, 9.9993e-01],
        [2.8984e-05, 9.9997e-01],
        [9.9991e-01, 9.3494e-05],
        [3.7090e-02, 9.6291e-01],
        [1.4411e-10, 1.0000e+00],
        [6.5518e-02, 9.3448e-01],
        [7.3333e-02, 9.2667e-01],
        [9.9962e-01, 3.8004e-04],
        [9.9865e-01, 1.3468e-03],
        [1.5623e-08, 1.0000e+00],
        [1.8675e-10, 1.0000e+00],
        [9.9992e-01, 7.7923e-05],
        [3.7160e-13, 1.0000e+00],
        [1.8586e-04, 9.9981e-01],
        [9.8827e-06, 9.9999e-01],
        [4.6677e-04, 9.9953e-01],
        [9.9994e-01, 5.5924e-05],
        [9.4029e-01, 5.9706e-02],
        [1.2357e-06, 1.0000e+00],
        [9.9981e-01, 1.9181e-04],
        [1.6629e-02, 9.8337e-01],
        [1.9256e-08, 1.0000e+00],
        [9.9218e-01, 7.8176e-03],
        [9.7719e-01, 2.2815e-02],
        [1.2815e-01, 8.7185e-01],
        [2.0111e-10, 1.0000e+00],
        [9.2855e-05, 9.9991e-01],
        [9.9647e-01, 3.5257e-03],
        [2.1548e-09, 1.0000e+00],
        [2.0382e-06, 1.0000e+00],
        [9.9975e-01, 2.5221e-04],
        [9.9949e-01, 5.0587e-04],
        [4.8962e-10, 1.0000e+00],
        [9.4774e-06, 9.9999e-01],
        [9.9988e-01, 1.2200e-04],
        [1.9426e-07, 1.0000e+00],
        [4.2359e-05, 9.9996e-01],
        [8.9127e-04, 9.9911e-01],
        [1.6748e-08, 1.0000e+00],
        [1.1497e-06, 1.0000e+00],
        [1.5568e-07, 1.0000e+00],
        [2.6076e-06, 1.0000e+00],
        [9.9994e-01, 5.7385e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4075e-07, 1.0000e+00],
        [1.1238e-08, 1.0000e+00],
        [9.9941e-01, 5.9427e-04],
        [6.0082e-04, 9.9940e-01],
        [1.6717e-07, 1.0000e+00],
        [9.9988e-01, 1.2467e-04],
        [4.7315e-03, 9.9527e-01],
        [1.0000e+00, 1.0072e-06],
        [6.4386e-07, 1.0000e+00],
        [9.9988e-01, 1.1514e-04],
        [1.0000e+00, 4.7605e-06],
        [4.5035e-06, 1.0000e+00],
        [2.4219e-06, 1.0000e+00],
        [1.9719e-03, 9.9803e-01],
        [1.9701e-05, 9.9998e-01],
        [2.6735e-05, 9.9997e-01],
        [9.9994e-01, 5.5265e-05],
        [3.5561e-06, 1.0000e+00],
        [1.6981e-04, 9.9983e-01],
        [3.8234e-04, 9.9962e-01],
        [1.2331e-05, 9.9999e-01],
        [9.9999e-01, 6.2827e-06],
        [9.9999e-01, 1.2735e-05],
        [2.0684e-07, 1.0000e+00],
        [1.1365e-07, 1.0000e+00],
        [9.9993e-01, 7.4193e-05],
        [4.9877e-04, 9.9950e-01],
        [2.3294e-06, 1.0000e+00],
        [7.3315e-06, 9.9999e-01],
        [9.9981e-01, 1.8871e-04],
        [1.4578e-08, 1.0000e+00],
        [4.6546e-10, 1.0000e+00],
        [2.0585e-05, 9.9998e-01],
        [7.7086e-12, 1.0000e+00],
        [2.5003e-10, 1.0000e+00],
        [4.4059e-04, 9.9956e-01],
        [2.1865e-09, 1.0000e+00],
        [7.5028e-08, 1.0000e+00],
        [1.0871e-08, 1.0000e+00],
        [6.5541e-07, 1.0000e+00],
        [9.9984e-01, 1.6183e-04],
        [9.9994e-01, 6.0306e-05],
        [1.9934e-05, 9.9998e-01],
        [9.6004e-01, 3.9958e-02],
        [1.0233e-04, 9.9990e-01],
        [5.8495e-05, 9.9994e-01],
        [1.5055e-06, 1.0000e+00],
        [1.6037e-03, 9.9840e-01],
        [1.8535e-02, 9.8147e-01],
        [1.2195e-04, 9.9988e-01],
        [6.5915e-07, 1.0000e+00],
        [9.9331e-01, 6.6863e-03],
        [6.3496e-08, 1.0000e+00],
        [2.3716e-06, 1.0000e+00],
        [8.7526e-04, 9.9912e-01],
        [6.6915e-03, 9.9331e-01],
        [3.2864e-04, 9.9967e-01],
        [1.8602e-05, 9.9998e-01],
        [2.3856e-11, 1.0000e+00],
        [1.4529e-06, 1.0000e+00],
        [1.5124e-05, 9.9998e-01],
        [7.3917e-06, 9.9999e-01],
        [1.4995e-11, 1.0000e+00],
        [9.2881e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.0673e-10, 1.0000e+00],
        [1.0000e+00, 4.2554e-06],
        [9.0995e-07, 1.0000e+00],
        [7.5920e-08, 1.0000e+00],
        [1.2286e-07, 1.0000e+00],
        [9.9986e-01, 1.4490e-04],
        [7.6899e-05, 9.9992e-01],
        [3.5682e-11, 1.0000e+00],
        [2.8379e-07, 1.0000e+00],
        [1.4822e-05, 9.9999e-01],
        [2.2427e-05, 9.9998e-01],
        [1.0410e-05, 9.9999e-01],
        [1.0008e-01, 8.9992e-01],
        [2.2610e-07, 1.0000e+00],
        [2.3623e-07, 1.0000e+00],
        [6.7422e-05, 9.9993e-01],
        [1.5694e-01, 8.4306e-01],
        [9.9997e-01, 3.4602e-05],
        [7.9720e-01, 2.0280e-01],
        [2.1133e-10, 1.0000e+00],
        [8.5474e-06, 9.9999e-01],
        [6.1724e-05, 9.9994e-01],
        [9.9937e-01, 6.2715e-04],
        [7.8378e-04, 9.9922e-01],
        [3.5303e-06, 1.0000e+00],
        [2.5562e-08, 1.0000e+00],
        [2.2389e-06, 1.0000e+00],
        [1.1734e-07, 1.0000e+00],
        [7.3086e-05, 9.9993e-01],
        [1.2746e-06, 1.0000e+00],
        [4.9388e-03, 9.9506e-01],
        [3.1105e-03, 9.9689e-01],
        [9.9994e-01, 5.5225e-05],
        [3.1055e-06, 1.0000e+00],
        [3.5411e-09, 1.0000e+00],
        [9.3012e-02, 9.0699e-01],
        [1.1264e-04, 9.9989e-01],
        [9.9964e-01, 3.5757e-04],
        [1.4336e-06, 1.0000e+00],
        [9.9994e-01, 6.1822e-05],
        [8.2386e-04, 9.9918e-01],
        [4.4278e-06, 1.0000e+00],
        [9.9961e-01, 3.8651e-04],
        [1.0929e-08, 1.0000e+00],
        [2.1658e-07, 1.0000e+00],
        [2.2810e-10, 1.0000e+00],
        [2.2831e-02, 9.7717e-01],
        [9.9173e-05, 9.9990e-01],
        [9.9997e-01, 2.5014e-05],
        [9.3545e-01, 6.4554e-02],
        [3.4809e-07, 1.0000e+00],
        [6.0602e-06, 9.9999e-01],
        [3.2506e-07, 1.0000e+00],
        [3.4103e-02, 9.6590e-01],
        [2.8198e-04, 9.9972e-01],
        [4.4116e-11, 1.0000e+00],
        [8.3319e-09, 1.0000e+00],
        [7.6433e-07, 1.0000e+00],
        [9.9976e-01, 2.3503e-04],
        [5.1083e-06, 9.9999e-01],
        [8.6486e-01, 1.3514e-01],
        [9.9977e-01, 2.2789e-04],
        [1.1913e-04, 9.9988e-01],
        [1.5734e-02, 9.8427e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1633e-10, 1.0000e+00],
        [9.9951e-01, 4.8707e-04],
        [1.0547e-08, 1.0000e+00],
        [9.3031e-05, 9.9991e-01],
        [3.7928e-03, 9.9621e-01],
        [1.3488e-05, 9.9999e-01],
        [3.1260e-05, 9.9997e-01],
        [1.7311e-05, 9.9998e-01],
        [2.0790e-06, 1.0000e+00],
        [1.4065e-07, 1.0000e+00],
        [1.4316e-07, 1.0000e+00],
        [1.7333e-01, 8.2667e-01],
        [1.6813e-08, 1.0000e+00],
        [5.0483e-04, 9.9950e-01],
        [9.9982e-01, 1.7996e-04],
        [9.1486e-03, 9.9085e-01],
        [9.5542e-05, 9.9990e-01],
        [1.3530e-04, 9.9986e-01],
        [9.0743e-10, 1.0000e+00],
        [2.9566e-08, 1.0000e+00],
        [4.1727e-08, 1.0000e+00],
        [3.9822e-07, 1.0000e+00],
        [9.9996e-01, 4.3720e-05],
        [4.5961e-06, 1.0000e+00],
        [9.9683e-01, 3.1737e-03],
        [1.1583e-07, 1.0000e+00],
        [1.8515e-08, 1.0000e+00],
        [9.9873e-01, 1.2651e-03],
        [1.4797e-03, 9.9852e-01],
        [5.3230e-07, 1.0000e+00],
        [2.6305e-07, 1.0000e+00],
        [1.1454e-05, 9.9999e-01],
        [3.0000e-07, 1.0000e+00],
        [8.0990e-09, 1.0000e+00],
        [6.3523e-05, 9.9994e-01],
        [1.9386e-09, 1.0000e+00],
        [9.6894e-08, 1.0000e+00],
        [9.9997e-01, 2.8764e-05],
        [9.9983e-01, 1.7015e-04],
        [1.3453e-08, 1.0000e+00],
        [1.0094e-02, 9.8991e-01],
        [5.5669e-08, 1.0000e+00],
        [9.9975e-01, 2.5445e-04],
        [1.6866e-09, 1.0000e+00],
        [6.0505e-07, 1.0000e+00],
        [9.9493e-01, 5.0693e-03],
        [9.9992e-01, 8.2230e-05],
        [9.9996e-01, 3.8562e-05],
        [3.4558e-09, 1.0000e+00],
        [1.9583e-10, 1.0000e+00],
        [4.4604e-07, 1.0000e+00],
        [9.9996e-01, 3.6469e-05],
        [9.9997e-01, 2.5641e-05],
        [7.9565e-06, 9.9999e-01],
        [4.5959e-06, 1.0000e+00],
        [8.5135e-06, 9.9999e-01],
        [9.9911e-01, 8.8958e-04],
        [9.4567e-01, 5.4327e-02],
        [1.4227e-06, 1.0000e+00],
        [2.3786e-06, 1.0000e+00],
        [9.9884e-01, 1.1620e-03],
        [1.0650e-08, 1.0000e+00],
        [7.1507e-08, 1.0000e+00],
        [1.6884e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9983e-01, 1.6634e-04],
        [6.6562e-08, 1.0000e+00],
        [9.9755e-01, 2.4503e-03],
        [3.4892e-07, 1.0000e+00],
        [6.7339e-04, 9.9933e-01],
        [1.5749e-11, 1.0000e+00],
        [2.9451e-06, 1.0000e+00],
        [2.1320e-07, 1.0000e+00],
        [6.2888e-10, 1.0000e+00],
        [9.9997e-01, 3.1961e-05],
        [8.5914e-08, 1.0000e+00],
        [9.9706e-01, 2.9418e-03],
        [1.2232e-04, 9.9988e-01],
        [6.2331e-02, 9.3767e-01],
        [1.3897e-06, 1.0000e+00],
        [1.0014e-07, 1.0000e+00],
        [9.9988e-01, 1.2022e-04],
        [1.4749e-11, 1.0000e+00],
        [2.4821e-11, 1.0000e+00],
        [9.9983e-01, 1.6622e-04],
        [3.4481e-05, 9.9997e-01],
        [9.9996e-01, 3.5885e-05],
        [9.0469e-01, 9.5308e-02],
        [9.9999e-01, 1.2419e-05],
        [3.2058e-05, 9.9997e-01],
        [1.7782e-06, 1.0000e+00],
        [9.9987e-01, 1.2605e-04],
        [9.9988e-01, 1.1839e-04],
        [9.9991e-01, 9.1719e-05],
        [1.0623e-05, 9.9999e-01],
        [9.2960e-01, 7.0402e-02],
        [1.3708e-06, 1.0000e+00],
        [1.0000e+00, 3.6839e-06],
        [1.1871e-09, 1.0000e+00],
        [1.1107e-06, 1.0000e+00],
        [9.9982e-01, 1.8491e-04],
        [1.1345e-07, 1.0000e+00],
        [2.4848e-09, 1.0000e+00],
        [2.1108e-05, 9.9998e-01],
        [4.8479e-04, 9.9952e-01],
        [9.9997e-01, 3.1392e-05],
        [1.5653e-05, 9.9998e-01],
        [9.9995e-01, 5.2040e-05],
        [1.3505e-06, 1.0000e+00],
        [7.3722e-06, 9.9999e-01],
        [1.4502e-05, 9.9999e-01],
        [4.0135e-04, 9.9960e-01],
        [9.9998e-01, 2.3545e-05],
        [9.9229e-01, 7.7070e-03],
        [9.9968e-01, 3.1881e-04],
        [2.1706e-03, 9.9783e-01],
        [9.9995e-01, 4.9887e-05],
        [9.9996e-01, 3.6400e-05],
        [4.1619e-04, 9.9958e-01],
        [6.9150e-06, 9.9999e-01],
        [1.9443e-08, 1.0000e+00],
        [9.9992e-01, 7.9315e-05],
        [1.0219e-05, 9.9999e-01],
        [4.9705e-14, 1.0000e+00],
        [2.7725e-10, 1.0000e+00],
        [2.3829e-07, 1.0000e+00],
        [9.9037e-01, 9.6343e-03],
        [1.2546e-11, 1.0000e+00],
        [3.2780e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7955e-08, 1.0000e+00],
        [9.9999e-01, 7.7142e-06],
        [3.0407e-02, 9.6959e-01],
        [9.9965e-01, 3.5137e-04],
        [1.3016e-05, 9.9999e-01],
        [5.5003e-08, 1.0000e+00],
        [6.6788e-09, 1.0000e+00],
        [3.8858e-04, 9.9961e-01],
        [6.5481e-01, 3.4519e-01],
        [9.9994e-01, 6.4292e-05],
        [9.9999e-01, 1.3674e-05],
        [6.5553e-02, 9.3445e-01],
        [1.3023e-04, 9.9987e-01],
        [9.6873e-05, 9.9990e-01],
        [1.7649e-05, 9.9998e-01],
        [9.9879e-01, 1.2071e-03],
        [9.9902e-01, 9.7638e-04],
        [1.6364e-04, 9.9984e-01],
        [1.4270e-07, 1.0000e+00],
        [3.7435e-09, 1.0000e+00],
        [1.6703e-04, 9.9983e-01],
        [2.3611e-07, 1.0000e+00],
        [1.0000e+00, 4.6063e-08],
        [9.9491e-01, 5.0856e-03],
        [2.9296e-09, 1.0000e+00],
        [1.5037e-04, 9.9985e-01],
        [9.7529e-06, 9.9999e-01],
        [1.6361e-04, 9.9984e-01],
        [3.0408e-07, 1.0000e+00],
        [9.1375e-06, 9.9999e-01],
        [9.9997e-01, 2.6398e-05],
        [9.9991e-01, 8.7977e-05],
        [9.6090e-03, 9.9039e-01],
        [1.1200e-02, 9.8880e-01],
        [9.9609e-01, 3.9103e-03],
        [2.2161e-09, 1.0000e+00],
        [2.9239e-04, 9.9971e-01],
        [2.9653e-05, 9.9997e-01],
        [1.1844e-04, 9.9988e-01],
        [9.9945e-01, 5.4905e-04],
        [9.5853e-07, 1.0000e+00],
        [4.8535e-06, 1.0000e+00],
        [9.9986e-01, 1.4469e-04],
        [9.9999e-01, 8.4676e-06],
        [2.2888e-07, 1.0000e+00],
        [2.4751e-08, 1.0000e+00],
        [9.9965e-01, 3.4995e-04],
        [1.6340e-05, 9.9998e-01],
        [1.8267e-02, 9.8173e-01],
        [2.2230e-10, 1.0000e+00],
        [2.3449e-03, 9.9766e-01],
        [9.9991e-01, 9.0637e-05],
        [4.0169e-08, 1.0000e+00],
        [1.1546e-05, 9.9999e-01],
        [9.9991e-01, 9.3610e-05],
        [4.8935e-09, 1.0000e+00],
        [4.7620e-04, 9.9952e-01],
        [3.2145e-04, 9.9968e-01],
        [4.0306e-06, 1.0000e+00],
        [9.9999e-01, 1.3974e-05],
        [2.5163e-08, 1.0000e+00],
        [1.0626e-03, 9.9894e-01],
        [9.9987e-01, 1.3271e-04],
        [6.1461e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5761e-06, 1.0000e+00],
        [6.1228e-01, 3.8772e-01],
        [1.4789e-07, 1.0000e+00],
        [7.3050e-06, 9.9999e-01],
        [5.0514e-06, 9.9999e-01],
        [1.9266e-02, 9.8073e-01],
        [2.4233e-07, 1.0000e+00],
        [9.9998e-01, 1.9760e-05],
        [9.7549e-03, 9.9024e-01],
        [3.3617e-03, 9.9664e-01],
        [1.5814e-07, 1.0000e+00],
        [9.9991e-01, 8.9381e-05],
        [2.9046e-02, 9.7095e-01],
        [4.7052e-05, 9.9995e-01],
        [2.6623e-03, 9.9734e-01],
        [1.4106e-04, 9.9986e-01],
        [4.8828e-05, 9.9995e-01],
        [9.7317e-01, 2.6835e-02],
        [1.8327e-06, 1.0000e+00],
        [9.9993e-01, 7.2699e-05],
        [3.7869e-07, 1.0000e+00],
        [4.5317e-01, 5.4683e-01],
        [9.9989e-01, 1.1495e-04],
        [9.9948e-01, 5.1527e-04],
        [1.1708e-09, 1.0000e+00],
        [3.9230e-02, 9.6077e-01],
        [1.3208e-08, 1.0000e+00],
        [8.5507e-05, 9.9991e-01],
        [8.4248e-08, 1.0000e+00],
        [3.6087e-08, 1.0000e+00],
        [1.5945e-04, 9.9984e-01],
        [3.9126e-05, 9.9996e-01],
        [9.9996e-01, 3.6808e-05],
        [3.5033e-08, 1.0000e+00],
        [9.9992e-01, 7.8443e-05],
        [1.7253e-04, 9.9983e-01],
        [3.8746e-03, 9.9613e-01],
        [3.6108e-04, 9.9964e-01],
        [2.3736e-11, 1.0000e+00],
        [2.5192e-09, 1.0000e+00],
        [3.7624e-02, 9.6238e-01],
        [9.9997e-01, 3.4729e-05],
        [9.9955e-01, 4.4889e-04],
        [9.9977e-01, 2.3030e-04],
        [4.9108e-04, 9.9951e-01],
        [1.7756e-12, 1.0000e+00],
        [8.6607e-12, 1.0000e+00],
        [2.0971e-06, 1.0000e+00],
        [4.9613e-10, 1.0000e+00],
        [1.5740e-08, 1.0000e+00],
        [1.0080e-08, 1.0000e+00],
        [1.0039e-05, 9.9999e-01],
        [2.1609e-09, 1.0000e+00],
        [3.0759e-06, 1.0000e+00],
        [2.0653e-04, 9.9979e-01],
        [2.1375e-08, 1.0000e+00],
        [2.0815e-03, 9.9792e-01],
        [6.0436e-04, 9.9940e-01],
        [1.1556e-06, 1.0000e+00],
        [1.2771e-05, 9.9999e-01],
        [1.0869e-05, 9.9999e-01],
        [3.0672e-03, 9.9693e-01],
        [3.7700e-07, 1.0000e+00],
        [4.3995e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.5004e-01, 4.4996e-01],
        [3.0606e-06, 1.0000e+00],
        [9.9997e-01, 3.4709e-05],
        [3.1190e-08, 1.0000e+00],
        [2.4838e-11, 1.0000e+00],
        [4.7396e-04, 9.9953e-01],
        [9.9985e-01, 1.5270e-04],
        [4.3887e-01, 5.6113e-01],
        [8.2203e-05, 9.9992e-01],
        [1.2233e-02, 9.8777e-01],
        [1.8891e-10, 1.0000e+00],
        [2.8144e-06, 1.0000e+00],
        [2.2362e-07, 1.0000e+00],
        [9.4332e-06, 9.9999e-01],
        [1.0768e-09, 1.0000e+00],
        [1.0702e-03, 9.9893e-01],
        [8.3849e-09, 1.0000e+00],
        [9.9998e-01, 2.4059e-05],
        [1.5653e-06, 1.0000e+00],
        [5.7184e-10, 1.0000e+00],
        [2.2880e-02, 9.7712e-01],
        [6.0845e-02, 9.3915e-01],
        [6.0072e-06, 9.9999e-01],
        [2.2643e-06, 1.0000e+00],
        [9.9915e-01, 8.5390e-04],
        [9.9737e-01, 2.6288e-03],
        [4.1533e-06, 1.0000e+00],
        [4.5522e-03, 9.9545e-01],
        [4.8901e-09, 1.0000e+00],
        [9.9945e-01, 5.4719e-04],
        [2.9315e-08, 1.0000e+00],
        [9.9993e-01, 6.8428e-05],
        [8.0006e-10, 1.0000e+00],
        [1.0000e+00, 1.3265e-06],
        [2.6272e-05, 9.9997e-01],
        [1.0692e-07, 1.0000e+00],
        [9.9976e-01, 2.4061e-04],
        [2.7648e-03, 9.9724e-01],
        [1.0850e-07, 1.0000e+00],
        [7.7687e-02, 9.2231e-01],
        [1.1897e-04, 9.9988e-01],
        [3.7700e-06, 1.0000e+00],
        [1.3807e-06, 1.0000e+00],
        [4.9213e-05, 9.9995e-01],
        [2.3258e-09, 1.0000e+00],
        [3.4551e-07, 1.0000e+00],
        [9.0926e-09, 1.0000e+00],
        [7.9353e-05, 9.9992e-01],
        [6.4844e-07, 1.0000e+00],
        [2.6500e-04, 9.9973e-01],
        [2.6936e-03, 9.9731e-01],
        [5.7699e-03, 9.9423e-01],
        [9.9796e-01, 2.0371e-03],
        [9.9993e-01, 6.8481e-05],
        [9.9996e-01, 3.6562e-05],
        [9.9995e-01, 5.2130e-05],
        [6.6134e-05, 9.9993e-01],
        [5.7904e-07, 1.0000e+00],
        [2.0315e-05, 9.9998e-01],
        [1.8435e-08, 1.0000e+00],
        [9.9992e-01, 7.8345e-05],
        [6.2812e-10, 1.0000e+00],
        [1.4898e-07, 1.0000e+00],
        [5.5536e-01, 4.4464e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.4470e-01, 5.5295e-02],
        [6.5384e-09, 1.0000e+00],
        [5.4119e-04, 9.9946e-01],
        [7.1634e-10, 1.0000e+00],
        [9.9482e-01, 5.1769e-03],
        [5.4481e-07, 1.0000e+00],
        [9.9988e-01, 1.1531e-04],
        [1.9751e-05, 9.9998e-01],
        [1.9239e-08, 1.0000e+00],
        [6.6148e-08, 1.0000e+00],
        [9.9990e-01, 9.5368e-05],
        [9.3465e-01, 6.5346e-02],
        [9.9986e-01, 1.4410e-04],
        [3.8208e-08, 1.0000e+00],
        [4.9582e-06, 9.9999e-01],
        [2.6162e-07, 1.0000e+00],
        [9.9992e-01, 8.0712e-05],
        [6.7346e-08, 1.0000e+00],
        [3.7770e-11, 1.0000e+00],
        [1.2117e-03, 9.9879e-01],
        [1.3473e-06, 1.0000e+00],
        [9.9989e-01, 1.1218e-04],
        [5.9361e-07, 1.0000e+00],
        [9.9869e-01, 1.3147e-03],
        [5.4296e-05, 9.9995e-01],
        [1.0096e-10, 1.0000e+00],
        [6.8958e-07, 1.0000e+00],
        [3.7024e-08, 1.0000e+00],
        [6.5760e-07, 1.0000e+00],
        [9.9989e-01, 1.1218e-04],
        [1.6383e-05, 9.9998e-01],
        [9.9993e-01, 6.9407e-05],
        [3.0895e-08, 1.0000e+00],
        [8.1735e-09, 1.0000e+00],
        [1.5491e-02, 9.8451e-01],
        [1.0338e-08, 1.0000e+00],
        [2.7232e-12, 1.0000e+00],
        [3.0542e-10, 1.0000e+00],
        [1.0132e-10, 1.0000e+00],
        [3.4148e-05, 9.9997e-01],
        [2.6984e-03, 9.9730e-01],
        [1.6767e-08, 1.0000e+00],
        [6.2877e-04, 9.9937e-01],
        [9.9956e-01, 4.4434e-04],
        [2.5073e-05, 9.9997e-01],
        [9.2747e-05, 9.9991e-01],
        [1.6294e-04, 9.9984e-01],
        [1.6357e-05, 9.9998e-01],
        [3.4977e-07, 1.0000e+00],
        [8.4491e-11, 1.0000e+00],
        [2.1605e-07, 1.0000e+00],
        [3.9322e-10, 1.0000e+00],
        [9.9996e-01, 3.7495e-05],
        [1.4835e-06, 1.0000e+00],
        [9.9992e-01, 8.1115e-05],
        [9.9312e-01, 6.8796e-03],
        [9.9997e-01, 3.1288e-05],
        [2.8013e-08, 1.0000e+00],
        [3.8737e-04, 9.9961e-01],
        [9.8527e-01, 1.4732e-02],
        [1.8600e-04, 9.9981e-01],
        [9.9975e-01, 2.4987e-04],
        [8.5068e-09, 1.0000e+00],
        [9.9979e-01, 2.1063e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.9091e-05, 9.9998e-01],
        [9.9550e-01, 4.4997e-03],
        [6.8171e-04, 9.9932e-01],
        [9.9785e-01, 2.1466e-03],
        [9.9689e-01, 3.1145e-03],
        [9.9991e-01, 9.4169e-05],
        [4.4717e-03, 9.9553e-01],
        [9.9718e-01, 2.8228e-03],
        [1.5223e-07, 1.0000e+00],
        [7.6451e-06, 9.9999e-01],
        [2.0652e-07, 1.0000e+00],
        [9.9447e-06, 9.9999e-01],
        [9.9994e-01, 6.1389e-05],
        [1.6863e-08, 1.0000e+00],
        [5.8476e-05, 9.9994e-01],
        [9.9979e-01, 2.0847e-04],
        [7.4127e-05, 9.9993e-01],
        [3.8302e-08, 1.0000e+00],
        [3.1945e-06, 1.0000e+00],
        [9.9811e-01, 1.8854e-03],
        [9.9996e-01, 4.2112e-05],
        [9.8002e-01, 1.9979e-02],
        [9.9986e-01, 1.3675e-04]], grad_fn=<SoftmaxBackward0>)
Epoch 9, loss: 0.3318
tensor([[1.3097e-08, 1.0000e+00],
        [4.3593e-06, 1.0000e+00],
        [2.9546e-07, 1.0000e+00],
        [4.4692e-06, 1.0000e+00],
        [2.1499e-08, 1.0000e+00],
        [5.7561e-09, 1.0000e+00],
        [1.2878e-01, 8.7122e-01],
        [2.9655e-10, 1.0000e+00],
        [9.6827e-04, 9.9903e-01],
        [9.9093e-01, 9.0702e-03],
        [5.5616e-09, 1.0000e+00],
        [1.5966e-07, 1.0000e+00],
        [8.3907e-01, 1.6093e-01],
        [9.9945e-01, 5.4953e-04],
        [9.6706e-06, 9.9999e-01],
        [5.9649e-06, 9.9999e-01],
        [1.0690e-11, 1.0000e+00],
        [1.0947e-05, 9.9999e-01],
        [6.4260e-05, 9.9994e-01],
        [7.8974e-07, 1.0000e+00],
        [2.7658e-02, 9.7234e-01],
        [1.0764e-06, 1.0000e+00],
        [1.1205e-08, 1.0000e+00],
        [7.7358e-05, 9.9992e-01],
        [1.1744e-03, 9.9883e-01],
        [6.5007e-05, 9.9994e-01],
        [2.8995e-05, 9.9997e-01],
        [6.5398e-09, 1.0000e+00],
        [2.3472e-10, 1.0000e+00],
        [4.9863e-09, 1.0000e+00],
        [2.3047e-11, 1.0000e+00],
        [5.4612e-01, 4.5388e-01],
        [1.7556e-06, 1.0000e+00],
        [6.2396e-09, 1.0000e+00],
        [9.9990e-01, 9.7944e-05],
        [2.0620e-08, 1.0000e+00],
        [2.0831e-07, 1.0000e+00],
        [7.5694e-11, 1.0000e+00],
        [9.9945e-01, 5.5188e-04],
        [8.4355e-09, 1.0000e+00],
        [9.8823e-04, 9.9901e-01],
        [2.4584e-12, 1.0000e+00],
        [7.0716e-09, 1.0000e+00],
        [7.9939e-09, 1.0000e+00],
        [9.9994e-01, 5.8288e-05],
        [3.3902e-08, 1.0000e+00],
        [7.5665e-10, 1.0000e+00],
        [1.1403e-06, 1.0000e+00],
        [9.9980e-01, 1.9567e-04],
        [9.9994e-01, 5.7316e-05],
        [5.2976e-06, 9.9999e-01],
        [4.3932e-05, 9.9996e-01],
        [2.3389e-08, 1.0000e+00],
        [1.6708e-10, 1.0000e+00],
        [9.9867e-01, 1.3250e-03],
        [5.9271e-12, 1.0000e+00],
        [9.9977e-01, 2.3309e-04],
        [9.9973e-01, 2.7066e-04],
        [9.9986e-01, 1.3647e-04],
        [4.3629e-08, 1.0000e+00],
        [7.4866e-08, 1.0000e+00],
        [3.0407e-07, 1.0000e+00],
        [7.1399e-06, 9.9999e-01],
        [4.4710e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.9524e-10, 1.0000e+00],
        [2.2145e-05, 9.9998e-01],
        [3.5451e-07, 1.0000e+00],
        [1.1105e-08, 1.0000e+00],
        [2.7099e-08, 1.0000e+00],
        [8.8058e-01, 1.1942e-01],
        [9.9438e-01, 5.6162e-03],
        [4.2115e-11, 1.0000e+00],
        [9.9991e-01, 9.0667e-05],
        [1.3165e-03, 9.9868e-01],
        [6.1541e-07, 1.0000e+00],
        [3.4375e-07, 1.0000e+00],
        [9.6478e-01, 3.5222e-02],
        [3.1094e-04, 9.9969e-01],
        [2.2407e-05, 9.9998e-01],
        [2.3149e-09, 1.0000e+00],
        [1.4631e-09, 1.0000e+00],
        [3.4866e-08, 1.0000e+00],
        [1.0838e-05, 9.9999e-01],
        [9.3588e-05, 9.9991e-01],
        [4.1116e-07, 1.0000e+00],
        [5.9494e-08, 1.0000e+00],
        [2.3414e-11, 1.0000e+00],
        [4.2486e-09, 1.0000e+00],
        [3.0529e-04, 9.9969e-01],
        [2.2058e-08, 1.0000e+00],
        [6.1986e-09, 1.0000e+00],
        [9.9992e-01, 7.5217e-05],
        [7.4878e-10, 1.0000e+00],
        [3.3764e-06, 1.0000e+00],
        [9.4519e-09, 1.0000e+00],
        [9.9659e-01, 3.4097e-03],
        [9.9980e-01, 2.0061e-04],
        [1.5881e-07, 1.0000e+00],
        [8.9269e-10, 1.0000e+00],
        [6.7735e-07, 1.0000e+00],
        [3.0768e-08, 1.0000e+00],
        [9.4568e-10, 1.0000e+00],
        [8.9747e-12, 1.0000e+00],
        [1.2669e-08, 1.0000e+00],
        [4.4012e-06, 1.0000e+00],
        [4.3048e-10, 1.0000e+00],
        [1.5714e-06, 1.0000e+00],
        [1.3121e-07, 1.0000e+00],
        [6.7450e-01, 3.2550e-01],
        [8.6323e-05, 9.9991e-01],
        [1.1403e-04, 9.9989e-01],
        [2.0453e-11, 1.0000e+00],
        [2.4432e-09, 1.0000e+00],
        [9.4947e-11, 1.0000e+00],
        [9.8227e-01, 1.7729e-02],
        [6.6380e-05, 9.9993e-01],
        [3.2868e-09, 1.0000e+00],
        [2.0747e-03, 9.9793e-01],
        [3.9368e-11, 1.0000e+00],
        [9.9908e-01, 9.2061e-04],
        [9.9973e-01, 2.6746e-04],
        [8.4443e-09, 1.0000e+00],
        [7.4019e-08, 1.0000e+00],
        [1.2114e-01, 8.7886e-01],
        [6.4981e-05, 9.9994e-01],
        [1.6551e-09, 1.0000e+00],
        [9.9992e-01, 7.9807e-05],
        [9.9982e-01, 1.7728e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4811e-04, 9.9985e-01],
        [1.6866e-04, 9.9983e-01],
        [9.9994e-01, 5.5706e-05],
        [3.2379e-08, 1.0000e+00],
        [9.9318e-01, 6.8191e-03],
        [8.2371e-06, 9.9999e-01],
        [1.6213e-06, 1.0000e+00],
        [9.0368e-07, 1.0000e+00],
        [9.9954e-01, 4.5899e-04],
        [9.9998e-01, 2.2867e-05],
        [9.9863e-01, 1.3718e-03],
        [2.0328e-07, 1.0000e+00],
        [4.5392e-08, 1.0000e+00],
        [9.9978e-01, 2.2278e-04],
        [3.2379e-08, 1.0000e+00],
        [1.1404e-02, 9.8860e-01],
        [5.7033e-09, 1.0000e+00],
        [1.5212e-05, 9.9998e-01],
        [7.2931e-06, 9.9999e-01],
        [1.0796e-07, 1.0000e+00],
        [9.9978e-01, 2.1903e-04],
        [6.7722e-10, 1.0000e+00],
        [1.0102e-04, 9.9990e-01],
        [1.4637e-04, 9.9985e-01],
        [9.7471e-01, 2.5288e-02],
        [1.0000e+00, 2.8636e-06],
        [1.2271e-04, 9.9988e-01],
        [1.5282e-07, 1.0000e+00],
        [6.3960e-10, 1.0000e+00],
        [1.2083e-08, 1.0000e+00],
        [1.7432e-06, 1.0000e+00],
        [1.6712e-05, 9.9998e-01],
        [8.5093e-09, 1.0000e+00],
        [3.2111e-08, 1.0000e+00],
        [5.3420e-06, 9.9999e-01],
        [3.5302e-07, 1.0000e+00],
        [1.0070e-08, 1.0000e+00],
        [9.2075e-09, 1.0000e+00],
        [4.4115e-06, 1.0000e+00],
        [9.0591e-10, 1.0000e+00],
        [9.9541e-01, 4.5940e-03],
        [8.0581e-09, 1.0000e+00],
        [2.1576e-08, 1.0000e+00],
        [2.6054e-10, 1.0000e+00],
        [7.1134e-01, 2.8866e-01],
        [2.6441e-09, 1.0000e+00],
        [5.7980e-05, 9.9994e-01],
        [9.9999e-01, 1.2841e-05],
        [2.6426e-06, 1.0000e+00],
        [2.4913e-04, 9.9975e-01],
        [4.2956e-05, 9.9996e-01],
        [4.9704e-04, 9.9950e-01],
        [2.8796e-05, 9.9997e-01],
        [1.3144e-06, 1.0000e+00],
        [1.4527e-05, 9.9999e-01],
        [7.0552e-08, 1.0000e+00],
        [9.7547e-09, 1.0000e+00],
        [5.5978e-08, 1.0000e+00],
        [1.3305e-05, 9.9999e-01],
        [2.9107e-09, 1.0000e+00],
        [1.9333e-11, 1.0000e+00],
        [2.0231e-05, 9.9998e-01],
        [3.9023e-10, 1.0000e+00],
        [2.2114e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2801e-04, 9.9987e-01],
        [6.7680e-07, 1.0000e+00],
        [1.3822e-08, 1.0000e+00],
        [1.6848e-07, 1.0000e+00],
        [2.5245e-11, 1.0000e+00],
        [3.1417e-10, 1.0000e+00],
        [3.8478e-05, 9.9996e-01],
        [1.7564e-05, 9.9998e-01],
        [4.5072e-09, 1.0000e+00],
        [5.0732e-05, 9.9995e-01],
        [8.2003e-05, 9.9992e-01],
        [2.6704e-12, 1.0000e+00],
        [4.5384e-11, 1.0000e+00],
        [3.2168e-05, 9.9997e-01],
        [4.1793e-04, 9.9958e-01],
        [3.2499e-09, 1.0000e+00],
        [4.9687e-09, 1.0000e+00],
        [3.5969e-09, 1.0000e+00],
        [9.9881e-01, 1.1942e-03],
        [9.9989e-01, 1.0795e-04],
        [1.9055e-06, 1.0000e+00],
        [2.3367e-02, 9.7663e-01],
        [3.0726e-06, 1.0000e+00],
        [9.9996e-01, 4.3556e-05],
        [4.9916e-11, 1.0000e+00],
        [9.3859e-08, 1.0000e+00],
        [7.3110e-05, 9.9993e-01],
        [1.4060e-07, 1.0000e+00],
        [2.2567e-08, 1.0000e+00],
        [9.9986e-01, 1.3607e-04],
        [3.3645e-07, 1.0000e+00],
        [9.9994e-01, 5.8080e-05],
        [2.9804e-10, 1.0000e+00],
        [6.3322e-13, 1.0000e+00],
        [3.2336e-07, 1.0000e+00],
        [6.5127e-06, 9.9999e-01],
        [9.9621e-01, 3.7933e-03],
        [9.2957e-06, 9.9999e-01],
        [3.2522e-12, 1.0000e+00],
        [9.9988e-01, 1.1719e-04],
        [9.7313e-06, 9.9999e-01],
        [9.8363e-01, 1.6369e-02],
        [2.2822e-07, 1.0000e+00],
        [3.6382e-09, 1.0000e+00],
        [1.2541e-09, 1.0000e+00],
        [9.9990e-01, 1.0253e-04],
        [9.9989e-01, 1.1377e-04],
        [3.2252e-09, 1.0000e+00],
        [4.4537e-05, 9.9996e-01],
        [9.7508e-12, 1.0000e+00],
        [7.4757e-08, 1.0000e+00],
        [9.9806e-01, 1.9419e-03],
        [6.2943e-06, 9.9999e-01],
        [2.2954e-08, 1.0000e+00],
        [9.9926e-09, 1.0000e+00],
        [6.2936e-08, 1.0000e+00],
        [8.1435e-01, 1.8565e-01],
        [9.9978e-01, 2.1735e-04],
        [4.1297e-07, 1.0000e+00],
        [5.2047e-10, 1.0000e+00],
        [1.6609e-09, 1.0000e+00],
        [1.2207e-06, 1.0000e+00],
        [9.8182e-10, 1.0000e+00],
        [3.0199e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9850e-01, 1.5044e-03],
        [2.0983e-08, 1.0000e+00],
        [9.9995e-01, 4.8160e-05],
        [1.1325e-05, 9.9999e-01],
        [1.1653e-08, 1.0000e+00],
        [1.1293e-08, 1.0000e+00],
        [9.9974e-01, 2.5595e-04],
        [4.4366e-05, 9.9996e-01],
        [8.0556e-10, 1.0000e+00],
        [9.9989e-01, 1.0739e-04],
        [3.3398e-08, 1.0000e+00],
        [4.6657e-12, 1.0000e+00],
        [1.6668e-02, 9.8333e-01],
        [5.5082e-10, 1.0000e+00],
        [9.9889e-01, 1.1120e-03],
        [9.9988e-01, 1.2051e-04],
        [9.9877e-01, 1.2269e-03],
        [8.0434e-05, 9.9992e-01],
        [9.9737e-01, 2.6264e-03],
        [2.8837e-06, 1.0000e+00],
        [2.9725e-08, 1.0000e+00],
        [8.6332e-01, 1.3668e-01],
        [3.7394e-08, 1.0000e+00],
        [6.9913e-06, 9.9999e-01],
        [1.0941e-09, 1.0000e+00],
        [1.0000e+00, 7.2495e-07],
        [9.9266e-01, 7.3421e-03],
        [9.9971e-01, 2.8522e-04],
        [2.4943e-03, 9.9751e-01],
        [1.2481e-07, 1.0000e+00],
        [1.5355e-09, 1.0000e+00],
        [6.5420e-05, 9.9993e-01],
        [1.9122e-05, 9.9998e-01],
        [3.6939e-03, 9.9631e-01],
        [9.9988e-01, 1.2314e-04],
        [5.8132e-07, 1.0000e+00],
        [8.8330e-09, 1.0000e+00],
        [2.8095e-07, 1.0000e+00],
        [9.8040e-01, 1.9599e-02],
        [5.0162e-07, 1.0000e+00],
        [9.9854e-01, 1.4584e-03],
        [9.7789e-06, 9.9999e-01],
        [4.2612e-04, 9.9957e-01],
        [1.7848e-10, 1.0000e+00],
        [1.5800e-07, 1.0000e+00],
        [7.6892e-07, 1.0000e+00],
        [8.9293e-10, 1.0000e+00],
        [9.6448e-06, 9.9999e-01],
        [9.9445e-01, 5.5481e-03],
        [4.5157e-09, 1.0000e+00],
        [7.5718e-10, 1.0000e+00],
        [2.2806e-07, 1.0000e+00],
        [1.1117e-09, 1.0000e+00],
        [4.3331e-10, 1.0000e+00],
        [1.2169e-02, 9.8783e-01],
        [9.9327e-01, 6.7320e-03],
        [3.7580e-08, 1.0000e+00],
        [1.8952e-08, 1.0000e+00],
        [1.0813e-08, 1.0000e+00],
        [1.7626e-07, 1.0000e+00],
        [4.2041e-07, 1.0000e+00],
        [2.7240e-05, 9.9997e-01],
        [1.7367e-04, 9.9983e-01],
        [8.6967e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9991e-01, 9.0364e-05],
        [9.9986e-01, 1.3653e-04],
        [1.5371e-04, 9.9985e-01],
        [4.4166e-13, 1.0000e+00],
        [8.5055e-05, 9.9991e-01],
        [9.9980e-01, 1.9590e-04],
        [3.6890e-11, 1.0000e+00],
        [1.8883e-08, 1.0000e+00],
        [5.3757e-09, 1.0000e+00],
        [3.7212e-04, 9.9963e-01],
        [9.9986e-01, 1.4030e-04],
        [1.0174e-03, 9.9898e-01],
        [4.1553e-07, 1.0000e+00],
        [9.9778e-01, 2.2227e-03],
        [7.1197e-09, 1.0000e+00],
        [3.7335e-07, 1.0000e+00],
        [1.2341e-10, 1.0000e+00],
        [2.3541e-06, 1.0000e+00],
        [3.7366e-01, 6.2634e-01],
        [9.9666e-01, 3.3358e-03],
        [9.9995e-01, 4.8426e-05],
        [1.0971e-03, 9.9890e-01],
        [1.8022e-06, 1.0000e+00],
        [3.2704e-05, 9.9997e-01],
        [6.6208e-08, 1.0000e+00],
        [8.9994e-08, 1.0000e+00],
        [7.4107e-07, 1.0000e+00],
        [4.5845e-10, 1.0000e+00],
        [9.9999e-01, 5.8322e-06],
        [1.7914e-08, 1.0000e+00],
        [8.4821e-08, 1.0000e+00],
        [9.3701e-12, 1.0000e+00],
        [3.0103e-10, 1.0000e+00],
        [6.0278e-09, 1.0000e+00],
        [9.9970e-01, 2.9896e-04],
        [1.5191e-11, 1.0000e+00],
        [5.9853e-05, 9.9994e-01],
        [6.1633e-11, 1.0000e+00],
        [3.1545e-04, 9.9968e-01],
        [4.8533e-06, 1.0000e+00],
        [1.3474e-07, 1.0000e+00],
        [9.9421e-01, 5.7881e-03],
        [8.4943e-08, 1.0000e+00],
        [1.1721e-08, 1.0000e+00],
        [9.9990e-01, 9.6690e-05],
        [3.7672e-08, 1.0000e+00],
        [3.3713e-08, 1.0000e+00],
        [9.9988e-01, 1.1794e-04],
        [3.4416e-09, 1.0000e+00],
        [2.3746e-05, 9.9998e-01],
        [9.9970e-01, 3.0337e-04],
        [3.3723e-09, 1.0000e+00],
        [6.2300e-08, 1.0000e+00],
        [9.9819e-01, 1.8095e-03],
        [1.0000e+00, 2.4217e-06],
        [2.3597e-05, 9.9998e-01],
        [1.4958e-09, 1.0000e+00],
        [6.2402e-05, 9.9994e-01],
        [1.3095e-10, 1.0000e+00],
        [1.1512e-10, 1.0000e+00],
        [8.1828e-11, 1.0000e+00],
        [3.3898e-03, 9.9661e-01],
        [3.0901e-12, 1.0000e+00],
        [9.4935e-03, 9.9051e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9935e-01, 6.5329e-04],
        [9.7977e-06, 9.9999e-01],
        [9.7573e-06, 9.9999e-01],
        [1.9443e-05, 9.9998e-01],
        [4.5865e-03, 9.9541e-01],
        [9.7027e-01, 2.9734e-02],
        [2.3827e-07, 1.0000e+00],
        [9.9808e-01, 1.9218e-03],
        [1.9066e-04, 9.9981e-01],
        [1.0105e-04, 9.9990e-01],
        [4.0918e-07, 1.0000e+00],
        [2.5321e-07, 1.0000e+00],
        [9.7661e-01, 2.3389e-02],
        [5.9267e-04, 9.9941e-01],
        [1.7670e-11, 1.0000e+00],
        [9.9986e-01, 1.3833e-04],
        [6.0408e-08, 1.0000e+00],
        [9.2905e-01, 7.0950e-02],
        [4.9866e-12, 1.0000e+00],
        [7.8217e-07, 1.0000e+00],
        [1.1624e-05, 9.9999e-01],
        [1.6489e-03, 9.9835e-01],
        [2.4571e-06, 1.0000e+00],
        [9.9996e-01, 3.9038e-05],
        [2.2064e-04, 9.9978e-01],
        [1.4747e-07, 1.0000e+00],
        [2.8327e-08, 1.0000e+00],
        [1.5889e-06, 1.0000e+00],
        [1.7423e-11, 1.0000e+00],
        [1.4340e-05, 9.9999e-01],
        [2.3444e-06, 1.0000e+00],
        [4.1664e-03, 9.9583e-01],
        [4.9473e-03, 9.9505e-01],
        [1.5648e-07, 1.0000e+00],
        [4.6919e-09, 1.0000e+00],
        [9.9998e-01, 1.8197e-05],
        [9.6402e-02, 9.0360e-01],
        [3.2635e-09, 1.0000e+00],
        [9.9983e-01, 1.6773e-04],
        [3.5577e-12, 1.0000e+00],
        [2.9729e-10, 1.0000e+00],
        [1.3471e-07, 1.0000e+00],
        [3.4201e-05, 9.9997e-01],
        [9.9837e-01, 1.6273e-03],
        [9.8424e-07, 1.0000e+00],
        [9.9979e-01, 2.1387e-04],
        [2.4431e-07, 1.0000e+00],
        [9.9998e-01, 1.7669e-05],
        [1.3902e-04, 9.9986e-01],
        [1.8004e-05, 9.9998e-01],
        [9.9980e-01, 2.0164e-04],
        [1.6255e-11, 1.0000e+00],
        [6.4874e-07, 1.0000e+00],
        [1.6849e-07, 1.0000e+00],
        [9.7184e-01, 2.8164e-02],
        [9.9953e-01, 4.7443e-04],
        [1.3820e-10, 1.0000e+00],
        [1.5096e-10, 1.0000e+00],
        [1.8681e-06, 1.0000e+00],
        [9.9965e-01, 3.5467e-04],
        [9.9996e-01, 3.5960e-05],
        [1.4337e-03, 9.9857e-01],
        [2.8787e-05, 9.9997e-01],
        [9.7720e-01, 2.2798e-02]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9991e-01, 8.9262e-05],
        [1.0492e-08, 1.0000e+00],
        [8.3603e-03, 9.9164e-01],
        [1.8371e-07, 1.0000e+00],
        [2.4326e-03, 9.9757e-01],
        [9.9984e-01, 1.5512e-04],
        [3.1994e-05, 9.9997e-01],
        [5.0523e-09, 1.0000e+00],
        [1.1606e-05, 9.9999e-01],
        [9.8616e-01, 1.3838e-02],
        [2.8362e-07, 1.0000e+00],
        [4.8597e-09, 1.0000e+00],
        [1.0006e-08, 1.0000e+00],
        [3.2861e-09, 1.0000e+00],
        [1.1377e-08, 1.0000e+00],
        [9.9989e-01, 1.1476e-04],
        [1.7873e-07, 1.0000e+00],
        [4.3685e-05, 9.9996e-01],
        [7.7420e-06, 9.9999e-01],
        [2.4929e-07, 1.0000e+00],
        [9.6032e-07, 1.0000e+00],
        [1.9095e-05, 9.9998e-01],
        [1.4493e-09, 1.0000e+00],
        [7.9189e-04, 9.9921e-01],
        [8.9883e-01, 1.0117e-01],
        [9.9980e-01, 2.0080e-04],
        [3.2856e-09, 1.0000e+00],
        [3.3615e-01, 6.6385e-01],
        [3.3932e-09, 1.0000e+00],
        [5.9458e-12, 1.0000e+00],
        [9.9995e-01, 4.8561e-05],
        [9.9992e-01, 8.2785e-05],
        [3.4117e-09, 1.0000e+00],
        [9.9992e-01, 8.3559e-05],
        [2.5678e-08, 1.0000e+00],
        [2.9297e-06, 1.0000e+00],
        [2.7259e-03, 9.9727e-01],
        [4.6876e-07, 1.0000e+00],
        [1.7269e-07, 1.0000e+00],
        [9.9985e-01, 1.5125e-04],
        [1.2574e-07, 1.0000e+00],
        [1.3712e-03, 9.9863e-01],
        [1.3197e-10, 1.0000e+00],
        [9.9956e-01, 4.4258e-04],
        [4.1787e-03, 9.9582e-01],
        [9.9999e-01, 6.2741e-06],
        [1.6104e-08, 1.0000e+00],
        [3.7561e-13, 1.0000e+00],
        [9.9868e-01, 1.3211e-03],
        [9.7130e-03, 9.9029e-01],
        [1.8383e-09, 1.0000e+00],
        [9.9985e-01, 1.5203e-04],
        [1.4976e-03, 9.9850e-01],
        [9.9994e-01, 5.7294e-05],
        [8.2941e-01, 1.7059e-01],
        [1.2830e-05, 9.9999e-01],
        [2.9702e-02, 9.7030e-01],
        [3.0472e-05, 9.9997e-01],
        [5.0079e-05, 9.9995e-01],
        [9.8634e-06, 9.9999e-01],
        [9.6480e-10, 1.0000e+00],
        [2.6841e-06, 1.0000e+00],
        [1.6709e-08, 1.0000e+00],
        [5.9640e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.5038e-08, 1.0000e+00],
        [9.4322e-01, 5.6776e-02],
        [4.8978e-11, 1.0000e+00],
        [9.9988e-01, 1.2065e-04],
        [9.1051e-01, 8.9490e-02],
        [8.2935e-08, 1.0000e+00],
        [9.9840e-01, 1.5951e-03],
        [8.0101e-06, 9.9999e-01],
        [1.2830e-09, 1.0000e+00],
        [1.8662e-10, 1.0000e+00],
        [4.4700e-08, 1.0000e+00],
        [1.3541e-02, 9.8646e-01],
        [7.1048e-05, 9.9993e-01],
        [1.3794e-07, 1.0000e+00],
        [4.5157e-05, 9.9995e-01],
        [1.5835e-05, 9.9998e-01],
        [9.9992e-01, 7.8009e-05],
        [7.8574e-09, 1.0000e+00],
        [9.9987e-01, 1.3443e-04],
        [3.0592e-04, 9.9969e-01],
        [3.9654e-10, 1.0000e+00],
        [1.9206e-11, 1.0000e+00],
        [1.3121e-07, 1.0000e+00],
        [3.6903e-10, 1.0000e+00],
        [2.8878e-09, 1.0000e+00],
        [8.3313e-08, 1.0000e+00],
        [7.4872e-07, 1.0000e+00],
        [1.4437e-05, 9.9999e-01],
        [5.7206e-07, 1.0000e+00],
        [8.6152e-10, 1.0000e+00],
        [1.5037e-12, 1.0000e+00],
        [5.8604e-06, 9.9999e-01],
        [9.9994e-01, 6.3649e-05],
        [1.9010e-07, 1.0000e+00],
        [1.6505e-08, 1.0000e+00],
        [6.2568e-05, 9.9994e-01],
        [8.5398e-02, 9.1460e-01],
        [9.9958e-01, 4.1539e-04],
        [9.9996e-01, 3.5284e-05],
        [1.2259e-10, 1.0000e+00],
        [4.5065e-08, 1.0000e+00],
        [2.9651e-05, 9.9997e-01],
        [7.3251e-07, 1.0000e+00],
        [1.3942e-09, 1.0000e+00],
        [4.0693e-07, 1.0000e+00],
        [2.6934e-08, 1.0000e+00],
        [2.3971e-05, 9.9998e-01],
        [8.6991e-11, 1.0000e+00],
        [1.3255e-12, 1.0000e+00],
        [1.4400e-06, 1.0000e+00],
        [7.8015e-04, 9.9922e-01],
        [4.6515e-06, 1.0000e+00],
        [1.0000e+00, 2.3792e-06],
        [3.9083e-06, 1.0000e+00],
        [9.9996e-01, 3.9295e-05],
        [2.3122e-06, 1.0000e+00],
        [8.2966e-06, 9.9999e-01],
        [5.9183e-11, 1.0000e+00],
        [6.5859e-09, 1.0000e+00],
        [1.7724e-08, 1.0000e+00],
        [9.9978e-01, 2.1606e-04],
        [9.9982e-01, 1.8038e-04],
        [9.9996e-01, 4.2546e-05],
        [9.9995e-01, 4.5423e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[2.4095e-02, 9.7591e-01],
        [2.0758e-04, 9.9979e-01],
        [7.9842e-02, 9.2016e-01],
        [7.4891e-08, 1.0000e+00],
        [9.9978e-01, 2.1796e-04],
        [1.2555e-05, 9.9999e-01],
        [1.0918e-07, 1.0000e+00],
        [2.8746e-06, 1.0000e+00],
        [2.8207e-11, 1.0000e+00],
        [6.6620e-01, 3.3380e-01],
        [5.4730e-04, 9.9945e-01],
        [1.3807e-04, 9.9986e-01],
        [4.1691e-04, 9.9958e-01],
        [9.9995e-01, 4.6968e-05],
        [2.4987e-06, 1.0000e+00],
        [1.4196e-06, 1.0000e+00],
        [2.1821e-09, 1.0000e+00],
        [2.1079e-06, 1.0000e+00],
        [2.3426e-01, 7.6574e-01],
        [1.1403e-05, 9.9999e-01],
        [1.1804e-05, 9.9999e-01],
        [6.6692e-07, 1.0000e+00],
        [1.3554e-08, 1.0000e+00],
        [4.0697e-08, 1.0000e+00],
        [4.6176e-06, 1.0000e+00],
        [3.2653e-03, 9.9673e-01],
        [9.9996e-01, 3.5377e-05],
        [3.5293e-05, 9.9996e-01],
        [6.9204e-16, 1.0000e+00],
        [9.7904e-05, 9.9990e-01],
        [1.4453e-09, 1.0000e+00],
        [9.9998e-01, 1.9856e-05],
        [9.9993e-01, 6.6095e-05],
        [3.8798e-07, 1.0000e+00],
        [1.0000e+00, 2.5498e-06],
        [8.6963e-10, 1.0000e+00],
        [1.6994e-06, 1.0000e+00],
        [9.9986e-01, 1.3717e-04],
        [3.6990e-05, 9.9996e-01],
        [7.1207e-05, 9.9993e-01],
        [6.7172e-07, 1.0000e+00],
        [9.9997e-01, 3.1977e-05],
        [1.5169e-07, 1.0000e+00],
        [1.0000e+00, 4.3327e-07],
        [1.7390e-02, 9.8261e-01],
        [7.7840e-02, 9.2216e-01],
        [4.6734e-10, 1.0000e+00],
        [4.3591e-06, 1.0000e+00],
        [1.0814e-02, 9.8919e-01],
        [9.9973e-01, 2.7400e-04],
        [9.3512e-01, 6.4880e-02],
        [3.4146e-08, 1.0000e+00],
        [5.4023e-02, 9.4598e-01],
        [9.9990e-01, 1.0347e-04],
        [1.4382e-02, 9.8562e-01],
        [2.0414e-07, 1.0000e+00],
        [2.8325e-06, 1.0000e+00],
        [1.0946e-10, 1.0000e+00],
        [3.6856e-06, 1.0000e+00],
        [9.3883e-09, 1.0000e+00],
        [2.9134e-07, 1.0000e+00],
        [9.0261e-07, 1.0000e+00],
        [6.3614e-07, 1.0000e+00],
        [1.7333e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1676e-05, 9.9999e-01],
        [9.9983e-01, 1.6579e-04],
        [2.6171e-07, 1.0000e+00],
        [7.9185e-05, 9.9992e-01],
        [1.9342e-08, 1.0000e+00],
        [5.7441e-04, 9.9943e-01],
        [2.8143e-04, 9.9972e-01],
        [9.9638e-07, 1.0000e+00],
        [5.9175e-05, 9.9994e-01],
        [3.9569e-10, 1.0000e+00],
        [9.9987e-01, 1.3321e-04],
        [3.5098e-04, 9.9965e-01],
        [7.1105e-08, 1.0000e+00],
        [2.4788e-05, 9.9998e-01],
        [9.9867e-01, 1.3350e-03],
        [9.0386e-04, 9.9910e-01],
        [1.4934e-04, 9.9985e-01],
        [9.9987e-01, 1.3105e-04],
        [1.0743e-02, 9.8926e-01],
        [2.0019e-03, 9.9800e-01],
        [9.9999e-01, 6.8595e-06],
        [2.2825e-07, 1.0000e+00],
        [6.7331e-10, 1.0000e+00],
        [9.9611e-01, 3.8918e-03],
        [2.9049e-05, 9.9997e-01],
        [6.4523e-08, 1.0000e+00],
        [1.5955e-06, 1.0000e+00],
        [1.0947e-06, 1.0000e+00],
        [1.4382e-09, 1.0000e+00],
        [1.6249e-13, 1.0000e+00],
        [2.2572e-03, 9.9774e-01],
        [9.9969e-01, 3.0836e-04],
        [1.3561e-08, 1.0000e+00],
        [5.9104e-02, 9.4090e-01],
        [1.4350e-08, 1.0000e+00],
        [3.1462e-09, 1.0000e+00],
        [2.4755e-08, 1.0000e+00],
        [8.8694e-07, 1.0000e+00],
        [2.8153e-08, 1.0000e+00],
        [9.9976e-01, 2.3919e-04],
        [3.4974e-02, 9.6503e-01],
        [1.5539e-05, 9.9998e-01],
        [9.9676e-01, 3.2424e-03],
        [2.1720e-06, 1.0000e+00],
        [1.2431e-05, 9.9999e-01],
        [2.1726e-06, 1.0000e+00],
        [1.7012e-09, 1.0000e+00],
        [9.9640e-01, 3.6011e-03],
        [2.0365e-01, 7.9635e-01],
        [9.9986e-01, 1.4078e-04],
        [6.1651e-02, 9.3835e-01],
        [1.2892e-03, 9.9871e-01],
        [8.7768e-01, 1.2232e-01],
        [8.5652e-09, 1.0000e+00],
        [2.7938e-07, 1.0000e+00],
        [9.9918e-01, 8.1876e-04],
        [2.9850e-07, 1.0000e+00],
        [2.5617e-01, 7.4383e-01],
        [9.9987e-01, 1.2834e-04],
        [2.4498e-06, 1.0000e+00],
        [9.7139e-01, 2.8611e-02],
        [9.9994e-01, 5.8329e-05],
        [9.9994e-01, 6.0669e-05],
        [3.0920e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[7.9921e-08, 1.0000e+00],
        [1.0490e-03, 9.9895e-01],
        [9.9996e-01, 3.6484e-05],
        [1.7859e-15, 1.0000e+00],
        [7.9747e-01, 2.0253e-01],
        [1.0615e-10, 1.0000e+00],
        [1.9950e-08, 1.0000e+00],
        [9.9989e-01, 1.1207e-04],
        [1.2674e-07, 1.0000e+00],
        [2.0202e-06, 1.0000e+00],
        [2.3454e-09, 1.0000e+00],
        [6.3365e-07, 1.0000e+00],
        [3.5568e-08, 1.0000e+00],
        [2.4211e-09, 1.0000e+00],
        [1.7441e-08, 1.0000e+00],
        [1.2882e-06, 1.0000e+00],
        [1.9188e-05, 9.9998e-01],
        [8.9255e-10, 1.0000e+00],
        [1.2794e-06, 1.0000e+00],
        [9.9995e-01, 4.8518e-05],
        [1.5802e-06, 1.0000e+00],
        [1.8108e-01, 8.1892e-01],
        [5.6811e-06, 9.9999e-01],
        [8.3736e-03, 9.9163e-01],
        [9.9995e-01, 5.0364e-05],
        [6.1449e-09, 1.0000e+00],
        [9.9988e-01, 1.1577e-04],
        [8.2192e-07, 1.0000e+00],
        [6.0619e-10, 1.0000e+00],
        [2.8459e-10, 1.0000e+00],
        [3.6132e-10, 1.0000e+00],
        [1.1781e-10, 1.0000e+00],
        [9.9996e-01, 3.9404e-05],
        [6.2657e-09, 1.0000e+00],
        [9.9974e-01, 2.6141e-04],
        [1.5323e-07, 1.0000e+00],
        [9.9986e-01, 1.4293e-04],
        [1.7644e-06, 1.0000e+00],
        [9.9999e-01, 1.2126e-05],
        [2.2178e-06, 1.0000e+00],
        [1.3716e-06, 1.0000e+00],
        [5.1674e-11, 1.0000e+00],
        [2.7526e-08, 1.0000e+00],
        [1.6345e-04, 9.9984e-01],
        [2.0001e-06, 1.0000e+00],
        [8.3507e-04, 9.9916e-01],
        [1.5608e-10, 1.0000e+00],
        [3.2587e-08, 1.0000e+00],
        [2.1411e-06, 1.0000e+00],
        [1.7966e-07, 1.0000e+00],
        [7.8016e-10, 1.0000e+00],
        [7.1404e-09, 1.0000e+00],
        [1.0834e-04, 9.9989e-01],
        [9.9983e-01, 1.7224e-04],
        [1.1266e-03, 9.9887e-01],
        [1.9291e-09, 1.0000e+00],
        [7.7220e-07, 1.0000e+00],
        [6.0337e-05, 9.9994e-01],
        [9.9990e-01, 9.7077e-05],
        [2.2474e-03, 9.9775e-01],
        [5.5237e-10, 1.0000e+00],
        [9.9970e-01, 3.0046e-04],
        [2.5361e-10, 1.0000e+00],
        [1.2031e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.6977e-09, 1.0000e+00],
        [9.9995e-01, 4.9565e-05],
        [8.8094e-09, 1.0000e+00],
        [1.5378e-08, 1.0000e+00],
        [9.9990e-01, 9.7034e-05],
        [1.5647e-07, 1.0000e+00],
        [1.4797e-04, 9.9985e-01],
        [5.6336e-02, 9.4366e-01],
        [2.7420e-05, 9.9997e-01],
        [3.7434e-08, 1.0000e+00],
        [2.4457e-08, 1.0000e+00],
        [1.4429e-08, 1.0000e+00],
        [9.9996e-01, 4.0804e-05],
        [2.6849e-02, 9.7315e-01],
        [1.6056e-07, 1.0000e+00],
        [2.6716e-10, 1.0000e+00],
        [9.9987e-01, 1.3327e-04],
        [8.8728e-04, 9.9911e-01],
        [6.0191e-03, 9.9398e-01],
        [4.0793e-10, 1.0000e+00],
        [1.4145e-06, 1.0000e+00],
        [2.8189e-07, 1.0000e+00],
        [1.8067e-07, 1.0000e+00],
        [2.4585e-04, 9.9975e-01],
        [9.9986e-01, 1.4089e-04],
        [9.9995e-01, 5.2374e-05],
        [5.7408e-07, 1.0000e+00],
        [2.3000e-11, 1.0000e+00],
        [9.9992e-01, 7.7467e-05],
        [1.0690e-08, 1.0000e+00],
        [3.8530e-08, 1.0000e+00],
        [1.1188e-03, 9.9888e-01],
        [8.7903e-13, 1.0000e+00],
        [1.0659e-06, 1.0000e+00],
        [3.4996e-07, 1.0000e+00],
        [4.2057e-10, 1.0000e+00],
        [4.1463e-07, 1.0000e+00],
        [1.0115e-04, 9.9990e-01],
        [9.9894e-01, 1.0596e-03],
        [9.6562e-05, 9.9990e-01],
        [4.5530e-09, 1.0000e+00],
        [4.9332e-03, 9.9507e-01],
        [1.2387e-04, 9.9988e-01],
        [1.6077e-08, 1.0000e+00],
        [9.9982e-01, 1.8422e-04],
        [9.9985e-01, 1.5360e-04],
        [4.0915e-02, 9.5909e-01],
        [3.4177e-04, 9.9966e-01],
        [1.3109e-11, 1.0000e+00],
        [9.9992e-01, 7.6725e-05],
        [9.9988e-01, 1.1718e-04],
        [2.9855e-09, 1.0000e+00],
        [2.1903e-09, 1.0000e+00],
        [3.3957e-08, 1.0000e+00],
        [2.8659e-06, 1.0000e+00],
        [2.7796e-05, 9.9997e-01],
        [1.4578e-09, 1.0000e+00],
        [9.9996e-01, 3.7095e-05],
        [1.9475e-10, 1.0000e+00],
        [1.5018e-06, 1.0000e+00],
        [9.4941e-11, 1.0000e+00],
        [9.9979e-01, 2.0631e-04],
        [8.9202e-05, 9.9991e-01],
        [2.7172e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.2124e-05, 9.9991e-01],
        [1.1708e-04, 9.9988e-01],
        [4.1566e-09, 1.0000e+00],
        [1.2790e-06, 1.0000e+00],
        [9.9995e-01, 4.9785e-05],
        [9.9999e-01, 8.6269e-06],
        [1.4245e-04, 9.9986e-01],
        [8.5957e-01, 1.4043e-01],
        [2.0687e-03, 9.9793e-01],
        [9.9529e-01, 4.7117e-03],
        [5.8666e-13, 1.0000e+00],
        [3.9281e-04, 9.9961e-01],
        [3.2642e-09, 1.0000e+00],
        [1.3149e-08, 1.0000e+00],
        [9.9982e-01, 1.7506e-04],
        [3.6333e-13, 1.0000e+00],
        [1.0000e+00, 3.9774e-07],
        [2.3164e-07, 1.0000e+00],
        [2.1882e-07, 1.0000e+00],
        [1.7125e-06, 1.0000e+00],
        [7.1792e-05, 9.9993e-01],
        [6.9266e-07, 1.0000e+00],
        [3.8462e-06, 1.0000e+00],
        [9.9994e-01, 6.3349e-05],
        [4.8239e-06, 1.0000e+00],
        [2.4054e-08, 1.0000e+00],
        [9.6472e-06, 9.9999e-01],
        [1.0169e-05, 9.9999e-01],
        [5.4922e-05, 9.9995e-01],
        [8.6248e-07, 1.0000e+00],
        [7.1729e-09, 1.0000e+00],
        [2.1525e-08, 1.0000e+00],
        [2.9248e-10, 1.0000e+00],
        [4.6327e-11, 1.0000e+00],
        [4.4960e-05, 9.9996e-01],
        [1.2607e-11, 1.0000e+00],
        [5.9730e-01, 4.0270e-01],
        [1.7818e-04, 9.9982e-01],
        [1.3367e-10, 1.0000e+00],
        [1.1989e-05, 9.9999e-01],
        [9.9230e-01, 7.7008e-03],
        [7.9847e-08, 1.0000e+00],
        [2.6739e-06, 1.0000e+00],
        [9.9990e-01, 1.0136e-04],
        [8.1024e-05, 9.9992e-01],
        [2.3960e-13, 1.0000e+00],
        [6.9774e-12, 1.0000e+00],
        [1.2137e-05, 9.9999e-01],
        [5.3618e-10, 1.0000e+00],
        [9.9994e-01, 6.4741e-05],
        [9.9602e-01, 3.9824e-03],
        [9.9980e-01, 2.0358e-04],
        [7.2597e-05, 9.9993e-01],
        [7.8366e-01, 2.1634e-01],
        [2.1812e-04, 9.9978e-01],
        [7.5481e-08, 1.0000e+00],
        [5.4153e-05, 9.9995e-01],
        [5.3111e-06, 9.9999e-01],
        [5.6728e-06, 9.9999e-01],
        [1.9907e-07, 1.0000e+00],
        [7.7276e-07, 1.0000e+00],
        [1.6703e-05, 9.9998e-01],
        [9.9963e-01, 3.6506e-04],
        [1.2210e-11, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9992e-01, 8.4345e-05],
        [5.3126e-06, 9.9999e-01],
        [3.8113e-11, 1.0000e+00],
        [7.4312e-09, 1.0000e+00],
        [7.6802e-07, 1.0000e+00],
        [1.1350e-10, 1.0000e+00],
        [9.7895e-01, 2.1054e-02],
        [1.1782e-04, 9.9988e-01],
        [2.1052e-08, 1.0000e+00],
        [4.6581e-07, 1.0000e+00],
        [2.2620e-07, 1.0000e+00],
        [9.9985e-01, 1.4894e-04],
        [9.9988e-01, 1.1737e-04],
        [9.9988e-01, 1.2216e-04],
        [9.9986e-01, 1.3607e-04],
        [9.9881e-01, 1.1902e-03],
        [9.9990e-01, 9.9326e-05],
        [5.9049e-06, 9.9999e-01],
        [9.9969e-01, 3.1057e-04],
        [1.8216e-07, 1.0000e+00],
        [1.6278e-07, 1.0000e+00],
        [9.9995e-01, 5.1719e-05],
        [2.7133e-06, 1.0000e+00],
        [6.7335e-07, 1.0000e+00],
        [4.2278e-06, 1.0000e+00],
        [1.7513e-10, 1.0000e+00],
        [2.1340e-05, 9.9998e-01],
        [4.3772e-05, 9.9996e-01],
        [1.6204e-10, 1.0000e+00],
        [9.9998e-01, 2.4938e-05],
        [7.8188e-08, 1.0000e+00],
        [5.3068e-06, 9.9999e-01],
        [1.2902e-07, 1.0000e+00],
        [3.2214e-08, 1.0000e+00],
        [9.9967e-01, 3.2644e-04],
        [3.8064e-09, 1.0000e+00],
        [2.9229e-03, 9.9708e-01],
        [2.1048e-04, 9.9979e-01],
        [2.3334e-09, 1.0000e+00],
        [1.0835e-03, 9.9892e-01],
        [8.3848e-04, 9.9916e-01],
        [9.4866e-01, 5.1336e-02],
        [3.2637e-07, 1.0000e+00],
        [9.9981e-01, 1.8800e-04],
        [9.9994e-01, 5.7900e-05],
        [9.8250e-01, 1.7501e-02],
        [4.7415e-05, 9.9995e-01],
        [1.8098e-09, 1.0000e+00],
        [2.9695e-11, 1.0000e+00],
        [5.6013e-10, 1.0000e+00],
        [2.0719e-05, 9.9998e-01],
        [9.9988e-01, 1.1664e-04],
        [5.6627e-10, 1.0000e+00],
        [2.9067e-12, 1.0000e+00],
        [9.9980e-01, 1.9921e-04],
        [1.2326e-05, 9.9999e-01],
        [2.7918e-07, 1.0000e+00],
        [4.3582e-11, 1.0000e+00],
        [8.9662e-11, 1.0000e+00],
        [2.4212e-11, 1.0000e+00],
        [8.7000e-07, 1.0000e+00],
        [7.2524e-10, 1.0000e+00],
        [3.1456e-07, 1.0000e+00],
        [5.0685e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.2370e-08, 1.0000e+00],
        [4.2354e-12, 1.0000e+00],
        [9.4424e-01, 5.5764e-02],
        [5.2010e-07, 1.0000e+00],
        [2.7106e-07, 1.0000e+00],
        [6.9261e-04, 9.9931e-01],
        [1.4153e-08, 1.0000e+00],
        [3.4768e-12, 1.0000e+00],
        [9.9977e-01, 2.3185e-04],
        [4.2520e-09, 1.0000e+00],
        [1.4492e-08, 1.0000e+00],
        [4.0328e-06, 1.0000e+00],
        [3.2763e-08, 1.0000e+00],
        [9.7628e-08, 1.0000e+00],
        [4.0187e-10, 1.0000e+00],
        [9.9968e-01, 3.1795e-04],
        [4.1161e-06, 1.0000e+00],
        [9.9991e-01, 9.0522e-05],
        [9.9988e-01, 1.2368e-04],
        [9.9999e-01, 6.4102e-06],
        [1.0000e+00, 1.8760e-07],
        [4.9402e-11, 1.0000e+00],
        [9.9986e-01, 1.4339e-04],
        [2.7426e-07, 1.0000e+00],
        [8.2631e-01, 1.7369e-01],
        [3.3146e-07, 1.0000e+00],
        [9.9944e-01, 5.6477e-04],
        [3.8728e-11, 1.0000e+00],
        [9.2690e-04, 9.9907e-01],
        [3.9866e-02, 9.6013e-01],
        [9.8111e-01, 1.8892e-02],
        [6.9479e-08, 1.0000e+00],
        [2.4290e-10, 1.0000e+00],
        [1.6121e-10, 1.0000e+00],
        [9.9172e-01, 8.2811e-03],
        [9.9958e-01, 4.1845e-04],
        [9.9986e-01, 1.4400e-04],
        [9.9980e-01, 2.0283e-04],
        [9.9990e-01, 1.0071e-04],
        [4.2842e-08, 1.0000e+00],
        [1.4245e-03, 9.9858e-01],
        [9.1133e-08, 1.0000e+00],
        [1.0688e-06, 1.0000e+00],
        [9.9990e-01, 9.6999e-05],
        [9.9998e-01, 1.9597e-05],
        [1.0203e-11, 1.0000e+00],
        [3.7911e-12, 1.0000e+00],
        [1.8303e-10, 1.0000e+00],
        [1.4295e-12, 1.0000e+00],
        [9.9972e-01, 2.7796e-04],
        [1.7191e-02, 9.8281e-01],
        [9.9994e-01, 5.6447e-05],
        [9.9996e-01, 4.0223e-05],
        [2.2679e-04, 9.9977e-01],
        [1.6942e-07, 1.0000e+00],
        [2.7106e-07, 1.0000e+00],
        [2.7903e-09, 1.0000e+00],
        [7.7618e-04, 9.9922e-01],
        [1.3869e-06, 1.0000e+00],
        [2.0058e-05, 9.9998e-01],
        [3.2720e-03, 9.9673e-01],
        [1.4992e-06, 1.0000e+00],
        [9.9997e-01, 2.5904e-05],
        [1.2988e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.5970e-09, 1.0000e+00],
        [6.5909e-10, 1.0000e+00],
        [2.8738e-07, 1.0000e+00],
        [7.5155e-06, 9.9999e-01],
        [6.2617e-11, 1.0000e+00],
        [4.5265e-08, 1.0000e+00],
        [6.2899e-05, 9.9994e-01],
        [6.0324e-09, 1.0000e+00],
        [1.0507e-04, 9.9989e-01],
        [6.5856e-10, 1.0000e+00],
        [1.4662e-07, 1.0000e+00],
        [9.9420e-13, 1.0000e+00],
        [3.8029e-06, 1.0000e+00],
        [8.8530e-01, 1.1470e-01],
        [1.2816e-06, 1.0000e+00],
        [3.6855e-10, 1.0000e+00],
        [6.6572e-10, 1.0000e+00],
        [1.1403e-03, 9.9886e-01],
        [2.8905e-08, 1.0000e+00],
        [1.5897e-07, 1.0000e+00],
        [1.8056e-09, 1.0000e+00],
        [2.7332e-06, 1.0000e+00],
        [9.9961e-01, 3.8975e-04],
        [9.8819e-01, 1.1814e-02],
        [9.8294e-01, 1.7057e-02],
        [9.9957e-01, 4.3477e-04],
        [3.4958e-10, 1.0000e+00],
        [9.9994e-01, 5.8420e-05],
        [5.3298e-12, 1.0000e+00],
        [2.0595e-03, 9.9794e-01],
        [9.9993e-01, 6.7273e-05],
        [3.3515e-10, 1.0000e+00],
        [4.4568e-05, 9.9996e-01],
        [1.6211e-11, 1.0000e+00],
        [9.8893e-01, 1.1075e-02],
        [1.3130e-06, 1.0000e+00],
        [1.2957e-07, 1.0000e+00],
        [9.9542e-01, 4.5812e-03],
        [9.9987e-01, 1.3039e-04],
        [3.5212e-09, 1.0000e+00],
        [9.2110e-01, 7.8902e-02],
        [3.8878e-05, 9.9996e-01],
        [9.9984e-01, 1.6235e-04],
        [9.9987e-01, 1.3260e-04],
        [5.2314e-08, 1.0000e+00],
        [9.9981e-01, 1.8645e-04],
        [9.9982e-01, 1.7589e-04],
        [9.9982e-01, 1.7813e-04],
        [9.9992e-01, 8.1630e-05],
        [1.2963e-11, 1.0000e+00],
        [1.2550e-03, 9.9874e-01],
        [1.1083e-03, 9.9889e-01],
        [9.9932e-01, 6.7704e-04],
        [1.0871e-08, 1.0000e+00],
        [5.1192e-09, 1.0000e+00],
        [2.6301e-06, 1.0000e+00],
        [4.3047e-04, 9.9957e-01],
        [9.8693e-01, 1.3074e-02],
        [5.0379e-05, 9.9995e-01],
        [4.0696e-09, 1.0000e+00],
        [9.9931e-01, 6.9374e-04],
        [9.9995e-01, 4.6362e-05],
        [9.0178e-14, 1.0000e+00],
        [3.5214e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0164e-05, 9.9999e-01],
        [9.9988e-01, 1.1948e-04],
        [9.9987e-01, 1.3046e-04],
        [9.6547e-09, 1.0000e+00],
        [3.1984e-08, 1.0000e+00],
        [7.8681e-07, 1.0000e+00],
        [7.6919e-12, 1.0000e+00],
        [2.1076e-05, 9.9998e-01],
        [5.4874e-07, 1.0000e+00],
        [1.0777e-02, 9.8922e-01],
        [1.1137e-09, 1.0000e+00],
        [9.9952e-01, 4.8133e-04],
        [1.0969e-03, 9.9890e-01],
        [9.9776e-01, 2.2425e-03],
        [1.5889e-08, 1.0000e+00],
        [1.7645e-02, 9.8235e-01],
        [1.3317e-08, 1.0000e+00],
        [8.2093e-04, 9.9918e-01],
        [9.9992e-01, 8.1125e-05],
        [1.4596e-08, 1.0000e+00],
        [9.9775e-07, 1.0000e+00],
        [9.9781e-01, 2.1923e-03],
        [6.9906e-09, 1.0000e+00],
        [3.9251e-13, 1.0000e+00],
        [2.1980e-05, 9.9998e-01],
        [7.0087e-01, 2.9913e-01],
        [9.1432e-06, 9.9999e-01],
        [1.7428e-07, 1.0000e+00],
        [5.8811e-02, 9.4119e-01],
        [3.5724e-11, 1.0000e+00],
        [9.9965e-01, 3.4542e-04],
        [3.0364e-04, 9.9970e-01],
        [4.2585e-10, 1.0000e+00],
        [4.2089e-09, 1.0000e+00],
        [2.7305e-03, 9.9727e-01],
        [9.9997e-01, 2.7900e-05],
        [9.9996e-01, 3.5804e-05],
        [2.3766e-02, 9.7623e-01],
        [1.9224e-08, 1.0000e+00],
        [9.8429e-01, 1.5707e-02],
        [1.0261e-08, 1.0000e+00],
        [1.5759e-06, 1.0000e+00],
        [1.0341e-10, 1.0000e+00],
        [3.6639e-07, 1.0000e+00],
        [3.8893e-01, 6.1107e-01],
        [9.3856e-01, 6.1443e-02],
        [4.9703e-08, 1.0000e+00],
        [2.6493e-11, 1.0000e+00],
        [1.7141e-07, 1.0000e+00],
        [9.9471e-01, 5.2921e-03],
        [5.4802e-08, 1.0000e+00],
        [1.3226e-07, 1.0000e+00],
        [2.8077e-06, 1.0000e+00],
        [1.5216e-07, 1.0000e+00],
        [9.6716e-01, 3.2836e-02],
        [1.3023e-06, 1.0000e+00],
        [3.9681e-06, 1.0000e+00],
        [6.8951e-13, 1.0000e+00],
        [9.9126e-09, 1.0000e+00],
        [9.9988e-01, 1.1743e-04],
        [3.0721e-06, 1.0000e+00],
        [9.6471e-04, 9.9904e-01],
        [1.4933e-02, 9.8507e-01],
        [9.9602e-01, 3.9840e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8466e-01, 1.5341e-02],
        [3.9666e-05, 9.9996e-01],
        [9.9998e-01, 1.6300e-05],
        [1.3213e-07, 1.0000e+00],
        [4.4462e-09, 1.0000e+00],
        [2.2046e-08, 1.0000e+00],
        [9.9995e-01, 5.0733e-05],
        [9.8087e-08, 1.0000e+00],
        [1.1462e-03, 9.9885e-01],
        [9.7064e-01, 2.9364e-02],
        [6.8402e-10, 1.0000e+00],
        [2.9267e-01, 7.0733e-01],
        [1.4489e-10, 1.0000e+00],
        [1.0157e-03, 9.9898e-01],
        [6.6380e-09, 1.0000e+00],
        [9.9993e-01, 7.3598e-05],
        [1.3678e-03, 9.9863e-01],
        [4.9752e-07, 1.0000e+00],
        [2.8705e-03, 9.9713e-01],
        [3.1817e-09, 1.0000e+00],
        [2.7148e-04, 9.9973e-01],
        [2.0430e-07, 1.0000e+00],
        [4.0837e-12, 1.0000e+00],
        [9.9981e-01, 1.8769e-04],
        [1.1642e-04, 9.9988e-01],
        [6.5554e-05, 9.9993e-01],
        [3.3566e-13, 1.0000e+00],
        [7.6922e-07, 1.0000e+00],
        [2.1390e-04, 9.9979e-01],
        [9.9983e-01, 1.7174e-04],
        [9.9995e-01, 5.4788e-05],
        [3.3048e-04, 9.9967e-01],
        [4.5494e-08, 1.0000e+00],
        [4.8090e-03, 9.9519e-01],
        [9.9928e-01, 7.1940e-04],
        [3.5796e-04, 9.9964e-01],
        [3.4230e-07, 1.0000e+00],
        [1.5580e-10, 1.0000e+00],
        [9.6008e-05, 9.9990e-01],
        [2.0669e-11, 1.0000e+00],
        [9.9991e-01, 9.2903e-05],
        [2.3612e-07, 1.0000e+00],
        [1.2291e-09, 1.0000e+00],
        [5.5901e-06, 9.9999e-01],
        [1.9972e-11, 1.0000e+00],
        [7.1500e-10, 1.0000e+00],
        [7.0183e-13, 1.0000e+00],
        [1.8797e-07, 1.0000e+00],
        [1.5484e-12, 1.0000e+00],
        [9.9990e-01, 1.0203e-04],
        [9.9976e-01, 2.3701e-04],
        [2.7048e-07, 1.0000e+00],
        [9.5820e-01, 4.1804e-02],
        [9.9988e-01, 1.1649e-04],
        [5.4972e-01, 4.5028e-01],
        [2.0109e-08, 1.0000e+00],
        [9.8105e-05, 9.9990e-01],
        [1.1257e-05, 9.9999e-01],
        [2.8345e-06, 1.0000e+00],
        [9.9992e-01, 7.7283e-05],
        [1.5931e-05, 9.9998e-01],
        [9.9939e-01, 6.0735e-04],
        [1.9480e-08, 1.0000e+00],
        [1.2885e-04, 9.9987e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4324e-08, 1.0000e+00],
        [1.3407e-08, 1.0000e+00],
        [9.9987e-01, 1.3440e-04],
        [3.4446e-08, 1.0000e+00],
        [9.9981e-01, 1.8536e-04],
        [9.9957e-01, 4.2952e-04],
        [9.9994e-01, 6.4667e-05],
        [3.3917e-13, 1.0000e+00],
        [9.9989e-01, 1.1331e-04],
        [6.1749e-07, 1.0000e+00],
        [9.9974e-01, 2.5687e-04],
        [7.8027e-11, 1.0000e+00],
        [3.3626e-06, 1.0000e+00],
        [1.2607e-03, 9.9874e-01],
        [3.6041e-06, 1.0000e+00],
        [2.9540e-08, 1.0000e+00],
        [2.7716e-07, 1.0000e+00],
        [1.9591e-04, 9.9980e-01],
        [3.1443e-05, 9.9997e-01],
        [6.1431e-05, 9.9994e-01],
        [2.0684e-10, 1.0000e+00],
        [9.9994e-01, 6.3664e-05],
        [2.7788e-11, 1.0000e+00],
        [1.2906e-07, 1.0000e+00],
        [1.4880e-12, 1.0000e+00],
        [4.0876e-10, 1.0000e+00],
        [2.1152e-03, 9.9788e-01],
        [6.8011e-07, 1.0000e+00],
        [1.8909e-09, 1.0000e+00],
        [9.9993e-01, 7.1150e-05],
        [9.9730e-01, 2.6951e-03],
        [9.9950e-01, 5.0445e-04],
        [3.2413e-11, 1.0000e+00],
        [9.9935e-01, 6.5449e-04],
        [9.9630e-01, 3.7033e-03],
        [7.4039e-06, 9.9999e-01],
        [6.8589e-05, 9.9993e-01],
        [4.7637e-12, 1.0000e+00],
        [1.6101e-06, 1.0000e+00],
        [3.0340e-09, 1.0000e+00],
        [5.6264e-04, 9.9944e-01],
        [2.2901e-03, 9.9771e-01],
        [5.6002e-07, 1.0000e+00],
        [7.1877e-07, 1.0000e+00],
        [2.1407e-04, 9.9979e-01],
        [9.9987e-01, 1.2769e-04],
        [9.9890e-01, 1.0989e-03],
        [9.9397e-01, 6.0311e-03],
        [2.0561e-08, 1.0000e+00],
        [9.9987e-01, 1.3458e-04],
        [1.9430e-06, 1.0000e+00],
        [2.8004e-05, 9.9997e-01],
        [9.9999e-01, 1.3032e-05],
        [9.6708e-06, 9.9999e-01],
        [9.9979e-01, 2.1457e-04],
        [1.0701e-07, 1.0000e+00],
        [9.9997e-01, 3.4127e-05],
        [5.5998e-07, 1.0000e+00],
        [1.1346e-09, 1.0000e+00],
        [4.8993e-07, 1.0000e+00],
        [2.0748e-04, 9.9979e-01],
        [1.5625e-09, 1.0000e+00],
        [4.4622e-09, 1.0000e+00],
        [1.6348e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.3863e-03, 9.9561e-01],
        [9.9989e-01, 1.0739e-04],
        [1.4486e-06, 1.0000e+00],
        [1.5880e-08, 1.0000e+00],
        [1.4207e-11, 1.0000e+00],
        [3.5594e-05, 9.9996e-01],
        [6.2506e-07, 1.0000e+00],
        [9.8953e-01, 1.0468e-02],
        [2.9615e-01, 7.0385e-01],
        [1.3661e-03, 9.9863e-01],
        [1.0000e+00, 3.2203e-07],
        [1.4566e-12, 1.0000e+00],
        [2.7616e-07, 1.0000e+00],
        [2.8554e-04, 9.9971e-01],
        [6.9772e-04, 9.9930e-01],
        [9.9996e-01, 3.5650e-05],
        [7.6159e-08, 1.0000e+00],
        [1.5614e-09, 1.0000e+00],
        [2.5046e-05, 9.9997e-01],
        [1.5575e-06, 1.0000e+00],
        [2.1741e-04, 9.9978e-01],
        [2.1990e-05, 9.9998e-01],
        [7.1617e-06, 9.9999e-01],
        [1.5649e-10, 1.0000e+00],
        [5.8849e-04, 9.9941e-01],
        [4.6552e-07, 1.0000e+00],
        [7.1225e-06, 9.9999e-01],
        [6.2935e-01, 3.7065e-01],
        [1.0000e+00, 2.3605e-06],
        [5.8974e-06, 9.9999e-01],
        [5.9933e-08, 1.0000e+00],
        [9.9919e-01, 8.0510e-04],
        [1.5518e-05, 9.9998e-01],
        [1.0737e-11, 1.0000e+00],
        [7.9328e-05, 9.9992e-01],
        [5.3668e-06, 9.9999e-01],
        [2.5995e-05, 9.9997e-01],
        [6.7891e-03, 9.9321e-01],
        [9.9996e-01, 3.8764e-05],
        [2.3530e-06, 1.0000e+00],
        [2.0179e-08, 1.0000e+00],
        [1.4952e-04, 9.9985e-01],
        [8.7097e-05, 9.9991e-01],
        [9.9985e-01, 1.4673e-04],
        [9.9982e-01, 1.7636e-04],
        [8.4916e-11, 1.0000e+00],
        [7.5698e-06, 9.9999e-01],
        [8.4673e-10, 1.0000e+00],
        [2.2247e-08, 1.0000e+00],
        [9.7909e-01, 2.0913e-02],
        [9.9793e-01, 2.0669e-03],
        [1.1331e-06, 1.0000e+00],
        [4.4804e-07, 1.0000e+00],
        [2.0376e-04, 9.9980e-01],
        [1.6516e-09, 1.0000e+00],
        [9.9296e-01, 7.0363e-03],
        [6.5602e-08, 1.0000e+00],
        [9.9979e-01, 2.1171e-04],
        [9.9992e-01, 8.3617e-05],
        [7.3702e-12, 1.0000e+00],
        [9.9711e-01, 2.8915e-03],
        [9.9963e-01, 3.6668e-04],
        [6.7661e-10, 1.0000e+00],
        [2.8283e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.2742e-01, 2.7258e-01],
        [8.8300e-11, 1.0000e+00],
        [7.3389e-09, 1.0000e+00],
        [4.2731e-12, 1.0000e+00],
        [9.9255e-01, 7.4513e-03],
        [1.6510e-05, 9.9998e-01],
        [3.5851e-05, 9.9996e-01],
        [7.9569e-01, 2.0431e-01],
        [9.9995e-01, 5.0199e-05],
        [6.4512e-10, 1.0000e+00],
        [7.9716e-03, 9.9203e-01],
        [2.5805e-05, 9.9997e-01],
        [2.0250e-03, 9.9798e-01],
        [4.9974e-06, 9.9999e-01],
        [2.5431e-04, 9.9975e-01],
        [1.4337e-07, 1.0000e+00],
        [1.5681e-08, 1.0000e+00],
        [5.6084e-09, 1.0000e+00],
        [3.3039e-01, 6.6961e-01],
        [1.2831e-08, 1.0000e+00],
        [1.5977e-02, 9.8402e-01],
        [9.9973e-01, 2.6597e-04],
        [9.9988e-01, 1.2255e-04],
        [1.7314e-09, 1.0000e+00],
        [2.0760e-02, 9.7924e-01],
        [4.3197e-10, 1.0000e+00],
        [7.2410e-08, 1.0000e+00],
        [3.2708e-07, 1.0000e+00],
        [1.0193e-01, 8.9807e-01],
        [2.6429e-10, 1.0000e+00],
        [9.2566e-11, 1.0000e+00],
        [9.9995e-01, 4.7324e-05],
        [1.2091e-08, 1.0000e+00],
        [6.4881e-07, 1.0000e+00],
        [1.8903e-02, 9.8110e-01],
        [7.8048e-12, 1.0000e+00],
        [2.3036e-08, 1.0000e+00],
        [1.1439e-03, 9.9886e-01],
        [5.4624e-07, 1.0000e+00],
        [4.1961e-10, 1.0000e+00],
        [3.7202e-08, 1.0000e+00],
        [9.9999e-01, 4.9643e-06],
        [9.9981e-01, 1.9416e-04],
        [4.1961e-10, 1.0000e+00],
        [8.6177e-08, 1.0000e+00],
        [5.3019e-07, 1.0000e+00],
        [6.7080e-03, 9.9329e-01],
        [9.9995e-01, 5.4759e-05],
        [2.0758e-10, 1.0000e+00],
        [1.3980e-06, 1.0000e+00],
        [5.7494e-07, 1.0000e+00],
        [2.1484e-07, 1.0000e+00],
        [2.1713e-11, 1.0000e+00],
        [1.5403e-09, 1.0000e+00],
        [3.0004e-04, 9.9970e-01],
        [8.0710e-05, 9.9992e-01],
        [9.9993e-01, 6.6756e-05],
        [1.3618e-11, 1.0000e+00],
        [6.7847e-05, 9.9993e-01],
        [2.0360e-06, 1.0000e+00],
        [8.9631e-08, 1.0000e+00],
        [9.9963e-01, 3.7438e-04],
        [5.1390e-07, 1.0000e+00],
        [4.1004e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5250e-06, 1.0000e+00],
        [6.5144e-07, 1.0000e+00],
        [9.6445e-06, 9.9999e-01],
        [9.9995e-01, 5.3154e-05],
        [3.1324e-03, 9.9687e-01],
        [1.6101e-06, 1.0000e+00],
        [4.2590e-11, 1.0000e+00],
        [2.6536e-09, 1.0000e+00],
        [1.8612e-07, 1.0000e+00],
        [2.6705e-05, 9.9997e-01],
        [9.9839e-01, 1.6142e-03],
        [9.9985e-01, 1.4975e-04],
        [2.3199e-06, 1.0000e+00],
        [8.9874e-08, 1.0000e+00],
        [1.4197e-04, 9.9986e-01],
        [1.1154e-08, 1.0000e+00],
        [9.9989e-01, 1.0813e-04],
        [4.0722e-06, 1.0000e+00],
        [9.9967e-01, 3.3259e-04],
        [5.5471e-06, 9.9999e-01],
        [9.9968e-01, 3.1553e-04],
        [9.9733e-01, 2.6706e-03],
        [3.4297e-08, 1.0000e+00],
        [2.5857e-05, 9.9997e-01],
        [1.3264e-08, 1.0000e+00],
        [9.9951e-01, 4.8960e-04],
        [4.3962e-05, 9.9996e-01],
        [9.9991e-01, 8.6784e-05],
        [2.7355e-02, 9.7265e-01],
        [2.0031e-10, 1.0000e+00],
        [6.6068e-08, 1.0000e+00],
        [9.9802e-01, 1.9792e-03],
        [6.8467e-06, 9.9999e-01],
        [1.4436e-07, 1.0000e+00],
        [5.9467e-07, 1.0000e+00],
        [9.9994e-01, 5.6201e-05],
        [9.9986e-01, 1.3886e-04],
        [2.1033e-01, 7.8967e-01],
        [5.0050e-08, 1.0000e+00],
        [1.3803e-06, 1.0000e+00],
        [6.4103e-07, 1.0000e+00],
        [7.1533e-08, 1.0000e+00],
        [7.8796e-09, 1.0000e+00],
        [4.7820e-09, 1.0000e+00],
        [1.4020e-02, 9.8598e-01],
        [5.1712e-06, 9.9999e-01],
        [9.9834e-01, 1.6559e-03],
        [5.0363e-06, 9.9999e-01],
        [9.9998e-01, 1.9365e-05],
        [3.3088e-08, 1.0000e+00],
        [6.9407e-09, 1.0000e+00],
        [7.8021e-03, 9.9220e-01],
        [9.9990e-01, 9.6489e-05],
        [1.0278e-03, 9.9897e-01],
        [4.1672e-03, 9.9583e-01],
        [9.2580e-05, 9.9991e-01],
        [1.3911e-07, 1.0000e+00],
        [3.5987e-04, 9.9964e-01],
        [9.9829e-01, 1.7113e-03],
        [9.3732e-08, 1.0000e+00],
        [9.9916e-01, 8.4045e-04],
        [9.9983e-01, 1.7068e-04],
        [9.9999e-01, 5.7376e-06],
        [1.0201e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.4003e-09, 1.0000e+00],
        [1.4328e-01, 8.5672e-01],
        [3.4775e-06, 1.0000e+00],
        [1.9227e-09, 1.0000e+00],
        [6.2950e-07, 1.0000e+00],
        [9.9999e-01, 8.4617e-06],
        [1.1478e-04, 9.9989e-01],
        [9.9971e-01, 2.8745e-04],
        [2.5333e-03, 9.9747e-01],
        [2.1628e-08, 1.0000e+00],
        [2.4514e-07, 1.0000e+00],
        [3.4151e-04, 9.9966e-01],
        [2.3396e-07, 1.0000e+00],
        [1.3455e-03, 9.9865e-01],
        [6.3067e-07, 1.0000e+00],
        [2.1293e-04, 9.9979e-01],
        [9.9391e-01, 6.0898e-03],
        [5.0700e-04, 9.9949e-01],
        [9.9994e-01, 5.5523e-05],
        [9.9987e-01, 1.2945e-04],
        [1.3231e-06, 1.0000e+00],
        [6.3778e-05, 9.9994e-01],
        [2.8727e-08, 1.0000e+00],
        [5.6176e-02, 9.4382e-01],
        [9.8365e-02, 9.0163e-01],
        [8.0359e-01, 1.9641e-01],
        [9.9976e-01, 2.3681e-04],
        [9.9935e-01, 6.4950e-04],
        [5.8147e-05, 9.9994e-01],
        [1.3249e-08, 1.0000e+00],
        [1.9586e-06, 1.0000e+00],
        [5.3657e-08, 1.0000e+00],
        [1.3572e-11, 1.0000e+00],
        [9.9997e-01, 2.8975e-05],
        [1.9565e-03, 9.9804e-01],
        [9.9997e-01, 3.1714e-05],
        [7.0084e-03, 9.9299e-01],
        [1.0755e-06, 1.0000e+00],
        [2.5633e-07, 1.0000e+00],
        [2.0684e-05, 9.9998e-01],
        [5.5193e-06, 9.9999e-01],
        [4.9490e-06, 9.9999e-01],
        [9.9990e-01, 9.8130e-05],
        [1.0011e-02, 9.8999e-01],
        [9.9976e-01, 2.4092e-04],
        [9.9939e-01, 6.1277e-04],
        [9.3307e-08, 1.0000e+00],
        [6.8209e-07, 1.0000e+00],
        [2.8602e-06, 1.0000e+00],
        [6.2092e-08, 1.0000e+00],
        [1.1366e-07, 1.0000e+00],
        [9.9988e-01, 1.2209e-04],
        [2.0621e-05, 9.9998e-01],
        [1.7611e-08, 1.0000e+00],
        [9.9972e-01, 2.7937e-04],
        [1.7840e-05, 9.9998e-01],
        [2.8916e-07, 1.0000e+00],
        [9.9996e-01, 3.5129e-05],
        [9.9992e-01, 8.1800e-05],
        [2.0503e-10, 1.0000e+00],
        [9.4352e-07, 1.0000e+00],
        [9.9978e-01, 2.1903e-04],
        [1.3628e-09, 1.0000e+00],
        [9.9977e-01, 2.2607e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[2.9775e-08, 1.0000e+00],
        [1.6437e-05, 9.9998e-01],
        [1.9503e-03, 9.9805e-01],
        [8.2301e-09, 1.0000e+00],
        [4.4487e-03, 9.9555e-01],
        [4.2590e-03, 9.9574e-01],
        [4.0623e-04, 9.9959e-01],
        [4.0512e-07, 1.0000e+00],
        [9.6435e-08, 1.0000e+00],
        [1.3788e-10, 1.0000e+00],
        [3.5055e-08, 1.0000e+00],
        [3.4801e-07, 1.0000e+00],
        [2.0966e-04, 9.9979e-01],
        [9.9970e-01, 3.0320e-04],
        [9.9998e-01, 2.3936e-05],
        [5.9316e-02, 9.4068e-01],
        [6.4290e-05, 9.9994e-01],
        [4.8501e-10, 1.0000e+00],
        [9.9961e-01, 3.8514e-04],
        [8.4598e-11, 1.0000e+00],
        [3.1454e-12, 1.0000e+00],
        [4.8689e-05, 9.9995e-01],
        [2.7088e-06, 1.0000e+00],
        [2.6464e-10, 1.0000e+00],
        [9.9988e-01, 1.2286e-04],
        [9.9981e-01, 1.8688e-04],
        [1.1928e-12, 1.0000e+00],
        [3.1790e-03, 9.9682e-01],
        [8.4298e-04, 9.9916e-01],
        [1.9987e-09, 1.0000e+00],
        [5.5676e-09, 1.0000e+00],
        [1.3688e-07, 1.0000e+00],
        [9.9970e-01, 2.9769e-04],
        [8.8043e-10, 1.0000e+00],
        [9.5662e-01, 4.3381e-02],
        [9.9998e-01, 1.5276e-05],
        [1.2361e-05, 9.9999e-01],
        [6.7579e-07, 1.0000e+00],
        [9.6658e-12, 1.0000e+00],
        [1.4436e-05, 9.9999e-01],
        [1.2247e-07, 1.0000e+00],
        [2.6020e-09, 1.0000e+00],
        [9.1392e-06, 9.9999e-01],
        [4.5077e-11, 1.0000e+00],
        [2.0962e-01, 7.9038e-01],
        [1.2988e-01, 8.7012e-01],
        [3.0103e-10, 1.0000e+00],
        [6.5917e-04, 9.9934e-01],
        [9.9973e-01, 2.6674e-04],
        [1.8032e-04, 9.9982e-01],
        [5.2998e-04, 9.9947e-01],
        [1.4454e-10, 1.0000e+00],
        [7.7158e-04, 9.9923e-01],
        [1.3612e-03, 9.9864e-01],
        [5.5753e-09, 1.0000e+00],
        [1.2369e-06, 1.0000e+00],
        [8.7231e-10, 1.0000e+00],
        [9.9992e-01, 8.4718e-05],
        [5.6567e-04, 9.9943e-01],
        [1.6456e-11, 1.0000e+00],
        [9.9923e-01, 7.7332e-04],
        [9.7675e-01, 2.3248e-02],
        [1.9280e-05, 9.9998e-01],
        [6.7385e-05, 9.9993e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8820e-09, 1.0000e+00],
        [3.5320e-05, 9.9996e-01],
        [6.8385e-06, 9.9999e-01],
        [9.9941e-01, 5.9034e-04],
        [1.1665e-01, 8.8335e-01],
        [9.3870e-01, 6.1302e-02],
        [2.8826e-09, 1.0000e+00],
        [9.0049e-06, 9.9999e-01],
        [9.9993e-01, 6.5129e-05],
        [9.9981e-01, 1.8735e-04],
        [1.0157e-11, 1.0000e+00],
        [9.9868e-01, 1.3218e-03],
        [1.9088e-06, 1.0000e+00],
        [1.4295e-08, 1.0000e+00],
        [2.9306e-09, 1.0000e+00],
        [1.4543e-05, 9.9999e-01],
        [2.5973e-08, 1.0000e+00],
        [1.1665e-06, 1.0000e+00],
        [9.9819e-01, 1.8130e-03],
        [9.9958e-01, 4.1549e-04],
        [1.0317e-07, 1.0000e+00],
        [5.8775e-06, 9.9999e-01],
        [7.0608e-06, 9.9999e-01],
        [2.3243e-02, 9.7676e-01],
        [9.7964e-01, 2.0357e-02],
        [1.0930e-05, 9.9999e-01],
        [6.9478e-07, 1.0000e+00],
        [7.2231e-08, 1.0000e+00],
        [3.4175e-10, 1.0000e+00],
        [2.7356e-06, 1.0000e+00],
        [4.1367e-05, 9.9996e-01],
        [1.5511e-07, 1.0000e+00],
        [9.9995e-01, 4.9245e-05],
        [2.9860e-03, 9.9701e-01],
        [9.9979e-01, 2.1439e-04],
        [1.0489e-08, 1.0000e+00],
        [9.9906e-01, 9.4412e-04],
        [1.8226e-07, 1.0000e+00],
        [3.4385e-06, 1.0000e+00],
        [1.9515e-04, 9.9980e-01],
        [6.2140e-01, 3.7860e-01],
        [1.3393e-07, 1.0000e+00],
        [9.9921e-01, 7.9355e-04],
        [2.4428e-04, 9.9976e-01],
        [9.9995e-01, 5.4792e-05],
        [1.1150e-06, 1.0000e+00],
        [9.9992e-01, 7.5233e-05],
        [9.9986e-01, 1.3527e-04],
        [7.3158e-09, 1.0000e+00],
        [3.7993e-06, 1.0000e+00],
        [8.7044e-05, 9.9991e-01],
        [8.7283e-09, 1.0000e+00],
        [9.9995e-01, 5.2541e-05],
        [1.7055e-06, 1.0000e+00],
        [2.1676e-06, 1.0000e+00],
        [8.4503e-01, 1.5497e-01],
        [9.9992e-01, 7.8502e-05],
        [9.9990e-01, 9.5244e-05],
        [9.9989e-01, 1.1455e-04],
        [9.9985e-01, 1.5162e-04],
        [9.7454e-01, 2.5457e-02],
        [9.9570e-01, 4.3024e-03],
        [9.9980e-01, 2.0064e-04],
        [1.5310e-03, 9.9847e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2387e-08, 1.0000e+00],
        [1.7545e-05, 9.9998e-01],
        [3.2636e-05, 9.9997e-01],
        [6.2736e-01, 3.7264e-01],
        [9.9989e-01, 1.0956e-04],
        [2.8346e-04, 9.9972e-01],
        [7.3360e-03, 9.9266e-01],
        [8.0106e-06, 9.9999e-01],
        [5.8295e-08, 1.0000e+00],
        [5.6970e-09, 1.0000e+00],
        [4.9011e-09, 1.0000e+00],
        [4.9437e-02, 9.5056e-01],
        [2.6566e-03, 9.9734e-01],
        [2.2106e-14, 1.0000e+00],
        [8.0559e-03, 9.9194e-01],
        [9.9997e-01, 3.4008e-05],
        [2.2906e-02, 9.7709e-01],
        [1.7132e-01, 8.2868e-01],
        [1.2327e-09, 1.0000e+00],
        [5.2811e-06, 9.9999e-01],
        [2.0653e-05, 9.9998e-01],
        [9.9991e-01, 9.1836e-05],
        [9.9991e-01, 9.3881e-05],
        [3.9453e-03, 9.9605e-01],
        [9.9995e-01, 5.5005e-05],
        [7.5680e-10, 1.0000e+00],
        [9.9992e-01, 7.7550e-05],
        [2.1914e-07, 1.0000e+00],
        [5.4125e-01, 4.5875e-01],
        [9.2495e-03, 9.9075e-01],
        [9.9984e-01, 1.6239e-04],
        [9.9991e-01, 9.3034e-05],
        [2.6181e-09, 1.0000e+00],
        [1.2939e-04, 9.9987e-01],
        [1.1434e-03, 9.9886e-01],
        [8.8703e-08, 1.0000e+00],
        [7.2808e-06, 9.9999e-01],
        [9.9991e-01, 9.3412e-05],
        [6.4986e-04, 9.9935e-01],
        [9.9989e-01, 1.0931e-04],
        [9.9992e-01, 8.0116e-05],
        [5.9731e-11, 1.0000e+00],
        [1.1121e-06, 1.0000e+00],
        [3.2715e-06, 1.0000e+00],
        [9.9983e-01, 1.6568e-04],
        [2.5675e-06, 1.0000e+00],
        [9.9995e-01, 4.9389e-05],
        [1.3196e-02, 9.8680e-01],
        [2.9371e-06, 1.0000e+00],
        [9.9998e-01, 2.2103e-05],
        [6.5344e-10, 1.0000e+00],
        [1.2546e-10, 1.0000e+00],
        [1.4700e-05, 9.9999e-01],
        [2.9192e-07, 1.0000e+00],
        [9.9985e-01, 1.5364e-04],
        [6.0126e-08, 1.0000e+00],
        [9.1325e-08, 1.0000e+00],
        [3.8197e-07, 1.0000e+00],
        [9.9912e-01, 8.8089e-04],
        [9.6516e-01, 3.4836e-02],
        [9.9993e-01, 7.3082e-05],
        [1.3428e-01, 8.6572e-01],
        [2.4447e-08, 1.0000e+00],
        [9.9997e-01, 2.8191e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.8752e-09, 1.0000e+00],
        [9.9999e-01, 1.4033e-05],
        [5.1018e-05, 9.9995e-01],
        [9.9993e-01, 6.5934e-05],
        [9.8196e-09, 1.0000e+00],
        [9.9985e-01, 1.4706e-04],
        [8.5947e-10, 1.0000e+00],
        [3.6880e-08, 1.0000e+00],
        [2.0511e-05, 9.9998e-01],
        [4.1691e-07, 1.0000e+00],
        [1.5324e-05, 9.9998e-01],
        [4.2705e-09, 1.0000e+00],
        [3.2262e-08, 1.0000e+00],
        [4.3954e-06, 1.0000e+00],
        [7.6861e-08, 1.0000e+00],
        [8.5966e-03, 9.9140e-01],
        [9.0206e-07, 1.0000e+00],
        [1.8428e-10, 1.0000e+00],
        [9.9976e-01, 2.4076e-04],
        [1.1752e-07, 1.0000e+00],
        [4.1383e-09, 1.0000e+00],
        [9.9904e-01, 9.6236e-04],
        [9.9992e-01, 8.1134e-05],
        [1.0000e+00, 1.3188e-07],
        [9.9963e-01, 3.7147e-04],
        [2.8007e-07, 1.0000e+00],
        [1.1261e-01, 8.8739e-01],
        [5.5942e-10, 1.0000e+00],
        [9.9989e-01, 1.1156e-04],
        [2.2468e-03, 9.9775e-01],
        [6.9811e-09, 1.0000e+00],
        [1.8924e-03, 9.9811e-01],
        [9.9987e-01, 1.3111e-04],
        [2.8445e-05, 9.9997e-01],
        [1.1998e-06, 1.0000e+00],
        [7.0212e-07, 1.0000e+00],
        [2.7334e-02, 9.7267e-01],
        [3.7161e-06, 1.0000e+00],
        [6.5287e-06, 9.9999e-01],
        [1.3546e-06, 1.0000e+00],
        [7.4523e-09, 1.0000e+00],
        [9.7010e-06, 9.9999e-01],
        [3.7411e-10, 1.0000e+00],
        [3.7591e-10, 1.0000e+00],
        [1.2024e-05, 9.9999e-01],
        [9.9991e-01, 9.2537e-05],
        [7.6499e-04, 9.9924e-01],
        [7.8848e-01, 2.1152e-01],
        [1.4118e-06, 1.0000e+00],
        [2.5515e-03, 9.9745e-01],
        [5.4838e-08, 1.0000e+00],
        [6.9184e-03, 9.9308e-01],
        [6.1922e-01, 3.8078e-01],
        [1.8930e-03, 9.9811e-01],
        [1.6542e-07, 1.0000e+00],
        [8.5070e-05, 9.9991e-01],
        [9.9990e-01, 1.0008e-04],
        [2.6400e-07, 1.0000e+00],
        [1.0000e+00, 3.4465e-06],
        [1.1781e-08, 1.0000e+00],
        [9.9858e-01, 1.4249e-03],
        [9.9897e-01, 1.0295e-03],
        [3.5010e-09, 1.0000e+00],
        [2.6818e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.2628e-07, 1.0000e+00],
        [7.0671e-09, 1.0000e+00],
        [9.5055e-02, 9.0495e-01],
        [1.7288e-08, 1.0000e+00],
        [2.2539e-09, 1.0000e+00],
        [7.5350e-02, 9.2465e-01],
        [9.8616e-01, 1.3839e-02],
        [4.8338e-05, 9.9995e-01],
        [4.3116e-08, 1.0000e+00],
        [1.5162e-05, 9.9998e-01],
        [1.7807e-07, 1.0000e+00],
        [7.2091e-01, 2.7909e-01],
        [1.1520e-03, 9.9885e-01],
        [3.4327e-04, 9.9966e-01],
        [9.9978e-01, 2.2070e-04],
        [9.6662e-08, 1.0000e+00],
        [4.7641e-03, 9.9524e-01],
        [1.2662e-08, 1.0000e+00],
        [4.5017e-07, 1.0000e+00],
        [1.2320e-05, 9.9999e-01],
        [1.1570e-07, 1.0000e+00],
        [1.4814e-04, 9.9985e-01],
        [1.1447e-06, 1.0000e+00],
        [3.5544e-04, 9.9964e-01],
        [2.2721e-05, 9.9998e-01],
        [9.9601e-01, 3.9938e-03],
        [3.5233e-03, 9.9648e-01],
        [9.9999e-01, 6.8931e-06],
        [7.1261e-09, 1.0000e+00],
        [4.7864e-10, 1.0000e+00],
        [9.5490e-06, 9.9999e-01],
        [2.9491e-04, 9.9971e-01],
        [2.2981e-05, 9.9998e-01],
        [6.9252e-06, 9.9999e-01],
        [6.2299e-04, 9.9938e-01],
        [9.9979e-01, 2.1258e-04],
        [3.1892e-07, 1.0000e+00],
        [2.9698e-04, 9.9970e-01],
        [9.9980e-01, 1.9975e-04],
        [2.1096e-04, 9.9979e-01],
        [9.9996e-01, 3.9609e-05],
        [9.9999e-01, 1.1561e-05],
        [9.9987e-01, 1.2771e-04],
        [2.9218e-07, 1.0000e+00],
        [3.4154e-04, 9.9966e-01],
        [9.4244e-10, 1.0000e+00],
        [5.5774e-05, 9.9994e-01],
        [1.1581e-03, 9.9884e-01],
        [9.9983e-01, 1.6938e-04],
        [9.9987e-01, 1.2618e-04],
        [9.9953e-01, 4.7444e-04],
        [5.8890e-08, 1.0000e+00],
        [7.1729e-08, 1.0000e+00],
        [1.1076e-07, 1.0000e+00],
        [1.1561e-03, 9.9884e-01],
        [3.3707e-08, 1.0000e+00],
        [9.9995e-01, 4.5850e-05],
        [4.2354e-07, 1.0000e+00],
        [9.8962e-01, 1.0377e-02],
        [9.9974e-01, 2.6152e-04],
        [7.8885e-07, 1.0000e+00],
        [9.9996e-01, 4.4513e-05],
        [1.9400e-07, 1.0000e+00],
        [2.3493e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.3073e-06, 1.0000e+00],
        [3.0154e-04, 9.9970e-01],
        [5.3457e-02, 9.4654e-01],
        [4.8661e-02, 9.5134e-01],
        [3.6294e-06, 1.0000e+00],
        [2.6953e-04, 9.9973e-01],
        [9.9200e-01, 8.0019e-03],
        [1.7035e-06, 1.0000e+00],
        [6.6280e-06, 9.9999e-01],
        [3.0810e-04, 9.9969e-01],
        [5.6373e-07, 1.0000e+00],
        [4.1200e-04, 9.9959e-01],
        [9.6494e-01, 3.5065e-02],
        [3.6259e-08, 1.0000e+00],
        [5.0095e-04, 9.9950e-01],
        [2.5369e-10, 1.0000e+00],
        [9.9995e-01, 4.6627e-05],
        [1.4546e-03, 9.9855e-01],
        [9.9992e-01, 8.0190e-05],
        [4.0753e-07, 1.0000e+00],
        [9.2568e-01, 7.4318e-02],
        [3.5801e-09, 1.0000e+00],
        [1.8000e-03, 9.9820e-01],
        [1.3015e-09, 1.0000e+00],
        [7.3297e-11, 1.0000e+00],
        [9.9990e-01, 1.0259e-04],
        [8.9282e-07, 1.0000e+00],
        [9.9996e-01, 3.9401e-05],
        [9.9997e-01, 3.2675e-05],
        [2.4753e-03, 9.9752e-01],
        [2.9651e-05, 9.9997e-01],
        [9.9984e-01, 1.5655e-04],
        [2.5213e-09, 1.0000e+00],
        [7.7568e-01, 2.2432e-01],
        [9.9996e-01, 3.8996e-05],
        [4.8419e-01, 5.1581e-01],
        [2.9670e-04, 9.9970e-01],
        [5.4100e-04, 9.9946e-01],
        [9.9965e-01, 3.4769e-04],
        [3.5516e-05, 9.9996e-01],
        [9.9957e-01, 4.3064e-04],
        [1.3407e-13, 1.0000e+00],
        [9.9990e-01, 9.6886e-05],
        [1.6699e-05, 9.9998e-01],
        [1.8431e-05, 9.9998e-01],
        [9.9989e-01, 1.1143e-04],
        [3.2188e-08, 1.0000e+00],
        [8.5515e-10, 1.0000e+00],
        [1.7104e-03, 9.9829e-01],
        [1.8526e-05, 9.9998e-01],
        [9.9996e-01, 3.5257e-05],
        [3.5077e-04, 9.9965e-01],
        [6.8089e-12, 1.0000e+00],
        [9.9999e-01, 9.1004e-06],
        [4.1283e-03, 9.9587e-01],
        [5.0426e-07, 1.0000e+00],
        [1.6445e-04, 9.9984e-01],
        [6.5454e-06, 9.9999e-01],
        [8.9632e-11, 1.0000e+00],
        [6.7324e-07, 1.0000e+00],
        [3.6623e-06, 1.0000e+00],
        [3.2242e-05, 9.9997e-01],
        [1.0331e-05, 9.9999e-01],
        [4.4307e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.1083e-06, 1.0000e+00],
        [9.7305e-01, 2.6950e-02],
        [8.7689e-05, 9.9991e-01],
        [1.2389e-09, 1.0000e+00],
        [1.8425e-04, 9.9982e-01],
        [3.0553e-04, 9.9969e-01],
        [3.6823e-05, 9.9996e-01],
        [6.2749e-01, 3.7251e-01],
        [9.9994e-01, 5.6628e-05],
        [1.0367e-07, 1.0000e+00],
        [2.6452e-07, 1.0000e+00],
        [3.2756e-10, 1.0000e+00],
        [1.8905e-07, 1.0000e+00],
        [6.3506e-07, 1.0000e+00],
        [7.9289e-09, 1.0000e+00],
        [3.5807e-08, 1.0000e+00],
        [7.3080e-10, 1.0000e+00],
        [7.5626e-07, 1.0000e+00],
        [3.8430e-06, 1.0000e+00],
        [1.5775e-04, 9.9984e-01],
        [1.0000e+00, 1.7988e-06],
        [2.4379e-05, 9.9998e-01],
        [3.2733e-02, 9.6727e-01],
        [1.2900e-10, 1.0000e+00],
        [6.2567e-09, 1.0000e+00],
        [5.8011e-05, 9.9994e-01],
        [3.4559e-05, 9.9997e-01],
        [2.4191e-07, 1.0000e+00],
        [9.8617e-01, 1.3829e-02],
        [2.3466e-05, 9.9998e-01],
        [4.0596e-06, 1.0000e+00],
        [9.9988e-01, 1.1928e-04],
        [2.2796e-07, 1.0000e+00],
        [1.5039e-11, 1.0000e+00],
        [5.6637e-04, 9.9943e-01],
        [9.9852e-01, 1.4796e-03],
        [9.9665e-01, 3.3544e-03],
        [2.4140e-06, 1.0000e+00],
        [1.5813e-04, 9.9984e-01],
        [9.8957e-01, 1.0432e-02],
        [2.8106e-10, 1.0000e+00],
        [9.9997e-01, 3.4285e-05],
        [1.7036e-03, 9.9830e-01],
        [1.0920e-08, 1.0000e+00],
        [5.7343e-06, 9.9999e-01],
        [2.7852e-08, 1.0000e+00],
        [1.0961e-04, 9.9989e-01],
        [3.1797e-02, 9.6820e-01],
        [9.2392e-07, 1.0000e+00],
        [9.9988e-01, 1.1741e-04],
        [1.2782e-03, 9.9872e-01],
        [4.2135e-08, 1.0000e+00],
        [9.9988e-01, 1.2058e-04],
        [4.3056e-04, 9.9957e-01],
        [1.5365e-03, 9.9846e-01],
        [5.1274e-05, 9.9995e-01],
        [9.9142e-01, 8.5780e-03],
        [1.4440e-03, 9.9856e-01],
        [1.7373e-05, 9.9998e-01],
        [8.1785e-01, 1.8215e-01],
        [5.3745e-07, 1.0000e+00],
        [1.2636e-06, 1.0000e+00],
        [4.0083e-10, 1.0000e+00],
        [1.2229e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.8183e-10, 1.0000e+00],
        [9.9978e-01, 2.2344e-04],
        [9.1877e-05, 9.9991e-01],
        [1.0633e-07, 1.0000e+00],
        [6.1806e-06, 9.9999e-01],
        [5.4998e-10, 1.0000e+00],
        [7.1881e-07, 1.0000e+00],
        [1.6983e-06, 1.0000e+00],
        [2.5493e-07, 1.0000e+00],
        [1.7853e-06, 1.0000e+00],
        [1.1166e-08, 1.0000e+00],
        [1.4975e-08, 1.0000e+00],
        [9.9968e-01, 3.2377e-04],
        [1.2049e-10, 1.0000e+00],
        [1.0827e-04, 9.9989e-01],
        [9.9980e-01, 1.9889e-04],
        [6.6905e-08, 1.0000e+00],
        [5.2987e-06, 9.9999e-01],
        [3.4235e-08, 1.0000e+00],
        [1.9974e-02, 9.8003e-01],
        [9.9912e-01, 8.7839e-04],
        [7.3004e-04, 9.9927e-01],
        [9.9993e-01, 6.5362e-05],
        [9.9990e-01, 1.0092e-04],
        [9.7929e-11, 1.0000e+00],
        [1.3782e-08, 1.0000e+00],
        [1.4365e-05, 9.9999e-01],
        [1.6729e-13, 1.0000e+00],
        [4.4507e-06, 1.0000e+00],
        [2.4460e-07, 1.0000e+00],
        [2.8209e-10, 1.0000e+00],
        [9.6981e-01, 3.0192e-02],
        [7.4021e-05, 9.9993e-01],
        [1.1818e-08, 1.0000e+00],
        [9.9989e-01, 1.1181e-04],
        [9.9992e-01, 8.0613e-05],
        [6.5181e-08, 1.0000e+00],
        [4.1425e-03, 9.9586e-01],
        [2.3864e-12, 1.0000e+00],
        [9.9985e-01, 1.5486e-04],
        [5.0437e-07, 1.0000e+00],
        [1.1076e-07, 1.0000e+00],
        [3.6040e-07, 1.0000e+00],
        [7.1984e-06, 9.9999e-01],
        [3.0734e-05, 9.9997e-01],
        [1.9895e-05, 9.9998e-01],
        [2.7450e-08, 1.0000e+00],
        [1.1261e-05, 9.9999e-01],
        [6.8284e-10, 1.0000e+00],
        [4.9777e-01, 5.0223e-01],
        [4.5226e-07, 1.0000e+00],
        [3.8453e-06, 1.0000e+00],
        [6.5300e-09, 1.0000e+00],
        [8.2144e-11, 1.0000e+00],
        [5.4170e-06, 9.9999e-01],
        [1.3165e-05, 9.9999e-01],
        [1.1425e-05, 9.9999e-01],
        [4.6953e-06, 1.0000e+00],
        [5.4123e-07, 1.0000e+00],
        [9.6879e-06, 9.9999e-01],
        [9.0150e-07, 1.0000e+00],
        [5.2092e-13, 1.0000e+00],
        [9.9991e-01, 9.4539e-05],
        [9.9947e-01, 5.2863e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[7.0070e-08, 1.0000e+00],
        [9.6965e-01, 3.0351e-02],
        [4.6650e-11, 1.0000e+00],
        [9.9947e-01, 5.2621e-04],
        [9.9911e-01, 8.9465e-04],
        [2.8183e-07, 1.0000e+00],
        [5.4570e-13, 1.0000e+00],
        [7.8697e-09, 1.0000e+00],
        [2.9503e-05, 9.9997e-01],
        [9.6990e-09, 1.0000e+00],
        [1.2492e-09, 1.0000e+00],
        [9.9863e-01, 1.3696e-03],
        [9.9937e-01, 6.3178e-04],
        [1.7455e-08, 1.0000e+00],
        [5.2706e-11, 1.0000e+00],
        [8.3766e-01, 1.6234e-01],
        [1.1842e-08, 1.0000e+00],
        [4.0303e-06, 1.0000e+00],
        [9.9996e-01, 3.9192e-05],
        [7.9484e-08, 1.0000e+00],
        [3.1895e-07, 1.0000e+00],
        [6.7413e-09, 1.0000e+00],
        [6.3462e-01, 3.6538e-01],
        [8.6190e-04, 9.9914e-01],
        [9.9987e-01, 1.3311e-04],
        [1.3985e-08, 1.0000e+00],
        [6.9919e-12, 1.0000e+00],
        [9.9993e-01, 7.0723e-05],
        [9.9959e-01, 4.0968e-04],
        [1.1777e-11, 1.0000e+00],
        [5.9215e-09, 1.0000e+00],
        [5.5966e-16, 1.0000e+00],
        [9.9979e-01, 2.1102e-04],
        [3.7313e-06, 1.0000e+00],
        [3.0269e-06, 1.0000e+00],
        [1.2398e-03, 9.9876e-01],
        [1.8020e-04, 9.9982e-01],
        [9.9983e-01, 1.6701e-04],
        [9.9983e-01, 1.6653e-04],
        [9.9986e-01, 1.3568e-04],
        [4.9523e-07, 1.0000e+00],
        [9.9962e-01, 3.8216e-04],
        [2.9771e-08, 1.0000e+00],
        [1.7996e-12, 1.0000e+00],
        [1.0142e-06, 1.0000e+00],
        [8.3659e-08, 1.0000e+00],
        [1.7052e-12, 1.0000e+00],
        [5.0377e-06, 9.9999e-01],
        [6.4722e-04, 9.9935e-01],
        [9.9996e-01, 4.4277e-05],
        [5.4638e-05, 9.9995e-01],
        [7.6215e-08, 1.0000e+00],
        [8.6517e-10, 1.0000e+00],
        [1.6413e-07, 1.0000e+00],
        [4.6076e-08, 1.0000e+00],
        [9.3765e-01, 6.2347e-02],
        [8.9238e-08, 1.0000e+00],
        [1.4579e-04, 9.9985e-01],
        [9.2596e-05, 9.9991e-01],
        [9.9997e-01, 3.4573e-05],
        [9.4449e-01, 5.5514e-02],
        [3.8570e-02, 9.6143e-01],
        [3.3435e-08, 1.0000e+00],
        [4.0566e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.7049e-05, 9.9997e-01],
        [3.8579e-05, 9.9996e-01],
        [9.9972e-01, 2.8295e-04],
        [9.9999e-01, 5.9763e-06],
        [1.2063e-03, 9.9879e-01],
        [2.6967e-07, 1.0000e+00],
        [2.7918e-04, 9.9972e-01],
        [9.9976e-01, 2.4040e-04],
        [7.7632e-08, 1.0000e+00],
        [5.3923e-09, 1.0000e+00],
        [3.6125e-10, 1.0000e+00],
        [6.3512e-13, 1.0000e+00],
        [6.7546e-12, 1.0000e+00],
        [4.6577e-11, 1.0000e+00],
        [5.4140e-06, 9.9999e-01],
        [9.9746e-01, 2.5384e-03],
        [4.2339e-03, 9.9577e-01],
        [9.9305e-01, 6.9489e-03],
        [9.9863e-01, 1.3653e-03],
        [9.9949e-01, 5.1346e-04],
        [9.9829e-01, 1.7060e-03],
        [3.1085e-08, 1.0000e+00],
        [5.4152e-11, 1.0000e+00],
        [3.5475e-09, 1.0000e+00],
        [9.3926e-01, 6.0745e-02],
        [1.7866e-11, 1.0000e+00],
        [5.7938e-05, 9.9994e-01],
        [6.5545e-07, 1.0000e+00],
        [9.7025e-14, 1.0000e+00],
        [2.6759e-08, 1.0000e+00],
        [6.4684e-05, 9.9994e-01],
        [6.8724e-07, 1.0000e+00],
        [9.9995e-01, 5.1853e-05],
        [2.0966e-06, 1.0000e+00],
        [1.0300e-06, 1.0000e+00],
        [1.1253e-04, 9.9989e-01],
        [2.2171e-08, 1.0000e+00],
        [5.6197e-08, 1.0000e+00],
        [7.8002e-01, 2.1998e-01],
        [9.9983e-01, 1.6775e-04],
        [1.5812e-10, 1.0000e+00],
        [2.2171e-10, 1.0000e+00],
        [7.3365e-03, 9.9266e-01],
        [9.9996e-01, 3.9696e-05],
        [1.9152e-07, 1.0000e+00],
        [2.3323e-08, 1.0000e+00],
        [4.9126e-07, 1.0000e+00],
        [4.5663e-06, 1.0000e+00],
        [3.5639e-10, 1.0000e+00],
        [1.8243e-06, 1.0000e+00],
        [1.0778e-09, 1.0000e+00],
        [2.2980e-06, 1.0000e+00],
        [4.5686e-08, 1.0000e+00],
        [7.2693e-09, 1.0000e+00],
        [9.9993e-01, 7.0972e-05],
        [3.9950e-05, 9.9996e-01],
        [1.9560e-05, 9.9998e-01],
        [4.6314e-07, 1.0000e+00],
        [1.0354e-04, 9.9990e-01],
        [9.9557e-08, 1.0000e+00],
        [2.2359e-07, 1.0000e+00],
        [9.9958e-01, 4.2024e-04],
        [2.6256e-09, 1.0000e+00],
        [9.9954e-01, 4.5737e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[7.2439e-08, 1.0000e+00],
        [2.2968e-09, 1.0000e+00],
        [9.9036e-01, 9.6360e-03],
        [2.2653e-10, 1.0000e+00],
        [2.0292e-11, 1.0000e+00],
        [2.8168e-11, 1.0000e+00],
        [9.9993e-01, 6.7761e-05],
        [1.5271e-10, 1.0000e+00],
        [1.5161e-05, 9.9998e-01],
        [3.7489e-12, 1.0000e+00],
        [2.1154e-06, 1.0000e+00],
        [5.1566e-08, 1.0000e+00],
        [4.7067e-11, 1.0000e+00],
        [1.7735e-08, 1.0000e+00],
        [1.3499e-01, 8.6501e-01],
        [2.7320e-12, 1.0000e+00],
        [3.0419e-07, 1.0000e+00],
        [9.9972e-01, 2.7991e-04],
        [2.2756e-08, 1.0000e+00],
        [9.5357e-07, 1.0000e+00],
        [4.7206e-05, 9.9995e-01],
        [3.3486e-12, 1.0000e+00],
        [3.9937e-10, 1.0000e+00],
        [3.3451e-05, 9.9997e-01],
        [1.2919e-13, 1.0000e+00],
        [2.0454e-11, 1.0000e+00],
        [9.9997e-01, 2.9816e-05],
        [9.4808e-01, 5.1921e-02],
        [2.1995e-03, 9.9780e-01],
        [3.7014e-07, 1.0000e+00],
        [4.3231e-01, 5.6769e-01],
        [9.9538e-01, 4.6181e-03],
        [1.3984e-10, 1.0000e+00],
        [8.9485e-07, 1.0000e+00],
        [3.2170e-06, 1.0000e+00],
        [2.5824e-10, 1.0000e+00],
        [8.1098e-09, 1.0000e+00],
        [2.6698e-03, 9.9733e-01],
        [1.9502e-07, 1.0000e+00],
        [1.7293e-06, 1.0000e+00],
        [1.8757e-03, 9.9812e-01],
        [5.1795e-07, 1.0000e+00],
        [5.2058e-10, 1.0000e+00],
        [9.9956e-01, 4.3582e-04],
        [9.6996e-01, 3.0037e-02],
        [9.9988e-01, 1.2384e-04],
        [6.5015e-09, 1.0000e+00],
        [1.6998e-07, 1.0000e+00],
        [9.9996e-01, 4.3128e-05],
        [1.7910e-09, 1.0000e+00],
        [8.8232e-07, 1.0000e+00],
        [1.9398e-09, 1.0000e+00],
        [3.2101e-05, 9.9997e-01],
        [9.9350e-01, 6.4981e-03],
        [5.7797e-10, 1.0000e+00],
        [9.9997e-01, 3.3170e-05],
        [1.8340e-09, 1.0000e+00],
        [2.7647e-11, 1.0000e+00],
        [1.1815e-08, 1.0000e+00],
        [9.9994e-01, 5.9269e-05],
        [2.0339e-12, 1.0000e+00],
        [1.4464e-04, 9.9986e-01],
        [3.7987e-08, 1.0000e+00],
        [5.0868e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.3013e-09, 1.0000e+00],
        [5.6663e-06, 9.9999e-01],
        [3.7252e-07, 1.0000e+00],
        [9.9981e-01, 1.9132e-04],
        [2.2293e-08, 1.0000e+00],
        [1.7103e-05, 9.9998e-01],
        [9.5740e-08, 1.0000e+00],
        [4.1370e-08, 1.0000e+00],
        [9.4195e-09, 1.0000e+00],
        [4.8659e-09, 1.0000e+00],
        [7.6643e-11, 1.0000e+00],
        [3.9067e-11, 1.0000e+00],
        [4.6345e-07, 1.0000e+00],
        [9.9993e-01, 6.9316e-05],
        [2.9811e-05, 9.9997e-01],
        [1.1347e-04, 9.9989e-01],
        [1.3498e-10, 1.0000e+00],
        [2.4444e-04, 9.9976e-01],
        [5.0251e-05, 9.9995e-01],
        [2.0696e-09, 1.0000e+00],
        [1.1612e-06, 1.0000e+00],
        [3.0101e-04, 9.9970e-01],
        [9.2449e-01, 7.5509e-02],
        [9.9980e-01, 1.9511e-04],
        [8.5227e-03, 9.9148e-01],
        [6.4681e-09, 1.0000e+00],
        [9.5017e-01, 4.9827e-02],
        [7.6655e-06, 9.9999e-01],
        [3.2160e-10, 1.0000e+00],
        [3.7171e-08, 1.0000e+00],
        [2.5386e-09, 1.0000e+00],
        [9.9957e-01, 4.2690e-04],
        [9.8727e-11, 1.0000e+00],
        [1.5671e-09, 1.0000e+00],
        [5.5993e-02, 9.4401e-01],
        [1.3443e-06, 1.0000e+00],
        [1.7688e-09, 1.0000e+00],
        [1.3256e-06, 1.0000e+00],
        [9.9987e-01, 1.3354e-04],
        [4.4079e-01, 5.5921e-01],
        [3.9404e-08, 1.0000e+00],
        [1.7580e-07, 1.0000e+00],
        [9.9970e-01, 3.0278e-04],
        [2.0252e-07, 1.0000e+00],
        [2.2830e-08, 1.0000e+00],
        [1.7379e-07, 1.0000e+00],
        [1.1271e-09, 1.0000e+00],
        [9.9929e-01, 7.0540e-04],
        [2.4988e-04, 9.9975e-01],
        [7.8461e-01, 2.1539e-01],
        [9.5208e-04, 9.9905e-01],
        [4.1161e-07, 1.0000e+00],
        [7.4065e-08, 1.0000e+00],
        [9.9989e-01, 1.1153e-04],
        [2.0973e-10, 1.0000e+00],
        [7.6745e-07, 1.0000e+00],
        [1.1092e-03, 9.9889e-01],
        [6.5099e-07, 1.0000e+00],
        [4.5817e-06, 1.0000e+00],
        [1.1083e-07, 1.0000e+00],
        [9.4807e-01, 5.1929e-02],
        [4.6634e-09, 1.0000e+00],
        [8.4651e-06, 9.9999e-01],
        [2.1393e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.6089e-10, 1.0000e+00],
        [2.1803e-11, 1.0000e+00],
        [9.9983e-01, 1.7103e-04],
        [3.1716e-06, 1.0000e+00],
        [2.8628e-05, 9.9997e-01],
        [3.3984e-07, 1.0000e+00],
        [2.9674e-04, 9.9970e-01],
        [4.4366e-07, 1.0000e+00],
        [3.3979e-13, 1.0000e+00],
        [9.9996e-01, 4.3024e-05],
        [9.9871e-01, 1.2859e-03],
        [1.4534e-13, 1.0000e+00],
        [2.1795e-01, 7.8205e-01],
        [1.4696e-07, 1.0000e+00],
        [2.7189e-08, 1.0000e+00],
        [7.4511e-07, 1.0000e+00],
        [4.7102e-08, 1.0000e+00],
        [1.9988e-06, 1.0000e+00],
        [4.1994e-09, 1.0000e+00],
        [7.4127e-10, 1.0000e+00],
        [2.2217e-08, 1.0000e+00],
        [4.9690e-11, 1.0000e+00],
        [9.9992e-01, 8.4375e-05],
        [3.7758e-11, 1.0000e+00],
        [3.6062e-09, 1.0000e+00],
        [9.9975e-01, 2.4525e-04],
        [9.7757e-03, 9.9022e-01],
        [9.9996e-01, 3.9331e-05],
        [5.7591e-01, 4.2409e-01],
        [1.2514e-07, 1.0000e+00],
        [3.9369e-11, 1.0000e+00],
        [7.0706e-09, 1.0000e+00],
        [4.2078e-05, 9.9996e-01],
        [1.3427e-12, 1.0000e+00],
        [3.3995e-07, 1.0000e+00],
        [1.1415e-08, 1.0000e+00],
        [9.9981e-01, 1.8758e-04],
        [5.5439e-05, 9.9994e-01],
        [2.0188e-12, 1.0000e+00],
        [1.5590e-07, 1.0000e+00],
        [7.9930e-08, 1.0000e+00],
        [9.9975e-01, 2.5257e-04],
        [5.3630e-11, 1.0000e+00],
        [2.4540e-05, 9.9998e-01],
        [9.6518e-12, 1.0000e+00],
        [1.7801e-08, 1.0000e+00],
        [5.2384e-07, 1.0000e+00],
        [3.4151e-10, 1.0000e+00],
        [4.2103e-01, 5.7897e-01],
        [1.1296e-06, 1.0000e+00],
        [3.2020e-06, 1.0000e+00],
        [2.1482e-07, 1.0000e+00],
        [9.9989e-01, 1.0874e-04],
        [1.3414e-01, 8.6586e-01],
        [9.9989e-01, 1.1173e-04],
        [2.7921e-06, 1.0000e+00],
        [9.9992e-01, 7.6266e-05],
        [9.3222e-08, 1.0000e+00],
        [1.9999e-04, 9.9980e-01],
        [9.9983e-01, 1.6762e-04],
        [4.0212e-08, 1.0000e+00],
        [9.8340e-01, 1.6599e-02],
        [1.7654e-14, 1.0000e+00],
        [1.0000e+00, 3.4128e-06]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7675e-03, 9.9823e-01],
        [2.2064e-11, 1.0000e+00],
        [9.9992e-01, 7.9773e-05],
        [2.0387e-05, 9.9998e-01],
        [4.2298e-07, 1.0000e+00],
        [9.4655e-06, 9.9999e-01],
        [1.7519e-09, 1.0000e+00],
        [5.0172e-08, 1.0000e+00],
        [1.6052e-06, 1.0000e+00],
        [2.9041e-05, 9.9997e-01],
        [3.6487e-10, 1.0000e+00],
        [9.9759e-01, 2.4145e-03],
        [2.5493e-10, 1.0000e+00],
        [1.0484e-09, 1.0000e+00],
        [9.7640e-01, 2.3595e-02],
        [2.1523e-07, 1.0000e+00],
        [3.1181e-06, 1.0000e+00],
        [1.0824e-07, 1.0000e+00],
        [2.1693e-07, 1.0000e+00],
        [9.9911e-01, 8.9366e-04],
        [9.6312e-05, 9.9990e-01],
        [8.1267e-05, 9.9992e-01],
        [5.5295e-05, 9.9994e-01],
        [1.8633e-04, 9.9981e-01],
        [1.1651e-10, 1.0000e+00],
        [9.9547e-01, 4.5325e-03],
        [3.4635e-04, 9.9965e-01],
        [9.9976e-01, 2.3966e-04],
        [9.9093e-02, 9.0091e-01],
        [2.5923e-09, 1.0000e+00],
        [7.5290e-05, 9.9992e-01],
        [2.6361e-07, 1.0000e+00],
        [6.8912e-08, 1.0000e+00],
        [3.2848e-10, 1.0000e+00],
        [9.9993e-01, 7.4187e-05],
        [2.1123e-13, 1.0000e+00],
        [2.0285e-09, 1.0000e+00],
        [9.9998e-01, 2.3248e-05],
        [4.2737e-08, 1.0000e+00],
        [1.8947e-09, 1.0000e+00],
        [9.9997e-01, 3.4617e-05],
        [2.0815e-07, 1.0000e+00],
        [6.6482e-01, 3.3518e-01],
        [7.5475e-06, 9.9999e-01],
        [4.5956e-06, 1.0000e+00],
        [3.5492e-04, 9.9965e-01],
        [1.6872e-09, 1.0000e+00],
        [9.9998e-01, 2.0320e-05],
        [3.0900e-03, 9.9691e-01],
        [1.3581e-05, 9.9999e-01],
        [4.5459e-05, 9.9995e-01],
        [8.4438e-07, 1.0000e+00],
        [9.9999e-01, 1.4141e-05],
        [4.6006e-04, 9.9954e-01],
        [2.5391e-06, 1.0000e+00],
        [9.7181e-01, 2.8194e-02],
        [2.6523e-10, 1.0000e+00],
        [3.7586e-08, 1.0000e+00],
        [9.9989e-01, 1.1319e-04],
        [5.6597e-01, 4.3403e-01],
        [2.8747e-12, 1.0000e+00],
        [8.2990e-07, 1.0000e+00],
        [1.2752e-10, 1.0000e+00],
        [3.0480e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.1539e-11, 1.0000e+00],
        [9.9995e-01, 5.2238e-05],
        [5.0768e-08, 1.0000e+00],
        [1.2125e-07, 1.0000e+00],
        [9.7506e-01, 2.4941e-02],
        [9.9468e-01, 5.3230e-03],
        [1.1776e-10, 1.0000e+00],
        [9.9989e-01, 1.1282e-04],
        [3.2808e-06, 1.0000e+00],
        [3.0377e-10, 1.0000e+00],
        [1.7020e-12, 1.0000e+00],
        [1.2504e-11, 1.0000e+00],
        [1.7584e-09, 1.0000e+00],
        [8.6799e-06, 9.9999e-01],
        [9.9966e-01, 3.3622e-04],
        [9.2741e-04, 9.9907e-01],
        [1.0005e-05, 9.9999e-01],
        [2.1913e-05, 9.9998e-01],
        [9.9996e-01, 4.4675e-05],
        [1.9555e-06, 1.0000e+00],
        [1.2336e-06, 1.0000e+00],
        [1.0627e-07, 1.0000e+00],
        [3.9277e-10, 1.0000e+00],
        [9.9395e-01, 6.0491e-03],
        [4.8370e-06, 1.0000e+00],
        [9.0957e-06, 9.9999e-01],
        [1.9226e-05, 9.9998e-01],
        [9.9956e-01, 4.3804e-04],
        [6.1863e-07, 1.0000e+00],
        [9.9988e-01, 1.1635e-04],
        [9.9987e-01, 1.3012e-04],
        [9.6894e-01, 3.1062e-02],
        [9.0547e-06, 9.9999e-01],
        [9.9863e-01, 1.3706e-03],
        [1.7491e-05, 9.9998e-01],
        [5.6744e-07, 1.0000e+00],
        [3.2364e-07, 1.0000e+00],
        [9.9996e-01, 4.4950e-05],
        [9.0080e-06, 9.9999e-01],
        [2.3839e-03, 9.9762e-01],
        [9.9990e-01, 9.7639e-05],
        [7.1604e-02, 9.2840e-01],
        [9.9996e-01, 4.1520e-05],
        [8.9856e-01, 1.0144e-01],
        [5.1474e-08, 1.0000e+00],
        [1.6958e-08, 1.0000e+00],
        [1.3786e-08, 1.0000e+00],
        [7.9781e-01, 2.0219e-01],
        [8.5353e-14, 1.0000e+00],
        [4.9734e-05, 9.9995e-01],
        [4.2555e-06, 1.0000e+00],
        [5.4191e-06, 9.9999e-01],
        [9.9263e-01, 7.3657e-03],
        [1.5848e-06, 1.0000e+00],
        [3.5247e-02, 9.6475e-01],
        [2.5780e-06, 1.0000e+00],
        [7.0447e-10, 1.0000e+00],
        [2.6227e-06, 1.0000e+00],
        [9.9994e-01, 5.8436e-05],
        [5.6720e-07, 1.0000e+00],
        [7.5691e-07, 1.0000e+00],
        [1.4624e-06, 1.0000e+00],
        [1.6585e-12, 1.0000e+00],
        [1.8212e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.8536e-07, 1.0000e+00],
        [1.1116e-06, 1.0000e+00],
        [1.0841e-06, 1.0000e+00],
        [8.5370e-01, 1.4630e-01],
        [1.1213e-04, 9.9989e-01],
        [9.9993e-01, 7.2020e-05],
        [3.7870e-07, 1.0000e+00],
        [2.1218e-09, 1.0000e+00],
        [1.2967e-04, 9.9987e-01],
        [1.0205e-01, 8.9795e-01],
        [4.0261e-08, 1.0000e+00],
        [6.9469e-07, 1.0000e+00],
        [1.2675e-04, 9.9987e-01],
        [9.9999e-01, 5.2872e-06],
        [1.0000e+00, 1.7307e-06],
        [3.0344e-04, 9.9970e-01],
        [1.4931e-09, 1.0000e+00],
        [3.4629e-04, 9.9965e-01],
        [9.4686e-01, 5.3141e-02],
        [2.5160e-10, 1.0000e+00],
        [9.9993e-01, 6.6352e-05],
        [2.5870e-08, 1.0000e+00],
        [9.9994e-01, 6.4144e-05],
        [6.4674e-02, 9.3533e-01],
        [9.9995e-01, 5.1372e-05],
        [9.9474e-01, 5.2552e-03],
        [1.7430e-03, 9.9826e-01],
        [1.7474e-05, 9.9998e-01],
        [5.8282e-11, 1.0000e+00],
        [9.3341e-01, 6.6591e-02],
        [1.2171e-03, 9.9878e-01],
        [2.9428e-07, 1.0000e+00],
        [6.4658e-04, 9.9935e-01],
        [5.2026e-07, 1.0000e+00],
        [3.6954e-08, 1.0000e+00],
        [1.0402e-06, 1.0000e+00],
        [9.9907e-01, 9.2713e-04],
        [8.4483e-05, 9.9992e-01],
        [9.9995e-01, 4.6346e-05],
        [2.1513e-03, 9.9785e-01],
        [1.3129e-05, 9.9999e-01],
        [1.1971e-08, 1.0000e+00],
        [1.1392e-08, 1.0000e+00],
        [9.9743e-01, 2.5674e-03],
        [6.7917e-05, 9.9993e-01],
        [1.6058e-09, 1.0000e+00],
        [1.8091e-07, 1.0000e+00],
        [9.9995e-01, 4.9460e-05],
        [9.9953e-01, 4.6888e-04],
        [2.5770e-04, 9.9974e-01],
        [2.0276e-04, 9.9980e-01],
        [4.3920e-06, 1.0000e+00],
        [1.8410e-03, 9.9816e-01],
        [4.0048e-06, 1.0000e+00],
        [1.2391e-03, 9.9876e-01],
        [1.6079e-08, 1.0000e+00],
        [9.9997e-01, 2.8658e-05],
        [9.9997e-01, 3.4236e-05],
        [2.8216e-05, 9.9997e-01],
        [1.7955e-03, 9.9820e-01],
        [4.0635e-08, 1.0000e+00],
        [1.0339e-06, 1.0000e+00],
        [1.5835e-08, 1.0000e+00],
        [9.9971e-01, 2.8846e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7068e-03, 9.9829e-01],
        [1.3561e-05, 9.9999e-01],
        [1.4749e-07, 1.0000e+00],
        [1.6997e-04, 9.9983e-01],
        [1.6659e-08, 1.0000e+00],
        [1.7437e-10, 1.0000e+00],
        [6.7059e-09, 1.0000e+00],
        [3.7311e-07, 1.0000e+00],
        [9.9920e-01, 7.9530e-04],
        [5.1361e-09, 1.0000e+00],
        [1.0634e-05, 9.9999e-01],
        [6.0225e-07, 1.0000e+00],
        [4.1741e-01, 5.8259e-01],
        [4.3424e-09, 1.0000e+00],
        [1.9351e-07, 1.0000e+00],
        [1.3022e-07, 1.0000e+00],
        [7.7092e-01, 2.2908e-01],
        [5.5996e-09, 1.0000e+00],
        [8.1609e-03, 9.9184e-01],
        [1.5695e-02, 9.8431e-01],
        [9.9996e-01, 4.3099e-05],
        [2.2289e-05, 9.9998e-01],
        [7.4657e-07, 1.0000e+00],
        [4.6127e-03, 9.9539e-01],
        [3.1284e-06, 1.0000e+00],
        [5.4992e-06, 9.9999e-01],
        [9.9996e-01, 3.6906e-05],
        [5.4606e-08, 1.0000e+00],
        [9.9953e-01, 4.6566e-04],
        [4.1692e-10, 1.0000e+00],
        [5.5472e-10, 1.0000e+00],
        [9.9993e-01, 7.0759e-05],
        [3.7456e-04, 9.9963e-01],
        [5.6386e-09, 1.0000e+00],
        [9.3809e-04, 9.9906e-01],
        [9.9999e-01, 6.2886e-06],
        [4.3206e-04, 9.9957e-01],
        [9.8350e-06, 9.9999e-01],
        [9.5076e-05, 9.9990e-01],
        [1.3533e-11, 1.0000e+00],
        [7.2851e-06, 9.9999e-01],
        [9.9994e-01, 5.6467e-05],
        [1.5267e-09, 1.0000e+00],
        [1.6503e-08, 1.0000e+00],
        [9.4406e-08, 1.0000e+00],
        [2.2278e-03, 9.9777e-01],
        [2.3141e-06, 1.0000e+00],
        [3.9371e-07, 1.0000e+00],
        [3.8155e-06, 1.0000e+00],
        [6.6183e-04, 9.9934e-01],
        [1.2269e-08, 1.0000e+00],
        [9.9993e-01, 6.8434e-05],
        [9.7886e-01, 2.1145e-02],
        [9.9988e-01, 1.2354e-04],
        [1.2829e-02, 9.8717e-01],
        [7.7277e-08, 1.0000e+00],
        [1.2071e-09, 1.0000e+00],
        [2.3847e-09, 1.0000e+00],
        [1.2642e-08, 1.0000e+00],
        [8.7843e-09, 1.0000e+00],
        [1.8957e-07, 1.0000e+00],
        [2.1193e-04, 9.9979e-01],
        [2.8095e-06, 1.0000e+00],
        [3.8698e-01, 6.1302e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[8.0806e-01, 1.9194e-01],
        [4.0323e-02, 9.5968e-01],
        [5.2018e-07, 1.0000e+00],
        [1.9232e-10, 1.0000e+00],
        [5.1031e-10, 1.0000e+00],
        [9.9971e-01, 2.9385e-04],
        [9.9999e-01, 6.7400e-06],
        [4.7781e-04, 9.9952e-01],
        [7.2746e-04, 9.9927e-01],
        [7.1171e-07, 1.0000e+00],
        [9.9891e-01, 1.0945e-03],
        [9.9984e-01, 1.5879e-04],
        [8.5168e-06, 9.9999e-01],
        [1.6552e-05, 9.9998e-01],
        [9.9998e-01, 2.1251e-05],
        [6.2778e-05, 9.9994e-01],
        [9.3595e-08, 1.0000e+00],
        [4.5431e-04, 9.9955e-01],
        [1.1111e-05, 9.9999e-01],
        [2.8079e-07, 1.0000e+00],
        [2.5800e-05, 9.9997e-01],
        [9.9049e-03, 9.9010e-01],
        [3.2457e-07, 1.0000e+00],
        [9.9461e-01, 5.3885e-03],
        [9.9994e-01, 6.3517e-05],
        [1.9012e-08, 1.0000e+00],
        [1.7269e-06, 1.0000e+00],
        [7.0011e-08, 1.0000e+00],
        [1.6190e-05, 9.9998e-01],
        [5.3183e-06, 9.9999e-01],
        [2.3257e-08, 1.0000e+00],
        [9.9983e-01, 1.6625e-04],
        [9.9996e-01, 4.3546e-05],
        [1.7643e-07, 1.0000e+00],
        [9.9992e-01, 7.5093e-05],
        [9.9998e-01, 2.2198e-05],
        [9.9994e-01, 6.0257e-05],
        [3.9943e-05, 9.9996e-01],
        [6.2079e-06, 9.9999e-01],
        [8.2441e-06, 9.9999e-01],
        [9.9998e-01, 1.5071e-05],
        [2.4915e-09, 1.0000e+00],
        [2.1092e-03, 9.9789e-01],
        [9.9761e-01, 2.3915e-03],
        [6.0949e-07, 1.0000e+00],
        [2.9410e-05, 9.9997e-01],
        [9.9999e-01, 5.9409e-06],
        [1.0977e-08, 1.0000e+00],
        [9.9986e-01, 1.3971e-04],
        [9.7119e-05, 9.9990e-01],
        [6.6022e-07, 1.0000e+00],
        [6.5660e-17, 1.0000e+00],
        [2.1140e-04, 9.9979e-01],
        [3.4966e-07, 1.0000e+00],
        [9.9988e-01, 1.1708e-04],
        [1.5837e-04, 9.9984e-01],
        [3.5261e-05, 9.9996e-01],
        [9.9990e-01, 9.6233e-05],
        [3.8652e-04, 9.9961e-01],
        [4.4111e-04, 9.9956e-01],
        [8.4218e-07, 1.0000e+00],
        [8.0654e-09, 1.0000e+00],
        [1.4322e-02, 9.8568e-01],
        [9.9997e-01, 2.5573e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[4.9894e-05, 9.9995e-01],
        [6.3001e-05, 9.9994e-01],
        [1.0336e-07, 1.0000e+00],
        [3.7660e-01, 6.2340e-01],
        [5.2332e-07, 1.0000e+00],
        [3.7343e-08, 1.0000e+00],
        [9.9996e-01, 3.8206e-05],
        [2.8633e-05, 9.9997e-01],
        [9.9996e-01, 3.7786e-05],
        [6.2965e-03, 9.9370e-01],
        [2.8574e-08, 1.0000e+00],
        [5.9805e-01, 4.0195e-01],
        [9.9305e-05, 9.9990e-01],
        [8.1617e-04, 9.9918e-01],
        [1.1884e-04, 9.9988e-01],
        [9.8442e-06, 9.9999e-01],
        [9.2669e-03, 9.9073e-01],
        [3.7616e-07, 1.0000e+00],
        [4.8872e-06, 1.0000e+00],
        [1.1799e-08, 1.0000e+00],
        [9.9998e-01, 1.8586e-05],
        [2.3279e-07, 1.0000e+00],
        [1.3283e-02, 9.8672e-01],
        [4.4902e-06, 1.0000e+00],
        [3.4167e-05, 9.9997e-01],
        [1.3824e-08, 1.0000e+00],
        [7.7154e-06, 9.9999e-01],
        [3.9323e-08, 1.0000e+00],
        [1.0000e+00, 4.7845e-06],
        [9.9997e-01, 2.6565e-05],
        [9.9997e-01, 2.5496e-05],
        [3.7935e-05, 9.9996e-01],
        [5.0537e-06, 9.9999e-01],
        [4.1416e-03, 9.9586e-01],
        [9.6510e-09, 1.0000e+00],
        [9.9995e-01, 4.9861e-05],
        [4.2820e-09, 1.0000e+00],
        [4.5304e-04, 9.9955e-01],
        [6.0498e-04, 9.9940e-01],
        [9.7896e-08, 1.0000e+00],
        [6.2231e-04, 9.9938e-01],
        [1.0263e-04, 9.9990e-01],
        [1.0310e-03, 9.9897e-01],
        [7.1438e-01, 2.8562e-01],
        [1.2847e-01, 8.7153e-01],
        [9.9990e-01, 1.0418e-04],
        [6.3418e-09, 1.0000e+00],
        [3.0439e-07, 1.0000e+00],
        [1.6480e-02, 9.8352e-01],
        [9.9992e-01, 8.0218e-05],
        [1.3340e-08, 1.0000e+00],
        [6.3518e-08, 1.0000e+00],
        [9.9994e-01, 5.5899e-05],
        [2.6510e-07, 1.0000e+00],
        [2.3213e-09, 1.0000e+00],
        [2.3821e-05, 9.9998e-01],
        [1.5342e-03, 9.9847e-01],
        [3.3242e-08, 1.0000e+00],
        [3.2901e-09, 1.0000e+00],
        [9.9996e-01, 3.7780e-05],
        [7.2987e-05, 9.9993e-01],
        [9.7961e-06, 9.9999e-01],
        [1.2671e-07, 1.0000e+00],
        [4.1832e-03, 9.9582e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.4845e-08, 1.0000e+00],
        [2.3156e-05, 9.9998e-01],
        [9.9992e-01, 7.5167e-05],
        [9.9997e-01, 2.5508e-05],
        [1.8764e-08, 1.0000e+00],
        [5.9505e-06, 9.9999e-01],
        [4.9744e-06, 9.9999e-01],
        [2.0692e-05, 9.9998e-01],
        [4.9194e-09, 1.0000e+00],
        [2.4560e-02, 9.7544e-01],
        [9.9993e-01, 6.9393e-05],
        [2.9254e-04, 9.9971e-01],
        [3.8910e-01, 6.1090e-01],
        [9.9987e-01, 1.2788e-04],
        [2.0332e-07, 1.0000e+00],
        [2.5234e-03, 9.9748e-01],
        [7.4637e-09, 1.0000e+00],
        [6.3333e-08, 1.0000e+00],
        [1.1446e-03, 9.9886e-01],
        [1.9716e-07, 1.0000e+00],
        [1.0009e-05, 9.9999e-01],
        [9.9958e-01, 4.1570e-04],
        [9.9983e-01, 1.6659e-04],
        [1.6643e-07, 1.0000e+00],
        [1.1416e-04, 9.9989e-01],
        [9.9991e-01, 8.7257e-05],
        [9.9999e-01, 5.3137e-06],
        [2.0109e-03, 9.9799e-01],
        [4.8157e-08, 1.0000e+00],
        [9.9817e-01, 1.8346e-03],
        [4.9811e-06, 9.9999e-01],
        [5.5755e-04, 9.9944e-01],
        [4.0676e-07, 1.0000e+00],
        [3.0903e-05, 9.9997e-01],
        [2.7015e-05, 9.9997e-01],
        [7.1105e-06, 9.9999e-01],
        [1.5891e-07, 1.0000e+00],
        [9.9995e-01, 4.7426e-05],
        [9.9978e-01, 2.1901e-04],
        [2.0477e-06, 1.0000e+00],
        [9.9993e-01, 6.6339e-05],
        [4.6056e-02, 9.5394e-01],
        [6.8633e-06, 9.9999e-01],
        [3.1748e-02, 9.6825e-01],
        [1.3135e-11, 1.0000e+00],
        [3.3919e-09, 1.0000e+00],
        [1.7724e-06, 1.0000e+00],
        [8.6894e-05, 9.9991e-01],
        [1.9625e-03, 9.9804e-01],
        [1.7522e-06, 1.0000e+00],
        [8.5711e-05, 9.9991e-01],
        [1.2825e-07, 1.0000e+00],
        [3.0416e-05, 9.9997e-01],
        [9.9996e-01, 3.8910e-05],
        [1.4158e-04, 9.9986e-01],
        [9.9978e-01, 2.2173e-04],
        [3.6653e-04, 9.9963e-01],
        [9.1389e-07, 1.0000e+00],
        [7.4812e-05, 9.9993e-01],
        [1.9645e-04, 9.9980e-01],
        [1.5005e-11, 1.0000e+00],
        [5.4677e-05, 9.9995e-01],
        [8.3485e-07, 1.0000e+00],
        [2.4793e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9981e-01, 1.9029e-04],
        [3.0816e-04, 9.9969e-01],
        [3.9226e-01, 6.0774e-01],
        [4.4647e-03, 9.9554e-01],
        [2.3066e-07, 1.0000e+00],
        [8.6855e-10, 1.0000e+00],
        [5.7056e-05, 9.9994e-01],
        [2.0081e-06, 1.0000e+00],
        [6.3840e-08, 1.0000e+00],
        [4.6895e-06, 1.0000e+00],
        [8.5991e-01, 1.4009e-01],
        [2.4287e-04, 9.9976e-01],
        [6.4734e-09, 1.0000e+00],
        [1.1783e-04, 9.9988e-01],
        [9.9991e-01, 9.1343e-05],
        [9.1576e-01, 8.4241e-02],
        [3.1067e-07, 1.0000e+00],
        [3.7475e-04, 9.9963e-01],
        [9.9993e-01, 6.6331e-05],
        [2.8437e-10, 1.0000e+00],
        [3.9408e-05, 9.9996e-01],
        [1.0395e-06, 1.0000e+00],
        [1.8530e-08, 1.0000e+00],
        [9.9943e-01, 5.6656e-04],
        [1.5779e-03, 9.9842e-01],
        [8.2420e-05, 9.9992e-01],
        [9.9987e-01, 1.3108e-04],
        [1.3433e-06, 1.0000e+00],
        [5.7764e-05, 9.9994e-01],
        [9.8347e-07, 1.0000e+00],
        [9.9308e-01, 6.9195e-03],
        [9.2968e-01, 7.0322e-02],
        [2.5124e-08, 1.0000e+00],
        [6.8972e-09, 1.0000e+00],
        [7.1015e-10, 1.0000e+00],
        [1.0652e-01, 8.9348e-01],
        [1.2054e-09, 1.0000e+00],
        [9.9978e-01, 2.1802e-04],
        [8.5274e-04, 9.9915e-01],
        [8.1226e-04, 9.9919e-01],
        [9.9993e-01, 6.7913e-05],
        [2.3117e-10, 1.0000e+00],
        [4.1033e-03, 9.9590e-01],
        [5.0879e-06, 9.9999e-01],
        [1.8982e-05, 9.9998e-01],
        [1.6795e-08, 1.0000e+00],
        [9.9990e-01, 1.0097e-04],
        [2.5375e-09, 1.0000e+00],
        [3.2205e-02, 9.6780e-01],
        [7.4483e-05, 9.9993e-01],
        [9.9996e-01, 4.2314e-05],
        [1.9367e-06, 1.0000e+00],
        [5.6713e-10, 1.0000e+00],
        [4.4761e-06, 1.0000e+00],
        [1.7916e-07, 1.0000e+00],
        [9.9999e-01, 1.2881e-05],
        [7.4067e-01, 2.5933e-01],
        [1.2065e-05, 9.9999e-01],
        [7.8728e-09, 1.0000e+00],
        [2.4026e-03, 9.9760e-01],
        [2.5724e-09, 1.0000e+00],
        [3.5949e-01, 6.4051e-01],
        [5.0717e-02, 9.4928e-01],
        [6.2435e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2310e-06, 1.0000e+00],
        [1.0696e-07, 1.0000e+00],
        [3.6271e-02, 9.6373e-01],
        [9.9996e-01, 4.4810e-05],
        [8.4420e-03, 9.9156e-01],
        [1.4734e-02, 9.8527e-01],
        [2.9244e-04, 9.9971e-01],
        [1.7419e-09, 1.0000e+00],
        [1.3942e-03, 9.9861e-01],
        [2.1599e-01, 7.8401e-01],
        [4.0239e-08, 1.0000e+00],
        [9.9971e-01, 2.8834e-04],
        [1.9699e-04, 9.9980e-01],
        [2.6527e-06, 1.0000e+00],
        [5.2704e-10, 1.0000e+00],
        [9.9995e-01, 4.9982e-05],
        [3.3305e-03, 9.9667e-01],
        [7.2915e-10, 1.0000e+00],
        [9.9997e-01, 3.2334e-05],
        [1.9691e-03, 9.9803e-01],
        [2.1923e-07, 1.0000e+00],
        [9.9992e-01, 7.8360e-05],
        [3.8396e-08, 1.0000e+00],
        [9.9997e-01, 3.1639e-05],
        [4.9663e-06, 9.9999e-01],
        [3.0354e-04, 9.9970e-01],
        [9.4330e-06, 9.9999e-01],
        [9.9993e-01, 7.0209e-05],
        [1.5518e-13, 1.0000e+00],
        [9.9997e-01, 3.0000e-05],
        [6.5927e-06, 9.9999e-01],
        [9.9998e-01, 2.4757e-05],
        [3.3617e-04, 9.9966e-01],
        [4.1829e-03, 9.9582e-01],
        [1.2927e-06, 1.0000e+00],
        [4.6316e-09, 1.0000e+00],
        [3.4226e-06, 1.0000e+00],
        [4.1355e-05, 9.9996e-01],
        [1.0161e-03, 9.9898e-01],
        [1.8725e-06, 1.0000e+00],
        [9.8468e-01, 1.5324e-02],
        [4.9392e-04, 9.9951e-01],
        [3.2827e-05, 9.9997e-01],
        [9.9980e-01, 1.9639e-04],
        [9.9992e-01, 8.5039e-05],
        [5.7078e-03, 9.9429e-01],
        [1.0915e-08, 1.0000e+00],
        [9.9997e-01, 3.4776e-05],
        [9.9994e-01, 6.3221e-05],
        [2.7026e-07, 1.0000e+00],
        [9.9858e-01, 1.4180e-03],
        [3.6089e-05, 9.9996e-01],
        [7.5908e-03, 9.9241e-01],
        [2.2813e-07, 1.0000e+00],
        [9.9976e-01, 2.3537e-04],
        [4.6821e-08, 1.0000e+00],
        [1.9553e-04, 9.9980e-01],
        [2.1047e-05, 9.9998e-01],
        [2.8399e-05, 9.9997e-01],
        [5.2152e-04, 9.9948e-01],
        [8.0122e-05, 9.9992e-01],
        [1.1516e-05, 9.9999e-01],
        [4.4566e-03, 9.9554e-01],
        [9.9996e-01, 4.4810e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.4811e-02, 9.8519e-01],
        [9.9993e-01, 7.2440e-05],
        [1.1710e-07, 1.0000e+00],
        [1.0104e-07, 1.0000e+00],
        [9.9998e-01, 2.2792e-05],
        [9.9996e-01, 3.8598e-05],
        [1.7488e-12, 1.0000e+00],
        [1.8036e-02, 9.8196e-01],
        [2.1938e-11, 1.0000e+00],
        [9.9993e-01, 7.3168e-05],
        [1.2207e-03, 9.9878e-01],
        [1.6770e-03, 9.9832e-01],
        [5.7305e-09, 1.0000e+00],
        [2.1295e-08, 1.0000e+00],
        [3.3379e-01, 6.6621e-01],
        [5.7305e-09, 1.0000e+00],
        [3.9625e-09, 1.0000e+00],
        [2.2272e-06, 1.0000e+00],
        [9.9994e-01, 6.3096e-05],
        [3.2311e-06, 1.0000e+00],
        [3.1295e-08, 1.0000e+00],
        [4.3414e-07, 1.0000e+00],
        [1.3292e-07, 1.0000e+00],
        [3.2801e-02, 9.6720e-01],
        [1.2116e-04, 9.9988e-01],
        [4.4301e-04, 9.9956e-01],
        [1.4241e-07, 1.0000e+00],
        [4.8607e-05, 9.9995e-01],
        [5.5980e-06, 9.9999e-01],
        [6.4539e-10, 1.0000e+00],
        [9.9995e-01, 5.4299e-05],
        [6.8437e-01, 3.1563e-01],
        [9.9990e-01, 9.9193e-05],
        [6.1557e-07, 1.0000e+00],
        [9.9996e-01, 3.8798e-05],
        [6.7345e-06, 9.9999e-01],
        [9.9823e-01, 1.7670e-03],
        [4.8820e-08, 1.0000e+00],
        [9.9997e-01, 3.3480e-05],
        [9.9997e-01, 3.0120e-05],
        [1.5859e-01, 8.4141e-01],
        [1.6354e-04, 9.9984e-01],
        [9.9985e-01, 1.4763e-04],
        [1.2216e-07, 1.0000e+00],
        [3.1209e-07, 1.0000e+00],
        [2.1282e-08, 1.0000e+00],
        [3.8043e-01, 6.1957e-01],
        [4.7730e-05, 9.9995e-01],
        [9.9970e-01, 2.9935e-04],
        [3.2705e-04, 9.9967e-01],
        [1.3873e-06, 1.0000e+00],
        [9.9623e-01, 3.7665e-03],
        [7.7962e-05, 9.9992e-01],
        [5.4877e-05, 9.9995e-01],
        [6.0887e-08, 1.0000e+00],
        [9.9986e-01, 1.4461e-04],
        [9.9984e-01, 1.5573e-04],
        [9.7584e-06, 9.9999e-01],
        [5.6644e-03, 9.9434e-01],
        [3.8019e-08, 1.0000e+00],
        [2.5735e-09, 1.0000e+00],
        [7.2980e-07, 1.0000e+00],
        [4.9458e-06, 1.0000e+00],
        [9.9974e-01, 2.5687e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[5.2493e-06, 9.9999e-01],
        [3.1066e-04, 9.9969e-01],
        [9.2674e-08, 1.0000e+00],
        [9.9514e-01, 4.8634e-03],
        [8.4722e-08, 1.0000e+00],
        [9.9939e-01, 6.0983e-04],
        [1.0393e-04, 9.9990e-01],
        [9.9993e-01, 7.3183e-05],
        [8.4088e-07, 1.0000e+00],
        [3.0339e-01, 6.9661e-01],
        [2.5202e-09, 1.0000e+00],
        [7.5742e-06, 9.9999e-01],
        [2.5763e-09, 1.0000e+00],
        [9.9995e-01, 4.9550e-05],
        [9.9893e-01, 1.0743e-03],
        [1.3652e-06, 1.0000e+00],
        [2.3678e-04, 9.9976e-01],
        [1.4689e-03, 9.9853e-01],
        [2.7563e-08, 1.0000e+00],
        [2.4677e-07, 1.0000e+00],
        [9.3000e-01, 7.0002e-02],
        [3.7421e-07, 1.0000e+00],
        [4.2456e-07, 1.0000e+00],
        [9.8962e-01, 1.0381e-02],
        [3.4409e-07, 1.0000e+00],
        [9.9997e-01, 2.9300e-05],
        [2.7157e-08, 1.0000e+00],
        [5.9872e-03, 9.9401e-01],
        [2.5665e-05, 9.9997e-01],
        [9.9973e-01, 2.7008e-04],
        [1.0892e-02, 9.8911e-01],
        [3.2372e-02, 9.6763e-01],
        [2.4033e-06, 1.0000e+00],
        [4.4594e-07, 1.0000e+00],
        [8.0380e-08, 1.0000e+00],
        [9.3286e-05, 9.9991e-01],
        [8.1868e-08, 1.0000e+00],
        [4.8102e-02, 9.5190e-01],
        [2.7041e-06, 1.0000e+00],
        [9.8482e-01, 1.5182e-02],
        [6.1683e-08, 1.0000e+00],
        [1.6357e-06, 1.0000e+00],
        [1.2715e-04, 9.9987e-01],
        [2.4510e-07, 1.0000e+00],
        [3.7378e-01, 6.2622e-01],
        [9.9994e-01, 6.1106e-05],
        [9.9998e-01, 1.8969e-05],
        [1.3872e-08, 1.0000e+00],
        [1.8707e-10, 1.0000e+00],
        [1.0419e-08, 1.0000e+00],
        [1.2680e-09, 1.0000e+00],
        [3.1753e-02, 9.6825e-01],
        [2.3256e-09, 1.0000e+00],
        [9.0052e-04, 9.9910e-01],
        [9.9995e-01, 5.4971e-05],
        [9.9996e-01, 4.1810e-05],
        [3.7722e-05, 9.9996e-01],
        [2.3556e-02, 9.7644e-01],
        [9.9996e-01, 3.9193e-05],
        [9.9964e-01, 3.5723e-04],
        [9.9985e-01, 1.5307e-04],
        [8.9041e-04, 9.9911e-01],
        [1.3723e-07, 1.0000e+00],
        [1.1840e-05, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9995e-01, 4.9846e-05],
        [3.5330e-04, 9.9965e-01],
        [1.5321e-08, 1.0000e+00],
        [2.3260e-12, 1.0000e+00],
        [1.5513e-08, 1.0000e+00],
        [3.2475e-08, 1.0000e+00],
        [9.9997e-01, 2.6703e-05],
        [2.0858e-07, 1.0000e+00],
        [9.9971e-01, 2.8665e-04],
        [9.9966e-01, 3.3513e-04],
        [2.9302e-07, 1.0000e+00],
        [1.5806e-08, 1.0000e+00],
        [8.8713e-10, 1.0000e+00],
        [4.0956e-04, 9.9959e-01],
        [1.1928e-01, 8.8072e-01],
        [5.0715e-06, 9.9999e-01],
        [8.7589e-03, 9.9124e-01],
        [2.1689e-07, 1.0000e+00],
        [9.9991e-01, 9.0781e-05],
        [1.5275e-06, 1.0000e+00],
        [1.0439e-04, 9.9990e-01],
        [5.5159e-07, 1.0000e+00],
        [9.8103e-10, 1.0000e+00],
        [1.9992e-07, 1.0000e+00],
        [1.3681e-01, 8.6319e-01],
        [9.9990e-01, 1.0270e-04],
        [1.1293e-03, 9.9887e-01],
        [2.7963e-01, 7.2037e-01],
        [4.2766e-04, 9.9957e-01],
        [4.0236e-05, 9.9996e-01],
        [3.5150e-04, 9.9965e-01],
        [1.4392e-02, 9.8561e-01],
        [2.5957e-07, 1.0000e+00],
        [8.3725e-04, 9.9916e-01],
        [2.1257e-07, 1.0000e+00],
        [5.0322e-08, 1.0000e+00],
        [4.7310e-01, 5.2690e-01],
        [1.3664e-07, 1.0000e+00],
        [4.3300e-02, 9.5670e-01],
        [1.0000e+00, 4.3373e-08],
        [5.0317e-07, 1.0000e+00],
        [8.8181e-09, 1.0000e+00],
        [1.0828e-09, 1.0000e+00],
        [1.7142e-11, 1.0000e+00],
        [9.9632e-01, 3.6824e-03],
        [3.7333e-07, 1.0000e+00],
        [1.8872e-10, 1.0000e+00],
        [1.2616e-07, 1.0000e+00],
        [9.9719e-01, 2.8064e-03],
        [1.7227e-09, 1.0000e+00],
        [1.6445e-02, 9.8356e-01],
        [2.0266e-06, 1.0000e+00],
        [2.4496e-04, 9.9976e-01],
        [9.7777e-01, 2.2225e-02],
        [1.6356e-06, 1.0000e+00],
        [1.9621e-09, 1.0000e+00],
        [4.1544e-02, 9.5846e-01],
        [1.9891e-05, 9.9998e-01],
        [1.1146e-04, 9.9989e-01],
        [2.8898e-04, 9.9971e-01],
        [1.4058e-04, 9.9986e-01],
        [1.9489e-05, 9.9998e-01],
        [8.4877e-06, 9.9999e-01],
        [7.7029e-02, 9.2297e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[5.8679e-05, 9.9994e-01],
        [3.2639e-05, 9.9997e-01],
        [9.9953e-01, 4.6959e-04],
        [9.9986e-01, 1.3999e-04],
        [2.9620e-07, 1.0000e+00],
        [9.9667e-01, 3.3273e-03],
        [2.8311e-09, 1.0000e+00],
        [3.4834e-06, 1.0000e+00],
        [2.1906e-07, 1.0000e+00],
        [9.9984e-01, 1.5840e-04],
        [9.9989e-01, 1.0923e-04],
        [9.9911e-01, 8.9023e-04],
        [2.9747e-07, 1.0000e+00],
        [1.2119e-12, 1.0000e+00],
        [1.9267e-13, 1.0000e+00],
        [2.5055e-12, 1.0000e+00],
        [6.2697e-08, 1.0000e+00],
        [9.9448e-01, 5.5229e-03],
        [9.9902e-01, 9.8382e-04],
        [2.3232e-07, 1.0000e+00],
        [9.9967e-01, 3.3044e-04],
        [9.8284e-01, 1.7159e-02],
        [1.1453e-03, 9.9885e-01],
        [9.9996e-01, 3.9971e-05],
        [6.1294e-07, 1.0000e+00],
        [2.5614e-09, 1.0000e+00],
        [1.0551e-06, 1.0000e+00],
        [3.0205e-07, 1.0000e+00],
        [2.5368e-06, 1.0000e+00],
        [7.6617e-06, 9.9999e-01],
        [9.2575e-08, 1.0000e+00],
        [1.7578e-06, 1.0000e+00],
        [9.9995e-01, 5.2630e-05],
        [7.4168e-10, 1.0000e+00],
        [4.1240e-06, 1.0000e+00],
        [9.9611e-01, 3.8940e-03],
        [6.7126e-10, 1.0000e+00],
        [9.9994e-01, 5.6201e-05],
        [9.9982e-01, 1.8201e-04],
        [2.0822e-07, 1.0000e+00],
        [9.9990e-01, 1.0237e-04],
        [9.9996e-01, 4.4121e-05],
        [9.1364e-05, 9.9991e-01],
        [4.8583e-06, 1.0000e+00],
        [9.9990e-01, 1.0086e-04],
        [2.3077e-06, 1.0000e+00],
        [5.7633e-11, 1.0000e+00],
        [1.2636e-07, 1.0000e+00],
        [9.9767e-01, 2.3281e-03],
        [1.8063e-08, 1.0000e+00],
        [9.0166e-07, 1.0000e+00],
        [1.0921e-08, 1.0000e+00],
        [2.9861e-03, 9.9701e-01],
        [2.3595e-06, 1.0000e+00],
        [9.9994e-01, 6.3295e-05],
        [1.0975e-07, 1.0000e+00],
        [9.5464e-05, 9.9990e-01],
        [7.4248e-10, 1.0000e+00],
        [9.9993e-01, 7.4132e-05],
        [4.8933e-06, 1.0000e+00],
        [1.4306e-06, 1.0000e+00],
        [9.9998e-01, 1.9727e-05],
        [3.5910e-11, 1.0000e+00],
        [9.9989e-01, 1.1034e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.5367e-06, 1.0000e+00],
        [8.1201e-06, 9.9999e-01],
        [7.5814e-09, 1.0000e+00],
        [1.3794e-01, 8.6206e-01],
        [1.6273e-08, 1.0000e+00],
        [3.9351e-03, 9.9606e-01],
        [5.4273e-07, 1.0000e+00],
        [9.9995e-01, 5.4017e-05],
        [9.9994e-01, 6.2358e-05],
        [1.4880e-05, 9.9999e-01],
        [1.2545e-04, 9.9987e-01],
        [3.4556e-05, 9.9997e-01],
        [8.1931e-07, 1.0000e+00],
        [1.0235e-06, 1.0000e+00],
        [1.0513e-06, 1.0000e+00],
        [1.3036e-10, 1.0000e+00],
        [9.9985e-01, 1.5371e-04],
        [9.9998e-01, 1.7767e-05],
        [9.6292e-07, 1.0000e+00],
        [9.9992e-01, 8.4693e-05],
        [3.0136e-11, 1.0000e+00],
        [1.5891e-03, 9.9841e-01],
        [9.9983e-01, 1.6630e-04],
        [2.2424e-11, 1.0000e+00],
        [5.9045e-09, 1.0000e+00],
        [2.3981e-09, 1.0000e+00],
        [4.6480e-11, 1.0000e+00],
        [3.4706e-03, 9.9653e-01],
        [9.9999e-01, 5.8236e-06],
        [3.8796e-09, 1.0000e+00],
        [9.9955e-01, 4.4822e-04],
        [1.7675e-07, 1.0000e+00],
        [9.9981e-01, 1.8573e-04],
        [5.3409e-06, 9.9999e-01],
        [2.0348e-03, 9.9797e-01],
        [1.6736e-08, 1.0000e+00],
        [1.9817e-03, 9.9802e-01],
        [5.3670e-11, 1.0000e+00],
        [9.9955e-01, 4.5224e-04],
        [3.7376e-14, 1.0000e+00],
        [9.9627e-07, 1.0000e+00],
        [2.6349e-02, 9.7365e-01],
        [7.7078e-05, 9.9992e-01],
        [1.6764e-03, 9.9832e-01],
        [9.8790e-01, 1.2099e-02],
        [1.2856e-05, 9.9999e-01],
        [7.5707e-01, 2.4293e-01],
        [9.9988e-01, 1.2150e-04],
        [1.2170e-10, 1.0000e+00],
        [3.1412e-04, 9.9969e-01],
        [4.6305e-01, 5.3695e-01],
        [1.0720e-08, 1.0000e+00],
        [9.9991e-01, 9.4706e-05],
        [1.9987e-11, 1.0000e+00],
        [9.9980e-01, 1.9845e-04],
        [1.5601e-05, 9.9998e-01],
        [3.0590e-06, 1.0000e+00],
        [8.0298e-13, 1.0000e+00],
        [6.3183e-10, 1.0000e+00],
        [5.9479e-08, 1.0000e+00],
        [1.6701e-06, 1.0000e+00],
        [1.3788e-15, 1.0000e+00],
        [1.8400e-10, 1.0000e+00],
        [9.9617e-01, 3.8260e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9919e-01, 8.0814e-04],
        [6.6323e-06, 9.9999e-01],
        [7.0389e-08, 1.0000e+00],
        [3.5677e-07, 1.0000e+00],
        [5.6181e-12, 1.0000e+00],
        [2.3120e-08, 1.0000e+00],
        [1.4265e-08, 1.0000e+00],
        [3.4444e-04, 9.9966e-01],
        [2.0377e-06, 1.0000e+00],
        [9.9709e-01, 2.9124e-03],
        [6.8251e-06, 9.9999e-01],
        [2.1435e-07, 1.0000e+00],
        [2.5017e-07, 1.0000e+00],
        [2.4552e-08, 1.0000e+00],
        [9.9990e-01, 9.9763e-05],
        [9.9976e-01, 2.4175e-04],
        [4.6675e-07, 1.0000e+00],
        [4.3013e-06, 1.0000e+00],
        [8.4772e-07, 1.0000e+00],
        [6.6280e-12, 1.0000e+00],
        [9.9984e-01, 1.6181e-04],
        [2.8285e-06, 1.0000e+00],
        [4.3490e-05, 9.9996e-01],
        [1.0298e-07, 1.0000e+00],
        [5.9809e-01, 4.0191e-01],
        [2.5186e-09, 1.0000e+00],
        [9.9994e-01, 6.2862e-05],
        [9.9991e-01, 8.8389e-05],
        [3.0330e-10, 1.0000e+00],
        [2.9323e-11, 1.0000e+00],
        [9.9045e-01, 9.5472e-03],
        [9.9101e-01, 8.9863e-03],
        [2.0274e-11, 1.0000e+00],
        [9.9990e-01, 1.0336e-04],
        [3.3987e-10, 1.0000e+00],
        [6.5953e-06, 9.9999e-01],
        [9.9998e-01, 2.0238e-05],
        [3.6520e-04, 9.9963e-01],
        [3.8985e-08, 1.0000e+00],
        [2.3714e-08, 1.0000e+00],
        [2.1255e-05, 9.9998e-01],
        [2.2111e-08, 1.0000e+00],
        [9.2186e-08, 1.0000e+00],
        [9.9982e-01, 1.7719e-04],
        [5.7045e-01, 4.2955e-01],
        [1.7754e-07, 1.0000e+00],
        [9.9991e-01, 8.6242e-05],
        [3.5913e-07, 1.0000e+00],
        [1.4499e-10, 1.0000e+00],
        [3.4116e-09, 1.0000e+00],
        [1.0291e-06, 1.0000e+00],
        [9.9991e-01, 8.5197e-05],
        [1.9547e-05, 9.9998e-01],
        [9.9973e-01, 2.6714e-04],
        [2.9242e-06, 1.0000e+00],
        [7.7163e-07, 1.0000e+00],
        [9.9987e-01, 1.3340e-04],
        [3.1648e-07, 1.0000e+00],
        [7.7560e-06, 9.9999e-01],
        [2.7824e-03, 9.9722e-01],
        [4.3814e-13, 1.0000e+00],
        [1.0707e-07, 1.0000e+00],
        [3.4093e-03, 9.9659e-01],
        [1.0881e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.5877e-07, 1.0000e+00],
        [1.8513e-06, 1.0000e+00],
        [8.0984e-08, 1.0000e+00],
        [9.3650e-07, 1.0000e+00],
        [9.9963e-01, 3.6750e-04],
        [9.9981e-01, 1.8901e-04],
        [1.5351e-08, 1.0000e+00],
        [5.8714e-05, 9.9994e-01],
        [3.0544e-07, 1.0000e+00],
        [5.4330e-07, 1.0000e+00],
        [1.4599e-12, 1.0000e+00],
        [9.9976e-01, 2.4294e-04],
        [9.9502e-01, 4.9826e-03],
        [7.8156e-04, 9.9922e-01],
        [2.0041e-06, 1.0000e+00],
        [1.1585e-03, 9.9884e-01],
        [1.1408e-04, 9.9989e-01],
        [8.9956e-09, 1.0000e+00],
        [1.3809e-05, 9.9999e-01],
        [7.7106e-01, 2.2894e-01],
        [7.9321e-01, 2.0679e-01],
        [2.6071e-08, 1.0000e+00],
        [3.4489e-10, 1.0000e+00],
        [2.5616e-13, 1.0000e+00],
        [5.7556e-04, 9.9942e-01],
        [2.3770e-10, 1.0000e+00],
        [8.9481e-03, 9.9105e-01],
        [9.7346e-01, 2.6542e-02],
        [9.9993e-01, 6.6001e-05],
        [1.0484e-06, 1.0000e+00],
        [1.0000e+00, 3.8001e-09],
        [1.2200e-08, 1.0000e+00],
        [4.1936e-09, 1.0000e+00],
        [1.0000e+00, 1.3139e-06],
        [1.2846e-09, 1.0000e+00],
        [2.8525e-10, 1.0000e+00],
        [9.9988e-01, 1.2048e-04],
        [7.1026e-11, 1.0000e+00],
        [5.9304e-10, 1.0000e+00],
        [7.4825e-09, 1.0000e+00],
        [4.4738e-01, 5.5262e-01],
        [9.9989e-01, 1.0881e-04],
        [9.9980e-01, 2.0035e-04],
        [8.3642e-07, 1.0000e+00],
        [9.9990e-01, 9.8430e-05],
        [9.9964e-01, 3.5495e-04],
        [1.1113e-08, 1.0000e+00],
        [2.0660e-09, 1.0000e+00],
        [2.3661e-10, 1.0000e+00],
        [7.2896e-03, 9.9271e-01],
        [6.8818e-08, 1.0000e+00],
        [8.5805e-06, 9.9999e-01],
        [9.9033e-05, 9.9990e-01],
        [9.9951e-01, 4.9085e-04],
        [7.9446e-01, 2.0554e-01],
        [6.9914e-07, 1.0000e+00],
        [5.5272e-15, 1.0000e+00],
        [9.9693e-01, 3.0657e-03],
        [9.9834e-01, 1.6560e-03],
        [1.6208e-10, 1.0000e+00],
        [4.5904e-09, 1.0000e+00],
        [9.9998e-01, 2.4467e-05],
        [8.8378e-04, 9.9912e-01],
        [2.1637e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.7970e-01, 7.2030e-01],
        [7.1725e-11, 1.0000e+00],
        [1.0933e-06, 1.0000e+00],
        [5.9948e-05, 9.9994e-01],
        [1.4743e-10, 1.0000e+00],
        [8.2384e-09, 1.0000e+00],
        [9.8699e-01, 1.3013e-02],
        [5.4415e-08, 1.0000e+00],
        [1.4862e-10, 1.0000e+00],
        [9.9999e-01, 1.0676e-05],
        [1.0580e-07, 1.0000e+00],
        [2.3339e-09, 1.0000e+00],
        [9.8604e-10, 1.0000e+00],
        [7.5150e-07, 1.0000e+00],
        [9.9860e-01, 1.3991e-03],
        [1.7297e-02, 9.8270e-01],
        [5.5621e-07, 1.0000e+00],
        [1.2398e-07, 1.0000e+00],
        [4.3095e-06, 1.0000e+00],
        [2.6130e-06, 1.0000e+00],
        [9.8524e-01, 1.4756e-02],
        [6.1575e-09, 1.0000e+00],
        [6.3548e-09, 1.0000e+00],
        [5.6798e-11, 1.0000e+00],
        [9.9771e-01, 2.2865e-03],
        [1.5324e-11, 1.0000e+00],
        [3.1906e-03, 9.9681e-01],
        [1.3048e-05, 9.9999e-01],
        [1.6069e-11, 1.0000e+00],
        [9.2215e-11, 1.0000e+00],
        [1.8099e-09, 1.0000e+00],
        [4.3774e-12, 1.0000e+00],
        [3.3536e-08, 1.0000e+00],
        [1.2236e-06, 1.0000e+00],
        [6.1742e-11, 1.0000e+00],
        [2.7289e-05, 9.9997e-01],
        [4.4404e-03, 9.9556e-01],
        [3.6027e-05, 9.9996e-01],
        [3.4851e-06, 1.0000e+00],
        [9.9884e-01, 1.1633e-03],
        [7.1107e-10, 1.0000e+00],
        [1.0096e-06, 1.0000e+00],
        [8.9475e-08, 1.0000e+00],
        [9.9995e-01, 4.6479e-05],
        [6.2968e-06, 9.9999e-01],
        [1.5508e-09, 1.0000e+00],
        [7.4218e-10, 1.0000e+00],
        [6.6744e-01, 3.3256e-01],
        [5.7338e-06, 9.9999e-01],
        [5.5206e-03, 9.9448e-01],
        [9.9986e-01, 1.4468e-04],
        [9.9979e-01, 2.0683e-04],
        [1.4862e-03, 9.9851e-01],
        [4.5497e-10, 1.0000e+00],
        [1.1738e-05, 9.9999e-01],
        [9.9991e-01, 9.2506e-05],
        [3.3573e-08, 1.0000e+00],
        [5.8472e-11, 1.0000e+00],
        [9.9997e-01, 3.0867e-05],
        [7.9148e-09, 1.0000e+00],
        [2.1037e-05, 9.9998e-01],
        [2.0099e-11, 1.0000e+00],
        [1.2925e-08, 1.0000e+00],
        [1.3600e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[4.1174e-07, 1.0000e+00],
        [2.6173e-03, 9.9738e-01],
        [9.9987e-01, 1.2943e-04],
        [2.5423e-09, 1.0000e+00],
        [1.8746e-11, 1.0000e+00],
        [3.7481e-06, 1.0000e+00],
        [4.7147e-05, 9.9995e-01],
        [1.5346e-06, 1.0000e+00],
        [8.2404e-09, 1.0000e+00],
        [8.8316e-10, 1.0000e+00],
        [2.1339e-06, 1.0000e+00],
        [9.9976e-01, 2.3691e-04],
        [4.7796e-10, 1.0000e+00],
        [3.1260e-10, 1.0000e+00],
        [5.0125e-11, 1.0000e+00],
        [7.4396e-07, 1.0000e+00],
        [7.5029e-07, 1.0000e+00],
        [9.6353e-01, 3.6471e-02],
        [6.3213e-07, 1.0000e+00],
        [5.3220e-04, 9.9947e-01],
        [1.2374e-10, 1.0000e+00],
        [2.8829e-10, 1.0000e+00],
        [9.9942e-12, 1.0000e+00],
        [7.7622e-07, 1.0000e+00],
        [9.9980e-01, 1.9879e-04],
        [3.4902e-06, 1.0000e+00],
        [6.4432e-08, 1.0000e+00],
        [7.7622e-07, 1.0000e+00],
        [1.8138e-06, 1.0000e+00],
        [9.9985e-01, 1.4509e-04],
        [2.9147e-02, 9.7085e-01],
        [1.2202e-06, 1.0000e+00],
        [8.5756e-07, 1.0000e+00],
        [9.9818e-01, 1.8220e-03],
        [4.6774e-08, 1.0000e+00],
        [9.9999e-01, 7.8654e-06],
        [8.8687e-01, 1.1313e-01],
        [5.3920e-08, 1.0000e+00],
        [5.1834e-12, 1.0000e+00],
        [3.6227e-12, 1.0000e+00],
        [4.9952e-02, 9.5005e-01],
        [2.4220e-12, 1.0000e+00],
        [1.3259e-06, 1.0000e+00],
        [1.9452e-07, 1.0000e+00],
        [9.8394e-01, 1.6057e-02],
        [1.0197e-01, 8.9803e-01],
        [1.7632e-14, 1.0000e+00],
        [1.8094e-09, 1.0000e+00],
        [3.1260e-10, 1.0000e+00],
        [9.9996e-01, 3.9351e-05],
        [9.9995e-01, 4.5757e-05],
        [5.9247e-10, 1.0000e+00],
        [2.3469e-04, 9.9977e-01],
        [2.1340e-06, 1.0000e+00],
        [1.8126e-12, 1.0000e+00],
        [9.9907e-01, 9.2559e-04],
        [1.5874e-05, 9.9998e-01],
        [5.9046e-05, 9.9994e-01],
        [7.7109e-06, 9.9999e-01],
        [1.5606e-09, 1.0000e+00],
        [1.5060e-06, 1.0000e+00],
        [4.4461e-11, 1.0000e+00],
        [4.7111e-04, 9.9953e-01],
        [9.9987e-01, 1.3273e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9768e-10, 1.0000e+00],
        [9.9968e-01, 3.2100e-04],
        [3.9531e-09, 1.0000e+00],
        [4.1026e-05, 9.9996e-01],
        [5.2584e-03, 9.9474e-01],
        [4.5368e-11, 1.0000e+00],
        [3.1114e-01, 6.8886e-01],
        [9.9998e-01, 1.5947e-05],
        [9.9977e-01, 2.3091e-04],
        [1.6740e-12, 1.0000e+00],
        [1.3255e-07, 1.0000e+00],
        [2.3318e-12, 1.0000e+00],
        [1.6561e-01, 8.3439e-01],
        [8.0202e-07, 1.0000e+00],
        [3.3582e-12, 1.0000e+00],
        [5.6957e-08, 1.0000e+00],
        [1.1871e-05, 9.9999e-01],
        [4.3812e-09, 1.0000e+00],
        [3.0834e-07, 1.0000e+00],
        [1.6740e-12, 1.0000e+00],
        [6.2877e-08, 1.0000e+00],
        [1.4765e-09, 1.0000e+00],
        [7.5294e-08, 1.0000e+00],
        [9.8787e-01, 1.2131e-02],
        [7.2446e-13, 1.0000e+00],
        [9.6987e-12, 1.0000e+00],
        [9.9990e-01, 1.0015e-04],
        [3.9065e-06, 1.0000e+00],
        [2.1064e-07, 1.0000e+00],
        [3.5403e-09, 1.0000e+00],
        [1.4693e-10, 1.0000e+00],
        [4.7751e-12, 1.0000e+00],
        [1.3207e-04, 9.9987e-01],
        [6.0396e-06, 9.9999e-01],
        [6.2027e-08, 1.0000e+00],
        [2.8823e-12, 1.0000e+00],
        [9.9987e-01, 1.2606e-04],
        [2.6617e-06, 1.0000e+00],
        [9.6726e-09, 1.0000e+00],
        [3.0812e-05, 9.9997e-01],
        [1.0029e-12, 1.0000e+00],
        [4.7432e-06, 1.0000e+00],
        [9.9200e-01, 8.0042e-03],
        [1.7718e-10, 1.0000e+00],
        [9.9974e-01, 2.6481e-04],
        [7.1540e-08, 1.0000e+00],
        [5.5674e-05, 9.9994e-01],
        [9.6967e-03, 9.9030e-01],
        [2.1154e-09, 1.0000e+00],
        [6.7847e-06, 9.9999e-01],
        [6.7402e-02, 9.3260e-01],
        [9.9992e-01, 7.8137e-05],
        [2.4922e-10, 1.0000e+00],
        [1.2366e-05, 9.9999e-01],
        [3.4892e-07, 1.0000e+00],
        [2.9547e-08, 1.0000e+00],
        [9.9992e-01, 7.5422e-05],
        [4.0059e-07, 1.0000e+00],
        [2.6234e-07, 1.0000e+00],
        [9.9999e-01, 9.6124e-06],
        [1.7371e-04, 9.9983e-01],
        [9.9999e-01, 9.9688e-06],
        [9.5907e-04, 9.9904e-01],
        [9.9039e-01, 9.6074e-03]], grad_fn=<SoftmaxBackward0>)
tensor([[9.8978e-01, 1.0223e-02],
        [2.0789e-09, 1.0000e+00],
        [6.3843e-02, 9.3616e-01],
        [2.7933e-07, 1.0000e+00],
        [8.8969e-04, 9.9911e-01],
        [1.6127e-06, 1.0000e+00],
        [6.6887e-06, 9.9999e-01],
        [1.1696e-06, 1.0000e+00],
        [3.0281e-07, 1.0000e+00],
        [3.3698e-09, 1.0000e+00],
        [1.1028e-03, 9.9890e-01],
        [4.7348e-12, 1.0000e+00],
        [5.5324e-07, 1.0000e+00],
        [1.1349e-08, 1.0000e+00],
        [4.6339e-09, 1.0000e+00],
        [9.7497e-01, 2.5032e-02],
        [1.2460e-09, 1.0000e+00],
        [1.6601e-05, 9.9998e-01],
        [9.9792e-01, 2.0764e-03],
        [9.9898e-01, 1.0231e-03],
        [9.9953e-01, 4.7300e-04],
        [5.3146e-01, 4.6854e-01],
        [1.1459e-07, 1.0000e+00],
        [5.9182e-06, 9.9999e-01],
        [6.6645e-10, 1.0000e+00],
        [2.3593e-06, 1.0000e+00],
        [2.7082e-11, 1.0000e+00],
        [8.4350e-03, 9.9156e-01],
        [1.2085e-04, 9.9988e-01],
        [2.5733e-07, 1.0000e+00],
        [4.3791e-07, 1.0000e+00],
        [7.3993e-01, 2.6007e-01],
        [2.8782e-09, 1.0000e+00],
        [2.7082e-11, 1.0000e+00],
        [3.1224e-12, 1.0000e+00],
        [4.7936e-07, 1.0000e+00],
        [4.1562e-03, 9.9584e-01],
        [6.3541e-06, 9.9999e-01],
        [2.6547e-13, 1.0000e+00],
        [9.9985e-01, 1.4662e-04],
        [2.0289e-08, 1.0000e+00],
        [6.5448e-08, 1.0000e+00],
        [1.2066e-07, 1.0000e+00],
        [2.4305e-05, 9.9998e-01],
        [3.9975e-07, 1.0000e+00],
        [9.9998e-01, 2.2110e-05],
        [9.9971e-01, 2.9100e-04],
        [9.2962e-10, 1.0000e+00],
        [1.3674e-12, 1.0000e+00],
        [9.9991e-01, 8.6048e-05],
        [5.1385e-07, 1.0000e+00],
        [2.9473e-07, 1.0000e+00],
        [8.2928e-09, 1.0000e+00],
        [9.9997e-01, 3.4076e-05],
        [4.0033e-05, 9.9996e-01],
        [1.1098e-08, 1.0000e+00],
        [2.1024e-11, 1.0000e+00],
        [2.6011e-07, 1.0000e+00],
        [9.0530e-13, 1.0000e+00],
        [1.9028e-06, 1.0000e+00],
        [4.3563e-05, 9.9996e-01],
        [9.2193e-08, 1.0000e+00],
        [7.3401e-07, 1.0000e+00],
        [1.6567e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9991e-01, 8.6705e-05],
        [1.6276e-04, 9.9984e-01],
        [7.0797e-06, 9.9999e-01],
        [2.7471e-06, 1.0000e+00],
        [1.0438e-06, 1.0000e+00],
        [9.9990e-01, 1.0321e-04],
        [1.2802e-07, 1.0000e+00],
        [1.1955e-09, 1.0000e+00],
        [5.0315e-08, 1.0000e+00],
        [2.5200e-05, 9.9997e-01],
        [2.8640e-09, 1.0000e+00],
        [9.9990e-01, 1.0075e-04],
        [9.5127e-10, 1.0000e+00],
        [3.3183e-08, 1.0000e+00],
        [2.2173e-06, 1.0000e+00],
        [9.9981e-01, 1.9127e-04],
        [3.9049e-07, 1.0000e+00],
        [1.3093e-09, 1.0000e+00],
        [4.4919e-05, 9.9996e-01],
        [1.2820e-07, 1.0000e+00],
        [1.6520e-06, 1.0000e+00],
        [4.5218e-08, 1.0000e+00],
        [2.0950e-01, 7.9050e-01],
        [1.8270e-08, 1.0000e+00],
        [7.7606e-01, 2.2394e-01],
        [3.5810e-09, 1.0000e+00],
        [5.4656e-08, 1.0000e+00],
        [5.8264e-10, 1.0000e+00],
        [9.9996e-01, 4.4621e-05],
        [1.7835e-09, 1.0000e+00],
        [9.1269e-05, 9.9991e-01],
        [2.1507e-06, 1.0000e+00],
        [9.7974e-01, 2.0258e-02],
        [9.9891e-01, 1.0940e-03],
        [2.6106e-04, 9.9974e-01],
        [9.9975e-01, 2.4797e-04],
        [8.9847e-04, 9.9910e-01],
        [4.6768e-07, 1.0000e+00],
        [3.1197e-07, 1.0000e+00],
        [6.7646e-02, 9.3235e-01],
        [9.4256e-07, 1.0000e+00],
        [9.9991e-01, 8.7498e-05],
        [1.6301e-06, 1.0000e+00],
        [1.0354e-03, 9.9896e-01],
        [3.0986e-03, 9.9690e-01],
        [2.5304e-04, 9.9975e-01],
        [1.4545e-06, 1.0000e+00],
        [2.1376e-06, 1.0000e+00],
        [9.3472e-09, 1.0000e+00],
        [6.8628e-09, 1.0000e+00],
        [7.3431e-10, 1.0000e+00],
        [1.0690e-05, 9.9999e-01],
        [9.9999e-01, 1.1325e-05],
        [1.4103e-10, 1.0000e+00],
        [9.9969e-01, 3.1388e-04],
        [1.1883e-08, 1.0000e+00],
        [9.9988e-01, 1.2275e-04],
        [3.9543e-06, 1.0000e+00],
        [6.0081e-11, 1.0000e+00],
        [4.3395e-06, 1.0000e+00],
        [3.1466e-05, 9.9997e-01],
        [4.4449e-05, 9.9996e-01],
        [1.6500e-07, 1.0000e+00],
        [3.2958e-05, 9.9997e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3911e-04, 9.9976e-01],
        [2.0767e-03, 9.9792e-01],
        [8.3590e-14, 1.0000e+00],
        [8.8382e-09, 1.0000e+00],
        [2.1467e-07, 1.0000e+00],
        [9.9951e-01, 4.8513e-04],
        [3.6742e-13, 1.0000e+00],
        [1.6229e-08, 1.0000e+00],
        [8.2917e-01, 1.7083e-01],
        [1.0000e+00, 8.9625e-07],
        [1.7574e-09, 1.0000e+00],
        [9.4371e-12, 1.0000e+00],
        [9.1260e-06, 9.9999e-01],
        [2.9247e-05, 9.9997e-01],
        [1.9917e-11, 1.0000e+00],
        [3.5285e-10, 1.0000e+00],
        [1.9454e-06, 1.0000e+00],
        [2.1343e-04, 9.9979e-01],
        [3.7693e-08, 1.0000e+00],
        [2.0801e-03, 9.9792e-01],
        [2.5365e-06, 1.0000e+00],
        [9.9035e-01, 9.6451e-03],
        [1.2560e-07, 1.0000e+00],
        [3.6499e-08, 1.0000e+00],
        [9.9992e-01, 8.2376e-05],
        [2.8549e-07, 1.0000e+00],
        [9.9968e-01, 3.1609e-04],
        [9.9295e-01, 7.0485e-03],
        [7.6714e-07, 1.0000e+00],
        [9.0128e-07, 1.0000e+00],
        [9.1593e-07, 1.0000e+00],
        [1.4339e-08, 1.0000e+00],
        [1.9451e-10, 1.0000e+00],
        [9.9985e-01, 1.4756e-04],
        [6.2792e-09, 1.0000e+00],
        [1.6612e-12, 1.0000e+00],
        [9.9975e-01, 2.4670e-04],
        [8.9130e-11, 1.0000e+00],
        [1.3307e-08, 1.0000e+00],
        [9.9997e-01, 3.0711e-05],
        [5.6944e-08, 1.0000e+00],
        [5.5154e-06, 9.9999e-01],
        [9.9998e-01, 1.8615e-05],
        [7.1896e-04, 9.9928e-01],
        [9.3974e-02, 9.0603e-01],
        [2.4019e-12, 1.0000e+00],
        [9.9989e-01, 1.1033e-04],
        [9.9765e-01, 2.3500e-03],
        [2.8183e-06, 1.0000e+00],
        [6.2039e-01, 3.7961e-01],
        [1.7380e-06, 1.0000e+00],
        [3.6040e-12, 1.0000e+00],
        [1.0000e+00, 5.4860e-08],
        [9.9997e-01, 2.7743e-05],
        [9.5640e-08, 1.0000e+00],
        [1.2837e-01, 8.7163e-01],
        [1.7970e-10, 1.0000e+00],
        [9.9992e-01, 8.3486e-05],
        [3.6065e-06, 1.0000e+00],
        [9.9993e-01, 7.0572e-05],
        [2.5224e-06, 1.0000e+00],
        [9.9901e-01, 9.8548e-04],
        [9.9992e-01, 7.8057e-05],
        [9.9903e-01, 9.6972e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2943e-10, 1.0000e+00],
        [7.2872e-08, 1.0000e+00],
        [9.9945e-01, 5.5476e-04],
        [5.8223e-07, 1.0000e+00],
        [1.7936e-08, 1.0000e+00],
        [2.1217e-09, 1.0000e+00],
        [3.0204e-11, 1.0000e+00],
        [3.9230e-06, 1.0000e+00],
        [9.3335e-08, 1.0000e+00],
        [1.6717e-04, 9.9983e-01],
        [1.8152e-06, 1.0000e+00],
        [8.7219e-09, 1.0000e+00],
        [3.5052e-08, 1.0000e+00],
        [5.7415e-07, 1.0000e+00],
        [1.2356e-02, 9.8764e-01],
        [8.6929e-09, 1.0000e+00],
        [3.0075e-02, 9.6993e-01],
        [1.2065e-07, 1.0000e+00],
        [7.7704e-07, 1.0000e+00],
        [9.5496e-09, 1.0000e+00],
        [9.7073e-11, 1.0000e+00],
        [1.8112e-08, 1.0000e+00],
        [6.2896e-07, 1.0000e+00],
        [9.9978e-01, 2.1961e-04],
        [7.3316e-05, 9.9993e-01],
        [2.0056e-07, 1.0000e+00],
        [5.5125e-13, 1.0000e+00],
        [9.9986e-01, 1.4373e-04],
        [7.4752e-03, 9.9252e-01],
        [9.1488e-03, 9.9085e-01],
        [3.0975e-07, 1.0000e+00],
        [8.4160e-12, 1.0000e+00],
        [9.9997e-01, 2.6403e-05],
        [9.9990e-01, 9.5599e-05],
        [8.5903e-09, 1.0000e+00],
        [4.4221e-05, 9.9996e-01],
        [9.9997e-01, 3.4731e-05],
        [3.5142e-07, 1.0000e+00],
        [1.5815e-10, 1.0000e+00],
        [2.5359e-06, 1.0000e+00],
        [9.9995e-01, 4.6832e-05],
        [9.0919e-08, 1.0000e+00],
        [2.6943e-07, 1.0000e+00],
        [2.6943e-02, 9.7306e-01],
        [4.8401e-09, 1.0000e+00],
        [9.5367e-01, 4.6332e-02],
        [3.7507e-08, 1.0000e+00],
        [1.0684e-09, 1.0000e+00],
        [6.4342e-05, 9.9994e-01],
        [4.4107e-05, 9.9996e-01],
        [1.5235e-10, 1.0000e+00],
        [1.0955e-08, 1.0000e+00],
        [9.9703e-01, 2.9735e-03],
        [9.9994e-01, 5.8165e-05],
        [4.3584e-07, 1.0000e+00],
        [5.7600e-07, 1.0000e+00],
        [3.1206e-08, 1.0000e+00],
        [1.1061e-05, 9.9999e-01],
        [2.8129e-07, 1.0000e+00],
        [2.2388e-09, 1.0000e+00],
        [9.9998e-01, 1.6195e-05],
        [9.9985e-01, 1.4843e-04],
        [3.7829e-05, 9.9996e-01],
        [5.2014e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.6770e-09, 1.0000e+00],
        [2.9984e-06, 1.0000e+00],
        [4.7095e-07, 1.0000e+00],
        [5.3875e-04, 9.9946e-01],
        [2.9065e-07, 1.0000e+00],
        [5.3936e-05, 9.9995e-01],
        [3.1930e-09, 1.0000e+00],
        [2.1612e-08, 1.0000e+00],
        [1.7869e-09, 1.0000e+00],
        [4.1117e-05, 9.9996e-01],
        [1.3850e-05, 9.9999e-01],
        [9.8648e-01, 1.3523e-02],
        [1.4783e-04, 9.9985e-01],
        [8.2563e-08, 1.0000e+00],
        [8.3584e-06, 9.9999e-01],
        [2.2938e-03, 9.9771e-01],
        [3.0379e-08, 1.0000e+00],
        [5.7422e-10, 1.0000e+00],
        [9.9990e-01, 9.5754e-05],
        [5.8634e-05, 9.9994e-01],
        [1.1742e-04, 9.9988e-01],
        [1.6035e-07, 1.0000e+00],
        [2.0377e-06, 1.0000e+00],
        [9.9939e-01, 6.1347e-04],
        [2.6931e-02, 9.7307e-01],
        [8.2014e-06, 9.9999e-01],
        [2.5684e-06, 1.0000e+00],
        [4.3190e-10, 1.0000e+00],
        [2.0326e-11, 1.0000e+00],
        [2.6414e-04, 9.9974e-01],
        [9.6723e-09, 1.0000e+00],
        [9.9980e-01, 1.9639e-04],
        [8.8909e-09, 1.0000e+00],
        [9.6017e-05, 9.9990e-01],
        [9.9995e-01, 5.1021e-05],
        [3.4475e-09, 1.0000e+00],
        [2.1273e-06, 1.0000e+00],
        [9.0967e-01, 9.0329e-02],
        [1.7767e-11, 1.0000e+00],
        [1.1992e-09, 1.0000e+00],
        [8.5940e-12, 1.0000e+00],
        [1.2937e-04, 9.9987e-01],
        [4.6876e-05, 9.9995e-01],
        [5.2406e-06, 9.9999e-01],
        [9.9961e-01, 3.8881e-04],
        [9.0834e-03, 9.9092e-01],
        [6.5432e-08, 1.0000e+00],
        [1.2815e-09, 1.0000e+00],
        [9.9992e-01, 7.7078e-05],
        [8.8635e-08, 1.0000e+00],
        [1.0188e-09, 1.0000e+00],
        [5.1958e-08, 1.0000e+00],
        [9.0853e-09, 1.0000e+00],
        [2.6468e-04, 9.9974e-01],
        [1.4305e-09, 1.0000e+00],
        [9.9995e-01, 5.4921e-05],
        [7.7521e-11, 1.0000e+00],
        [1.0000e+00, 6.7001e-07],
        [6.6575e-10, 1.0000e+00],
        [3.0662e-05, 9.9997e-01],
        [9.3208e-06, 9.9999e-01],
        [9.9983e-01, 1.7472e-04],
        [1.3573e-10, 1.0000e+00],
        [9.9994e-01, 6.3053e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.0817e-08, 1.0000e+00],
        [8.1449e-08, 1.0000e+00],
        [9.9971e-01, 2.8806e-04],
        [3.0931e-05, 9.9997e-01],
        [1.0140e-03, 9.9899e-01],
        [7.1044e-01, 2.8956e-01],
        [2.2677e-06, 1.0000e+00],
        [9.9972e-01, 2.7774e-04],
        [2.1274e-04, 9.9979e-01],
        [2.2277e-11, 1.0000e+00],
        [9.9992e-01, 7.8369e-05],
        [1.7612e-03, 9.9824e-01],
        [9.2601e-07, 1.0000e+00],
        [1.0000e+00, 1.3467e-07],
        [5.0173e-01, 4.9827e-01],
        [2.2069e-05, 9.9998e-01],
        [9.8837e-01, 1.1629e-02],
        [9.9935e-01, 6.5054e-04],
        [2.2698e-09, 1.0000e+00],
        [4.3600e-09, 1.0000e+00],
        [5.4678e-10, 1.0000e+00],
        [1.8564e-04, 9.9981e-01],
        [1.1583e-09, 1.0000e+00],
        [9.9995e-01, 5.4608e-05],
        [1.1932e-09, 1.0000e+00],
        [2.2292e-09, 1.0000e+00],
        [5.1848e-11, 1.0000e+00],
        [9.9993e-01, 7.0647e-05],
        [9.9985e-01, 1.5295e-04],
        [9.9413e-01, 5.8717e-03],
        [5.1848e-11, 1.0000e+00],
        [1.5316e-02, 9.8468e-01],
        [3.3243e-09, 1.0000e+00],
        [9.7421e-06, 9.9999e-01],
        [6.5773e-05, 9.9993e-01],
        [1.8850e-03, 9.9812e-01],
        [4.1286e-05, 9.9996e-01],
        [9.8942e-01, 1.0577e-02],
        [9.9981e-01, 1.8807e-04],
        [9.9996e-01, 4.0950e-05],
        [1.5983e-09, 1.0000e+00],
        [6.7745e-11, 1.0000e+00],
        [5.6221e-03, 9.9438e-01],
        [1.0370e-10, 1.0000e+00],
        [1.2576e-09, 1.0000e+00],
        [1.1226e-05, 9.9999e-01],
        [2.5144e-07, 1.0000e+00],
        [1.4532e-10, 1.0000e+00],
        [8.9546e-06, 9.9999e-01],
        [3.0494e-10, 1.0000e+00],
        [9.9997e-01, 2.9079e-05],
        [1.1559e-07, 1.0000e+00],
        [9.9985e-01, 1.5469e-04],
        [1.1955e-04, 9.9988e-01],
        [6.3590e-08, 1.0000e+00],
        [3.8525e-06, 1.0000e+00],
        [4.7484e-07, 1.0000e+00],
        [3.6665e-12, 1.0000e+00],
        [4.8758e-08, 1.0000e+00],
        [9.8070e-01, 1.9296e-02],
        [1.1221e-05, 9.9999e-01],
        [1.3785e-05, 9.9999e-01],
        [4.3774e-09, 1.0000e+00],
        [3.3202e-10, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.0938e-15, 1.0000e+00],
        [7.5709e-08, 1.0000e+00],
        [9.4831e-06, 9.9999e-01],
        [1.7649e-06, 1.0000e+00],
        [4.2629e-07, 1.0000e+00],
        [9.9993e-01, 6.7670e-05],
        [2.3866e-05, 9.9998e-01],
        [1.8725e-04, 9.9981e-01],
        [9.9806e-01, 1.9366e-03],
        [2.2552e-03, 9.9774e-01],
        [1.9164e-08, 1.0000e+00],
        [1.7789e-07, 1.0000e+00],
        [9.9990e-01, 9.6422e-05],
        [5.4311e-07, 1.0000e+00],
        [1.9204e-10, 1.0000e+00],
        [6.8490e-10, 1.0000e+00],
        [1.0657e-06, 1.0000e+00],
        [2.9898e-09, 1.0000e+00],
        [3.2532e-09, 1.0000e+00],
        [7.6871e-08, 1.0000e+00],
        [9.9985e-01, 1.4931e-04],
        [2.8631e-09, 1.0000e+00],
        [9.4526e-09, 1.0000e+00],
        [9.4187e-01, 5.8125e-02],
        [1.4520e-11, 1.0000e+00],
        [5.3715e-06, 9.9999e-01],
        [7.4961e-08, 1.0000e+00],
        [5.3817e-06, 9.9999e-01],
        [1.3968e-02, 9.8603e-01],
        [8.9549e-07, 1.0000e+00],
        [2.8619e-13, 1.0000e+00],
        [3.9823e-09, 1.0000e+00],
        [3.8750e-03, 9.9612e-01],
        [4.2206e-09, 1.0000e+00],
        [7.6773e-10, 1.0000e+00],
        [9.9965e-01, 3.4761e-04],
        [9.9952e-01, 4.7700e-04],
        [9.9975e-01, 2.4903e-04],
        [9.9973e-01, 2.7256e-04],
        [5.2581e-09, 1.0000e+00],
        [1.2918e-06, 1.0000e+00],
        [4.3647e-04, 9.9956e-01],
        [9.9995e-01, 5.3066e-05],
        [9.9997e-01, 3.2929e-05],
        [1.6295e-06, 1.0000e+00],
        [2.6710e-07, 1.0000e+00],
        [8.9068e-03, 9.9109e-01],
        [9.9994e-01, 6.1562e-05],
        [2.5297e-06, 1.0000e+00],
        [9.9979e-01, 2.1242e-04],
        [9.8868e-01, 1.1324e-02],
        [1.0521e-09, 1.0000e+00],
        [1.0498e-05, 9.9999e-01],
        [9.9987e-01, 1.2521e-04],
        [4.2804e-09, 1.0000e+00],
        [1.7462e-04, 9.9983e-01],
        [9.9997e-01, 2.8418e-05],
        [2.4254e-09, 1.0000e+00],
        [9.9994e-01, 5.6460e-05],
        [4.7950e-07, 1.0000e+00],
        [9.9998e-01, 1.8697e-05],
        [2.9059e-05, 9.9997e-01],
        [1.9876e-11, 1.0000e+00],
        [1.3009e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7340e-02, 9.8266e-01],
        [9.9974e-01, 2.6014e-04],
        [7.4667e-05, 9.9993e-01],
        [6.3331e-02, 9.3667e-01],
        [3.6631e-09, 1.0000e+00],
        [9.7345e-01, 2.6550e-02],
        [9.9954e-01, 4.5536e-04],
        [9.8469e-01, 1.5309e-02],
        [1.9726e-09, 1.0000e+00],
        [9.9993e-01, 7.3344e-05],
        [1.1076e-08, 1.0000e+00],
        [3.5911e-02, 9.6409e-01],
        [9.9781e-01, 2.1922e-03],
        [9.9989e-01, 1.1198e-04],
        [5.3019e-07, 1.0000e+00],
        [4.2147e-06, 1.0000e+00],
        [4.9110e-05, 9.9995e-01],
        [3.6111e-06, 1.0000e+00],
        [1.7152e-11, 1.0000e+00],
        [9.9999e-01, 1.3076e-05],
        [4.8627e-09, 1.0000e+00],
        [1.3589e-01, 8.6411e-01],
        [5.9127e-12, 1.0000e+00],
        [7.8031e-06, 9.9999e-01],
        [2.2683e-07, 1.0000e+00],
        [1.1448e-11, 1.0000e+00],
        [9.8945e-01, 1.0550e-02],
        [9.9888e-01, 1.1156e-03],
        [3.8013e-11, 1.0000e+00],
        [4.5150e-10, 1.0000e+00],
        [7.9085e-07, 1.0000e+00],
        [5.1463e-07, 1.0000e+00],
        [9.9985e-01, 1.5381e-04],
        [1.0156e-08, 1.0000e+00],
        [9.9989e-01, 1.1425e-04],
        [1.1488e-05, 9.9999e-01],
        [2.6390e-08, 1.0000e+00],
        [7.7859e-01, 2.2141e-01],
        [2.8201e-05, 9.9997e-01],
        [1.4337e-08, 1.0000e+00],
        [4.0829e-04, 9.9959e-01],
        [6.0946e-07, 1.0000e+00],
        [9.9987e-01, 1.2668e-04],
        [9.2995e-09, 1.0000e+00],
        [3.0666e-09, 1.0000e+00],
        [7.4781e-08, 1.0000e+00],
        [1.1359e-09, 1.0000e+00],
        [9.9856e-01, 1.4408e-03],
        [2.0186e-04, 9.9980e-01],
        [9.9912e-01, 8.8352e-04],
        [3.2735e-03, 9.9673e-01],
        [3.6782e-06, 1.0000e+00],
        [9.9984e-01, 1.6438e-04],
        [8.8855e-01, 1.1145e-01],
        [9.0950e-05, 9.9991e-01],
        [1.0000e+00, 1.2672e-06],
        [2.8637e-09, 1.0000e+00],
        [9.9993e-01, 6.5401e-05],
        [3.2363e-05, 9.9997e-01],
        [1.8529e-09, 1.0000e+00],
        [9.9214e-01, 7.8562e-03],
        [3.1419e-08, 1.0000e+00],
        [1.2012e-04, 9.9988e-01],
        [5.1325e-05, 9.9995e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.3010e-05, 9.9999e-01],
        [2.9149e-07, 1.0000e+00],
        [9.9999e-01, 1.1437e-05],
        [1.3073e-06, 1.0000e+00],
        [9.9985e-01, 1.4745e-04],
        [1.6243e-08, 1.0000e+00],
        [1.1408e-06, 1.0000e+00],
        [1.5708e-08, 1.0000e+00],
        [4.3896e-03, 9.9561e-01],
        [1.4448e-12, 1.0000e+00],
        [9.9818e-01, 1.8157e-03],
        [2.6293e-04, 9.9974e-01],
        [6.7501e-05, 9.9993e-01],
        [9.2686e-01, 7.3141e-02],
        [2.5391e-06, 1.0000e+00],
        [2.6146e-09, 1.0000e+00],
        [3.5720e-06, 1.0000e+00],
        [1.1272e-08, 1.0000e+00],
        [1.9912e-10, 1.0000e+00],
        [1.2784e-04, 9.9987e-01],
        [4.2062e-09, 1.0000e+00],
        [8.2618e-07, 1.0000e+00],
        [9.9965e-01, 3.4728e-04],
        [3.4482e-05, 9.9997e-01],
        [2.1506e-09, 1.0000e+00],
        [3.9619e-08, 1.0000e+00],
        [5.8097e-05, 9.9994e-01],
        [2.4771e-07, 1.0000e+00],
        [5.4127e-08, 1.0000e+00],
        [1.5661e-07, 1.0000e+00],
        [9.9998e-01, 2.1131e-05],
        [2.0900e-06, 1.0000e+00],
        [6.9711e-13, 1.0000e+00],
        [8.0549e-09, 1.0000e+00],
        [1.3536e-08, 1.0000e+00],
        [1.4033e-01, 8.5967e-01],
        [2.1898e-06, 1.0000e+00],
        [7.3909e-07, 1.0000e+00],
        [9.5476e-01, 4.5237e-02],
        [1.0524e-04, 9.9989e-01],
        [9.9992e-01, 8.1025e-05],
        [9.9997e-01, 3.4860e-05],
        [9.9911e-01, 8.8835e-04],
        [3.4218e-07, 1.0000e+00],
        [9.9999e-01, 9.6167e-06],
        [8.6685e-06, 9.9999e-01],
        [3.1488e-06, 1.0000e+00],
        [9.5081e-06, 9.9999e-01],
        [9.9932e-01, 6.8035e-04],
        [9.5373e-01, 4.6268e-02],
        [1.7306e-11, 1.0000e+00],
        [9.9990e-01, 1.0273e-04],
        [1.8370e-12, 1.0000e+00],
        [2.3224e-09, 1.0000e+00],
        [9.9269e-01, 7.3051e-03],
        [9.9998e-01, 1.8135e-05],
        [9.9986e-01, 1.4355e-04],
        [9.9977e-01, 2.2733e-04],
        [9.8436e-07, 1.0000e+00],
        [4.0876e-05, 9.9996e-01],
        [1.8801e-07, 1.0000e+00],
        [1.2035e-10, 1.0000e+00],
        [7.4592e-12, 1.0000e+00],
        [9.9989e-01, 1.1153e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[6.0975e-08, 1.0000e+00],
        [9.9969e-01, 3.1047e-04],
        [1.0000e+00, 2.4204e-06],
        [4.8010e-09, 1.0000e+00],
        [3.7815e-08, 1.0000e+00],
        [6.4069e-09, 1.0000e+00],
        [2.1483e-05, 9.9998e-01],
        [8.0381e-08, 1.0000e+00],
        [1.9146e-09, 1.0000e+00],
        [4.6213e-03, 9.9538e-01],
        [2.0744e-09, 1.0000e+00],
        [1.1445e-04, 9.9989e-01],
        [1.9237e-03, 9.9808e-01],
        [9.9996e-01, 4.3198e-05],
        [6.9624e-07, 1.0000e+00],
        [3.6169e-07, 1.0000e+00],
        [2.6638e-03, 9.9734e-01],
        [7.2857e-13, 1.0000e+00],
        [1.6027e-05, 9.9998e-01],
        [9.9988e-01, 1.1528e-04],
        [1.0478e-06, 1.0000e+00],
        [1.4779e-04, 9.9985e-01],
        [6.5638e-04, 9.9934e-01],
        [1.1995e-02, 9.8801e-01],
        [9.9637e-01, 3.6329e-03],
        [1.2297e-06, 1.0000e+00],
        [2.8218e-05, 9.9997e-01],
        [6.7599e-09, 1.0000e+00],
        [7.2119e-04, 9.9928e-01],
        [8.2386e-08, 1.0000e+00],
        [9.9946e-01, 5.4282e-04],
        [4.1625e-10, 1.0000e+00],
        [2.6908e-11, 1.0000e+00],
        [6.0035e-09, 1.0000e+00],
        [4.4750e-03, 9.9552e-01],
        [8.2891e-08, 1.0000e+00],
        [9.5685e-01, 4.3145e-02],
        [1.1440e-06, 1.0000e+00],
        [3.6716e-08, 1.0000e+00],
        [9.9989e-01, 1.1261e-04],
        [4.3197e-03, 9.9568e-01],
        [9.9974e-01, 2.6213e-04],
        [5.8952e-09, 1.0000e+00],
        [3.0333e-06, 1.0000e+00],
        [9.9991e-01, 8.8416e-05],
        [9.9999e-01, 1.1449e-05],
        [9.9981e-01, 1.9382e-04],
        [9.9998e-01, 2.2244e-05],
        [9.1408e-01, 8.5915e-02],
        [9.9934e-01, 6.5687e-04],
        [9.9999e-01, 7.2776e-06],
        [1.2613e-11, 1.0000e+00],
        [3.0431e-04, 9.9970e-01],
        [7.0252e-06, 9.9999e-01],
        [6.3947e-01, 3.6053e-01],
        [1.0753e-08, 1.0000e+00],
        [9.6492e-01, 3.5079e-02],
        [7.5506e-06, 9.9999e-01],
        [1.1756e-09, 1.0000e+00],
        [2.4771e-03, 9.9752e-01],
        [9.9905e-01, 9.4705e-04],
        [1.0287e-03, 9.9897e-01],
        [4.9051e-09, 1.0000e+00],
        [8.8675e-12, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[5.9364e-09, 1.0000e+00],
        [3.1726e-09, 1.0000e+00],
        [1.4596e-08, 1.0000e+00],
        [7.1521e-05, 9.9993e-01],
        [8.5240e-08, 1.0000e+00],
        [1.9278e-07, 1.0000e+00],
        [9.9987e-01, 1.2813e-04],
        [1.6077e-06, 1.0000e+00],
        [9.7138e-06, 9.9999e-01],
        [5.9364e-09, 1.0000e+00],
        [2.2941e-07, 1.0000e+00],
        [9.9994e-01, 5.6441e-05],
        [9.9954e-01, 4.6202e-04],
        [2.8062e-10, 1.0000e+00],
        [3.7817e-07, 1.0000e+00],
        [5.9817e-05, 9.9994e-01],
        [5.9931e-03, 9.9401e-01],
        [1.5387e-08, 1.0000e+00],
        [1.1215e-05, 9.9999e-01],
        [8.6477e-07, 1.0000e+00],
        [1.8637e-09, 1.0000e+00],
        [9.9995e-01, 5.4145e-05],
        [1.2046e-05, 9.9999e-01],
        [1.0744e-09, 1.0000e+00],
        [3.3817e-07, 1.0000e+00],
        [6.8357e-09, 1.0000e+00],
        [9.9995e-01, 5.1507e-05],
        [8.5882e-15, 1.0000e+00],
        [6.0546e-01, 3.9454e-01],
        [6.6515e-07, 1.0000e+00],
        [9.9997e-01, 3.0740e-05],
        [1.3674e-05, 9.9999e-01],
        [9.9976e-01, 2.4200e-04],
        [9.9872e-01, 1.2787e-03],
        [3.9140e-07, 1.0000e+00],
        [1.3386e-09, 1.0000e+00],
        [8.6212e-06, 9.9999e-01],
        [3.5480e-06, 1.0000e+00],
        [9.9971e-01, 2.9493e-04],
        [6.5047e-09, 1.0000e+00],
        [2.7154e-08, 1.0000e+00],
        [1.2468e-04, 9.9988e-01],
        [5.6760e-07, 1.0000e+00],
        [7.2472e-01, 2.7528e-01],
        [1.5842e-03, 9.9842e-01],
        [2.0405e-05, 9.9998e-01],
        [1.7722e-03, 9.9823e-01],
        [7.7575e-04, 9.9922e-01],
        [5.4985e-03, 9.9450e-01],
        [2.8854e-09, 1.0000e+00],
        [9.9116e-01, 8.8444e-03],
        [9.2634e-08, 1.0000e+00],
        [1.9576e-11, 1.0000e+00],
        [1.4651e-08, 1.0000e+00],
        [6.4703e-02, 9.3530e-01],
        [9.7531e-01, 2.4689e-02],
        [7.6009e-06, 9.9999e-01],
        [1.0804e-07, 1.0000e+00],
        [9.9997e-01, 2.6960e-05],
        [1.2492e-06, 1.0000e+00],
        [2.0579e-09, 1.0000e+00],
        [9.0832e-07, 1.0000e+00],
        [1.8154e-07, 1.0000e+00],
        [9.9968e-01, 3.2127e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.1991e-08, 1.0000e+00],
        [1.1464e-03, 9.9885e-01],
        [6.8281e-09, 1.0000e+00],
        [7.8547e-10, 1.0000e+00],
        [2.0733e-07, 1.0000e+00],
        [9.9788e-01, 2.1199e-03],
        [2.7354e-07, 1.0000e+00],
        [9.9984e-01, 1.6114e-04],
        [6.5125e-10, 1.0000e+00],
        [6.0552e-01, 3.9448e-01],
        [3.0337e-09, 1.0000e+00],
        [6.4763e-01, 3.5237e-01],
        [9.9986e-01, 1.4356e-04],
        [9.9998e-01, 1.5584e-05],
        [9.9856e-01, 1.4422e-03],
        [3.8240e-11, 1.0000e+00],
        [2.7369e-06, 1.0000e+00],
        [9.2993e-01, 7.0074e-02],
        [7.4271e-09, 1.0000e+00],
        [1.0651e-04, 9.9989e-01],
        [1.1640e-05, 9.9999e-01],
        [1.4072e-06, 1.0000e+00],
        [1.5728e-06, 1.0000e+00],
        [4.9300e-08, 1.0000e+00],
        [9.9988e-01, 1.2058e-04],
        [1.1330e-06, 1.0000e+00],
        [5.2179e-09, 1.0000e+00],
        [1.0010e-08, 1.0000e+00],
        [2.4174e-08, 1.0000e+00],
        [1.6301e-04, 9.9984e-01],
        [3.7639e-05, 9.9996e-01],
        [5.2972e-04, 9.9947e-01],
        [6.0762e-06, 9.9999e-01],
        [9.3111e-01, 6.8894e-02],
        [1.5247e-10, 1.0000e+00],
        [1.5639e-06, 1.0000e+00],
        [3.9396e-11, 1.0000e+00],
        [9.9988e-01, 1.2042e-04],
        [4.3027e-10, 1.0000e+00],
        [1.0010e-08, 1.0000e+00],
        [5.5919e-05, 9.9994e-01],
        [9.7598e-09, 1.0000e+00],
        [4.3142e-09, 1.0000e+00],
        [2.9904e-09, 1.0000e+00],
        [6.3890e-01, 3.6110e-01],
        [2.2074e-06, 1.0000e+00],
        [6.8191e-06, 9.9999e-01],
        [1.9063e-06, 1.0000e+00],
        [9.4148e-09, 1.0000e+00],
        [1.2624e-04, 9.9987e-01],
        [9.9995e-01, 4.5636e-05],
        [9.0524e-14, 1.0000e+00],
        [1.7455e-07, 1.0000e+00],
        [8.9832e-06, 9.9999e-01],
        [9.9867e-01, 1.3315e-03],
        [1.1474e-05, 9.9999e-01],
        [3.8703e-01, 6.1297e-01],
        [1.5965e-01, 8.4035e-01],
        [8.2100e-09, 1.0000e+00],
        [8.1337e-10, 1.0000e+00],
        [9.9949e-01, 5.1067e-04],
        [9.9997e-01, 2.8993e-05],
        [1.0726e-10, 1.0000e+00],
        [9.9992e-01, 7.8034e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2170e-09, 1.0000e+00],
        [9.9995e-01, 5.0985e-05],
        [9.9986e-01, 1.3980e-04],
        [3.9896e-04, 9.9960e-01],
        [5.2678e-09, 1.0000e+00],
        [5.8701e-10, 1.0000e+00],
        [1.9810e-07, 1.0000e+00],
        [9.5706e-01, 4.2935e-02],
        [9.9660e-01, 3.4041e-03],
        [1.1811e-01, 8.8189e-01],
        [1.6621e-04, 9.9983e-01],
        [1.2601e-03, 9.9874e-01],
        [9.8487e-01, 1.5130e-02],
        [3.7305e-10, 1.0000e+00],
        [9.9988e-01, 1.1868e-04],
        [9.9960e-01, 4.0076e-04],
        [9.9954e-01, 4.5745e-04],
        [9.9994e-01, 6.0919e-05],
        [9.9978e-01, 2.1583e-04],
        [1.4940e-07, 1.0000e+00],
        [1.3137e-05, 9.9999e-01],
        [5.4840e-12, 1.0000e+00],
        [1.9227e-05, 9.9998e-01],
        [1.4526e-03, 9.9855e-01],
        [4.1765e-04, 9.9958e-01],
        [2.2003e-08, 1.0000e+00],
        [5.6627e-06, 9.9999e-01],
        [1.7217e-06, 1.0000e+00],
        [9.9796e-01, 2.0373e-03],
        [1.8211e-08, 1.0000e+00],
        [3.2586e-10, 1.0000e+00],
        [1.2593e-04, 9.9987e-01],
        [3.7896e-10, 1.0000e+00],
        [9.9897e-01, 1.0309e-03],
        [1.8894e-05, 9.9998e-01],
        [1.8201e-05, 9.9998e-01],
        [6.0527e-09, 1.0000e+00],
        [1.6527e-04, 9.9983e-01],
        [1.5027e-03, 9.9850e-01],
        [9.9998e-01, 2.1294e-05],
        [1.1095e-03, 9.9889e-01],
        [1.9735e-02, 9.8027e-01],
        [5.1194e-06, 9.9999e-01],
        [9.9992e-01, 8.0032e-05],
        [1.3034e-03, 9.9870e-01],
        [5.3398e-12, 1.0000e+00],
        [9.9060e-01, 9.4007e-03],
        [5.8578e-06, 9.9999e-01],
        [1.6388e-06, 1.0000e+00],
        [1.4940e-07, 1.0000e+00],
        [3.8144e-09, 1.0000e+00],
        [8.0373e-08, 1.0000e+00],
        [9.9985e-01, 1.4638e-04],
        [2.0067e-07, 1.0000e+00],
        [1.7082e-09, 1.0000e+00],
        [4.5127e-11, 1.0000e+00],
        [4.1867e-11, 1.0000e+00],
        [2.0985e-02, 9.7902e-01],
        [1.3602e-07, 1.0000e+00],
        [9.9988e-01, 1.1782e-04],
        [3.6338e-10, 1.0000e+00],
        [9.9932e-01, 6.8007e-04],
        [8.9880e-10, 1.0000e+00],
        [1.1044e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.0885e-05, 9.9997e-01],
        [6.3034e-05, 9.9994e-01],
        [2.5506e-06, 1.0000e+00],
        [2.0412e-08, 1.0000e+00],
        [3.8472e-05, 9.9996e-01],
        [2.8097e-08, 1.0000e+00],
        [1.4144e-11, 1.0000e+00],
        [4.8723e-01, 5.1277e-01],
        [3.2350e-06, 1.0000e+00],
        [1.0756e-10, 1.0000e+00],
        [1.4094e-08, 1.0000e+00],
        [4.6380e-10, 1.0000e+00],
        [2.1764e-07, 1.0000e+00],
        [2.7407e-04, 9.9973e-01],
        [9.9991e-01, 9.1849e-05],
        [2.2797e-11, 1.0000e+00],
        [1.8611e-04, 9.9981e-01],
        [2.3351e-06, 1.0000e+00],
        [2.1497e-05, 9.9998e-01],
        [9.9993e-01, 7.0879e-05],
        [1.7350e-03, 9.9827e-01],
        [2.3183e-06, 1.0000e+00],
        [8.3742e-05, 9.9992e-01],
        [9.9918e-01, 8.1982e-04],
        [3.0501e-05, 9.9997e-01],
        [2.9260e-04, 9.9971e-01],
        [5.2533e-07, 1.0000e+00],
        [1.2996e-01, 8.7004e-01],
        [4.3173e-07, 1.0000e+00],
        [9.9989e-01, 1.1077e-04],
        [1.8829e-07, 1.0000e+00],
        [5.6250e-04, 9.9944e-01],
        [1.1688e-08, 1.0000e+00],
        [9.9981e-01, 1.8770e-04],
        [9.9985e-01, 1.4879e-04],
        [1.4862e-06, 1.0000e+00],
        [3.4386e-05, 9.9997e-01],
        [5.3819e-07, 1.0000e+00],
        [3.4565e-12, 1.0000e+00],
        [9.9986e-01, 1.3917e-04],
        [6.9062e-10, 1.0000e+00],
        [9.9983e-01, 1.7230e-04],
        [6.4176e-04, 9.9936e-01],
        [9.9327e-01, 6.7313e-03],
        [9.9995e-01, 5.3200e-05],
        [5.9864e-02, 9.4014e-01],
        [2.4253e-07, 1.0000e+00],
        [1.8683e-06, 1.0000e+00],
        [4.5423e-06, 1.0000e+00],
        [1.4219e-11, 1.0000e+00],
        [3.6878e-07, 1.0000e+00],
        [3.5237e-11, 1.0000e+00],
        [9.9989e-01, 1.0717e-04],
        [6.8976e-10, 1.0000e+00],
        [1.0599e-06, 1.0000e+00],
        [9.1096e-01, 8.9044e-02],
        [8.2218e-08, 1.0000e+00],
        [1.2897e-03, 9.9871e-01],
        [1.7692e-11, 1.0000e+00],
        [2.4295e-10, 1.0000e+00],
        [2.9009e-10, 1.0000e+00],
        [4.4182e-07, 1.0000e+00],
        [6.9001e-08, 1.0000e+00],
        [8.7014e-06, 9.9999e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7162e-04, 9.9983e-01],
        [1.1502e-11, 1.0000e+00],
        [3.3476e-08, 1.0000e+00],
        [7.3753e-11, 1.0000e+00],
        [2.0882e-03, 9.9791e-01],
        [5.3764e-06, 9.9999e-01],
        [3.4695e-05, 9.9997e-01],
        [6.9617e-06, 9.9999e-01],
        [1.6145e-06, 1.0000e+00],
        [9.9332e-07, 1.0000e+00],
        [2.2591e-05, 9.9998e-01],
        [1.8728e-02, 9.8127e-01],
        [3.5544e-06, 1.0000e+00],
        [5.4021e-05, 9.9995e-01],
        [2.9051e-09, 1.0000e+00],
        [8.1293e-08, 1.0000e+00],
        [3.5593e-02, 9.6441e-01],
        [6.8009e-10, 1.0000e+00],
        [9.9993e-01, 6.9523e-05],
        [9.9990e-01, 1.0060e-04],
        [9.9997e-01, 2.5612e-05],
        [6.8037e-08, 1.0000e+00],
        [4.9176e-06, 1.0000e+00],
        [1.0950e-10, 1.0000e+00],
        [1.0178e-05, 9.9999e-01],
        [3.3849e-10, 1.0000e+00],
        [4.1392e-03, 9.9586e-01],
        [9.9959e-01, 4.1096e-04],
        [9.9990e-01, 9.7407e-05],
        [3.8637e-03, 9.9614e-01],
        [4.7760e-12, 1.0000e+00],
        [7.2830e-01, 2.7170e-01],
        [8.9071e-01, 1.0929e-01],
        [1.1282e-09, 1.0000e+00],
        [4.5155e-10, 1.0000e+00],
        [9.9992e-01, 8.4162e-05],
        [4.2340e-07, 1.0000e+00],
        [9.0194e-08, 1.0000e+00],
        [1.0263e-11, 1.0000e+00],
        [3.0720e-06, 1.0000e+00],
        [9.9983e-01, 1.7426e-04],
        [9.2462e-07, 1.0000e+00],
        [7.7473e-01, 2.2527e-01],
        [9.9920e-01, 7.9854e-04],
        [9.3767e-11, 1.0000e+00],
        [8.1727e-01, 1.8273e-01],
        [2.0605e-02, 9.7940e-01],
        [2.2488e-05, 9.9998e-01],
        [1.8886e-01, 8.1114e-01],
        [6.0268e-10, 1.0000e+00],
        [8.3388e-04, 9.9917e-01],
        [1.9463e-06, 1.0000e+00],
        [2.3158e-01, 7.6842e-01],
        [9.9981e-01, 1.8577e-04],
        [3.3883e-06, 1.0000e+00],
        [9.8160e-01, 1.8395e-02],
        [3.6802e-10, 1.0000e+00],
        [1.1155e-08, 1.0000e+00],
        [1.5625e-10, 1.0000e+00],
        [3.0682e-06, 1.0000e+00],
        [6.9143e-08, 1.0000e+00],
        [4.2012e-07, 1.0000e+00],
        [1.4185e-08, 1.0000e+00],
        [5.5653e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9990e-01, 1.0068e-04],
        [3.5857e-08, 1.0000e+00],
        [5.2878e-08, 1.0000e+00],
        [5.8116e-02, 9.4188e-01],
        [5.3841e-12, 1.0000e+00],
        [2.5509e-06, 1.0000e+00],
        [9.9553e-01, 4.4742e-03],
        [9.9897e-01, 1.0298e-03],
        [1.8225e-10, 1.0000e+00],
        [6.3803e-05, 9.9994e-01],
        [9.9995e-01, 5.4421e-05],
        [3.2562e-08, 1.0000e+00],
        [1.2425e-02, 9.8758e-01],
        [9.9997e-01, 2.9066e-05],
        [9.9972e-01, 2.7629e-04],
        [9.9983e-01, 1.7146e-04],
        [9.9976e-01, 2.3778e-04],
        [6.3177e-07, 1.0000e+00],
        [4.0118e-06, 1.0000e+00],
        [9.9996e-01, 4.1512e-05],
        [3.3178e-07, 1.0000e+00],
        [8.6546e-12, 1.0000e+00],
        [3.3336e-10, 1.0000e+00],
        [1.9046e-07, 1.0000e+00],
        [1.2573e-04, 9.9987e-01],
        [2.9117e-08, 1.0000e+00],
        [9.9996e-01, 4.2502e-05],
        [7.0764e-06, 9.9999e-01],
        [1.2354e-01, 8.7646e-01],
        [1.8365e-06, 1.0000e+00],
        [9.9882e-01, 1.1839e-03],
        [6.8665e-04, 9.9931e-01],
        [1.4189e-10, 1.0000e+00],
        [8.5215e-07, 1.0000e+00],
        [7.6025e-03, 9.9240e-01],
        [4.1374e-07, 1.0000e+00],
        [3.9354e-06, 1.0000e+00],
        [9.9993e-01, 7.2740e-05],
        [3.7638e-08, 1.0000e+00],
        [9.9995e-01, 4.6547e-05],
        [4.7628e-07, 1.0000e+00],
        [8.0748e-10, 1.0000e+00],
        [6.9821e-07, 1.0000e+00],
        [1.8388e-08, 1.0000e+00],
        [1.6841e-07, 1.0000e+00],
        [7.0735e-12, 1.0000e+00],
        [2.3142e-02, 9.7686e-01],
        [6.0323e-07, 1.0000e+00],
        [9.9998e-01, 2.4735e-05],
        [9.6244e-01, 3.7557e-02],
        [7.4621e-09, 1.0000e+00],
        [9.9986e-01, 1.3812e-04],
        [3.5588e-08, 1.0000e+00],
        [1.4713e-09, 1.0000e+00],
        [2.4978e-05, 9.9997e-01],
        [2.1711e-03, 9.9783e-01],
        [9.9990e-01, 9.8058e-05],
        [3.0018e-06, 1.0000e+00],
        [4.3946e-08, 1.0000e+00],
        [4.0774e-04, 9.9959e-01],
        [5.1175e-06, 9.9999e-01],
        [1.3243e-05, 9.9999e-01],
        [1.2481e-08, 1.0000e+00],
        [2.6271e-09, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.6046e-04, 9.9974e-01],
        [1.2193e-11, 1.0000e+00],
        [1.4105e-08, 1.0000e+00],
        [1.0907e-05, 9.9999e-01],
        [9.9990e-01, 9.8060e-05],
        [5.6396e-02, 9.4360e-01],
        [3.3088e-11, 1.0000e+00],
        [9.9994e-01, 5.6505e-05],
        [2.9377e-05, 9.9997e-01],
        [2.9903e-07, 1.0000e+00],
        [1.0457e-05, 9.9999e-01],
        [9.4325e-05, 9.9991e-01],
        [9.9996e-01, 4.2411e-05],
        [9.9991e-01, 8.9845e-05],
        [7.6010e-05, 9.9992e-01],
        [3.7301e-06, 1.0000e+00],
        [1.6882e-05, 9.9998e-01],
        [8.2233e-02, 9.1777e-01],
        [9.9997e-01, 2.8246e-05],
        [5.6495e-07, 1.0000e+00],
        [9.9418e-01, 5.8234e-03],
        [3.3732e-08, 1.0000e+00],
        [3.6741e-09, 1.0000e+00],
        [4.2891e-03, 9.9571e-01],
        [3.7092e-07, 1.0000e+00],
        [4.4054e-07, 1.0000e+00],
        [2.2010e-03, 9.9780e-01],
        [9.9967e-01, 3.3232e-04],
        [4.3643e-05, 9.9996e-01],
        [9.4508e-08, 1.0000e+00],
        [1.8324e-09, 1.0000e+00],
        [1.9626e-06, 1.0000e+00],
        [9.9976e-01, 2.3748e-04],
        [1.5188e-03, 9.9848e-01],
        [1.0000e+00, 2.0743e-07],
        [1.9203e-08, 1.0000e+00],
        [9.9919e-01, 8.1269e-04],
        [2.3607e-10, 1.0000e+00],
        [6.2603e-08, 1.0000e+00],
        [6.7152e-07, 1.0000e+00],
        [2.4058e-06, 1.0000e+00],
        [5.9211e-08, 1.0000e+00],
        [3.5454e-07, 1.0000e+00],
        [7.7152e-08, 1.0000e+00],
        [9.9997e-01, 2.5832e-05],
        [2.9469e-07, 1.0000e+00],
        [5.9684e-08, 1.0000e+00],
        [1.9467e-07, 1.0000e+00],
        [1.8439e-05, 9.9998e-01],
        [1.7591e-05, 9.9998e-01],
        [1.2921e-10, 1.0000e+00],
        [1.9136e-03, 9.9809e-01],
        [5.6168e-08, 1.0000e+00],
        [9.9982e-01, 1.8220e-04],
        [2.1357e-06, 1.0000e+00],
        [1.0173e-06, 1.0000e+00],
        [4.3346e-04, 9.9957e-01],
        [1.6406e-06, 1.0000e+00],
        [2.5103e-04, 9.9975e-01],
        [2.6238e-04, 9.9974e-01],
        [7.3834e-05, 9.9993e-01],
        [5.0195e-07, 1.0000e+00],
        [4.4878e-06, 1.0000e+00],
        [1.9686e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9993e-01, 7.2991e-05],
        [1.0358e-10, 1.0000e+00],
        [2.3341e-08, 1.0000e+00],
        [3.6327e-07, 1.0000e+00],
        [8.5727e-06, 9.9999e-01],
        [9.9996e-01, 4.2277e-05],
        [9.3292e-09, 1.0000e+00],
        [8.3430e-05, 9.9992e-01],
        [3.8166e-08, 1.0000e+00],
        [9.9994e-01, 6.2232e-05],
        [9.9982e-01, 1.8051e-04],
        [3.8069e-07, 1.0000e+00],
        [2.7750e-04, 9.9972e-01],
        [3.7162e-08, 1.0000e+00],
        [8.2342e-02, 9.1766e-01],
        [4.5235e-05, 9.9995e-01],
        [8.2728e-05, 9.9992e-01],
        [9.9952e-01, 4.7859e-04],
        [2.7418e-09, 1.0000e+00],
        [7.9284e-10, 1.0000e+00],
        [9.0269e-03, 9.9097e-01],
        [1.9498e-03, 9.9805e-01],
        [9.9997e-01, 3.4611e-05],
        [1.3370e-02, 9.8663e-01],
        [5.0428e-06, 9.9999e-01],
        [2.6001e-04, 9.9974e-01],
        [3.8073e-10, 1.0000e+00],
        [3.1074e-07, 1.0000e+00],
        [1.1015e-05, 9.9999e-01],
        [3.6057e-05, 9.9996e-01],
        [9.9997e-01, 3.4078e-05],
        [3.4962e-04, 9.9965e-01],
        [9.5723e-07, 1.0000e+00],
        [3.5008e-06, 1.0000e+00],
        [4.5532e-06, 1.0000e+00],
        [8.9825e-06, 9.9999e-01],
        [2.9192e-10, 1.0000e+00],
        [4.4568e-12, 1.0000e+00],
        [9.9951e-01, 4.9461e-04],
        [2.1963e-09, 1.0000e+00],
        [1.4214e-05, 9.9999e-01],
        [9.9970e-01, 3.0388e-04],
        [1.0269e-07, 1.0000e+00],
        [1.9999e-08, 1.0000e+00],
        [9.3180e-01, 6.8203e-02],
        [1.4404e-01, 8.5596e-01],
        [5.2635e-06, 9.9999e-01],
        [9.9988e-01, 1.2281e-04],
        [1.2666e-07, 1.0000e+00],
        [1.2074e-01, 8.7926e-01],
        [1.1600e-12, 1.0000e+00],
        [4.9427e-05, 9.9995e-01],
        [4.0813e-09, 1.0000e+00],
        [9.9975e-01, 2.4600e-04],
        [1.3476e-11, 1.0000e+00],
        [2.8851e-09, 1.0000e+00],
        [9.9982e-01, 1.8210e-04],
        [9.9995e-01, 4.7646e-05],
        [4.3201e-11, 1.0000e+00],
        [3.8176e-07, 1.0000e+00],
        [1.2792e-06, 1.0000e+00],
        [6.2047e-03, 9.9380e-01],
        [2.9110e-06, 1.0000e+00],
        [1.2667e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.8206e-12, 1.0000e+00],
        [3.1188e-09, 1.0000e+00],
        [6.8375e-08, 1.0000e+00],
        [2.0145e-11, 1.0000e+00],
        [9.9834e-01, 1.6619e-03],
        [2.8393e-04, 9.9972e-01],
        [7.6694e-08, 1.0000e+00],
        [9.9995e-01, 4.7668e-05],
        [2.4878e-04, 9.9975e-01],
        [6.9435e-07, 1.0000e+00],
        [1.3109e-04, 9.9987e-01],
        [9.9967e-01, 3.3254e-04],
        [5.3217e-09, 1.0000e+00],
        [9.9977e-01, 2.2837e-04],
        [4.5681e-12, 1.0000e+00],
        [9.9979e-01, 2.0746e-04],
        [1.2454e-06, 1.0000e+00],
        [9.9962e-01, 3.8169e-04],
        [1.4318e-07, 1.0000e+00],
        [9.9996e-01, 4.3986e-05],
        [2.3394e-10, 1.0000e+00],
        [4.5882e-07, 1.0000e+00],
        [3.9997e-07, 1.0000e+00],
        [2.7963e-07, 1.0000e+00],
        [1.9603e-09, 1.0000e+00],
        [1.0663e-03, 9.9893e-01],
        [6.8035e-01, 3.1965e-01],
        [9.9997e-01, 2.6816e-05],
        [6.0153e-06, 9.9999e-01],
        [8.2512e-07, 1.0000e+00],
        [1.6002e-04, 9.9984e-01],
        [8.4858e-13, 1.0000e+00],
        [1.0038e-04, 9.9990e-01],
        [9.9816e-01, 1.8366e-03],
        [9.9988e-01, 1.2001e-04],
        [2.5250e-04, 9.9975e-01],
        [1.6982e-04, 9.9983e-01],
        [5.2384e-11, 1.0000e+00],
        [9.4664e-01, 5.3360e-02],
        [5.8749e-10, 1.0000e+00],
        [2.0848e-06, 1.0000e+00],
        [9.9924e-01, 7.6254e-04],
        [1.5114e-07, 1.0000e+00],
        [7.4250e-10, 1.0000e+00],
        [9.0475e-08, 1.0000e+00],
        [2.4430e-07, 1.0000e+00],
        [1.0160e-08, 1.0000e+00],
        [2.0123e-06, 1.0000e+00],
        [2.2632e-03, 9.9774e-01],
        [5.9414e-10, 1.0000e+00],
        [9.9993e-01, 6.6979e-05],
        [9.9977e-01, 2.2774e-04],
        [1.5194e-09, 1.0000e+00],
        [5.0907e-08, 1.0000e+00],
        [9.9974e-01, 2.6112e-04],
        [9.9998e-01, 2.2901e-05],
        [9.9990e-01, 9.7239e-05],
        [7.3676e-07, 1.0000e+00],
        [1.6951e-05, 9.9998e-01],
        [1.0982e-05, 9.9999e-01],
        [9.9568e-01, 4.3230e-03],
        [1.8025e-07, 1.0000e+00],
        [1.4533e-05, 9.9999e-01],
        [9.9956e-01, 4.3870e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2684e-04, 9.9987e-01],
        [2.6261e-08, 1.0000e+00],
        [9.2777e-09, 1.0000e+00],
        [3.1348e-10, 1.0000e+00],
        [3.2893e-09, 1.0000e+00],
        [9.5733e-05, 9.9990e-01],
        [9.6432e-06, 9.9999e-01],
        [6.8147e-09, 1.0000e+00],
        [2.8158e-08, 1.0000e+00],
        [4.6244e-07, 1.0000e+00],
        [1.4212e-11, 1.0000e+00],
        [9.6592e-01, 3.4079e-02],
        [6.9328e-10, 1.0000e+00],
        [9.9985e-01, 1.5088e-04],
        [9.9662e-01, 3.3817e-03],
        [9.9990e-01, 9.9878e-05],
        [6.3944e-08, 1.0000e+00],
        [8.5554e-01, 1.4446e-01],
        [3.4182e-10, 1.0000e+00],
        [4.9422e-07, 1.0000e+00],
        [2.8465e-07, 1.0000e+00],
        [8.7196e-09, 1.0000e+00],
        [9.9975e-01, 2.5226e-04],
        [1.8314e-08, 1.0000e+00],
        [2.8198e-07, 1.0000e+00],
        [7.2354e-03, 9.9276e-01],
        [5.4622e-06, 9.9999e-01],
        [7.0036e-09, 1.0000e+00],
        [1.9543e-07, 1.0000e+00],
        [6.8283e-11, 1.0000e+00],
        [1.4989e-06, 1.0000e+00],
        [9.9965e-01, 3.4510e-04],
        [9.9165e-10, 1.0000e+00],
        [1.4287e-03, 9.9857e-01],
        [9.9842e-01, 1.5768e-03],
        [1.1968e-11, 1.0000e+00],
        [2.0115e-04, 9.9980e-01],
        [9.9999e-01, 1.4764e-05],
        [1.1719e-08, 1.0000e+00],
        [9.9982e-01, 1.8290e-04],
        [1.3254e-05, 9.9999e-01],
        [8.4007e-08, 1.0000e+00],
        [9.5086e-01, 4.9143e-02],
        [2.2111e-07, 1.0000e+00],
        [1.9315e-12, 1.0000e+00],
        [7.3444e-11, 1.0000e+00],
        [1.0814e-04, 9.9989e-01],
        [6.0799e-06, 9.9999e-01],
        [5.9046e-10, 1.0000e+00],
        [3.2858e-05, 9.9997e-01],
        [1.1548e-04, 9.9988e-01],
        [1.6241e-06, 1.0000e+00],
        [5.5114e-07, 1.0000e+00],
        [1.1289e-03, 9.9887e-01],
        [1.7918e-08, 1.0000e+00],
        [9.9996e-01, 3.5759e-05],
        [9.9403e-01, 5.9690e-03],
        [5.0629e-07, 1.0000e+00],
        [3.2631e-08, 1.0000e+00],
        [9.9990e-01, 9.8764e-05],
        [9.9790e-01, 2.1010e-03],
        [9.9973e-01, 2.6957e-04],
        [1.2397e-06, 1.0000e+00],
        [9.9994e-01, 6.2869e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[2.7728e-07, 1.0000e+00],
        [1.3487e-05, 9.9999e-01],
        [4.2504e-07, 1.0000e+00],
        [4.0476e-09, 1.0000e+00],
        [9.9983e-01, 1.7441e-04],
        [1.2367e-09, 1.0000e+00],
        [1.2402e-04, 9.9988e-01],
        [3.8465e-05, 9.9996e-01],
        [3.4117e-06, 1.0000e+00],
        [9.9684e-01, 3.1560e-03],
        [9.7258e-01, 2.7425e-02],
        [7.2219e-11, 1.0000e+00],
        [2.9852e-05, 9.9997e-01],
        [9.9984e-01, 1.6426e-04],
        [3.1235e-05, 9.9997e-01],
        [9.6532e-01, 3.4679e-02],
        [1.6099e-08, 1.0000e+00],
        [9.9986e-01, 1.3677e-04],
        [1.0229e-12, 1.0000e+00],
        [1.0016e-04, 9.9990e-01],
        [2.8193e-10, 1.0000e+00],
        [3.6269e-06, 1.0000e+00],
        [9.3611e-06, 9.9999e-01],
        [2.0595e-07, 1.0000e+00],
        [1.1291e-09, 1.0000e+00],
        [2.1647e-01, 7.8353e-01],
        [9.9782e-01, 2.1767e-03],
        [8.0622e-07, 1.0000e+00],
        [2.7875e-06, 1.0000e+00],
        [7.0004e-08, 1.0000e+00],
        [1.9515e-10, 1.0000e+00],
        [6.4847e-03, 9.9352e-01],
        [1.9409e-07, 1.0000e+00],
        [7.6113e-09, 1.0000e+00],
        [6.9209e-05, 9.9993e-01],
        [1.4178e-05, 9.9999e-01],
        [1.8593e-07, 1.0000e+00],
        [1.9143e-07, 1.0000e+00],
        [3.3337e-09, 1.0000e+00],
        [5.1306e-07, 1.0000e+00],
        [4.1216e-06, 1.0000e+00],
        [9.5858e-07, 1.0000e+00],
        [7.4553e-11, 1.0000e+00],
        [5.3419e-01, 4.6581e-01],
        [9.9996e-01, 3.5458e-05],
        [1.4643e-05, 9.9999e-01],
        [6.5646e-10, 1.0000e+00],
        [1.8628e-11, 1.0000e+00],
        [7.8918e-09, 1.0000e+00],
        [9.3366e-10, 1.0000e+00],
        [9.9960e-01, 3.9941e-04],
        [3.1544e-08, 1.0000e+00],
        [4.3337e-10, 1.0000e+00],
        [1.6369e-07, 1.0000e+00],
        [9.9980e-01, 1.9658e-04],
        [5.0179e-03, 9.9498e-01],
        [2.4712e-06, 1.0000e+00],
        [8.8571e-09, 1.0000e+00],
        [7.5769e-01, 2.4231e-01],
        [1.7416e-09, 1.0000e+00],
        [1.5307e-05, 9.9998e-01],
        [8.3076e-07, 1.0000e+00],
        [5.9144e-09, 1.0000e+00],
        [9.9987e-01, 1.3256e-04]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2408e-06, 1.0000e+00],
        [4.3892e-05, 9.9996e-01],
        [2.0906e-03, 9.9791e-01],
        [9.9982e-01, 1.8003e-04],
        [9.9993e-01, 6.6045e-05],
        [2.7289e-08, 1.0000e+00],
        [5.8822e-08, 1.0000e+00],
        [1.1528e-04, 9.9988e-01],
        [8.2807e-09, 1.0000e+00],
        [9.9987e-01, 1.3122e-04],
        [9.9988e-01, 1.1771e-04],
        [5.0770e-06, 9.9999e-01],
        [9.7552e-09, 1.0000e+00],
        [2.2984e-06, 1.0000e+00],
        [1.0000e+00, 9.1232e-07],
        [9.9881e-01, 1.1945e-03],
        [1.3356e-11, 1.0000e+00],
        [2.6130e-07, 1.0000e+00],
        [4.5769e-04, 9.9954e-01],
        [9.9659e-01, 3.4134e-03],
        [8.9256e-05, 9.9991e-01],
        [6.4240e-10, 1.0000e+00],
        [9.9991e-01, 9.4159e-05],
        [9.9997e-01, 3.1127e-05],
        [2.1175e-07, 1.0000e+00],
        [1.5837e-06, 1.0000e+00],
        [1.8225e-06, 1.0000e+00],
        [7.1679e-06, 9.9999e-01],
        [4.5210e-05, 9.9995e-01],
        [5.6675e-08, 1.0000e+00],
        [9.9996e-01, 3.6249e-05],
        [9.9969e-01, 3.1270e-04],
        [3.1013e-13, 1.0000e+00],
        [1.1580e-11, 1.0000e+00],
        [1.0437e-12, 1.0000e+00],
        [6.9069e-09, 1.0000e+00],
        [1.5473e-07, 1.0000e+00],
        [9.5033e-01, 4.9672e-02],
        [6.9946e-05, 9.9993e-01],
        [1.9973e-06, 1.0000e+00],
        [2.0355e-04, 9.9980e-01],
        [1.7138e-11, 1.0000e+00],
        [9.9447e-01, 5.5258e-03],
        [1.7384e-08, 1.0000e+00],
        [9.9956e-01, 4.3892e-04],
        [1.5773e-05, 9.9998e-01],
        [2.8603e-09, 1.0000e+00],
        [9.0962e-06, 9.9999e-01],
        [3.3265e-07, 1.0000e+00],
        [9.9974e-01, 2.6039e-04],
        [1.1973e-04, 9.9988e-01],
        [9.9999e-01, 5.0693e-06],
        [1.3579e-05, 9.9999e-01],
        [4.7637e-11, 1.0000e+00],
        [4.2286e-08, 1.0000e+00],
        [1.6257e-08, 1.0000e+00],
        [9.9990e-01, 9.7249e-05],
        [2.1279e-02, 9.7872e-01],
        [2.6559e-14, 1.0000e+00],
        [9.9765e-01, 2.3539e-03],
        [1.1603e-08, 1.0000e+00],
        [1.7381e-06, 1.0000e+00],
        [1.5890e-03, 9.9841e-01],
        [4.8995e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9996e-01, 4.0774e-05],
        [9.9999e-01, 6.2444e-06],
        [3.6771e-11, 1.0000e+00],
        [1.2992e-10, 1.0000e+00],
        [3.3072e-05, 9.9997e-01],
        [3.3495e-07, 1.0000e+00],
        [9.9995e-01, 5.1947e-05],
        [7.8839e-08, 1.0000e+00],
        [2.4803e-06, 1.0000e+00],
        [2.2901e-06, 1.0000e+00],
        [1.2011e-08, 1.0000e+00],
        [9.9974e-01, 2.6111e-04],
        [9.9996e-01, 3.7442e-05],
        [5.7546e-10, 1.0000e+00],
        [5.8936e-06, 9.9999e-01],
        [9.9984e-01, 1.6306e-04],
        [2.0848e-09, 1.0000e+00],
        [1.9345e-05, 9.9998e-01],
        [5.1417e-11, 1.0000e+00],
        [2.6422e-05, 9.9997e-01],
        [1.0109e-10, 1.0000e+00],
        [4.0216e-01, 5.9784e-01],
        [2.0822e-09, 1.0000e+00],
        [9.9995e-01, 5.3422e-05],
        [2.2304e-08, 1.0000e+00],
        [7.8832e-06, 9.9999e-01],
        [9.9979e-01, 2.1301e-04],
        [9.9972e-01, 2.8006e-04],
        [9.9999e-01, 6.1000e-06],
        [9.9905e-01, 9.5150e-04],
        [1.4233e-04, 9.9986e-01],
        [8.4472e-11, 1.0000e+00],
        [1.0323e-12, 1.0000e+00],
        [4.7379e-07, 1.0000e+00],
        [6.5161e-09, 1.0000e+00],
        [5.8381e-06, 9.9999e-01],
        [1.0503e-07, 1.0000e+00],
        [6.8974e-08, 1.0000e+00],
        [9.9788e-01, 2.1176e-03],
        [9.9986e-01, 1.4077e-04],
        [3.3783e-07, 1.0000e+00],
        [7.9081e-08, 1.0000e+00],
        [1.9477e-07, 1.0000e+00],
        [2.8319e-07, 1.0000e+00],
        [3.8203e-08, 1.0000e+00],
        [1.0000e+00, 3.0971e-07],
        [3.8577e-08, 1.0000e+00],
        [9.9997e-01, 3.4430e-05],
        [3.5896e-02, 9.6410e-01],
        [2.3862e-06, 1.0000e+00],
        [1.1188e-02, 9.8881e-01],
        [2.4625e-07, 1.0000e+00],
        [4.4783e-07, 1.0000e+00],
        [1.8549e-08, 1.0000e+00],
        [3.8828e-09, 1.0000e+00],
        [3.5930e-08, 1.0000e+00],
        [2.8695e-05, 9.9997e-01],
        [5.5563e-09, 1.0000e+00],
        [3.7246e-08, 1.0000e+00],
        [1.4533e-04, 9.9985e-01],
        [4.3745e-11, 1.0000e+00],
        [3.6135e-04, 9.9964e-01],
        [7.8020e-08, 1.0000e+00],
        [9.9993e-01, 7.2636e-05]], grad_fn=<SoftmaxBackward0>)
tensor([[4.4898e-03, 9.9551e-01],
        [5.3111e-11, 1.0000e+00],
        [2.6615e-10, 1.0000e+00],
        [9.9929e-01, 7.1270e-04],
        [4.9247e-05, 9.9995e-01],
        [5.5291e-03, 9.9447e-01],
        [7.9695e-07, 1.0000e+00],
        [2.7064e-05, 9.9997e-01],
        [9.9988e-01, 1.1992e-04],
        [9.9990e-01, 9.8367e-05],
        [9.8874e-01, 1.1256e-02],
        [6.0503e-07, 1.0000e+00],
        [1.6192e-04, 9.9984e-01],
        [9.0134e-09, 1.0000e+00],
        [5.9926e-05, 9.9994e-01],
        [2.0692e-03, 9.9793e-01],
        [1.3540e-09, 1.0000e+00],
        [1.5551e-07, 1.0000e+00],
        [9.9974e-01, 2.5795e-04],
        [3.1540e-12, 1.0000e+00],
        [8.7859e-10, 1.0000e+00],
        [9.9785e-01, 2.1471e-03],
        [2.4735e-11, 1.0000e+00],
        [7.7134e-07, 1.0000e+00],
        [7.6116e-09, 1.0000e+00],
        [9.9998e-01, 2.0341e-05],
        [1.9421e-04, 9.9981e-01],
        [7.8053e-01, 2.1947e-01],
        [7.8300e-10, 1.0000e+00],
        [5.6977e-10, 1.0000e+00],
        [4.8854e-05, 9.9995e-01],
        [4.1135e-07, 1.0000e+00],
        [8.6960e-09, 1.0000e+00],
        [1.5136e-06, 1.0000e+00],
        [9.9986e-01, 1.4425e-04],
        [9.9998e-01, 2.3949e-05],
        [9.9992e-01, 8.1504e-05],
        [9.9994e-01, 6.2148e-05],
        [9.0460e-01, 9.5397e-02],
        [1.0637e-07, 1.0000e+00],
        [1.7365e-08, 1.0000e+00],
        [1.1130e-07, 1.0000e+00],
        [4.2544e-10, 1.0000e+00],
        [5.2969e-09, 1.0000e+00],
        [9.4898e-08, 1.0000e+00],
        [2.8939e-06, 1.0000e+00],
        [1.0558e-06, 1.0000e+00],
        [6.5932e-05, 9.9993e-01],
        [1.1663e-10, 1.0000e+00],
        [7.1560e-09, 1.0000e+00],
        [7.6648e-09, 1.0000e+00],
        [1.6433e-08, 1.0000e+00],
        [9.9996e-01, 3.7117e-05],
        [9.9993e-01, 7.1577e-05],
        [1.8006e-07, 1.0000e+00],
        [7.7481e-04, 9.9923e-01],
        [3.1523e-04, 9.9968e-01],
        [7.5763e-03, 9.9242e-01],
        [9.9993e-01, 7.2323e-05],
        [9.9994e-01, 6.0757e-05],
        [7.3756e-02, 9.2624e-01],
        [9.8230e-01, 1.7700e-02],
        [9.9987e-01, 1.2710e-04],
        [7.8495e-04, 9.9922e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[7.4326e-11, 1.0000e+00],
        [5.7428e-06, 9.9999e-01],
        [2.1396e-08, 1.0000e+00],
        [9.9995e-01, 5.1103e-05],
        [1.1917e-08, 1.0000e+00],
        [8.8660e-05, 9.9991e-01],
        [4.2076e-15, 1.0000e+00],
        [3.9825e-04, 9.9960e-01],
        [6.6110e-07, 1.0000e+00],
        [2.5632e-04, 9.9974e-01],
        [1.0000e+00, 4.7272e-08],
        [9.7990e-07, 1.0000e+00],
        [9.9986e-01, 1.4121e-04],
        [8.3831e-07, 1.0000e+00],
        [8.6009e-07, 1.0000e+00],
        [1.8202e-10, 1.0000e+00],
        [2.6486e-09, 1.0000e+00],
        [6.6649e-10, 1.0000e+00],
        [3.8770e-06, 1.0000e+00],
        [5.4651e-11, 1.0000e+00],
        [1.3622e-10, 1.0000e+00],
        [9.9985e-01, 1.4832e-04],
        [8.0989e-06, 9.9999e-01],
        [1.1778e-03, 9.9882e-01],
        [6.9293e-08, 1.0000e+00],
        [1.7494e-08, 1.0000e+00],
        [8.8506e-06, 9.9999e-01],
        [4.2641e-04, 9.9957e-01],
        [6.3440e-07, 1.0000e+00],
        [3.6914e-08, 1.0000e+00],
        [1.3022e-10, 1.0000e+00],
        [2.0999e-12, 1.0000e+00],
        [8.2417e-04, 9.9918e-01],
        [1.1086e-08, 1.0000e+00],
        [1.4230e-04, 9.9986e-01],
        [1.3250e-06, 1.0000e+00],
        [7.1042e-11, 1.0000e+00],
        [2.7941e-07, 1.0000e+00],
        [1.8485e-05, 9.9998e-01],
        [1.7537e-07, 1.0000e+00],
        [3.6945e-02, 9.6305e-01],
        [7.1868e-11, 1.0000e+00],
        [9.9992e-01, 8.2860e-05],
        [9.9996e-01, 3.9720e-05],
        [5.9338e-01, 4.0662e-01],
        [9.4314e-06, 9.9999e-01],
        [1.1677e-04, 9.9988e-01],
        [9.9926e-01, 7.3508e-04],
        [1.1880e-05, 9.9999e-01],
        [3.1163e-08, 1.0000e+00],
        [1.8184e-04, 9.9982e-01],
        [2.8401e-12, 1.0000e+00],
        [1.2404e-06, 1.0000e+00],
        [1.1442e-09, 1.0000e+00],
        [9.9992e-01, 8.3559e-05],
        [4.4221e-04, 9.9956e-01],
        [1.1227e-06, 1.0000e+00],
        [7.6183e-07, 1.0000e+00],
        [1.0460e-09, 1.0000e+00],
        [4.0814e-09, 1.0000e+00],
        [9.2252e-01, 7.7475e-02],
        [9.9971e-01, 2.9082e-04],
        [1.7968e-12, 1.0000e+00],
        [2.7794e-06, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9968e-01, 3.1683e-04],
        [3.5928e-06, 1.0000e+00],
        [2.3001e-03, 9.9770e-01],
        [8.6789e-11, 1.0000e+00],
        [5.5899e-06, 9.9999e-01],
        [1.8720e-12, 1.0000e+00],
        [9.9962e-01, 3.7851e-04],
        [9.9996e-01, 4.1717e-05],
        [1.0180e-04, 9.9990e-01],
        [2.2912e-08, 1.0000e+00],
        [9.9989e-01, 1.1214e-04],
        [4.2528e-06, 1.0000e+00],
        [9.9998e-01, 1.6111e-05],
        [1.4645e-09, 1.0000e+00],
        [9.9005e-01, 9.9451e-03],
        [9.5317e-01, 4.6831e-02],
        [1.8833e-08, 1.0000e+00],
        [1.0180e-04, 9.9990e-01],
        [1.5114e-01, 8.4886e-01],
        [8.3672e-09, 1.0000e+00],
        [9.7935e-12, 1.0000e+00],
        [3.2394e-10, 1.0000e+00],
        [6.7877e-02, 9.3212e-01],
        [9.9010e-01, 9.8962e-03],
        [2.8980e-08, 1.0000e+00],
        [1.0909e-09, 1.0000e+00],
        [1.3284e-05, 9.9999e-01],
        [5.0754e-07, 1.0000e+00],
        [6.7478e-08, 1.0000e+00],
        [9.0460e-08, 1.0000e+00],
        [6.0819e-10, 1.0000e+00],
        [9.9997e-01, 2.7877e-05],
        [9.9994e-01, 6.0261e-05],
        [1.5226e-06, 1.0000e+00],
        [1.3811e-11, 1.0000e+00],
        [9.9989e-01, 1.0531e-04],
        [1.0132e-02, 9.8987e-01],
        [2.9490e-07, 1.0000e+00],
        [1.7787e-06, 1.0000e+00],
        [8.0081e-05, 9.9992e-01],
        [4.7735e-07, 1.0000e+00],
        [9.9999e-01, 7.0162e-06],
        [1.5721e-07, 1.0000e+00],
        [9.9993e-01, 7.4918e-05],
        [4.1891e-04, 9.9958e-01],
        [2.0139e-06, 1.0000e+00],
        [7.2916e-04, 9.9927e-01],
        [7.6479e-07, 1.0000e+00],
        [3.8185e-04, 9.9962e-01],
        [8.4394e-11, 1.0000e+00],
        [9.7172e-04, 9.9903e-01],
        [3.9370e-08, 1.0000e+00],
        [4.7320e-01, 5.2680e-01],
        [9.9996e-01, 4.4163e-05],
        [9.8864e-01, 1.1363e-02],
        [9.9984e-01, 1.6008e-04],
        [7.3667e-04, 9.9926e-01],
        [1.8682e-05, 9.9998e-01],
        [1.5226e-09, 1.0000e+00],
        [9.9995e-01, 4.9772e-05],
        [3.0206e-08, 1.0000e+00],
        [9.9983e-01, 1.6982e-04],
        [1.1067e-04, 9.9989e-01],
        [2.4869e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[2.3936e-04, 9.9976e-01],
        [2.5962e-08, 1.0000e+00],
        [1.0000e+00, 2.2104e-07],
        [7.4678e-03, 9.9253e-01],
        [9.9803e-01, 1.9692e-03],
        [1.2033e-05, 9.9999e-01],
        [6.6850e-07, 1.0000e+00],
        [9.9996e-01, 4.1969e-05],
        [9.7675e-01, 2.3251e-02],
        [2.5363e-07, 1.0000e+00],
        [7.2501e-03, 9.9275e-01],
        [5.0100e-08, 1.0000e+00],
        [4.9417e-04, 9.9951e-01],
        [1.5181e-03, 9.9848e-01],
        [5.1282e-02, 9.4872e-01],
        [5.0579e-07, 1.0000e+00],
        [1.7755e-08, 1.0000e+00],
        [4.0454e-02, 9.5955e-01],
        [2.3218e-04, 9.9977e-01],
        [5.1781e-08, 1.0000e+00],
        [9.1037e-03, 9.9090e-01],
        [3.9593e-08, 1.0000e+00],
        [1.6861e-11, 1.0000e+00],
        [9.4486e-03, 9.9055e-01],
        [8.9252e-06, 9.9999e-01],
        [1.7431e-07, 1.0000e+00],
        [4.8196e-05, 9.9995e-01],
        [7.0689e-08, 1.0000e+00],
        [7.3429e-08, 1.0000e+00],
        [1.2368e-04, 9.9988e-01],
        [9.7196e-09, 1.0000e+00],
        [6.1545e-09, 1.0000e+00],
        [4.0923e-10, 1.0000e+00],
        [9.9968e-01, 3.2422e-04],
        [4.4554e-11, 1.0000e+00],
        [9.9996e-01, 4.1106e-05],
        [6.8666e-07, 1.0000e+00],
        [5.9061e-06, 9.9999e-01],
        [1.4165e-05, 9.9999e-01],
        [6.7727e-06, 9.9999e-01],
        [1.0000e+00, 3.7891e-06],
        [1.7312e-10, 1.0000e+00],
        [1.9540e-10, 1.0000e+00],
        [4.1226e-07, 1.0000e+00],
        [9.4834e-06, 9.9999e-01],
        [5.4333e-04, 9.9946e-01],
        [9.9998e-01, 1.6517e-05],
        [1.6222e-07, 1.0000e+00],
        [5.4983e-05, 9.9995e-01],
        [3.9116e-07, 1.0000e+00],
        [2.0168e-03, 9.9798e-01],
        [6.6335e-02, 9.3366e-01],
        [5.3503e-07, 1.0000e+00],
        [8.1073e-07, 1.0000e+00],
        [3.0161e-08, 1.0000e+00],
        [3.6182e-10, 1.0000e+00],
        [4.4008e-11, 1.0000e+00],
        [1.3474e-01, 8.6526e-01],
        [6.6260e-07, 1.0000e+00],
        [9.3562e-09, 1.0000e+00],
        [9.9994e-01, 6.0971e-05],
        [1.3578e-09, 1.0000e+00],
        [6.1543e-06, 9.9999e-01],
        [9.4504e-05, 9.9991e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2725e-08, 1.0000e+00],
        [3.8762e-10, 1.0000e+00],
        [1.0299e-05, 9.9999e-01],
        [1.0939e-06, 1.0000e+00],
        [9.9999e-01, 8.2022e-06],
        [4.6483e-13, 1.0000e+00],
        [1.0373e-04, 9.9990e-01],
        [5.5862e-09, 1.0000e+00],
        [9.9993e-01, 7.1425e-05],
        [7.5151e-10, 1.0000e+00],
        [5.4228e-01, 4.5772e-01],
        [9.9994e-01, 5.8641e-05],
        [1.2882e-06, 1.0000e+00],
        [8.7398e-08, 1.0000e+00],
        [1.0254e-06, 1.0000e+00],
        [9.9989e-01, 1.1288e-04],
        [3.4562e-09, 1.0000e+00],
        [9.3469e-01, 6.5313e-02],
        [9.9981e-01, 1.9444e-04],
        [9.9999e-01, 5.6481e-06],
        [2.3982e-06, 1.0000e+00],
        [4.7326e-08, 1.0000e+00],
        [5.0917e-06, 9.9999e-01],
        [1.9791e-06, 1.0000e+00],
        [2.5695e-04, 9.9974e-01],
        [9.9986e-01, 1.3681e-04],
        [8.7679e-07, 1.0000e+00],
        [1.5703e-11, 1.0000e+00],
        [6.3389e-05, 9.9994e-01],
        [1.0911e-08, 1.0000e+00],
        [1.0121e-05, 9.9999e-01],
        [3.3184e-05, 9.9997e-01],
        [1.4368e-04, 9.9986e-01],
        [3.3018e-04, 9.9967e-01],
        [7.0662e-06, 9.9999e-01],
        [1.4409e-07, 1.0000e+00],
        [4.6266e-04, 9.9954e-01],
        [2.3162e-07, 1.0000e+00],
        [6.2328e-11, 1.0000e+00],
        [4.2152e-10, 1.0000e+00],
        [1.7318e-05, 9.9998e-01],
        [9.7585e-07, 1.0000e+00],
        [2.9855e-02, 9.7014e-01],
        [6.0212e-11, 1.0000e+00],
        [1.5176e-06, 1.0000e+00],
        [2.8462e-03, 9.9715e-01],
        [2.6372e-08, 1.0000e+00],
        [1.5209e-05, 9.9998e-01],
        [9.4249e-05, 9.9991e-01],
        [6.4548e-09, 1.0000e+00],
        [9.9966e-01, 3.3615e-04],
        [7.1562e-06, 9.9999e-01],
        [1.1097e-04, 9.9989e-01],
        [9.9992e-01, 8.1973e-05],
        [4.6931e-09, 1.0000e+00],
        [6.8231e-07, 1.0000e+00],
        [9.9987e-01, 1.3048e-04],
        [9.9997e-01, 3.4474e-05],
        [1.6065e-06, 1.0000e+00],
        [9.9997e-01, 2.9375e-05],
        [2.7141e-08, 1.0000e+00],
        [4.4497e-03, 9.9555e-01],
        [1.2016e-04, 9.9988e-01],
        [1.6022e-05, 9.9998e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[9.9978e-01, 2.1803e-04],
        [9.0358e-09, 1.0000e+00],
        [4.6487e-07, 1.0000e+00],
        [6.4724e-12, 1.0000e+00],
        [1.5657e-07, 1.0000e+00],
        [7.1234e-07, 1.0000e+00],
        [1.2316e-06, 1.0000e+00],
        [4.9034e-05, 9.9995e-01],
        [9.9986e-01, 1.3816e-04],
        [7.3489e-11, 1.0000e+00],
        [5.3185e-09, 1.0000e+00],
        [1.5049e-06, 1.0000e+00],
        [3.6880e-04, 9.9963e-01],
        [2.2685e-06, 1.0000e+00],
        [5.9025e-09, 1.0000e+00],
        [9.7376e-05, 9.9990e-01],
        [2.9086e-04, 9.9971e-01],
        [6.8158e-06, 9.9999e-01],
        [1.0066e-08, 1.0000e+00],
        [1.0775e-02, 9.8922e-01],
        [9.9980e-01, 2.0057e-04],
        [3.8246e-08, 1.0000e+00],
        [6.1706e-07, 1.0000e+00],
        [1.1219e-09, 1.0000e+00],
        [2.3227e-08, 1.0000e+00],
        [7.3987e-10, 1.0000e+00],
        [2.2526e-04, 9.9977e-01],
        [7.0507e-05, 9.9993e-01],
        [1.6434e-05, 9.9998e-01],
        [4.2383e-04, 9.9958e-01],
        [5.1950e-04, 9.9948e-01],
        [4.4443e-07, 1.0000e+00],
        [1.8511e-04, 9.9981e-01],
        [8.5168e-01, 1.4832e-01],
        [9.4258e-11, 1.0000e+00],
        [6.8701e-11, 1.0000e+00],
        [2.3574e-07, 1.0000e+00],
        [1.5393e-02, 9.8461e-01],
        [7.2219e-09, 1.0000e+00],
        [6.7123e-07, 1.0000e+00],
        [1.4080e-07, 1.0000e+00],
        [9.9988e-01, 1.1844e-04],
        [5.3014e-08, 1.0000e+00],
        [9.9998e-01, 1.6971e-05],
        [1.4233e-05, 9.9999e-01],
        [9.9995e-01, 5.0695e-05],
        [6.5779e-07, 1.0000e+00],
        [6.4019e-10, 1.0000e+00],
        [1.0416e-01, 8.9584e-01],
        [9.9991e-01, 9.0689e-05],
        [3.2525e-09, 1.0000e+00],
        [1.2513e-06, 1.0000e+00],
        [9.9993e-01, 6.6282e-05],
        [1.9366e-09, 1.0000e+00],
        [2.6827e-07, 1.0000e+00],
        [9.9991e-01, 9.0295e-05],
        [5.2952e-05, 9.9995e-01],
        [2.2774e-08, 1.0000e+00],
        [1.3749e-10, 1.0000e+00],
        [9.9988e-01, 1.2329e-04],
        [9.6040e-08, 1.0000e+00],
        [2.3473e-11, 1.0000e+00],
        [8.9445e-08, 1.0000e+00],
        [5.8079e-08, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7360e-07, 1.0000e+00],
        [3.1761e-09, 1.0000e+00],
        [1.5446e-08, 1.0000e+00],
        [2.7713e-07, 1.0000e+00],
        [5.7238e-07, 1.0000e+00],
        [3.9446e-08, 1.0000e+00],
        [1.5752e-05, 9.9998e-01],
        [2.4945e-10, 1.0000e+00],
        [9.9834e-01, 1.6642e-03],
        [2.4980e-07, 1.0000e+00],
        [2.9966e-08, 1.0000e+00],
        [2.3571e-11, 1.0000e+00],
        [2.9877e-01, 7.0123e-01],
        [9.9976e-01, 2.3826e-04],
        [1.5114e-03, 9.9849e-01],
        [9.9999e-01, 1.1742e-05],
        [1.1273e-07, 1.0000e+00],
        [2.2861e-06, 1.0000e+00],
        [8.7514e-05, 9.9991e-01],
        [1.9284e-11, 1.0000e+00],
        [5.6760e-10, 1.0000e+00],
        [5.9642e-05, 9.9994e-01],
        [1.0480e-08, 1.0000e+00],
        [8.3092e-07, 1.0000e+00],
        [1.1724e-06, 1.0000e+00],
        [8.9102e-05, 9.9991e-01],
        [2.6771e-09, 1.0000e+00],
        [4.2152e-05, 9.9996e-01],
        [9.9997e-01, 2.7207e-05],
        [9.9997e-01, 2.8299e-05],
        [2.4131e-10, 1.0000e+00],
        [3.8302e-03, 9.9617e-01],
        [8.9383e-01, 1.0617e-01],
        [1.9539e-08, 1.0000e+00],
        [9.9959e-01, 4.1497e-04],
        [8.6306e-04, 9.9914e-01],
        [8.4848e-06, 9.9999e-01],
        [9.9987e-01, 1.3163e-04],
        [5.4566e-04, 9.9945e-01],
        [4.3470e-07, 1.0000e+00],
        [8.9383e-10, 1.0000e+00],
        [9.5657e-01, 4.3427e-02],
        [2.3472e-08, 1.0000e+00],
        [1.1382e-03, 9.9886e-01],
        [6.7759e-06, 9.9999e-01],
        [9.9507e-01, 4.9264e-03],
        [1.6504e-10, 1.0000e+00],
        [1.9293e-11, 1.0000e+00],
        [8.4925e-07, 1.0000e+00],
        [9.9997e-01, 2.7781e-05],
        [5.5397e-06, 9.9999e-01],
        [2.4085e-06, 1.0000e+00],
        [9.9997e-01, 2.7098e-05],
        [5.2518e-07, 1.0000e+00],
        [9.2030e-01, 7.9701e-02],
        [9.3100e-10, 1.0000e+00],
        [4.5028e-05, 9.9995e-01],
        [2.9221e-11, 1.0000e+00],
        [5.2320e-08, 1.0000e+00],
        [9.9996e-01, 4.1036e-05],
        [2.1389e-11, 1.0000e+00],
        [3.0784e-08, 1.0000e+00],
        [8.3547e-10, 1.0000e+00],
        [9.1251e-15, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[3.2687e-03, 9.9673e-01],
        [6.5974e-09, 1.0000e+00],
        [4.0200e-05, 9.9996e-01],
        [2.0832e-12, 1.0000e+00],
        [9.9988e-01, 1.1563e-04],
        [1.6941e-08, 1.0000e+00],
        [4.1129e-06, 1.0000e+00],
        [4.1921e-02, 9.5808e-01],
        [4.6490e-10, 1.0000e+00],
        [9.9979e-01, 2.0833e-04],
        [5.3915e-10, 1.0000e+00],
        [1.2927e-05, 9.9999e-01],
        [4.2463e-13, 1.0000e+00],
        [1.3689e-08, 1.0000e+00],
        [6.2600e-11, 1.0000e+00],
        [5.0033e-11, 1.0000e+00],
        [6.8727e-09, 1.0000e+00],
        [3.8243e-03, 9.9618e-01],
        [9.9982e-01, 1.8112e-04],
        [1.5412e-07, 1.0000e+00],
        [1.9591e-09, 1.0000e+00],
        [1.5144e-10, 1.0000e+00],
        [8.7378e-10, 1.0000e+00],
        [7.4466e-06, 9.9999e-01],
        [9.9991e-01, 9.0762e-05],
        [2.8235e-08, 1.0000e+00],
        [1.6179e-07, 1.0000e+00],
        [5.8758e-10, 1.0000e+00],
        [3.2208e-06, 1.0000e+00],
        [9.9989e-01, 1.0924e-04],
        [5.6057e-07, 1.0000e+00],
        [2.6179e-01, 7.3821e-01],
        [1.1427e-07, 1.0000e+00],
        [2.3579e-07, 1.0000e+00],
        [9.9993e-01, 6.5137e-05],
        [3.9817e-10, 1.0000e+00],
        [1.2233e-09, 1.0000e+00],
        [1.8885e-08, 1.0000e+00],
        [3.5725e-08, 1.0000e+00],
        [5.3909e-08, 1.0000e+00],
        [7.7709e-05, 9.9992e-01],
        [9.9994e-01, 6.3990e-05],
        [9.6375e-09, 1.0000e+00],
        [5.7421e-04, 9.9943e-01],
        [2.8742e-06, 1.0000e+00],
        [3.7276e-06, 1.0000e+00],
        [8.3232e-01, 1.6768e-01],
        [1.2760e-11, 1.0000e+00],
        [1.5083e-12, 1.0000e+00],
        [9.9964e-01, 3.6043e-04],
        [4.2077e-07, 1.0000e+00],
        [1.1040e-11, 1.0000e+00],
        [4.4779e-07, 1.0000e+00],
        [2.9626e-06, 1.0000e+00],
        [6.8550e-03, 9.9314e-01],
        [5.6080e-07, 1.0000e+00],
        [1.6635e-07, 1.0000e+00],
        [2.3369e-06, 1.0000e+00],
        [9.8927e-08, 1.0000e+00],
        [2.5464e-04, 9.9975e-01],
        [1.6242e-07, 1.0000e+00],
        [2.2135e-11, 1.0000e+00],
        [2.6364e-08, 1.0000e+00],
        [1.8162e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7682e-08, 1.0000e+00],
        [4.7595e-10, 1.0000e+00],
        [9.9995e-01, 5.4665e-05],
        [2.8804e-04, 9.9971e-01],
        [3.2062e-05, 9.9997e-01],
        [9.9989e-01, 1.1052e-04],
        [6.6757e-06, 9.9999e-01],
        [1.1249e-06, 1.0000e+00],
        [3.3441e-09, 1.0000e+00],
        [5.8921e-10, 1.0000e+00],
        [1.5450e-09, 1.0000e+00],
        [2.0429e-01, 7.9571e-01],
        [1.0042e-05, 9.9999e-01],
        [2.1401e-05, 9.9998e-01],
        [5.7855e-06, 9.9999e-01],
        [1.9095e-09, 1.0000e+00],
        [4.0778e-05, 9.9996e-01],
        [2.9448e-07, 1.0000e+00],
        [1.0887e-05, 9.9999e-01],
        [3.9718e-06, 1.0000e+00],
        [5.7887e-02, 9.4211e-01],
        [5.4598e-03, 9.9454e-01],
        [2.3090e-10, 1.0000e+00],
        [3.7224e-08, 1.0000e+00],
        [9.9991e-01, 9.4877e-05],
        [4.1550e-03, 9.9585e-01],
        [9.9816e-01, 1.8447e-03],
        [9.9970e-01, 2.9997e-04],
        [9.9986e-01, 1.4088e-04],
        [2.4560e-07, 1.0000e+00],
        [9.9674e-01, 3.2580e-03],
        [2.9247e-08, 1.0000e+00],
        [1.5403e-04, 9.9985e-01],
        [9.3342e-01, 6.6575e-02],
        [3.6083e-09, 1.0000e+00],
        [3.1345e-07, 1.0000e+00],
        [9.9993e-01, 7.3613e-05],
        [1.5348e-07, 1.0000e+00],
        [8.3400e-11, 1.0000e+00],
        [5.9391e-03, 9.9406e-01],
        [9.9993e-01, 6.6048e-05],
        [3.9335e-10, 1.0000e+00],
        [6.9086e-06, 9.9999e-01],
        [1.7272e-04, 9.9983e-01],
        [3.9762e-04, 9.9960e-01],
        [9.9992e-01, 7.6284e-05],
        [3.3150e-04, 9.9967e-01],
        [9.9968e-01, 3.1540e-04],
        [4.3413e-10, 1.0000e+00],
        [4.1288e-12, 1.0000e+00],
        [8.2758e-08, 1.0000e+00],
        [3.3783e-08, 1.0000e+00],
        [2.2692e-05, 9.9998e-01],
        [1.3282e-09, 1.0000e+00],
        [9.9961e-01, 3.9134e-04],
        [5.2735e-03, 9.9473e-01],
        [9.9997e-01, 3.3324e-05],
        [5.0988e-11, 1.0000e+00],
        [7.1516e-09, 1.0000e+00],
        [5.2966e-07, 1.0000e+00],
        [9.9998e-01, 2.2993e-05],
        [1.0945e-03, 9.9891e-01],
        [2.7161e-08, 1.0000e+00],
        [3.0009e-12, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[8.0365e-08, 1.0000e+00],
        [5.0334e-03, 9.9497e-01],
        [8.3198e-07, 1.0000e+00],
        [1.3405e-06, 1.0000e+00],
        [1.2774e-06, 1.0000e+00],
        [6.0145e-11, 1.0000e+00],
        [3.3084e-07, 1.0000e+00],
        [1.5282e-09, 1.0000e+00],
        [9.8006e-01, 1.9940e-02],
        [3.5029e-05, 9.9996e-01],
        [9.0788e-04, 9.9909e-01],
        [9.9779e-01, 2.2083e-03],
        [9.9998e-01, 2.4462e-05],
        [1.6518e-08, 1.0000e+00],
        [3.3936e-04, 9.9966e-01],
        [1.1716e-03, 9.9883e-01],
        [9.9962e-01, 3.8387e-04],
        [1.3779e-06, 1.0000e+00],
        [5.1134e-06, 9.9999e-01],
        [9.9982e-01, 1.8124e-04],
        [5.4513e-10, 1.0000e+00],
        [9.9991e-01, 8.9397e-05],
        [9.9992e-01, 7.5716e-05],
        [9.0077e-01, 9.9229e-02],
        [9.9998e-01, 2.4457e-05],
        [3.3758e-04, 9.9966e-01],
        [9.9904e-01, 9.6488e-04],
        [1.8552e-03, 9.9814e-01],
        [2.1580e-07, 1.0000e+00],
        [8.7971e-07, 1.0000e+00],
        [3.4833e-09, 1.0000e+00],
        [1.6044e-11, 1.0000e+00],
        [2.4638e-03, 9.9754e-01],
        [6.9580e-07, 1.0000e+00],
        [9.9997e-01, 2.6723e-05],
        [1.9434e-06, 1.0000e+00],
        [2.1991e-08, 1.0000e+00],
        [2.2485e-07, 1.0000e+00],
        [4.3355e-07, 1.0000e+00],
        [2.7453e-06, 1.0000e+00],
        [7.7092e-02, 9.2291e-01],
        [1.4160e-04, 9.9986e-01],
        [1.4516e-08, 1.0000e+00],
        [9.9871e-01, 1.2916e-03],
        [5.5836e-07, 1.0000e+00],
        [8.0406e-06, 9.9999e-01],
        [4.2424e-09, 1.0000e+00],
        [4.3170e-07, 1.0000e+00],
        [5.5468e-06, 9.9999e-01],
        [1.9610e-08, 1.0000e+00],
        [3.9172e-05, 9.9996e-01],
        [3.0441e-06, 1.0000e+00],
        [1.6405e-05, 9.9998e-01],
        [1.5156e-07, 1.0000e+00],
        [1.5023e-08, 1.0000e+00],
        [1.8260e-04, 9.9982e-01],
        [9.9995e-01, 4.8676e-05],
        [2.6539e-04, 9.9973e-01],
        [1.8231e-14, 1.0000e+00],
        [3.3935e-05, 9.9997e-01],
        [7.1942e-09, 1.0000e+00],
        [2.1653e-10, 1.0000e+00],
        [4.4049e-06, 1.0000e+00],
        [3.5695e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
tensor([[6.5762e-07, 1.0000e+00],
        [5.9194e-01, 4.0806e-01],
        [3.8613e-09, 1.0000e+00],
        [9.9996e-01, 4.3089e-05],
        [9.9880e-01, 1.1997e-03],
        [1.7253e-04, 9.9983e-01],
        [4.9621e-01, 5.0379e-01],
        [1.9557e-06, 1.0000e+00],
        [1.4999e-10, 1.0000e+00],
        [9.9900e-01, 9.9663e-04],
        [5.6321e-08, 1.0000e+00],
        [2.5328e-05, 9.9997e-01],
        [3.5917e-10, 1.0000e+00],
        [9.9987e-01, 1.2879e-04],
        [1.0000e+00, 1.0289e-06],
        [2.3996e-05, 9.9998e-01],
        [5.0439e-06, 9.9999e-01],
        [1.6124e-08, 1.0000e+00],
        [5.3738e-06, 9.9999e-01],
        [9.9983e-01, 1.6962e-04],
        [8.0822e-08, 1.0000e+00],
        [1.3586e-05, 9.9999e-01],
        [6.9372e-10, 1.0000e+00],
        [8.3850e-06, 9.9999e-01],
        [6.5368e-04, 9.9935e-01],
        [9.9920e-01, 7.9885e-04],
        [4.8005e-06, 1.0000e+00],
        [1.0082e-08, 1.0000e+00],
        [9.9957e-01, 4.2923e-04],
        [9.9999e-01, 8.8644e-06],
        [2.1649e-06, 1.0000e+00],
        [9.9998e-01, 1.7659e-05],
        [9.9980e-01, 1.9642e-04],
        [9.9960e-01, 4.0303e-04],
        [6.9821e-05, 9.9993e-01],
        [1.0365e-08, 1.0000e+00],
        [4.5501e-07, 1.0000e+00],
        [9.9995e-01, 4.5809e-05],
        [2.3387e-03, 9.9766e-01],
        [3.1046e-07, 1.0000e+00],
        [2.0840e-08, 1.0000e+00],
        [1.7960e-02, 9.8204e-01],
        [9.3986e-04, 9.9906e-01],
        [5.3715e-05, 9.9995e-01],
        [6.8739e-08, 1.0000e+00],
        [6.1156e-04, 9.9939e-01],
        [9.9966e-01, 3.3794e-04],
        [9.9813e-01, 1.8698e-03],
        [9.9990e-01, 1.0220e-04],
        [7.8486e-08, 1.0000e+00],
        [1.0000e+00, 3.6866e-06],
        [2.5328e-05, 9.9997e-01],
        [1.0803e-05, 9.9999e-01],
        [2.0776e-04, 9.9979e-01],
        [2.8729e-06, 1.0000e+00],
        [8.9559e-02, 9.1044e-01],
        [3.8088e-08, 1.0000e+00],
        [1.7091e-06, 1.0000e+00],
        [4.3660e-08, 1.0000e+00],
        [2.6552e-08, 1.0000e+00],
        [9.9925e-01, 7.4708e-04],
        [2.6461e-05, 9.9997e-01],
        [1.6449e-07, 1.0000e+00],
        [5.6809e-02, 9.4319e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.7119e-05, 9.9998e-01],
        [1.3649e-05, 9.9999e-01],
        [2.7641e-10, 1.0000e+00],
        [9.8765e-06, 9.9999e-01],
        [2.2469e-04, 9.9978e-01],
        [1.5747e-11, 1.0000e+00],
        [1.2210e-06, 1.0000e+00],
        [1.9654e-04, 9.9980e-01],
        [9.9996e-01, 4.2005e-05],
        [1.0000e+00, 2.9241e-06],
        [9.9975e-01, 2.5251e-04],
        [2.4965e-13, 1.0000e+00],
        [4.0800e-11, 1.0000e+00],
        [1.0338e-05, 9.9999e-01],
        [6.8904e-07, 1.0000e+00],
        [7.8574e-08, 1.0000e+00],
        [3.2617e-05, 9.9997e-01],
        [8.1849e-09, 1.0000e+00],
        [2.8668e-11, 1.0000e+00],
        [1.4240e-04, 9.9986e-01],
        [1.1395e-07, 1.0000e+00],
        [9.9996e-01, 4.0965e-05],
        [1.6242e-08, 1.0000e+00],
        [9.9994e-01, 6.1461e-05],
        [9.9970e-01, 2.9881e-04],
        [6.2362e-05, 9.9994e-01],
        [8.1698e-01, 1.8302e-01],
        [1.1293e-08, 1.0000e+00],
        [5.6742e-10, 1.0000e+00],
        [1.4769e-06, 1.0000e+00],
        [1.6543e-05, 9.9998e-01],
        [4.2107e-05, 9.9996e-01],
        [9.4033e-05, 9.9991e-01],
        [8.8092e-03, 9.9119e-01],
        [1.3924e-02, 9.8608e-01],
        [6.3788e-01, 3.6212e-01],
        [3.3856e-10, 1.0000e+00],
        [9.9995e-01, 5.3294e-05],
        [2.4052e-02, 9.7595e-01],
        [4.0924e-07, 1.0000e+00],
        [7.4027e-08, 1.0000e+00],
        [9.9995e-01, 4.7359e-05],
        [4.3230e-11, 1.0000e+00],
        [1.1214e-09, 1.0000e+00],
        [3.3802e-01, 6.6198e-01],
        [3.0414e-04, 9.9970e-01],
        [1.4805e-06, 1.0000e+00],
        [9.9998e-01, 1.9558e-05],
        [9.9987e-01, 1.2636e-04],
        [2.1138e-07, 1.0000e+00],
        [1.8424e-02, 9.8158e-01],
        [1.7303e-08, 1.0000e+00],
        [2.1627e-08, 1.0000e+00],
        [1.1591e-05, 9.9999e-01],
        [3.6747e-10, 1.0000e+00],
        [1.1497e-10, 1.0000e+00],
        [9.2001e-08, 1.0000e+00],
        [8.2934e-06, 9.9999e-01],
        [2.1525e-08, 1.0000e+00],
        [4.4200e-04, 9.9956e-01],
        [5.4454e-07, 1.0000e+00],
        [1.4876e-06, 1.0000e+00],
        [5.5135e-08, 1.0000e+00],
        [2.3807e-04, 9.9976e-01]], grad_fn=<SoftmaxBackward0>)
tensor([[1.2369e-08, 1.0000e+00],
        [6.6178e-09, 1.0000e+00],
        [2.2626e-09, 1.0000e+00],
        [1.9047e-12, 1.0000e+00],
        [2.0730e-10, 1.0000e+00],
        [9.7042e-11, 1.0000e+00],
        [9.9989e-01, 1.1349e-04],
        [3.0380e-05, 9.9997e-01],
        [1.0908e-04, 9.9989e-01],
        [2.4768e-07, 1.0000e+00],
        [2.5424e-07, 1.0000e+00],
        [1.7181e-05, 9.9998e-01],
        [7.5548e-08, 1.0000e+00],
        [2.2445e-03, 9.9776e-01],
        [1.4009e-07, 1.0000e+00],
        [1.1239e-10, 1.0000e+00],
        [8.7819e-01, 1.2181e-01],
        [9.9988e-01, 1.1497e-04],
        [5.3894e-10, 1.0000e+00],
        [9.9992e-01, 8.2254e-05],
        [1.5670e-09, 1.0000e+00],
        [9.8786e-01, 1.2141e-02],
        [1.0672e-07, 1.0000e+00]], grad_fn=<SoftmaxBackward0>)
Finished Training

correct = 0
total = 0
with torch.no_grad():
    for data in test_dataloader:
        inputs, labels = data
        outputs = model(inputs.float())
        _, predicted = torch.max(outputs.data, 1)
        total += labels.size(0)
        correct += (predicted == labels).sum().item()
print('Accuracy on test set: %d %%' % (100 * correct / total))

tensor([[1.7202e-10, 1.0000e+00],
        [1.2536e-05, 9.9999e-01],
        [1.6356e-06, 1.0000e+00],
        [8.5210e-09, 1.0000e+00],
        [2.6330e-06, 1.0000e+00],
        [3.1719e-11, 1.0000e+00],
        [7.5173e-12, 1.0000e+00],
        [1.4972e-09, 1.0000e+00],
        [9.9978e-01, 2.1951e-04],
        [9.9896e-01, 1.0420e-03],
        [2.6575e-11, 1.0000e+00],
        [3.9438e-01, 6.0562e-01],
        [2.0028e-10, 1.0000e+00],
        [2.2875e-11, 1.0000e+00],
        [9.0060e-01, 9.9397e-02],
        [1.6835e-04, 9.9983e-01],
        [6.5582e-07, 1.0000e+00],
        [1.1467e-06, 1.0000e+00],
        [7.9620e-11, 1.0000e+00],
        [5.2297e-07, 1.0000e+00],
        [2.2945e-09, 1.0000e+00],
        [1.8827e-14, 1.0000e+00],
        [3.6914e-12, 1.0000e+00],
        [1.0232e-05, 9.9999e-01],
        [9.9996e-01, 4.3504e-05],
        [1.2202e-07, 1.0000e+00],
        [9.5178e-08, 1.0000e+00],
        [1.0603e-02, 9.8940e-01],
        [9.4093e-01, 5.9069e-02],
        [9.9953e-01, 4.6727e-04],
        [4.0027e-05, 9.9996e-01],
        [1.0232e-05, 9.9999e-01],
        [9.9990e-01, 9.7410e-05],
        [6.9317e-10, 1.0000e+00],
        [7.1531e-01, 2.8469e-01],
        [4.5204e-08, 1.0000e+00],
        [1.7716e-03, 9.9823e-01],
        [9.5942e-06, 9.9999e-01],
        [7.0563e-06, 9.9999e-01],
        [2.9637e-08, 1.0000e+00],
        [4.3060e-07, 1.0000e+00],
        [9.8936e-09, 1.0000e+00],
        [5.7116e-03, 9.9429e-01],
        [6.6022e-10, 1.0000e+00],
        [9.8973e-05, 9.9990e-01],
        [1.1994e-04, 9.9988e-01],
        [1.5506e-11, 1.0000e+00],
        [9.9975e-01, 2.4858e-04],
        [4.5068e-05, 9.9995e-01],
        [4.7508e-09, 1.0000e+00],
        [1.3963e-08, 1.0000e+00],
        [1.5434e-04, 9.9985e-01],
        [1.1042e-05, 9.9999e-01],
        [5.0842e-11, 1.0000e+00],
        [1.4754e-07, 1.0000e+00],
        [1.3403e-07, 1.0000e+00],
        [4.5559e-07, 1.0000e+00],
        [4.8323e-02, 9.5168e-01],
        [3.0700e-07, 1.0000e+00],
        [1.0562e-09, 1.0000e+00],
        [9.9991e-01, 9.1597e-05],
        [5.3040e-04, 9.9947e-01],
        [9.9965e-01, 3.4706e-04],
        [9.9995e-01, 5.3858e-05]])
tensor([[4.1633e-12, 1.0000e+00],
        [6.7926e-01, 3.2074e-01],
        [1.2453e-03, 9.9875e-01],
        [9.9981e-01, 1.8949e-04],
        [3.2815e-10, 1.0000e+00],
        [1.9707e-11, 1.0000e+00],
        [3.9621e-13, 1.0000e+00],
        [1.9707e-11, 1.0000e+00],
        [1.8665e-02, 9.8133e-01],
        [5.1256e-11, 1.0000e+00],
        [9.9910e-01, 8.9839e-04],
        [4.1963e-04, 9.9958e-01],
        [1.3730e-07, 1.0000e+00],
        [2.9742e-12, 1.0000e+00],
        [8.1369e-07, 1.0000e+00],
        [9.9995e-01, 5.4186e-05],
        [9.9994e-01, 5.5641e-05],
        [2.6850e-08, 1.0000e+00],
        [1.7928e-04, 9.9982e-01],
        [2.9322e-09, 1.0000e+00],
        [9.6977e-07, 1.0000e+00],
        [9.9376e-01, 6.2373e-03],
        [9.9942e-01, 5.8043e-04],
        [1.7943e-05, 9.9998e-01],
        [1.9373e-06, 1.0000e+00],
        [5.7387e-08, 1.0000e+00],
        [9.9805e-01, 1.9493e-03],
        [9.9997e-01, 3.0799e-05],
        [6.6228e-06, 9.9999e-01],
        [6.6534e-08, 1.0000e+00],
        [9.9983e-01, 1.6921e-04],
        [7.0741e-07, 1.0000e+00],
        [5.9886e-05, 9.9994e-01],
        [1.5063e-06, 1.0000e+00],
        [3.7295e-08, 1.0000e+00],
        [9.9988e-01, 1.1825e-04],
        [7.5359e-01, 2.4641e-01],
        [9.9989e-01, 1.1214e-04],
        [6.5426e-06, 9.9999e-01],
        [1.0272e-07, 1.0000e+00],
        [9.4599e-10, 1.0000e+00],
        [8.9316e-05, 9.9991e-01],
        [3.1955e-13, 1.0000e+00],
        [2.5647e-09, 1.0000e+00],
        [8.5594e-04, 9.9914e-01],
        [1.0665e-10, 1.0000e+00],
        [5.5014e-08, 1.0000e+00],
        [5.2700e-04, 9.9947e-01],
        [2.0910e-08, 1.0000e+00],
        [4.3795e-14, 1.0000e+00],
        [9.9993e-01, 6.9620e-05],
        [8.1168e-10, 1.0000e+00],
        [9.9992e-01, 7.9322e-05],
        [9.9971e-01, 2.9203e-04],
        [9.6872e-01, 3.1279e-02],
        [1.6663e-10, 1.0000e+00],
        [6.7821e-05, 9.9993e-01],
        [2.7496e-10, 1.0000e+00],
        [9.9988e-01, 1.2034e-04],
        [1.1160e-06, 1.0000e+00],
        [8.8415e-05, 9.9991e-01],
        [1.5483e-10, 1.0000e+00],
        [1.3105e-12, 1.0000e+00],
        [1.3140e-03, 9.9869e-01]])
tensor([[1.2899e-08, 1.0000e+00],
        [9.9995e-01, 5.1878e-05],
        [1.9329e-03, 9.9807e-01],
        [2.8661e-05, 9.9997e-01],
        [9.9995e-01, 4.9032e-05],
        [1.3782e-04, 9.9986e-01],
        [9.9296e-01, 7.0404e-03],
        [5.0526e-01, 4.9474e-01],
        [9.9987e-01, 1.3229e-04],
        [7.0428e-03, 9.9296e-01],
        [1.5157e-09, 1.0000e+00],
        [9.4394e-07, 1.0000e+00],
        [4.0096e-08, 1.0000e+00],
        [6.0166e-07, 1.0000e+00],
        [3.8869e-05, 9.9996e-01],
        [9.9996e-01, 3.9988e-05],
        [3.0818e-10, 1.0000e+00],
        [2.4525e-08, 1.0000e+00],
        [3.6321e-07, 1.0000e+00],
        [1.2087e-08, 1.0000e+00],
        [5.9808e-06, 9.9999e-01],
        [2.3110e-07, 1.0000e+00],
        [2.2577e-09, 1.0000e+00],
        [7.0174e-08, 1.0000e+00],
        [1.9678e-05, 9.9998e-01],
        [1.8164e-07, 1.0000e+00],
        [4.8094e-06, 1.0000e+00],
        [9.9634e-01, 3.6568e-03],
        [1.5208e-05, 9.9998e-01],
        [9.9989e-01, 1.1227e-04],
        [7.7115e-11, 1.0000e+00],
        [1.1317e-06, 1.0000e+00],
        [4.2370e-03, 9.9576e-01],
        [2.4711e-07, 1.0000e+00],
        [1.4052e-04, 9.9986e-01],
        [9.9978e-01, 2.2135e-04],
        [6.6022e-10, 1.0000e+00],
        [5.0018e-07, 1.0000e+00],
        [6.2135e-08, 1.0000e+00],
        [9.9815e-01, 1.8476e-03],
        [2.0338e-04, 9.9980e-01],
        [1.1686e-10, 1.0000e+00],
        [8.1482e-07, 1.0000e+00],
        [6.5565e-06, 9.9999e-01],
        [4.8845e-05, 9.9995e-01],
        [2.1383e-10, 1.0000e+00],
        [2.0092e-12, 1.0000e+00],
        [9.9991e-01, 8.5367e-05],
        [1.0000e+00, 1.2982e-07],
        [9.9602e-01, 3.9821e-03],
        [9.9901e-01, 9.8763e-04],
        [8.4893e-07, 1.0000e+00],
        [2.0584e-10, 1.0000e+00],
        [3.1034e-06, 1.0000e+00],
        [9.9995e-01, 5.2120e-05],
        [9.9995e-01, 5.3451e-05],
        [2.0178e-06, 1.0000e+00],
        [3.0938e-08, 1.0000e+00],
        [3.2183e-07, 1.0000e+00],
        [9.9992e-01, 8.2386e-05],
        [9.9998e-01, 2.1354e-05],
        [1.0601e-06, 1.0000e+00],
        [3.1632e-08, 1.0000e+00],
        [1.1469e-07, 1.0000e+00]])
tensor([[9.9987e-01, 1.3117e-04],
        [8.1482e-07, 1.0000e+00],
        [1.2307e-02, 9.8769e-01],
        [9.5555e-05, 9.9990e-01],
        [9.9986e-01, 1.4005e-04],
        [3.5660e-11, 1.0000e+00],
        [5.4389e-09, 1.0000e+00],
        [3.2685e-07, 1.0000e+00],
        [6.9741e-07, 1.0000e+00],
        [9.9994e-01, 6.3540e-05],
        [3.2789e-10, 1.0000e+00],
        [6.3978e-08, 1.0000e+00],
        [9.9990e-01, 9.7314e-05],
        [9.9959e-01, 4.1339e-04],
        [1.8004e-05, 9.9998e-01],
        [9.0941e-06, 9.9999e-01],
        [1.7119e-09, 1.0000e+00],
        [9.9992e-01, 7.7148e-05],
        [9.9983e-01, 1.6945e-04],
        [1.0338e-05, 9.9999e-01],
        [3.4170e-12, 1.0000e+00],
        [1.3689e-08, 1.0000e+00],
        [2.4280e-05, 9.9998e-01],
        [3.3277e-13, 1.0000e+00],
        [1.9654e-05, 9.9998e-01],
        [7.1331e-11, 1.0000e+00],
        [4.6505e-01, 5.3495e-01],
        [3.3887e-08, 1.0000e+00],
        [1.6281e-13, 1.0000e+00],
        [3.6157e-05, 9.9996e-01],
        [1.0583e-09, 1.0000e+00],
        [6.5631e-09, 1.0000e+00],
        [9.9984e-01, 1.5877e-04],
        [1.0608e-06, 1.0000e+00],
        [8.0816e-10, 1.0000e+00],
        [8.9659e-08, 1.0000e+00],
        [9.9980e-01, 2.0301e-04],
        [2.0268e-07, 1.0000e+00],
        [3.9404e-11, 1.0000e+00],
        [1.3131e-04, 9.9987e-01],
        [3.8976e-09, 1.0000e+00],
        [9.9990e-01, 9.6786e-05],
        [9.9367e-01, 6.3258e-03],
        [1.5826e-11, 1.0000e+00],
        [1.4497e-07, 1.0000e+00],
        [9.9992e-01, 8.4039e-05],
        [1.1915e-06, 1.0000e+00],
        [2.3967e-06, 1.0000e+00],
        [1.4972e-09, 1.0000e+00],
        [2.4645e-07, 1.0000e+00],
        [1.3402e-09, 1.0000e+00],
        [2.1213e-08, 1.0000e+00],
        [9.9996e-01, 4.2852e-05],
        [3.9569e-09, 1.0000e+00],
        [9.9990e-01, 1.0322e-04],
        [1.0075e-07, 1.0000e+00],
        [2.6908e-10, 1.0000e+00],
        [9.9591e-01, 4.0942e-03],
        [7.4493e-06, 9.9999e-01],
        [9.9786e-01, 2.1421e-03],
        [1.6372e-10, 1.0000e+00],
        [6.9996e-07, 1.0000e+00],
        [3.7268e-12, 1.0000e+00],
        [7.8684e-11, 1.0000e+00]])
tensor([[2.5668e-09, 1.0000e+00],
        [3.6802e-02, 9.6320e-01],
        [1.4714e-10, 1.0000e+00],
        [3.6272e-07, 1.0000e+00],
        [1.0259e-09, 1.0000e+00],
        [9.9990e-01, 9.5715e-05],
        [1.7885e-04, 9.9982e-01],
        [1.0895e-08, 1.0000e+00],
        [3.2713e-02, 9.6729e-01],
        [1.4319e-02, 9.8568e-01],
        [9.9973e-01, 2.7224e-04],
        [1.5526e-07, 1.0000e+00],
        [3.2053e-06, 1.0000e+00],
        [1.5020e-07, 1.0000e+00],
        [9.9993e-01, 7.0196e-05],
        [2.8287e-08, 1.0000e+00],
        [3.0607e-12, 1.0000e+00],
        [9.9982e-01, 1.8064e-04],
        [1.1989e-05, 9.9999e-01],
        [1.4612e-05, 9.9999e-01],
        [1.3966e-03, 9.9860e-01],
        [1.6816e-04, 9.9983e-01],
        [2.5268e-08, 1.0000e+00],
        [1.4710e-08, 1.0000e+00],
        [1.3442e-12, 1.0000e+00],
        [3.8333e-08, 1.0000e+00],
        [1.5960e-06, 1.0000e+00],
        [2.4518e-01, 7.5482e-01],
        [4.1385e-10, 1.0000e+00],
        [9.9988e-01, 1.1811e-04],
        [9.9990e-01, 9.9362e-05],
        [1.8937e-08, 1.0000e+00],
        [9.9996e-01, 3.8535e-05],
        [9.9989e-01, 1.1227e-04],
        [9.9983e-01, 1.7181e-04],
        [9.0298e-11, 1.0000e+00],
        [2.0243e-04, 9.9980e-01],
        [1.2437e-08, 1.0000e+00],
        [2.9142e-06, 1.0000e+00],
        [2.2979e-07, 1.0000e+00],
        [9.9991e-01, 8.8176e-05],
        [1.2222e-11, 1.0000e+00],
        [9.9970e-01, 3.0296e-04],
        [4.9068e-07, 1.0000e+00],
        [9.9883e-01, 1.1710e-03],
        [4.6909e-09, 1.0000e+00],
        [7.0492e-11, 1.0000e+00],
        [9.9998e-01, 2.1934e-05],
        [8.1731e-07, 1.0000e+00],
        [2.4994e-06, 1.0000e+00],
        [1.4337e-07, 1.0000e+00],
        [3.4392e-09, 1.0000e+00],
        [1.9588e-10, 1.0000e+00],
        [5.2363e-02, 9.4764e-01],
        [2.2923e-04, 9.9977e-01],
        [3.1012e-04, 9.9969e-01],
        [9.9985e-01, 1.4686e-04],
        [1.7467e-10, 1.0000e+00],
        [1.8465e-10, 1.0000e+00],
        [9.9990e-01, 1.0210e-04],
        [1.4477e-07, 1.0000e+00],
        [8.0056e-11, 1.0000e+00],
        [2.5159e-05, 9.9997e-01],
        [1.1317e-06, 1.0000e+00]])
tensor([[1.1592e-10, 1.0000e+00],
        [2.0072e-07, 1.0000e+00],
        [5.5997e-07, 1.0000e+00],
        [9.9974e-01, 2.6457e-04],
        [2.4987e-04, 9.9975e-01]])
Accuracy on test set: 98 %