[TFF联邦学习]关于@tff.tf_computation修饰符的使用

本文详细介绍了在TFF框架中如何正确使用tf_computation装饰器来包裹TensorFlow计算逻辑,确保这些计算可以被序列化并在联合计算中使用。文章特别强调了避免直接在联合计算中使用TensorFlow操作的重要性,并提供了一个有效的示例。

1.作用

在TFF框架中,如果需要 使用tff.federated_computation 创建联邦计算的代码块,就必须使用tff.tf_computation修饰,不可以直接包含tensorflow的运算(例如创建tf.tensor以及相关的运算)。

需要注意的一个重要限制是,用 tff.federated_computation 修饰的 Python 函数体必须仅包含联合运算符,即它们不能直接包含 TensorFlow 运算。例如,您不能直接使用 tf.nest 接口添加一对联合值。 TensorFlow 代码必须限制在 tff.tf_computation 修饰的代码块中。只有以这种方式包装时,包装的 TensorFlow 代码才能在 tff.federated_computation 的主体中调用。

2.特点

用tff.tf_computation 修饰的 Python 方法后,方法会被序列化为 TensorFlow 图。所以在这个方法中,不可以包含已经被序列化成TensorFlow 的变量。否则会报错:

try:

  # Eager mode
  constant_10 = tf.constant(10.)

  @tff
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值