TensorFlow函数：tf.ones_like

时间：2022-09-03 07:45:30
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的张量.

相关文章