It should be
np.allclose(pred_x, pred_x_padded[:,:3,:])
Hi @khteh
Thiank you for pointing this out, this is actually incorrect, the last three elements to remove should have been with [-3:] and not [:3]
@lucas.coutinho can you please look into this correction