diff --git a/test/TorchSharpTest/TestTorchTensor.cs b/test/TorchSharpTest/TestTorchTensor.cs index 926d2e777..fc6f34ccc 100644 --- a/test/TorchSharpTest/TestTorchTensor.cs +++ b/test/TorchSharpTest/TestTorchTensor.cs @@ -7167,15 +7167,15 @@ public void Float64HFFT2() } [Fact] - [TestOf(nameof(fft.hfft2))] + [TestOf(nameof(fft.hfftn))] public void Float32HFFTN() { var input = torch.rand(new long[] { 5, 5, 5, 5 }, complex64); - var output = fft.hfft2(input); + var output = fft.hfftn(input); Assert.Equal(new long[] { 5, 5, 5, 8 }, output.shape); Assert.Equal(ScalarType.Float32, output.dtype); - var inverted = fft.ihfft2(output); + var inverted = fft.ihfftn(output); Assert.Equal(new long[] { 5, 5, 5, 5 }, inverted.shape); Assert.Equal(ScalarType.ComplexFloat32, inverted.dtype); }