File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -48,8 +48,10 @@ class RadialFourier3D(Transform):
4848
4949 Returns:
5050 Radial Fourier transform of input data. Shape depends on parameters:
51- - If radial_bins is None: complex tensor of same spatial shape as input
52- - If radial_bins is set: real tensor of shape (radial_bins,) for magnitude/phase
51+ - If radial_bins is None: same spatial shape as input; magnitude and phase
52+ (if both requested) are concatenated along the last dimension, doubling it.
53+ - If radial_bins is set: shape (..., radial_bins) or (..., 2*radial_bins) if both
54+ magnitude and phase are requested, preserving leading (batch/channel) dimensions.
5355
5456 Raises:
5557 ValueError: If max_frequency not in (0.0, 1.0], radial_bins < 1,
@@ -293,6 +295,7 @@ class RadialFourierFeatures3D(Transform):
293295 Args:
294296 n_bins_list: list of radial bin counts to compute.
295297 return_types: list of return types: 'magnitude', 'phase', or 'complex'.
298+ 'complex' returns both magnitude and phase concatenated as real values.
296299 normalize: if True, normalize the output.
297300
298301 Returns:
You can’t perform that action at this time.
0 commit comments