1. 程式人生 > >tf.ones_like() tf.zeros_like()用法

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

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]]