我爱编程

如何获取tensor的维度值

2018-05-13  本文已影响0人  ziyu123

  在TensorFlow的使用过程中,我们常常希望得到一个tensor的维度信息使用,具体的说,也就是现在有了一个tensor值,如何才能得到其shape信息,也就是维度值作为一个整数值使用呢?
  对于一个tensor值,我们很容易利用tensor.get_shape()tf.shape(tensor)来获取其shape。但是,这两种方法返回的shape信息都是Dimension 类型的,并非int32类型的。下面两种方法可以获得tensor shape的具体值。

 num_rows, num_cols  = X.get_shape().as_list()
 num_rows, num_cols  = map(lambda i: i.value, X.get_shape())

参考:https://stackoverflow.com/questions/40666316/how-to-get-tensorflow-tensor-dimensions-shape-as-int-values

上一篇 下一篇

猜你喜欢

热点阅读