I can format the output better to just leave the shapes if that is helpful?
The other “half” of the Markov output being the following:
K = {int} 16
args = {Namespace} Namespace(saved=False, num_steps=50, batch_size=30, sequence_len=104, hidden_dim=16, learning_rate=0.01, file_path=WindowsPath('D:/Work miscellaneous/Downloads/ws300_level3_MultiNB_kmers_5_15_train.csv'), seed=0)
batch = {Tensor: (30,)} tensor([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17,\n 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29])
data1 = {Tensor: (30, 52)} tensor([[ 8, 8, 8, ..., 8, 8, 8],\n [ 1, 1, 1, ..., 1, 1, 1],\n [ 1, 1, 1, ..., 1, 1, 1],\n ...,\n [ 2, 2, 2, ..., 2, 2, 2],\n [11, 11, 11, ..., 11, 11, 11],\n [ 6, 6, 6, ..., 6, 6, 6]])
data2 = {Tensor: (30, 52)} tensor([[ 8, 8, 8, ..., 8, 8, 8],\n [ 1, 1, 1, ..., 1, 1, 1],\n [ 1, 1, 1, ..., 1, 1, 1],\n ...,\n [ 2, 2, 2, ..., 2, 2, 2],\n [11, 11, 11, ..., 11, 11, 11],\n [ 6, 6, 6, ..., 6, 6, 6]])
data_dim = {int} 1
epsilonyx = {Tensor: (16, 16)} tensor([[0.6581, 0.4897, 0.3875, 0.1918, 0.8458, 0.1278, 0.7048, 0.3319, 0.2588,\n 0.5898, 0.2403, 0.6152, 0.5982, 0.1288, 0.5832, 0.7130],\n [0.6979, 0.4371, 0.0901, 0.4229, 0.6737, 0.3176, 0.6898, 0.8330, 0.2389,\n 0.5049, 0.7067, 0.5
lengths = {int} 52
muovtheta = {Tensor: ()} tensor(30.8225, grad_fn=<SumBackward0>)
muy = {Tensor: (16,)} tensor([0.0432, 0.0892, 0.0689, 0.1261, 0.0641, 0.0889, 0.0491, 0.0565, 0.0031,\n 0.0237, 0.0413, 0.0729, 0.0981, 0.1125, 0.0226, 0.0397],\n grad_fn=<DivBackward0>)
num_sequences = {int} 30
prob_init = {Tensor: (16,)} tensor([0.0149, 0.1650, 0.0505, 0.0636, 0.0403, 0.1764, 0.1661, 0.0204, 0.0018,\n 0.0084, 0.0403, 0.0365, 0.0825, 0.0764, 0.0376, 0.0192],\n grad_fn=<DivBackward0>)
probs_s = {Tensor: (1,)} tensor([0.0100], grad_fn=<ClampBackward1>)
probs_x = {Tensor: (16, 16)} tensor([[9.1021e-01, 8.3677e-03, 6.4676e-03, 1.1830e-02, 6.0128e-03, 8.3444e-03,\n 4.6042e-03, 5.3013e-03, 2.9463e-04, 2.2284e-03, 3.8784e-03, 6.8428e-03,\n 9.2069e-03, 1.0558e-02, 2.1251e-03, 3.7250e-03],\n [7.5782e-04, 9.8403e-01, 1.2
probs_x1_t = {Tensor: (2, 16, 16, 1, 1, 1, 1, 16)} tensor([[[[[[[[9.1021e-01, 8.3677e-03, 6.4676e-03, ..., 1.0558e-02,\n 2.1251e-03, 3.7250e-03]]]]],\n\n\n\n\n [[[[[7.5782e-04, 9.8403e-01, 1.2081e-03, ..., 1.9721e-03,\n 3.9695e-04, 6.9581e-04]]]]],\n\n\n\n\n [[[[[1.9157e-03, 3.9511e-03, 9.5874e-01, ..., 4.9851e-03,\n 1.0034e-03, 1.7589e-03]]]]],\n\n\n\n\n ...,\n\n\n\n\n [[[[[2.0654e-03, 4.2601e-03, 3.2927e-03, ..., 9.5760e-01,\n 1.0819e-03, 1.8964e-03]]]]],\n\n\n\n\n [[[[[8.4507e-04, 1.7430e-03, 1.3472e-03, ..., 2.1991e-03,\n 9.8089e-01, 7.7593e-04]]]]],\n\n\n\n\n [[[[[2.8926e-03, 5.9662e-03, 4.6114e-03, ..., 7.5274e-03,\n 1.5152e-03, 9.3575e-01]]]]]],\n\n\n\n\n\n [[[[[[9.1021e-01, 8.3677e-03, 6.4676e-03, ..., 1.0558e-02,\n 2.1251e-03, 3.7250e-03]]]]],\n\n\n\n\n [[[[[7.5782e-04, 9.8403e-01, 1.2081e-03, ..., 1.9721e-03,\n 3.9695e-04, 6.9581e-04]]]]],\n\n\n\n\n [[[[[1.9157e-03, 3.9511e-03, 9.58...
probs_x2_t = {Tensor: (2, 16, 16, 1, 1, 1, 1, 16)} tensor([[[[[[[[9.1021e-01, 8.3677e-03, 6.4676e-03, ..., 1.0558e-02,\n 2.1251e-03, 3.7250e-03]]]]],\n\n\n\n\n [[[[[9.1021e-01, 8.3677e-03, 6.4676e-03, ..., 1.0558e-02,\n 2.1251e-03, 3.7250e-03]]]]],\n\n\n\n\n [[[[[9.1021e-01, 8.3677e-03, 6.4676e-03, ..., 1.0558e-02,\n 2.1251e-03, 3.7250e-03]]]]],\n\n\n\n\n ...,\n\n\n\n\n [[[[[9.1021e-01, 8.3677e-03, 6.4676e-03, ..., 1.0558e-02,\n 2.1251e-03, 3.7250e-03]]]]],\n\n\n\n\n [[[[[9.1021e-01, 8.3677e-03, 6.4676e-03, ..., 1.0558e-02,\n 2.1251e-03, 3.7250e-03]]]]],\n\n\n\n\n [[[[[9.1021e-01, 8.3677e-03, 6.4676e-03, ..., 1.0558e-02,\n 2.1251e-03, 3.7250e-03]]]]]],\n\n\n\n\n\n [[[[[[7.5782e-04, 9.8403e-01, 1.2081e-03, ..., 1.9721e-03,\n 3.9695e-04, 6.9581e-04]]]]],\n\n\n\n\n [[[[[7.5782e-04, 9.8403e-01, 1.2081e-03, ..., 1.9721e-03,\n 3.9695e-04, 6.9581e-04]]]]],\n\n\n\n\n [[[[[7.5782e-04, 9.8403e-01, 1.20...
probs_y = {Tensor: (16, 16, 16, 16)} tensor([[[[4.6958e-01, 7.3870e-02, 3.2052e-02, ..., 2.1780e-02,\n 1.2162e-02, 2.5602e-03],\n [2.2006e-02, 1.9347e-01, 7.6457e-02, ..., 5.0959e-02,\n 8.6678e-02, 1.0542e-01],\n [7.8160e-02, 6.3512e-02, 1.9419e-01, ..., 4.8891e-02,\n 5.8090e-02, 6.8918e-02],\n ...,\n [3.4034e-02, 4.3634e-02, 5.1779e-02, ..., 1.8184e-01,\n 3.4730e-02, 5.1301e-02],\n [1.0392e-01, 2.2576e-03, 1.1820e-01, ..., 9.9569e-02,\n 1.5914e-01, 2.2821e-02],\n [2.4776e-02, 9.2053e-02, 8.9401e-02, ..., 7.1076e-02,\n 2.5648e-02, 1.1154e-01]],\n\n [[1.9399e-01, 1.1225e-01, 4.8705e-02, ..., 3.3096e-02,\n 1.8481e-02, 3.8905e-03],\n [3.1744e-01, 1.3503e-01, 5.3361e-02, ..., 3.5565e-02,\n 6.0494e-02, 7.3577e-02],\n [7.8160e-02, 6.3512e-02, 1.9419e-01, ..., 4.8891e-02,\n 5.8090e-02, 6.8918e-02],\n ...,\n [3.4034e-02, 4.3634e-02, 5.1779e-02, ..., 1.8184...
probs_y1_t = {Tensor: (16, 2, 16, 16, 1, 1, 1, 30, 16)} tensor([[[[[[[[[5.0197e-02, 2.9047e-02, 1.2603e-02, ..., 8.5640e-03,\n 4.7823e-03, 1.0067e-03],\n [9.4993e-02, 5.6528e-01, 2.3850e-02, ..., 1.6206e-02,\n 9.0499e-03, 1.9051e-03],\n [9.4993e-02, 5.6528e-01, 2.3850e-02, ..., 1.6206e-02,\n 9.0499e-03, 1.9051e-03],\n ...,\n [7.5179e-02, 4.3502e-02, 6.3133e-01, ..., 1.2826e-02,\n 7.1623e-03, 1.5077e-03],\n [1.1934e-01, 6.9057e-02, 2.9963e-02, ..., 2.0360e-02,\n 1.1370e-02, 2.3934e-03],\n [1.3673e-01, 7.9118e-02, 3.4329e-02, ..., 2.3327e-02,\n 1.3026e-02, 2.7421e-03]]]]],\n\n\n\n\n [[[[[1.9399e-01, 1.1225e-01, 4.8705e-02, ..., 3.3096e-02,\n 1.8481e-02, 3.8905e-03],\n [1.9399e-01, 1.1225e-01, 4.8705e-02, ..., 3.3096e-02,\n 1.8481e-02, 3.8905e-03],\n [1.9399e-01, 1.1225e-01, 4.8705e-02, ..., 3.3096e-02,\n ...
probs_y2_t = {Tensor: (16, 1, 2, 16, 16, 1, 1, 1, 30, 16)} tensor([[[[[[[[[[5.0197e-02, 2.9047e-02, 1.2603e-02, ..., 8.5640e-03,\n 4.7823e-03, 1.0067e-03],\n [9.4993e-02, 5.6528e-01, 2.3850e-02, ..., 1.6206e-02,\n 9.0499e-03, 1.9051e-03],\n [9.4993e-02, 5.6528e-01, 2.3850e-02, ..., 1.6206e-02,\n 9.0499e-03, 1.9051e-03],\n ...,\n [7.5179e-02, 4.3502e-02, 6.3133e-01, ..., 1.2826e-02,\n 7.1623e-03, 1.5077e-03],\n [1.1934e-01, 6.9057e-02, 2.9963e-02, ..., 2.0360e-02,\n 1.1370e-02, 2.3934e-03],\n [1.3673e-01, 7.9118e-02, 3.4329e-02, ..., 2.3327e-02,\n 1.3026e-02, 2.7421e-03]]]]],\n\n\n\n\n [[[[[5.0197e-02, 2.9047e-02, 1.2603e-02, ..., 8.5640e-03,\n 4.7823e-03, 1.0067e-03],\n [9.4993e-02, 5.6528e-01, 2.3850e-02, ..., 1.6206e-02,\n 9.0499e-03, 1.9051e-03],\n [9.4993e-02, 5.6528e-01, 2.3850e-02, ..., 1.6206e-...
s = {Tensor: (2, 1, 1, 1, 1, 1, 1, 1, 1, 1)} tensor([[[[[[[[[[0.]]]]]]]]],\n\n\n\n\n\n\n\n\n [[[[[[[[[1.]]]]]]]]]])
sequences = {Tensor: (30, 104, 1)} tensor([[[ 8],\n [ 8],\n [ 8],\n ...,\n [ 8],\n [ 8],\n [ 8]],\n\n [[ 1],\n [ 1],\n [ 1],\n ...,\n [ 1],\n [ 1],\n [ 1]],\n\n [[ 1],\n [ 1],\n [ 1],\n ...,\n [ 1],\n [ 1],\n [ 1]],\n\n ...,\n\n [[ 2],\n [ 2],\n [ 2],\n ...,\n [ 2],\n [ 2],\n [ 2]],\n\n [[11],\n [11],\n [11],\n ...,\n [11],\n [11],\n [11]],\n\n [[ 6],\n [ 6],\n [ 6],\n ...,\n [ 6],\n [ 6],\n [ 6]]])
sigma = {Tensor: (1,)} tensor([0.0100], grad_fn=<ClampBackward1>)
t = {int} 6
thetay = {Tensor: (16,)} tensor([0.0938, 0.0175, 0.0443, 0.0643, 0.0516, 0.0164, 0.0096, 0.0899, 0.0581,\n 0.0915, 0.0332, 0.0647, 0.0386, 0.0478, 0.0195, 0.0669],\n grad_fn=<ClampBackward1>)
tmuxgivy = {Tensor: (16, 16)} tensor([[1.9399e-01, 1.1225e-01, 4.8705e-02, 1.0722e-01, 5.1441e-02, 6.7815e-02,\n 1.1685e-01, 4.4357e-03, 2.2719e-02, 4.5800e-02, 3.7421e-02, 1.1431e-01,\n 2.1576e-02, 3.3096e-02, 1.8481e-02, 3.8905e-03],\n [2.2006e-02, 1.9347e-01, 7.6457e-02, 7.8490e-02, 5.5647e-02, 2.5763e-02,\n 6.1811e-02, 3.5054e-03, 1.4667e-02, 2.5612e-02, 8.6223e-02, 8.3864e-02,\n 2.9421e-02, 5.0959e-02, 8.6678e-02, 1.0542e-01],\n [7.8160e-02, 6.3512e-02, 1.9419e-01, 2.3008e-02, 6.6380e-02, 1.2572e-02,\n 1.7173e-02, 2.7049e-02, 8.1271e-02, 7.8455e-02, 2.2809e-02, 7.2857e-02,\n 8.6670e-02, 4.8891e-02, 5.8090e-02, 6.8918e-02],\n [8.2116e-02, 9.9337e-02, 1.1624e-02, 1.2332e-01, 7.0594e-02, 9.2666e-02,\n 9.4777e-02, 9.5392e-02, 6.0763e-02, 6.6092e-03, 5.5339e-02, 1.8973e-02,\n 3.4484e-03, 9.5703e-02, 8.9210e-02, 1.2528e-04],\n [7.1237e-02, 4.9897e-02, 5.0131e-02, 3.2538e-02, 1.9109e-01, 2.4464e-02,\n 8.2004e-02, 9.0351e-02, 1.0296...
x1 = {Tensor: (16, 1, 1, 1, 1, 1, 1, 1)} tensor([[[[[[[[ 0]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 1]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 2]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 3]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 4]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 5]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 6]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 7]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 8]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 9]]]]]]],\n\n\n\n\n\n\n [[[[[[[10]]]]]]],\n\n\n\n\n\n\n [[[[[[[11]]]]]]],\n\n\n\n\n\n\n [[[[[[[12]]]]]]],\n\n\n\n\n\n\n [[[[[[[13]]]]]]],\n\n\n\n\n\n\n [[[[[[[14]]]]]]],\n\n\n\n\n\n\n [[[[[[[15]]]]]]]])
x1_help = {Tensor: (2, 16, 16, 1, 1, 1, 1)} tensor([[[[[[[ 0]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 3]]]],\n\n\n\n [[[[ 4]]]],\n\n\n\n [[[[ 5]]]],\n\n\n\n [[[[ 6]]]],\n\n\n\n [[[[ 7]]]],\n\n\n\n [[[[ 8]]]],\n\n\n\n [[[[ 9]]]],\n\n\n\n [[[[10]]]],\n\n\n\n [[[[11]]]],\n\n\n\n [[[[12]]]],\n\n\n\n [[[[13]]]],\n\n\n\n [[[[14]]]],\n\n\n\n [[[[15]]]]],\n\n\n\n\n [[[[[ 0]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 3]]]],\n\n\n\n [[[[ 4]]]],\n\n\n\n [[[[ 5]]]],\n\n\n\n [[[[ 6]]]],\n\n\n\n [[[[ 7]]]],\n\n\n\n [[[[ 8]]]],\n\n\n\n [[[[ 9]]]],\n\n\n\n [[[[10]]]],\n\n\n\n [[[[11]]]],\n\n\n\n [[[[12]]]],\n\n\n\n [[[[13]]]],\n\n\n\n [[[[14]]]],\n\n\n\n [[[[15]]]]],\n\n\n\n\n [[[[[ 0]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 3]]]],\n\n\n\n [[[[ 4]]]],\n\n\n\n [[[[ 5]]]],\n\n\n\n [[[[ 6]]]],\n\n\n\n [[[[ 7]]]],...
x1_list = {list: 6} [tensor([[[ 0]],\n\n [[ 1]],\n\n [[ 2]],\n\n [[ 3]],\n\n [[ 4]],\n\n [[ 5]],\n\n [[ 6]],\n\n [[ 7]],\n\n [[ 8]],\n\n [[ 9]],\n\n [[10]],\n\n [[11]],\n\n [[12]],\n\n [[13]],\n\n [[14]],\n\n [[15]]]), tensor([[[[[ 0]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 3]]]],\n\n\n\n [[[[ 4]]]],\n\n\n\n [[[[ 5]]]],\n\n\n\n [[[[ 6]]]],\n\n\n\n [[[[ 7]]]],\n\n\n\n [[[[ 8]]]],\n\n\n\n [[[[ 9]]]],\n\n\n\n [[[[10]]]],\n\n\n\n [[[[11]]]],\n\n\n\n [[[[12]]]],\n\n\n\n [[[[13]]]],\n\n\n\n [[[[14]]]],\n\n\n\n [[[[15]]]]]), tensor([[[[[[[[ 0]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 1]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 2]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 3]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 4]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 5]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 6]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 7]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 8]]]]]]],\n\n\n\n\n\n\n [[[[[[[ 9]]]]]]],\n\n\n\n\n\n\n [[[[[[[10]]]]]]],\n\n\n\n\n\n\n ...
x2 = {Tensor: (16, 1, 1, 1, 1, 1, 1, 1, 1)} tensor([[[[[[[[[ 0]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 1]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 2]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 3]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 4]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 5]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 6]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 7]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 8]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 9]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[10]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[11]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[12]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[13]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[14]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[15]]]]]]]]])
x2_help = {Tensor: (2, 16, 16, 1, 1, 1, 1)} tensor([[[[[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]],\n\n\n\n [[[[ 0]]]]],\n\n\n\n\n [[[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]],\n\n\n\n [[[[ 1]]]]],\n\n\n\n\n [[[[[ 2]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 2]]]],\n\n\n\n [[[[ 2]]]],...
x2_list = {list: 6} [tensor([[[[ 0]]],\n\n\n [[[ 1]]],\n\n\n [[[ 2]]],\n\n\n [[[ 3]]],\n\n\n [[[ 4]]],\n\n\n [[[ 5]]],\n\n\n [[[ 6]]],\n\n\n [[[ 7]]],\n\n\n [[[ 8]]],\n\n\n [[[ 9]]],\n\n\n [[[10]]],\n\n\n [[[11]]],\n\n\n [[[12]]],\n\n\n [[[13]]],\n\n\n [[[14]]],\n\n\n [[[15]]]]), tensor([[[[[[ 0]]]]],\n\n\n\n\n [[[[[ 1]]]]],\n\n\n\n\n [[[[[ 2]]]]],\n\n\n\n\n [[[[[ 3]]]]],\n\n\n\n\n [[[[[ 4]]]]],\n\n\n\n\n [[[[[ 5]]]]],\n\n\n\n\n [[[[[ 6]]]]],\n\n\n\n\n [[[[[ 7]]]]],\n\n\n\n\n [[[[[ 8]]]]],\n\n\n\n\n [[[[[ 9]]]]],\n\n\n\n\n [[[[[10]]]]],\n\n\n\n\n [[[[[11]]]]],\n\n\n\n\n [[[[[12]]]]],\n\n\n\n\n [[[[[13]]]]],\n\n\n\n\n [[[[[14]]]]],\n\n\n\n\n [[[[[15]]]]]]), tensor([[[[[[[[[ 0]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 1]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 2]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 3]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 4]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 5]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 6]]]]]]]],\n\n\n\n\n\n\n\n [[[[[[[[ 7]]...
y1 = {Tensor: (30,)} tensor([ 8, 1, 1, 11, 15, 3, 10, 13, 14, 14, 11, 2, 1, 1, 7, 10, 3, 15,\n 7, 8, 1, 13, 15, 7, 12, 12, 6, 2, 11, 6])
y1_help = {Tensor: (2, 1, 1, 1, 1, 1, 30)} tensor([[[[[[[ 8, 1, 1, 11, 15, 3, 10, 13, 14, 14, 11, 2, 1, 1, 7, 10,\n 3, 15, 7, 8, 1, 13, 15, 7, 12, 12, 6, 2, 11, 6]]]]]],\n\n\n\n\n\n [[[[[[ 8, 1, 1, 11, 15, 3, 10, 13, 14, 14, 11, 2, 1, 1, 7, 10,\n 3, 15, 7, 8, 1, 13, 15, 7, 12, 12, 6, 2, 11, 6]]]]]]])
y2 = {Tensor: (30,)} tensor([ 8, 1, 1, 11, 15, 3, 10, 13, 14, 14, 11, 2, 1, 1, 7, 10, 3, 15,\n 7, 8, 1, 13, 15, 7, 12, 12, 6, 2, 11, 6])
y2_help = {Tensor: (2, 1, 1, 1, 1, 1, 30)} tensor([[[[[[[ 8, 1, 1, 11, 15, 3, 10, 13, 14, 14, 11, 2, 1, 1, 7, 10,\n 3, 15, 7, 8, 1, 13, 15, 7, 12, 12, 6, 2, 11, 6]]]]]],\n\n\n\n\n\n [[[[[[ 8, 1, 1, 11, 15, 3, 10, 13, 14, 14, 11, 2, 1, 1, 7, 10,\n 3, 15, 7, 8, 1, 13, 15, 7, 12, 12, 6, 2, 11, 6]]]]]]])