Torch 三维数据维度转换:将 [32, 7, 64] 变为 [32, 252, 64]
可以使用torch.reshape()函数将三维数据变为二维数据,然后再使用torch.transpose()函数进行维度调换。具体代码如下:
import torch
# 生成三维数据 [32, 7, 64]
data = torch.randn(32, 7, 64)
# 变为二维数据 [32, 252, 64]
reshaped_data = torch.reshape(data, (32, 252, 64))
print(reshaped_data.shape)
上述代码中,torch.reshape()函数将原始数据data变为维度为(32, 252, 64)的数据reshaped_data。最后打印输出reshaped_data的形状。
原文地址: https://www.cveoy.top/t/topic/p7SA 著作权归作者所有。请勿转载和采集!