Open Meng-Wei opened 5 years ago
I think the implementation of the squeeze layer is inappropriate. (Tensor.view) partitions the tensor vector directly in order, which is not desired.
Thanks for pointing this out. To keep backwards compatibility with old models we've added a flag to set which squeeze implementation to use. See 9b4850a.
I think the implementation of the squeeze layer is inappropriate. (Tensor.view) partitions the tensor vector directly in order, which is not desired.