def _fft(bottom, sequential, compute_size):
if sequential:
return sequential_batch_fft(bottom, compute_size)
else:
return tf.fft(bottom)
微信
公众号
扫码关注公众号