tensorflow如何平均几个IndexedSlicesValue?

时间:2016-08-18 12:11:21

标签: tensorflow

我在tensorflow中定义了一个RNN模型,gradientscompute_gradients之一的类型为IndexedSlices,而其他类型为tensor。在我session.run(compute_gradients ...)之后,返回值类型IndexedSlicesIndexedSlicesValue,那么我有两个问题:

  1. 我如何平均几个IndexedSlicesValue值?
  2. 如何序列化IndexedSlicesValue并通过套接字将其发送到另一台计算机?
  3. 非常感谢!

1 个答案:

答案 0 :(得分:1)

IndexedSlices实际上是使用一对密集张量的稀疏张量的编码。它可能来自tf.gather操作的梯度。这里有一些关于IndexedSlices的API文档可能有所帮助:docs

我不知道有多少代码可以直接使用IndexedSlices;通常它们是用作渐变代码一部分的内部细节。根据数据大小,使用它们的最简单方法可能是将它们转换为密集的Tensor并处理/发送它。