张量流中的批处理操作（例如，fft）

Question

我是一个tensorflow新手，所以这可能是一个愚蠢的问题。无论如何，假设我有一个二维张量，并且我想将其转换为行的傅立叶变换的张量。我假设类似numpy（tf.signal.fft(foo, axis=1)）的东西会工作并且有效地工作（也就是说，整个东西都将被交付给GPU，并且所有的查找将并行完成），但这是真的？