Iblis Lin
|
3d8b7250ae
|
add scalar mean
|
2017-10-31 10:42:32 +00:00 |
|
Iblis Lin
|
c43bda019b
|
TrackedArray: implement mean
```julia
julia> p
Tracked 2×3 Array{Float64,2}:
1.0 3.0 5.0
2.0 4.0 6.0
```
Before
```julia
julia> @benchmark Flux.Tracker.back!(sum($p, 2) ./ size($p, 2), ones(2, 1))
BenchmarkTools.Trial:
memory estimate: 3.44 KiB
allocs estimate: 75
--------------
minimum time: 20.438 μs (0.00% GC)
median time: 21.239 μs (0.00% GC)
mean time: 22.354 μs (1.68% GC)
maximum time: 3.811 ms (98.51% GC)
--------------
samples: 10000
evals/sample: 1
```
After
```julia
julia> @benchmark Flux.Tracker.back!(mean($p, 2), ones(2, 1))
BenchmarkTools.Trial:
memory estimate: 1008 bytes
allocs estimate: 21
--------------
minimum time: 5.973 μs (0.00% GC)
median time: 6.310 μs (0.00% GC)
mean time: 6.630 μs (1.96% GC)
maximum time: 680.709 μs (97.28% GC)
--------------
samples: 10000
evals/sample: 6
```
|
2017-10-30 16:21:02 +08:00 |
|
Mike J Innes
|
cca4d25a10
|
efficient traversal
|
2017-09-06 23:09:32 -04:00 |
|
Mike J Innes
|
7cfc42d166
|
grad refactor
|
2017-09-06 21:21:35 -04:00 |
|
Mike J Innes
|
4c12d18033
|
yet another vcat method
|
2017-09-05 19:25:42 -04:00 |
|
Mike J Innes
|
830d7fa611
|
vcat fix
|
2017-09-05 02:28:11 -04:00 |
|
Mike J Innes
|
363caeddc6
|
repmat forward
|
2017-09-05 02:12:53 -04:00 |
|
Mike J Innes
|
a322c07fc8
|
vcat back
|
2017-09-05 02:11:28 -04:00 |
|
Mike J Innes
|
8f4ccdd5ba
|
scalar getindex backprop
|
2017-09-03 17:10:23 -04:00 |
|
Mike J Innes
|
9642ae8cd6
|
basic recurrence
|
2017-09-03 02:12:44 -04:00 |
|
Mike J Innes
|
107d9daa8f
|
add some non-differentiable functions
|
2017-09-01 23:33:05 -04:00 |
|
Mike J Innes
|
892a779ed1
|
tracked transpose
|
2017-09-01 11:42:18 -04:00 |
|
Mike J Innes
|
7bba38274b
|
Merge branch 'master' of github.com:MikeInnes/Flux.jl
|
2017-08-28 01:41:11 +01:00 |
|
Mike J Innes
|
0b89e1374c
|
gpu-friendly
|
2017-08-28 01:40:59 +01:00 |
|
Mike J Innes
|
73166c52a0
|
cleaner broadcasting fix
|
2017-08-27 09:49:42 +01:00 |
|
Mike J Innes
|
e4e9794f5e
|
loss function gradients
|
2017-08-23 17:50:43 +01:00 |
|
Mike J Innes
|
60c3090981
|
broadcasting fix
|
2017-08-23 17:21:02 +01:00 |
|
Mike J Innes
|
23c5a1b163
|
softmax gradient
|
2017-08-23 02:03:17 +01:00 |
|
Mike J Innes
|
56ed6f5680
|
de-broadcasting
|
2017-08-23 00:25:19 +01:00 |
|
Mike J Innes
|
f43eef8d7d
|
consistent case
|
2017-08-22 15:13:24 +01:00 |
|