layer_poincare_normalize {tfaddons}

Project into the Poincare ball with norm <= 1.0 - epsilon

Description

Project into the Poincare ball with norm <= 1.0 - epsilon

Usage

layer_poincare_normalize(object, axis = 1, epsilon = 1e-05, ...)

Arguments

object

Model or layer object

axis

Axis along which to normalize. A scalar or a vector of integers.

epsilon

A small deviation from the edge of the unit sphere for numerical stability.

...

additional parameters to pass

Details

https://en.wikipedia.org/wiki/Poincare_ball_model Used in Poincare Embeddings for Learning Hierarchical Representations Maximilian Nickel, Douwe Kiela https://arxiv.org/pdf/1705.08039.pdf For a 1-D tensor with axis = 0, computes

Value

A tensor