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