TensorFlow中的tf.ones_like()和tf.zeros_like()函数详解
TensorFlow中的tf.ones_like（）和tf.zeros_like（）函数详解
- 1.tf.ones_like()
- a.官方注释
- b.参数解释
- c.例子
- 2.tf.zeros_like()
- a.官方注释
- b.参数解释
- c.例子
- 参考
在基于tensorlayer的SRGAN代码中碰到了这两个函数，记录一下：
d_loss1 = tl.cost.sigmoid_cross_entropy(logits_real, tf.ones_like(logits_real), name='d1')
d_loss2 = tl.cost.sigmoid_cross_entropy(logits_fake, tf.zeros_like(logits_fake), name='d2')
1.tf.ones_like()
a.官方注释
Creates a tensor with all elements set to 1.
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.
翻译一下就是：
创建一个将所有元素设置为1的张量。
给定一个tensor（tensor ），该操作返回一个具有和给定tensor相同形状（shape）和相同数据类型（dtype），但是所有的元素都被设置为1的tensor。也可以为返回的tensor指定一个新的数据类型。
b.参数解释
def ones_like(tensor, dtype=None, name=None, optimize=True):
...
return ret
- tensor：张量.
- dtype：返回的 Tensor 类型.必须是：float32,float64,int8,int16,int32,int64,uint8,complex64,complex128或bool.
- name：操作的名称(可选).
- optimize：如果为 true,则尝试静态确定 “张量” 的形状并将其编码为常量.
- return：返回一个所有元素设置为1的tensor
c.例子
tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
tf.ones_like(tensor) # [[1, 1, 1], [1, 1, 1]]
2.tf.zeros_like()
a.官方注释
Creates a tensor with all elements set to zero.
Given a single tensor (tensor), this operation returns a tensor of the
same type and shape as tensor with all elements set to zero. Optionally,
you can use dtype to specify a new type for the returned tensor.
翻译一下就是：
创建一个将所有元素设置为0的张量。
给定一个tensor（tensor ），该操作返回一个具有和给定tensor相同形状（shape）和相同数据类型（dtype），但是所有的元素都被设置为0的tensor。也可以为返回的tensor指定一个新的数据类型。
b.参数解释
def zeros_like(tensor, dtype=None, name=None, optimize=True):
...
return gen_array_ops._zeros_like(tensor, name=name)
- tensor：张量.
- dtype：返回的张量的类型,必须是以下类型之一：float16,float32,float64,int8,uint8,int16,uint16,int32,int64,complex64,complex128,bool或string.
- name：操作的名称(可选).
- optimize：如果为 true,则尝试静态确定 “张量” 的形状并将其编码为常量.
- return：返回一个所有元素设置为零的tensor
c.例子
tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
tf.zeros_like(tensor) # [[0, 0, 0], [0, 0, 0]]
参考
[1] Tensorflow深度学习之二十九：tf.ones_like()和tf.zeros_like()
[2] TensorFlow函数：tf.ones_like
[3] TensorFlow函数：tf.zeros_like
码字不易，如果您觉得有帮助，麻烦帮我点个赞~~