guides/understanding_masking_and_padding.py/TemporalSplit/call
class TemporalSplit:    def call(self, inputs):
        # Expect the input to be 3D and mask to be 2D, split the input tensor into 2
        # subtensors along the time axis (axis 1).
        return ops.split(inputs, 2, axis=1)