tf.keras.constraints.UnitNorm

View source on GitHub

Constrains the weights incident to each hidden unit to have unit norm.

Inherits From: Constraint

View aliases

Main aliases

tf.keras.constraints.unit_norm

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.constraints.UnitNorm, tf.compat.v1.keras.constraints.unit_norm, `tf.compat.v2.keras.constraints.UnitNorm`, `tf.compat.v2.keras.constraints.unit_norm`

tf.keras.constraints.UnitNorm(
    axis=0
)

Arguments

axis

integer, axis along which to calculate weight norms. For instance, in a Dense layer the weight matrix has shape (input_dim, output_dim), set axis to 0 to constrain each weight vector of length (input_dim,). In a Conv2D layer with data_format="channels_last", the weight tensor has shape (rows, cols, input_depth, output_depth), set axis to [0, 1, 2] to constrain the weights of each filter tensor of size (rows, cols, input_depth).

Methods

get_config

View source

get_config()

__call__

View source

__call__(
    w
)

Call self as a function.