开发者社区

文档建议反馈控制台

文章/答案/技术大牛

发布

社区首页 >问答首页 >tf.embedding_lookup_sparse的FLOPs

问tf.embedding_lookup_sparse的FLOPs
EN

Stack Overflow用户

提问于 2017-01-10 14:46:53

回答 1查看 135关注 0票数 0

在tensorflow中执行tf.embedding_lookup_sparse时，是否执行了浮点操作？如果是这样，是否有可能找到该操作所携带的FLOP的数量？

EN

回答 1

Stack Overflow用户

发布于 2017-01-12 03:07:53

根据传递的参数，可以执行任意数量的FLOP。虽然实际的array_ops.gather(...)调用不会执行FLOP，但在许多其他地方可能会发生计算，也可能不会发生计算。

在对输入张量进行分片时，将执行

整数运算。(可以是mod或div。)合并权重的组合器(如果不是将值合并在一起的组合器(可以是None、"sum"、"mean"或"sqrtn".

您可以在此处查看其实现：https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/embedding_ops.py#L203-L366

票数 1

EN

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/41562718

复制

相似问题