tf.keras.constraints.UnitNorm | TensorFlow v2.0.0 (original) (raw)

tf.keras.constraints.UnitNorm

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

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

SeeMigration 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.