Updated changelog

NVIDIA · Sep 9, 2024 · 9b6166a · 9b6166a
1 parent 9f71769
commit 9b6166a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Changelog.md b/Changelog.md
@@ -5,6 +5,7 @@
 ### v0.7.2
 
 * Added resampling modules for convenience
+* Changing behavior of distributed SHT to use `dim=-3` as channel dimension
 
 ### v0.7.1
 

diff --git a/torch_harmonics/distributed/distributed_sht.py b/torch_harmonics/distributed/distributed_sht.py
@@ -55,7 +55,7 @@ class DistributedRealSHT(nn.Module):
 
     def __init__(self, nlat, nlon, lmax=None, mmax=None, grid="lobatto", norm="ortho", csphase=True):
         """
-        Initializes the SHT Layer, precomputing the necessary quadrature weights
+        Distribtued SHT layer. Expects the last 3 dimensions of the input tensor to be channels, latitude, longitude.
 
         Parameters:
         nlat: input grid resolution in the latitudinal direction