问题类型
特性请求
你是否在TensorFlow Nightly版本中复现了这个bug?
否
问题来源
源代码
TensorFlow版本
最新版本
自定义代码
是
OS平台和发行版
- 无响应*
移动设备
- 无响应*
Python版本
- 无响应*
Bazel版本
- 无响应*
GCC/编译器版本
- 无响应*
CUDA/cuDNN版本
- 无响应*
GPU型号和内存大小
- 无响应*
当前行为?
我想在TensorFlow中添加一个rfftn
函数,因为它没有这个函数。
可独立复现问题的代码片段
-
相关日志输出
- 无响应*
2条答案
按热度按时间lokaqttq1#
你好,@rajveer43!
感谢你提出这个功能请求!
将rfftn函数添加到TF后端对于许多原因都是有益的。一旦实现,它将被优化以提高性能。
@sachinprasadhs 你能看一下这个吗?
谢谢!
vm0i2vca2#
已经在其他框架中了!例如:
pytorch
,JAX
,numpy
,paddlepaddle
。所以在TensorFlow中添加这个功能是有意义的!我也想为此创建一个PR。你好,@rajveer43!感谢你提出这个功能请求!
将rfftn函数添加到TensorFlow后端将对许多原因有益。一旦实现,它将针对性能进行优化。
@sachinprasadhs 请看一下这个好吗?谢谢!