def convert_to_batch_order(self, output, N, L, K, C):
output = output.view(L, N, K, C)
output = torch.transpose(output, 0,1)
return output.contiguous().view(N*L*K, C)
微信
公众号
扫码关注公众号