小编bbh*_*hat的帖子

为num_splits使用变量为tf.split()

是否可以将num_split参数的占位符输入用于tf.split()?

我理想地喜欢这样做:

num_splits = tf.placeholder(tf.int32)
inputs = tf.placeholder(tf.int32, [5, None])
split_inputs = tf.split(1, num_splits, inputs)
Run Code Online (Sandbox Code Playgroud)

我的方法可能有问题.我希望枚举可变形状张量的维度.谢谢!

python tensorflow

9
推荐指数
1
解决办法
6496
查看次数

标签 统计

python ×1

tensorflow ×1