TFF 是否序列化另一个库的函数?

Does TFF serializalize functions of another library?

我正在计划一个 TFF 方案,其中客户端除了权重之外还向服务器发送数据,比如它们的硬件信息(例如 CPU 频率)。为此,我需要调用第三方 python 库的函数,例如 psutils。是否可以序列化(使用 tff.tf_computation)此类函数? 如果没有,在我通过 gRPC 使用远程执行程序设置的情况下,实现此 objective 的解决方案是什么?

很遗憾,没有修改,这将无法工作。 TFF 使用 TensorFlow 图将计算逻辑序列化到远程机器上的 运行。 TFF 不解释远程计算机上的 Python 代码。

创建 TensorFlow 自定义操作可能是一个解决方案。这意味着编写 C++ 代码来检索 CPU 频率,然后 Python API 在计算构造期间将操作添加到 TensorFlow 图。 Create an op 的 TensorFlow 指南可以提供详细说明。