def test_FFT2D(self):
# only defined for gpu
if DEVICE == GPU:
t = tf.fft2d(self.random(3, 4, complex=True))
self.check(t)
微信
公众号
扫码关注公众号