tensorflow Addition of rfftn function in TF backend

t9aqgxwy  于 3个月前  发布在  其他
关注(0)|答案(2)|浏览(37)

问题类型

特性请求

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

问题来源

源代码

TensorFlow版本

最新版本

自定义代码

OS平台和发行版

  • 无响应*

移动设备

  • 无响应*

Python版本

  • 无响应*

Bazel版本

  • 无响应*

GCC/编译器版本

  • 无响应*

CUDA/cuDNN版本

  • 无响应*

GPU型号和内存大小

  • 无响应*

当前行为?

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

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

-

相关日志输出

  • 无响应*
lokaqttq

lokaqttq1#

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

vm0i2vca

vm0i2vca2#

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

相关问题