Tensorflow函数:tf.zeros

tf.zeros函数

tf.zeros(
    shape,
    dtype=tf.float32,
    name=None
)

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

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

该操作返回一个带有形状shape的类型为dtype张量,并且所有元素都设为零.

例如:

tf.zeros([3, 4], tf.int32)  # [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]

函数参数:

  • shape:整数、整数元组或类型为int32的1维Tensor的列表.
  • dtype:结果Tensor中元素的类型.
  • name:操作的名称(可选).

函数返回值:

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