22 lines
336 B
Julia
22 lines
336 B
Julia
|
using TensorFlow
|
||
|
|
Flux.loadtf()
|
||
|
|
|
||
|
|
@testset "TensorFlow" begin
|
||
|
|
|
||
|
|
xs = rand(20)
|
||
|
|
d = Affine(20, 10)
|
||
|
|
|
||
|
|
dt = tf(d)
|
||
|
|
@test d(xs) ≈ dt(xs)
|
||
|
|
|
||
|
|
@testset "Tensor interface" begin
|
||
|
|
sess = TensorFlow.Session()
|
||
|
|
X = placeholder(Float32)
|
||
|
|
Y = Tensor(d, X)
|
||
|
|
run(sess, initialize_all_variables())
|
||
|
|
|
||
|
|
@test run(sess, Y, Dict(X=>xs')) ≈ d(xs)'
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|