tf.ones_like

Создаёт тензор, заполненный единицами, с той же формой, что и входной.

tf.ones_like(
    input, dtype=None, name=None, layout=None
)

Используется в блокнотах

См. также tf.ones.

Для данного единственного тензора (tensor), эта операция возвращает тензор того же типа и формы, что и tensor, со всеми элементами, установленными в 1. Необязательно, вы можете использовать dtype, чтобы указать новый тип для возвращаемого тензора.

Например:

tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
tf.ones_like(tensor)
<tf.Tensor: shape=(2, 3), dtype=int32, numpy=
  array([[1, 1, 1],
         [1, 1, 1]], dtype=int32)>

Обратите внимание, что макет входного тензора не сохраняется, если операция используется внутри tf.function. Чтобы получить тензор с тем же макетом, что и входной, цепочку возвращённого значения прикрепите к dtensor.relayout_like.