博文
TensorFlow tensor shape dimensions
|
Q: How to get the dimensions (shape) of a tensor as integer values?
import tensorflow as tf
shape = [3,4,5]
tensor = tf.ones(shape=shape)
tensor.get_shape().as_list()
Sources:
https://www.tensorflow.org/guide/tensors
https://blog.sciencenet.cn/blog-578676-1129603.html
上一篇：cudnn Exception error with GpuDnnConv
下一篇：随笔一记
