在Tensorflow中将3D张量减小到1D张量

Question

让我们假设，我有一个三维张量的形状[a，b，c]。 我想在模型运行时提取一个维度，例如[4,2，c]，所以我最终会得到一维张量[c]。

a和b的参数存储在形状[a，1]和[b，1]的不同张量中，因此使用tf.slice不是一个选项，因为tf.slice只接受1维标量张量。

有什么想法吗？

谢谢！

Answer 1

您可能需要将张量与tf.split分开。

https://www.tensorflow.org/api_docs/python/tf/split