[PWGJE] Configurable GNN input feature transform function

[PWGJE] Configurable GNN input feature transform function #37447