search

rwth-i6 / i6_models

Collection of NN-Model parts
Mozilla Public License 2.0
1 stars 0 forks source link

Add mask_tensor util #35

Closed JackTemaki closed 1 year ago

JackTemaki commented 1 year ago

Utility to create a boolean tensor mask on the tensors device based on sequence lengths.

JackTemaki commented 1 year ago

Closing, as this specific signature seems not to be beneficial for people other than me.