I’m not sure how the target b_y is created, but it seems that it has a shape of [unknown, 100], which doesn’t fit the model output of [batch_size, 10]. Could you explain how the target is computed and what its dimensions/shapes represent?