333

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

 

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