Return the output of the wrapped PyTorch model for the given input,
along with a callback to handle the backward pass.
convert_inputs = model.get_attr("convert_inputs") or _convert_inputs
convert_outputs = model.get_attr("convert_outputs") or _convert_outputs
Xtorch, get_dX = convert_inputs(model, X, is_train)
After Change
Return the output of the wrapped PyTorch model for the given input,
along with a callback to handle the backward pass.
convert_inputs = model.get_attr("convert_inputs")
convert_outputs = model.get_attr("convert_outputs")
Xtorch, get_dX = convert_inputs(model, X, is_train)
Ytorch, torch_backprop = model.shims[0](Xtorch, is_train)
