Attend the Women in ML Symposium on December 7 Register now

tfp.experimental.joint_distribution_layers.AffineLayer

Stay organized with collections Save and categorize content based on your preferences.

Affine layer.

This represents a linear map: y = weights @ x + bias.

weights A floating point Tensor with shape [out_units, in_units].
bias A floating point Tensor with shape [out_units].

Methods

__call__

View source

Applies the layer to an input.

Args
x A floating point Tensor with shape [in_units].

Returns
y A floating point Tensor with shape [out_units].