elementwise_mean -> mean (thinking ahead to pytorch 1.0)
under the pytorch 1.0 nightly this test generates 

```
UserWarning: reduction='elementwise_mean' is deprecated, please use reduction='mean' instead.
```

so this PR fixes that.
oops, doesn't work under current pytorch, never mind