Skip to content

GitLab

Commit 9fb288c0 authored by Andrea Paris's avatar Andrea Paris Committed by Boris Bonev
Browse files

put back oneliner

parent 9336efe8
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_convolution.py
View file @ 9fb288c0
...@@ -47,6 +47,7 @@ if torch.cuda.is_available(): ...@@ -47,6 +47,7 @@ if torch.cuda.is_available():
def _normalize_convolution_tensor_dense(psi, quad_weights, transpose_normalization=False, basis_norm_mode="none", merge_quadrature=False, eps=1e-9): def _normalize_convolution_tensor_dense(psi, quad_weights, transpose_normalization=False, basis_norm_mode="none", merge_quadrature=False, eps=1e-9):
"""Discretely normalizes the convolution tensor."""
kernel_size, nlat_out, nlon_out, nlat_in, nlon_in = psi.shape kernel_size, nlat_out, nlon_out, nlat_in, nlon_in = psi.shape
correction_factor = nlon_out / nlon_in correction_factor = nlon_out / nlon_in
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment