在张量流中移动尺寸或公差

Question

我想用张量在张量流中执行移位维（matlab）或rollaxis（numpy）。 例如。形状为（512,7,7）到（7,7,512）的矩阵。重塑将无法正常工作，因为我将获得垃圾数据。 我怎么能在张量流中做到这一点？

Answer 1

谢谢，我会回答我的问题。 tf.transpose是要使用的功能。 perm参数允许我们设置新维度。 在我的情况下，烫发将是（1,2,0）因为我从[512,7,7]到[7,7,512]在两种情况下都保留了7s的顺序。

tf.transpose（x，[1,2,0]）