tf中tensor的大小输出

最新推荐文章于 2023-03-11 21:48:48 发布

frostjsy

最新推荐文章于 2023-03-11 21:48:48 发布

阅读量758

点赞数

分类专栏：环境配置&常见工具命令使用 tensorflow 文章标签： tensorflow 深度学习 python

本文链接：https://blog.csdn.net/u013069552/article/details/125371866

版权

环境配置&常见工具命令使用同时被 2 个专栏收录

49 篇文章 0 订阅

订阅专栏

tensorflow

14 篇文章 0 订阅

订阅专栏

本文档通过实例探讨了在TensorFlow中使用tf.shape(), x.shape和x.get_shape()获取Tensor维度的方法，并展示了在重塑张量时的不同形状操作。重点在于这三个属性在实际应用中的区别和用途。

摘要由CSDN通过智能技术生成

import tensorflow as tf
logits=tf.constant([[1.0,2.0,3.0],[1.0,2.0,3.0],[1.0,2.0,3.0]])
labels = tf.constant([[0.0, 0.0, 1.0], [0.0, 0.0, 1.0], [0.0, 0.0, 1.0]])


logits_ = tf.reshape(logits, [-1, 1])
labels_ = tf.reshape(labels, [-1, 1])


print(logits.shape.as_list())
print(labels.shape.as_list())

print(logits_.shape.as_list())
print(labels_.shape.as_list())

输出

[3, 3]
[3, 3]
[9, 1]
[9, 1]

【1】 [TensorFlow笔记] 获取Tensor的维度（tf.shape(x)、x.shape和x.get_shape()的区别）_手撕机的博客-CSDN博客_获取tensor的shape