此页面由 Cloud Translation API 翻译。
Switch to English

tff.learning.framework.build_encoded_broadcast_from_model

GitHub上查看源代码

构建StatefulBroadcastFn通过返回模型的权重model_fn

此方法创建一个SimpleEncoder通过创建模型的每一个重量model_fn ,由返回encoder_fn

model_fn 一个Python可调用不带参数的函数,返回一个tff.learning.Model
encoder_fn 一个Python可调用带一个参数,其被预期是一个tf.Tensor形状,并且DTYPE进行编码。该函数必须返回一个tensor_encoding.core.SimpleEncoder ,一个希望将tf.Tensor兼容类型作为输入到它的encode方法。

一个StatefulBroadcastFn编码和广播通过创建模型的权重model_fn

TypeError 如果model_fnencoder_fn是不可调用的对象。