I tried to locate it in documentation, but could not. Can anyone please explain how this “|> gpu” or “|> cpu” works? Is this |> an operator?
Here is example of the code:
m = Chain(
Conv((5,5), 3=>16, relu),
MaxPool((2,2)),
Conv((5,5), 16=>8, relu),
MaxPool((2,2)),
x -> reshape(x, :, size(x, 4)),
Dense(200, 120),
Dense(120, 84),
Dense(84, 10),
softmax) |> gpu
this is from: Flux – Deep Learning with Flux - A 60 Minute Blitz