python中base feature 保存.h5

import h5py

 

with h5py.File('attention_map.h5','w') as f:
    m = torch.squeeze(base_feat.detach())
    #pdb.set_trace()
    m=m.sum(0)/1024.0
    m=m.unsqueeze(2).expand(38,53,3)
    #pdb.set_trace()
    m = m.numpy()
    f.create_dataset('base_feat',shape=(38,53,3),data=m)