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]]) # 创建一个tensortensor1 = tf.ones_like(tensor) # 返回与所有元素设置为1的类型和形状相同的张量print(tensor1)

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