当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.zeros用法及代码示例


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

用法

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

参数

  • shape 整数的 list、整数的 tuple 或 int32 类型的一维 Tensor 。
  • dtype 结果 Tensor 中元素的 DType。
  • name 可选字符串。操作的名称。

返回

  • 一个Tensor，所有元素都设置为零。

另见tf.zeros_like、tf.ones、tf.fill、tf.eye。

此操作返回一个类型为 dtype 的张量，其形状为 shape，并且所有元素都设置为零。

tf.zeros([3, 4], tf.int32)
<tf.Tensor:shape=(3, 4), dtype=int32, numpy=
array([[0, 0, 0, 0],
       [0, 0, 0, 0],
       [0, 0, 0, 0]], dtype=int32)>

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.zeros。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。