tf.keras.constraints.UnitNorm

TensorFlow 1 version

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.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)`.

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2020-10-01 UTC.