函数功能个:返回张量的大小.此函数返回一个表示input中的元素数的整数.
函数原型
size(input, name=None, out_type=tf.int32)
举例
t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
b=tf.size(t)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
print(sess.run(b)) # 12