Generate and print a spiral of natural numbers up to N with fully configurable shape and spacing.
Examples:
$ python3 spiral.py 50 square 2
32 31 30 29 28 27 26 25
33 24
| # Finite-difference method from http://www.cosy.sbg.ac.at/events/parnum05/book/horak1.pdf | |
| # Parallel Numerical Solution of 2-D Heat Equation, Verena Horak & Peter Gruber (Parallel Numerics ’05, 47-56) | |
| @cython.boundscheck(False) | |
| def diffuse(initial_condition, int nstep, double c = 1.0, double delta_t_factor = 0.5): | |
| """Evolve `initial_condition` according to the 2D heat (diffusion) equation under periodic boundary conditions.""" | |
| # Short circut | |
| if nstep == 0: | |
| return initial_condition |
| # Author: https://gist.github.com/Linux-cpp-lisp | |
| from typing import Tuple, List | |
| import numpy as np | |
| import torch | |
| @torch.jit.script | |
| def torch_divmod(a, b) -> Tuple[torch.Tensor, torch.Tensor]: |