search

jinhong-ni / DEQFusion

PyTorch Implementation of Deep Equilibrium Multimodal Fusion
14 stars 2 forks source link

输入DEQFusion的图片数据处理 #5

Closed sMarch7th closed 3 months ago

sMarch7th commented 3 months ago

作者您好，请问一下，DEQFusion的输入是[batch_size,dim]的列表，是如何将[batch_dim,dim,high,weigh]的数据处理为[batch_size,dim]的形状的？

jinhong-ni commented 3 months ago

Simply flatten/sum/mean should work.