333

out = torch.cat((out, out[:, :, :, :padding_cols]), dim=3)

 

posted @ 2024-06-28 10:57  helloWorldhelloWorld  阅读(4)  评论(0编辑  收藏  举报