[Apache TVM Discuss] [Questions] What Relay op would allow for splitting a tensor along an axis?

["Steven S. Lyubomirsky via Apache TVM Discuss" <no...@discuss.tvm.ai>]

I'm curious if someone might be able to guide me as to how I might be able to implement a Relay equivalent to the Keras [`TimeDistributed` layer](https://keras.io/api/layers/recurrent_layers/time_distributed/). I need to import a Keras model that features it and I am willing to extend the Keras importer to support this.

Would it be correct to say that the TVM equivalent would split a tensor on the second dimension, apply the passed layer to each of the resulting tensors, and then concatenate them back together? If so, which operator should I use to implement the splitting? I don't have much experience with Keras and, as I said, I would be very glad to PR support for `TimeDistributed` to the Keras importer.





---
[Visit Topic](https://discuss.tvm.apache.org/t/what-relay-op-would-allow-for-splitting-a-tensor-along-an-axis/8478/1) to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/646562559420e44b6ca6061eae63175e26f2e20bd346b832f2d11d1bf87bbc92).

[Apache TVM Discuss] [Questions] What Relay op would allow for splitting a tensor along an axis?

["Steven S. Lyubomirsky via Apache TVM Discuss" <no...@discuss.tvm.ai>]

@altanh pointed out to me that it should be possible to use the `split` operator for what I proposed. I will see if I can add support for `TimeDistributed` into the Keras importer





---
[Visit Topic](https://discuss.tvm.apache.org/t/what-relay-op-would-allow-for-splitting-a-tensor-along-an-axis/8478/2) to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/9c982522d9eec086734ecc36dd88a8ca7395f75a002a83122e27f4dc769411ac).