Merge pull request #19 from SebastianM-C/outsize

Add input and output size functions
LuxDL · Feb 14, 2024 · feacac5 · feacac5
2 parents 774c6fc + a14133d
commit feacac5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/lib/LuxCore/Project.toml b/lib/LuxCore/Project.toml
@@ -1,7 +1,7 @@
 name = "LuxCore"
 uuid = "bb33d45b-7691-41d6-9220-0943567d0623"
 authors = ["Avik Pal <[email protected]> and contributors"]
-version = "0.1.7"
+version = "0.1.8"
 
 [deps]
 Functors = "d9f16b24-f501-4c13-a1f2-28368ffc5196"

diff --git a/lib/LuxCore/src/LuxCore.jl b/lib/LuxCore/src/LuxCore.jl
@@ -93,6 +93,20 @@ statelength(nt::Union{NamedTuple, Tuple}) = length(nt) == 0 ? 0 : sum(statelengt
 statelength(a::AbstractArray) = length(a)
 statelength(::Any) = 1
 
+"""
+    inputsize(layer)
+
+Return the input size of the layer.
+"""
+function inputsize end
+
+"""
+    outputsize(layer)
+
+Return the output size of the layer.
+"""
+function outputsize end
+
 """
     setup(rng::AbstractRNG, layer)