add IntOrIntTuple

yuehhua · yuehhua · commit ce63fb3f21b1 · 2021-03-26T00:42:03.000+08:00
diff --git a/lib/NNlibCUDA/src/NNlibCUDA.jl b/lib/NNlibCUDA/src/NNlibCUDA.jl
@@ -4,6 +4,8 @@ using NNlib
 using CUDA
 using Random, Statistics
 
+const IntOrIntTuple = Union{Integer, NTuple{N,<:Integer} where N}
+
 include("upsample.jl")
 include("batchedmul.jl")
 include("scatter.jl")
