我想在Tensorflow中获取占位符的尺寸。功能tf$shape
立即出现在我的脑海里。我毫不费力地使用tf$shape
来获得具有正面维度的占位符的形状。不过,我希望我的占位符能够接受不同尺寸的输入,所以我留下了一个尺寸为NULL
。现在,根据我的理解,当涉及到尺寸时,Tensorflow会等效地处理NULL
和-1
。然而,当我跑我的代码,我收到此错误:获取Tensorflow占位符的形状
W tensorflow/core/framework/op_kernel.cc:1148] Invalid argument: Shape [-1,2] has negative dimensions
下面是我的代码重复的例子:
a = tf$placeholder(tf$float32, shape = shape(NULL, 2L))
sess = tf$Session()
sess$run(tf$shape(a))
是上面引用的在我的代码或由造成一些错误事实上,tf$shape
不能占用负面维度的占位符?如果后者是真的,有没有办法让我得到负尺寸占位符的形状而不使用tf$shape
?
非常感谢! – nnguyen24