首页 > 其他 > 详细

tf.size()函数

时间:2020-03-23 15:07:31      阅读:54      评论:0      收藏:0      [点我收藏+]

函数功能个:返回张量的大小.此函数返回一个表示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

 

tf.size()函数

原文:https://www.cnblogs.com/tingtin/p/12552045.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!