Fix some naughty uses of reshape/flatten (#88999)
Mutating after reshape/flatten is bad! And it turns out
the corresponding view operations are guaranteed to work
too.
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88999
Approved by: https://github.com/albanD