interpreter.set_tensor(tensor_index=9, value=input_tensor) # run inference interpreter.invoke() # tensor_index is 0 because the output contains only a single value weight_inferred = interpreter.get_tensor(tensor_index=0)