Python tf.ones_like用法及代码示例

创建一个与输入具有相同形状的所有张量。

用法

tf.ones_like(
    input, dtype=None, name=None
)

参数

input 一个Tensor。
dtype 返回的 Tensor 的类型。必须是 float16 , float32 , float64 , int8 , uint8 , int16 , uint16 , int32 , int64 , complex64 , complex128 , bool 或 string 。
name 操作的名称(可选)。

一个Tensor，所有元素都设置为一个。

另见tf.ones。

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

例如：

tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
tf.ones_like(tensor)
<tf.Tensor:shape=(2, 3), dtype=int32, numpy=
  array([[1, 1, 1],
         [1, 1, 1]], dtype=int32)>

相关用法

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