Haha ! You were right ! It does work !
Here’s the test output:
Adding nb 0
seed: 20595

tensor([[ 1.0241, -0.2424],
    [ 1.8578, -1.2400],
    [ 1.1870, -1.7786],
    [ 2.1059,  0.0410],
    [-0.6069, -0.7865]])

tensor([-0.3170, -0.1979,  0.3467, -0.1865,  1.0867])

tensor([[-0.4042,  1.0822,  0.0599, -0.2872,  0.0582]])

tensor([-0.2743])

Adding nb 1

seed: 155213

tensor([[-0.8802, -0.0765],
    [-2.2938, -0.9288],
    [-1.2693, -0.8255],
    [ 0.0502, -0.4021],
    [-0.6032,  1.7732]])

tensor([ 0.0450, -0.4972,  1.1490,  1.3801,  2.4020])

tensor([[-0.7401,  1.1411,  0.3848, -0.8668, -0.6465]])

tensor([-0.7383])

seed: 20595

Character 0

tensor([[ 1.0241, -0.2424],
    [ 1.8578, -1.2400],
    [ 1.1870, -1.7786],
    [ 2.1059,  0.0410],
    [-0.6069, -0.7865]])

tensor([-0.3170, -0.1979,  0.3467, -0.1865,  1.0867])

tensor([[-0.4042,  1.0822,  0.0599, -0.2872,  0.0582]])

tensor([-0.2743])

seed: 155213

Character 1

tensor([[-0.8802, -0.0765],
    [-2.2938, -0.9288],
    [-1.2693, -0.8255],
    [ 0.0502, -0.4021],
    [-0.6032,  1.7732]])

tensor([ 0.0450, -0.4972,  1.1490,  1.3801,  2.4020])

tensor([[-0.7401,  1.1411,  0.3848, -0.8668, -0.6465]])

tensor([-0.7383])