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

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

函数

返回与所有元素设置为1的类型和形状相同的张量

tf.ones_like(
    tensor,  # 给定tensorflow张量
    dtype=None,  # 返回值的类型
    name=None,  # 操作的名称（可选）
    optimize=True  # 如果为 True，则尝试静态确定"张量"的形状并将其编码为常量
)

示例

import tensorflow as tf

tensor = tf.constant([[1, 2, 3], [4, 5, 6]])  # 创建一个tensor
tensor1 = tf.ones_like(tensor)  # 返回与所有元素设置为1的类型和形状相同的张量
print(tensor1)

tf.Tensor(
[[1 1 1]
 [1 1 1]], shape=(2, 3), dtype=int32)