阅读(17.1k) 书签 (0)

TensorFlow函数:tf.ones_like

2018-11-07 17:39 更新
tf.ones_like 函数
ones_like(
    tensor,
    dtype=None,
    name=None,
    optimize=True
)

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

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

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

给定一个张量(tensor),此操作返回与所有元素设置为1的张量相同的类型和形状的张量.可选地,可以为返回的张量指定一个新的类型(dtype).

例如:

# 'tensor' is [[1, 2, 3], [4, 5, 6]]
tf.ones_like(tensor) ==> [[1, 1, 1], [1, 1, 1]]

参数:

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

返回值:

该函数将返回所有元素设置为1的张量.