tf.keras.backend.ones_like

TensorFlow 2 version 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
)

x Keras variable or tensor.
dtype String, dtype of returned Keras variable. None uses the dtype of x.
name String, name for the variable to create.

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

Example:

from keras import backend as K
kvar = K.variable(np.random.random((2,3)))
kvar_ones = K.ones_like(kvar)
K.eval(kvar_ones)
array([[ 1.,  1.,  1.],
       [ 1.,  1.,  1.]], dtype=float32)