tfp.experimental.nn.Affine

Basic affine layer.

input_size ...
output_size ...
init_kernel_fn ... Default value: None (i.e., tfp.experimental.nn.initializers.glorot_uniform()).
init_bias_fn ... Default value: None (i.e., tf.initializers.zeros()).
make_kernel_bias_fn ... Default value: tfp.experimental.nn.util.make_kernel_bias.
dtype ... Default value: tf.float32.
batch_shape ... Default value: ().
activation_fn ... Default value: None.
name ... Default value: None (i.e., 'Affine').

activation_fn

also_track

bias

dtype

extra_loss

extra_result

kernel

name Returns the name of this module as passed or determined in the ctor.

name_scope Returns a tf.name_scope instance for this class.
submodules Sequence of all sub-modules.

Submodules are modules which are properties of this module, or found as properties of modules which are properties of this module (and so on).

a = tf.Module()
b = tf.Module()
c = tf.Module()
a.b = b
b.c = c
list(a.submodules) == [b, c]
True
list(b.submodules) == [c]
True
list(c.submodules) == []
True

trainable_variables Sequence of trainable variables owned by this module and its submodules.

variables Sequence of variables owned by this module and its submodules.

Methods

eval

View source

load

View source

save

View source

summary

View source

with_name_scope

Decorator to automatically enter the module name scope.

class MyModule(tf.Module):
  @tf.Module.with_name_scope
  def __call__(self, x):
    if not hasattr(self, 'w'):
      self.w = tf.Variable(tf.random.normal([x.shape[1], 3]))
    return tf.matmul(x, self.w)

Using the above module would produce tf.Variables and tf.Tensors whose names included the module name:

mod = MyModule()
mod(tf.ones([1, 2]))
<tf.Tensor: shape=(1, 3), dtype=float32, numpy=..., dtype=float32)>
mod.w
<tf.Variable 'my_module/Variable:0' shape=(2, 3) dtype=float32,
numpy=..., dtype=float32)>

Args
method The method to wrap.

Returns
The original method wrapped such that it enters the module's name scope.

__call__

View source

Call self as a function.