tf.size()函数

函数功能个：返回张量的大小.此函数返回一个表示input中的元素数的整数. 

函数原型

size(input, name=None, out_type=tf.int32)

import tensorflow as tf

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