TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ndims for shapes

Browse Source
This commit is contained in:
Mike Innes 2018-03-16 14:42:08 +00:00
parent 80e378d6c9
commit 1c5f8e3534
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/jit/shapes.jl
View File

@ -12,6 +12,7 @@ Shape{T}(dims::NTuple{N,Integer}) where {T,N} = Shape{T,N}(dims)
Base.size(s::Shape) = s.dims Base.size(s::Shape) = s.dims
Base.size(s::Shape, n) = s.dims[n] Base.size(s::Shape, n) = s.dims[n]
Base.ndims(s::Shape{T,N}) where {T,N} = N
Base.length(s::Shape) = prod(s.dims) Base.length(s::Shape) = prod(s.dims)
Base.eltype(s::Shape{T}) where T = T Base.eltype(s::Shape{T}) where T = T