tf.compat.v1.ones_like | TensorFlow v2.16.1 (original) (raw)
tf.compat.v1.ones_like
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.
For example:
tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
tf.ones_like(tensor) # [[1, 1, 1], [1, 1, 1]]
| Args | |
|---|---|
| tensor | A Tensor. |
| dtype | A type for the returned Tensor. Must be float32, float64,int8, uint8, int16, uint16, int32, int64, complex64,complex128 or bool. |
| name | A name for the operation (optional). |
| optimize | if true, attempt to statically determine the shape of 'tensor' and encode it as a constant. |
| Returns |
|---|
| A 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.