Thanks. This way it does not work actually, because I need the transformation function processes a, b and c separately. Modifying f to take the concatenatted_tensor as input while keeping its subparts a, b, c separately is quite complicate to do.