tristancazenave

Hi Yuandong, bk has size 2 and is equal to: tensor([1, 1]) v.squeeze has size 128 and is equal to: tensor([172, 75, 90, 177, 6, 147, 189, 71, 181, 165,...

Different runs give different result but the size of v.squeeze is 64 times the size of bk, and bk is always filled with ones.