Python tf.compat.v1.ones_like用法及代码示例

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

用法

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

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

给定单个张量 (tensor)，此操作返回与 tensor 具有相同类型和形状的张量，所有元素都设置为 1。您可以选择为返回的张量指定新类型 (dtype)。

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

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.compat.v1.ones_like。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。