get_shape函数主要用于获取一个张量的维度,并且输出张量 每个维度上面的值,如果是二维矩阵,也就是输出行和列的值,使用非常方便。
例如:
import tensorflow as tf;
with tf.Session() as sess:
A = tf.random_normal(shape=[3,4]) print A.get_shape()
print A.get_shape输出:
(3, 4)
<bound method Tensor.get_shape of <tf.Tensor 'random_normal:0' shape=(3, 4) dtype=float32>>
注意:第一个输出是一个元祖,就是数值,而第二输出就是一个张量的对象,里面包含更多的东西,在不同的情况下,使用不同的方式。如果你需要输出某一个维度上面的值那就用下面的这种方式就好了。
A.get_shape()[0]这就表示第一个维度。
本文介绍了TensorFlow中get_shape()函数的基本用法及其返回值的特点,通过实例展示了如何使用该函数来获取张量的维度信息。
6537

被折叠的 条评论
为什么被折叠?



