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

.* in tf

Browse Source
This commit is contained in:
Mike J Innes 2016-10-30 10:54:55 +00:00
parent a1b1d87767
commit ec1950b466
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/backend/tensorflow/graph.jl
View File

@ -9,6 +9,7 @@ cvalue(v::Vertex) = cvalue(value(v))
graph(x::Tensor) = x
graph(::typeof(*), args...) = *(args...)
graph(::typeof(.*), args...) = .*(args...)
graph(::typeof(+), args...) = +(args...)
graph(::typeof(softmax), x) = nn.softmax(x)
graph(::typeof(relu), x) = nn.relu(x)