tf.zeros_like

Создаёт тензор со всеми элементами, равными нулю.

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

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

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

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

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

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

tf.zeros_like(tensor, dtype=tf.float32)
<tf.Tensor: shape=(2, 3), dtype=float32, numpy=
array([[0., 0., 0.],
       [0., 0., 0.]], dtype=float32)>

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