Calculates the curl operator on a staggered three-dimensional field. The resulting vector field is a staggered grid. If the velocities of

Function: Set obstacles Input: sim: FluidSimulation object

Check if data and log directories exist, and create otherwise Args: Returns: None

Check if data files exist. Will NOT replace them.

Removes all obstacles and empty cells from the simulation domain.

cosine schedule as proposed in https://openreview.net/forum?id=-NEXDKk8gZ

cosine schedule as proposed in https://openreview.net/forum?id=-NEXDKk8gZ

cosine schedule as proposed in https://openreview.net/forum?id=-NEXDKk8gZ

merges b into a and return merged result NOTE: side effect: a is modified NOTE: tuples and arbitrary objects are not handled as it is totally

Computes the spatial divergence of a vector field from finite differences. :param vel: tensor of shape (batch size, spatial dimensions..., spatia

Function: Get x,y for turns Output: xs: list x-position for each turn ys: list y-position for each turn

Generate data for Burgers' equation Args: experiment (str): experiment string pde (dict): dictionary for PDEs at different re

Function: get absorb area

Returns a context manager for manipulating a default stack.