View source on GitHub
|
Creates a tensor with all elements set to 1.
tf.compat.v1.ones_like(
tensor, dtype=None, name=None, optimize=True
)
See also tf.ones.
Given a single tensor (tensor), this operation returns a tensor of the same
type and shape as tensor with all elements set to 1. Optionally, you can
specify a new type (dtype) for the returned tensor.
tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
tf.ones_like(tensor) # [[1, 1, 1], [1, 1, 1]]
tensor
Tensor.
dtype
Tensor. Must be float32, float64,
int8, uint8, int16, uint16, int32, int64, complex64,
complex128 or bool.
name
optimize
Tensor with all elements set to 1.
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.