peal.architectures.onnx_predictor.truncate_to_features

peal.architectures.onnx_predictor.truncate_to_features(src, dst, feature_tensor)[source]

Write a copy of src whose only output is feature_tensor (the input of the final linear layer found by find_final_linear()). Returns dst.