Fix dimensionality of log_std in GaussianMLPModule

[codecov[bot]]

Codecov Report

Merging #2266 (1ca4dd0) into master (fa0cd08) will increase coverage by 0.04%. The diff coverage is 84.61%.

@@            Coverage Diff             @@
##           master    #2266      +/-   ##
==========================================
+ Coverage   91.23%   91.28%   +0.04%     
==========================================
  Files         199      199              
  Lines       11726    10966     -760     
  Branches     1574     1387     -187     
==========================================
- Hits        10698    10010     -688     
+ Misses        757      693      -64     
+ Partials      271      263       -8     

Impacted Files	Coverage Δ
src/garage/torch/modules/gaussian_mlp_module.py	92.13% <84.61%> (+3.12%)	:arrow_up:
src/garage/torch/optimizers/optimizer_wrapper.py	93.75% <0.00%> (-6.25%)	:arrow_down:
src/garage/sampler/env_update.py	82.92% <0.00%> (-3.62%)	:arrow_down:
src/garage/torch/optimizers/differentiable_sgd.py	86.66% <0.00%> (-3.34%)	:arrow_down:
src/garage/torch/policies/stochastic_policy.py	88.63% <0.00%> (-3.17%)	:arrow_down:
src/garage/torch/algos/trpo.py	91.66% <0.00%> (-3.08%)	:arrow_down:
src/garage/torch/algos/dqn.py	93.07% <0.00%> (-2.09%)	:arrow_down:
src/garage/torch/_functions.py	92.42% <0.00%> (-1.92%)	:arrow_down:
src/garage/np/_functions.py	77.39% <0.00%> (-1.74%)	:arrow_down:
src/garage/torch/algos/sac.py	97.14% <0.00%> (-1.66%)	:arrow_down:
... and 7 more

---

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update fa0cd08...1ca4dd0. Read the comment docs.

[avnishn]

@krzentner ptal at @akashvelu's comments

[krzentner]

The only untested on CI line in the diff is GPU-only, so I'm merging this.