Split into custom layers

[alejandrojapkin]

Is it possible to create split custom layer-wise layers? If so, would you provide an example?
Here's my intended implementation:

[yechengxi]

You'd need to write a custom layer for these operations. You can find examples in the LSTM implementation. For this type of operations it may be easier for you to switch to platforms with automatic differentiation.