tensorflow Addition of rfftn function in TF backend

t9aqgxwy 于 22天前发布在其他

关注(0)|答案(2)|浏览(17)

问题类型

特性请求

你是否在TensorFlow Nightly版本中复现了这个bug?

否

问题来源

源代码

TensorFlow版本

自定义代码

是

OS平台和发行版

无响应*

移动设备

无响应*

Python版本

无响应*

Bazel版本

无响应*

GCC/编译器版本

无响应*

CUDA/cuDNN版本

无响应*

GPU型号和内存大小

无响应*

当前行为？

我想在TensorFlow中添加一个rfftn函数，因为它没有这个函数。

可独立复现问题的代码片段

2条答案

按热度按时间

lokaqttq1#

你好，@rajveer43!
感谢你提出这个功能请求！
将rfftn函数添加到TF后端对于许多原因都是有益的。一旦实现，它将被优化以提高性能。
@sachinprasadhs 你能看一下这个吗？
谢谢！

赞(0）回复(0）举报 22天前

vm0i2vca2#

已经在其他框架中了！例如：pytorch , JAX , numpy , paddlepaddle 。所以在TensorFlow中添加这个功能是有意义的！我也想为此创建一个PR。
你好，@rajveer43!感谢你提出这个功能请求！
将rfftn函数添加到TensorFlow后端将对许多原因有益。一旦实现，它将针对性能进行优化。
@sachinprasadhs 请看一下这个好吗？谢谢！