Bilinear doesn’t like the fact that x_c was transposed. You can replace the call with this: x = drop_cat(cat(x_c.contiguous(), x_n)). This basically just reshapes the data (in memory) to match what the transpose did.