您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

如何获取Tensorflow张量尺寸(形状)作为int值?

如何获取Tensorflow张量尺寸(形状)作为int值?

要将形状显示为整数列表,请执行tensor.get_shape().as_list()

要完成tf.shape()通话,请尝试tensor2 = tf.reshape(tensor, tf.TensorShape([num_rows*num_cols, 1]))。或者,您可以直接tensor2 = tf.reshape(tensor, tf.TensorShape([-1, 1]))在可以推断出其第一维的地方进行操作。

其他 2022/1/1 18:25:19 有471人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶