部分堆叠张量
Partially stacking Tensors
我从 PyTorch 中的卷积层输出了一些形状为 [b,c,h] 的输出，其中 b 是我的批次，c 是我的通道，h 是特征。我想在不改变批次的情况下将这些堆叠起来以馈送到完全连接的层中，因此它们的形状为 [b, c*h]。我该怎么做？
似乎简单的重塑或视图应该有效：
input.shape # [b,c,h]
reshaped = input.view(input.shape[0],-1) # [b,c*h]
我从 PyTorch 中的卷积层输出了一些形状为 [b,c,h] 的输出，其中 b 是我的批次，c 是我的通道，h 是特征。我想在不改变批次的情况下将这些堆叠起来以馈送到完全连接的层中，因此它们的形状为 [b, c*h]。我该怎么做？
似乎简单的重塑或视图应该有效：
input.shape # [b,c,h]
reshaped = input.view(input.shape[0],-1) # [b,c*h]