Tensorflow 替换tf.transpose？

由于网络中各层的性质，我必须进行三种转换：

Input to 1D convolution
1D Convolution to 2D Convolution Transpose
1D Convolution to Dense

这需要大量的时间

batch_time=121.467 seconds

对于批量为10的产品

以前的模型，大小相似，没有换位，在同一个数据集上，第一批需要10秒，之后每批需要0.4秒。在第一批之后，我上面的型号每批仍然需要超过4秒的时间

---

我想我设法把它缩小到我所做的换位。有办法解决这个问题吗

在代码中添加一个最小的示例。如果是python，我一定会尝试看一看。如果不是python，它仍然会共享您的代码，因为其他人很可能会提供帮助。好吧，它基本上是tesorflow中的4D张量，我想用计算成本较低的操作替换tf.transpose（张量[0,2,1,3]）。