peal.architectures.onnx_predictor.truncate_to_features¶
- peal.architectures.onnx_predictor.truncate_to_features(src, dst, feature_tensor)[source]¶
Write a copy of
srcwhose only output isfeature_tensor(the input of the final linear layer found byfind_final_linear()). Returnsdst.