transformers

d4ad3218 - Turns out we assume that the number of dimension in size is the same, in fx but it's not in testing env