tensorflow-ops-0.3.0.0: Friendly layer around TensorFlow bindings.

Safe Haskell	None
Language	Haskell2010

TensorFlow.Gradient

Synopsis

type GradientCompatible a = (Num a, OneOf '[Float, Complex Float, Complex Double] a)
gradients :: forall a v1 t m. (MonadBuild m, Rendered t, ToTensor t, GradientCompatible a) => Tensor v1 a -> [t a] -> m [Tensor Value a]

Documentation

type GradientCompatible a = (Num a, OneOf '[Float, Complex Float, Complex Double] a) Source #

gradients Source #

Arguments

:: forall a v1 t m. (MonadBuild m, Rendered t, ToTensor t, GradientCompatible a)
=> Tensor v1 a	The output of the graph.
-> [t a]	Tensors for which gradients are computed.
-> m [Tensor Value a]

Gradient of y w.r.t. each element of xs.