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.