tf.ones_like

2022-09-04 21:01:42 浏览数 (1)

代码语言：javascript复制

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

创建一个所有元素都设为1的张量。给定一个张量(张量)，这个操作返回一个与所有元素都设为1的张量类型和形状相同的张量。还可以为返回的张量指定一个新类型(dtype)。

例:

代码语言：javascript复制

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

参数:

tensor: 一个张量。
dtype: 返回张量的类型。必须是float32、float64、int8、uint8、int16、uint16、int32、int64、complex64、complex128orbool '。
name: 操作的名称(可选)。
optimize: 如果为真，尝试静态确定“张量”的形状，并将其编码为常量。

返回值:

0 人点赞