Closed awf closed 3 years ago
We were failing equality checks in the benchmark because knossos returns scalars as scalars, and Python converts float32 to double. This fix wraps all scalars as 1x1 tensors, which is consistent with PyTorch.
New test fails before, and now passes.
We were failing equality checks in the benchmark because knossos returns scalars as scalars, and Python converts float32 to double. This fix wraps all scalars as 1x1 tensors, which is consistent with PyTorch.
New test fails before, and now passes.