tf.ones_like() tf.zeros_like()用法

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Eric_LH/article/details/83794004

tf.ones_like | tf.zeros_like

tf.ones_like(tensor,dype=None,name=None)
tf.zeros_like(tensor,dype=None,name=None)
新建一个与给定的tensor类型大小一致的tensor，其所有元素为1和0

tensor=[[1, 2, 3], [4, 5, 6]] 
x = tf.ones_like(tensor) 
print(sess.run(x))

#[[1 1 1],
# [1 1 1]]