Closed jpata closed 7 months ago
[batch, num_elements, num_features]
@farakiko @erwulff please check if you agree with this change.
Looks good to me!
[batch, num_elements, num_features]