tf.keras.backend.ones_like

View source on GitHub

Instantiates an all-ones variable of the same shape as another tensor.

tf.keras.backend.ones_like(
    x, dtype=None, name=None
)

Arguments:

Returns:

A Keras variable with the shape of x filled with ones.

Example:

>>> kvar = tf.keras.backend.variable(np.random.random((2,3)))
>>> kvar_ones = tf.keras.backend.ones_like(kvar)
>>> tf.keras.backend.eval(kvar_ones)
array([[1.,  1.,  1.],
       [1.,  1.,  1.]], dtype=float32)