Merge pull request #35 from oxinabox/patch-1
Replace deprecated `unpack` with `unstack`
This commit is contained in:
commit
e37973c3d5
@ -20,7 +20,7 @@ graph(::typeof(softmax), x) = nn.softmax(x)
|
||||
graph(::typeof(relu), x) = nn.relu(x)
|
||||
graph(::typeof(σ), x) = nn.sigmoid(x)
|
||||
graph(::typeof(hcat), xs...) = concat(1, xs)
|
||||
graph(::typeof(seq), xs, n) = TensorFlow.unpack(xs, num = n, axis = 1)
|
||||
graph(::typeof(seq), xs, n) = TensorFlow.unstack(xs, num = n, axis = 1)
|
||||
graph(::typeof(sum), x, dim=nothing) = TensorFlow.reduce_sum(x;axis=dim)
|
||||
graph(::typeof(prod), x, dim=nothing) = TensorFlow.reduce_prod(x;axis=dim)
|
||||
graph(::typeof(min), x, dim=nothing) = TensorFlow.reduce_min(x;axis=dim)
|
||||
|
Loading…
Reference in New Issue
Block a user