Fatal编程技术网
  • tensorflow/
  • 在TensorFlow中挤压除一个轴以外的所有轴

在TensorFlow中挤压除一个轴以外的所有轴

tensorflow

在TensorFlow中挤压除一个轴以外的所有轴,tensorflow,tensor,Tensorflow,Tensor,如何挤压张量的特定轴以外的所有轴 输入大小（以及有多少个1维度）事先未知，但我知道我只想保留第一个和第二个轴（即，其余只能是1） e、 g.如果我有一个张量（1，4，1，1，1）的形状，我想把它压缩到（1，4）中，那么tf.重塑（tf.挤压（t），[1，-1]）会起作用吗？这里t是你的张量。如果其他轴总是1，实际上不需要挤压；您可以直接重塑形状。

如何挤压
张量的特定轴以外的所有轴

输入大小（以及有多少个
1
维度）事先未知，但我知道我只想保留第一个和第二个轴（即，其余只能是1）


e、 g.如果我有一个
张量
（1，4，1，1，1）
的形状
，我想把它压缩到
（1，4）
中，那么
tf.重塑（tf.挤压（t），[1，-1]）
会起作用吗？这里
t
是你的张量。如果其他轴总是1，实际上不需要挤压；您可以直接重塑形状。