oryx.experimental.matching.rules.Expression | Oryx | TensorFlow Install Learn Introduction New to TensorFlow? TensorFlow The core open source ML library For JavaScript TensorFlow.js for ML using JavaScript For Mobile & Edge TensorFlow Lite for mobile and edge devices For Production TensorFlow Extended for end-to-end ML components API TensorFlow (v2.10.0) Versions… TensorFlow.js TensorFlow Lite TFX Resources Models & datasets Pre-trained models and datasets built by Google and the community Tools Ecosystem of tools to help you use TensorFlow Libraries & extensions Libraries and extensions built on TensorFlow TensorFlow Certificate program Differentiate yourself by demonstrating your ML proficiency Learn ML Educational resources to learn the fundamentals of ML with TensorFlow Responsible AI Resources and tools to integrate Responsible AI practices into your ML workflow Community Join Explore ways to get involved Blog Stay up to date with all things TensorFlow Forum ↗ Discussion platform for the TensorFlow community Groups User groups, interest groups and mailing lists Contribute Guide for contributing to code and documentation Why TensorFlow About Case studies English 中文 – 简体 GitHub Sign in Probability Oryx Overview Guides & Tutorials API Install Learn More API More Resources More Overview Guides & Tutorials API Community More Why TensorFlow More GitHub oryx Overview oryx.bijectors Overview AbsoluteValue Ascending AutoCompositeTensorBijector AutoregressiveNetwork Bijector Blockwise Chain CholeskyOuterProduct CholeskyToInvCholesky Composition CorrelationCholesky Cumsum Exp Expm1 FillScaleTriL FillTriangular FrechetCDF GeneralizedExtremeValueCDF GeneralizedPareto GompertzCDF GumbelCDF Householder Identity Inline Invert IteratedSigmoidCentered JointMap KumaraswamyCDF LambertWTail Log Log1p MaskedAutoregressiveFlow MatrixInverseTriL MatvecLU MoyalCDF NormalCDF Pad Permute Power PowerTransform RationalQuadraticSpline RayleighCDF RealNVP Reciprocal Reshape Restructure Scale ScaleMatvecDiag ScaleMatvecLU ScaleMatvecLinearOperator ScaleMatvecLinearOperatorBlock ScaleMatvecTriL Shift ShiftedGompertzCDF Sigmoid Sinh SinhArcsinh SoftClip Softfloor SoftmaxCentered Softplus Softsign Split Square Tanh TransformDiagonal Transpose WeibullCDF masked_autoregressive_default_template masked_dense pack_sequence_as real_nvp_default_template tree_flatten oryx.core Overview FlatPrimitive HigherOrderPrimitive NonInvertibleError Pytree call_bind custom_inverse harvest ildj inverse inverse_and_ildj log_prob nest plant reap sow tie_all tie_in interpreters Overview harvest Overview HarvestTrace HarvestTracer call_and_reap inverse Overview core Overview InverseAndILDJ register_binary register_elementwise custom_inverse Overview rules Overview slice Overview log_prob Overview LogProbRules propagate Overview Cell Environment Equation propagate ppl Overview LogProbFunction block conditional graph_replace intervene joint_log_prob joint_sample log_prob make_effect_handler plate random_variable trace trace_log_prob effect_handler Overview plate_util Overview make_plate transformations Overview primitive Overview pytree Overview serialize Overview deserialize from_tuple serialize to_tuple state Overview ArraySpec Module assign call call_and_update init make_array_spec spec update variable api Overview function Overview FunctionModule module Overview trace_util Overview get_dynamic_context get_shaped_aval new_dynamic_context pv_like stage trees oryx.distributions Overview Autoregressive BatchBroadcast BatchReshape Bates Bernoulli Beta BetaBinomial BetaQuotient Binomial Blockwise Categorical Cauchy Chi Chi2 CholeskyLKJ DeterminantalPointProcess Deterministic Dirichlet DirichletMultinomial Distribution DoublesidedMaxwell Empirical ExpGamma ExpInverseGamma ExpRelaxedOneHotCategorical Exponential ExponentiallyModifiedGaussian FiniteDiscrete Gamma GammaGamma GaussianProcess GaussianProcessRegressionModel GeneralizedExtremeValue GeneralizedNormal GeneralizedPareto Geometric Gumbel HalfCauchy HalfNormal HalfStudentT HiddenMarkovModel Horseshoe Independent InverseGamma InverseGaussian JohnsonSU JointDistribution JointDistribution.Root JointDistributionCoroutine JointDistributionCoroutineAutoBatched JointDistributionNamed JointDistributionNamedAutoBatched JointDistributionSequential JointDistributionSequentialAutoBatched Kumaraswamy LKJ LambertWDistribution LambertWNormal Laplace LinearGaussianStateSpaceModel LogLogistic LogNormal Logistic LogitNormal MarkovChain Masked MatrixNormalLinearOperator MatrixTLinearOperator Mixture MixtureSameFamily Moyal Multinomial MultivariateNormalDiag MultivariateNormalDiagPlusLowRank MultivariateNormalFullCovariance MultivariateNormalLinearOperator MultivariateNormalTriL MultivariateStudentTLinearOperator NegativeBinomial Normal NormalInverseGaussian OneHotCategorical OrderedLogistic PERT Pareto PlackettLuce Poisson PoissonLogNormalQuadratureCompound PowerSpherical ProbitBernoulli QuantizedDistribution RegisterKL RelaxedBernoulli RelaxedOneHotCategorical ReparameterizationType Sample SigmoidBeta SinhArcsinh Skellam SphericalUniform StoppingRatioLogistic StudentT StudentTProcess StudentTProcessRegressionModel TransformedDistribution Triangular TruncatedCauchy TruncatedNormal TwoPieceNormal Uniform VariationalGaussianProcess VectorDeterministic VonMises VonMisesFisher Weibull WishartLinearOperator WishartTriL Zipf independent_joint_distribution_from_structure kl_divergence mvn_conjugate_linear_update normal_conjugates_known_scale_posterior normal_conjugates_known_scale_predictive quadrature_scheme_lognormal_gauss_hermite quadrature_scheme_lognormal_quantiles oryx.experimental Overview matching Overview jax_rewrite Overview Bindings BoundExpression CallPrimitive Continuation JaxExpression JaxVar Literal Params Part Primitive Success evaluate evaluate_jax_expression evaluate_tuple evaluate_value jaxpr_to_expressions make_bound_expression primitive_to_expression rewrite matcher Overview Choice MatchError Not Pattern Plus Segment Star Var is_match match match_all matcher rules Overview Expression bottom_up in_order iterated make_rule rewrite_subexpressions rule_list term_rewriter top_down tree_children tree_map mcmc Overview constrain hmc mala metropolis metropolis_hastings random_walk sample_chain kernels Overview utils Overview nn Overview AvgPooling BatchNorm Conv Deconv Dense Dropout Flatten Layer LayerParams LogSoftmax MaxPooling Relu Reshape Serial Softmax Softplus SumPooling Tanh Template base Overview create_parameter combinator Overview convolution Overview conv_info core Overview Activation normalization Overview pooling Overview Pooling reshape Overview optimizers Overview adam gradient_descent noisy_sgd optimize rmsprop sgd optix Overview add_noise apply_every chain clip clip_by_global_norm global_norm scale scale_by_adam scale_by_rms scale_by_schedule scale_by_stddev trace oryx.internal Overview test_util Overview TestCase TestCase.failureException oryx.util Overview get_summaries summary Introduction TensorFlow For JavaScript For Mobile & Edge For Production TensorFlow (v2.10.0) Versions… TensorFlow.js TensorFlow Lite TFX Models & datasets Tools Libraries & extensions TensorFlow Certificate program Learn ML Responsible AI Join Blog Forum ↗ Groups Contribute About Case studies TensorFlow Resources Probability Oryx API oryx.experimental.matching.rules.Expression Stay organized with collections Save and categorize content based on your preferences. View source on GitHub A class that is auto-registered with tree_map and tree_children. Inherits From: Pattern View aliases Main aliases oryx.experimental.matching.jax_rewrite.rules.Expression Subclassing Expression and implementing Expression.tree_map and Expression.tree_children will result in a class whose instances that can be recursively rewritten with rule combinators. Methods match View source