问题描述
我正在计划一个 TFF 方案，在该方案中，除了权重之外，客户端还向服务器发送数据，例如它们的硬件信息（例如 cpu 频率）。为此，我需要调用第三方 python 库的函数，如 psutils。是否可以序列化（使用 tff.tf_computation）这样的函数？
如果没有，在我通过 gRPC 使用远程执行程序设置的情况下，实现此目标的解决方案是什么？
解决方法
不幸的是，没有修改，这不起作用。 TFF 使用 TensorFlow 图来序列化计算逻辑以在远程机器上运行。 TFF 不解释远程机器上的 Python 代码。
也许可以通过创建 TensorFlow 自定义操作来解决。这意味着编写 C++ 代码来检索 CPU 频率，然后编写 Python API 以在计算构建期间将操作添加到 TensorFlow 图。 TensorFlow 的 Create an op 指南可以提供详细说明。