TensorFlow函数:tf.zeros_like

时间:2024-07-07 17:37:50

tf.zeros_like函数

tf.zeros_like(
tensor,
dtype=None,
name=None,
optimize=True
)

定义在:tensorflow/python/ops/array_ops.py.

请参阅指南:生成常量,序列和随机值>常量值张量

创建一个所有元素都设置为零的张量.

给定一个张量(tensor),该操作返回与所有元素设置为零的tensor具有相同类型和形状的张量.或者,您可以使用dtype指定返回张量的新类型.

例如:

tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
tf.zeros_like(tensor) # [[0, 0, 0], [0, 0, 0]]

函数参数:

  • tensor:一个张量.
  • dtype:返回的张量的类型,必须是以下类型之一:float16,float32,float64,int8,uint8,int16,uint16,int32,int64,complex64,complex128,bool或string.
  • name:操作的名称(可选).
  • optimize:如果为true,则尝试静态确定“张量”的形状并将其编码为常量.

返回:

tf.zeros_like函数返回将所有元素设置为零的张量.