MCPcopy Create free account

hub / github.com/AI4Science-WestlakeU/wdno / functions

Functions1,208 in github.com/AI4Science-WestlakeU/wdno

⨍ Functions 1,208◇ Types & classes 150↳ Endpoints 24
↓ 123 callersMethodreshape
(self, value, shape)
smoke/phi/math/base.py:31
↓ 116 callersMethodzeros
Creates a NumPy array of zeros of which the shape is determined by the dimensions of the simulation, the given element type and batch size. The shape
smoke/phi/flow.py:240
↓ 104 callersMethodsum
(self, value, axis=None)
smoke/phi/math/base.py:34
↓ 83 callersMethodmean
(self, value, axis=None)
smoke/phi/math/base.py:37
↓ 73 callersMethodpad
(self, value, pad_width, mode="constant", constant_values=0)
smoke/phi/math/base.py:25
↓ 64 callersMethodset_obstacle
(self, mask_or_size, origin=None)
smoke/phi/flow.py:171
↓ 57 callersMethodsqrt
(self, x)
smoke/phi/math/base.py:83
↓ 54 callersMethodadvect
Performs a semi-Lagrangian advection step, propagating the field through the velocity field. A backwards Euler step is performed and the smpl
smoke/phi/math/nd.py:407
↓ 46 callersMethodstack
(self, values, axis=0)
smoke/phi/math/base.py:19
↓ 40 callersMethodset_condition
(self, img, u: torch.Tensor, shape, condition_type)
burgers/ddpm_burgers/diffusion_1d.py:276
↓ 38 callersMethodzeros_like
(self, tensor)
smoke/phi/math/base.py:46
↓ 37 callersFunctionspatial_rank
Returns the number of spatial dimensions. Arrays are expected to be of the shape (batch size, spatial dimensions..., component size) The number of sp
smoke/phi/math/nd.py:5
↓ 36 callersMethodones
Creates a NumPy array of ones of which the shape is determined by the dimensions of the simulation, the given element type and batch size. The shape
smoke/phi/flow.py:254
↓ 34 callersMethodchoose_backend
(self, values)
smoke/phi/math/base.py:132
↓ 30 callersMethodsize
(self, datasource)
smoke/phi/data/data.py:97
↓ 25 callersMethodconcat
(self, values, axis)
smoke/phi/math/base.py:22
↓ 22 callersMethodload
(self, milestone)
smoke/ddpm/diffusion_2d.py:1201
↓ 21 callersMethodabs
(self, x)
smoke/phi/math/base.py:62
↓ 20 callersMethodplaceholder
(self, element_type=1, name=None, batch_size=None, dtype=np.float32)
smoke/phi/tf/flow.py:96
↓ 19 callersMethodinfo
(self, message)
smoke/phi/model.py:254
↓ 19 callersMethodsave
(self, milestone)
smoke/ddpm/diffusion_2d.py:1187
↓ 18 callersFunctionconv
(n, filters, kernel_size, strides=[1,1,1,1], padding="VALID", activation=None, name=None, kernel_initializer=N
smoke/phi/experimental.py:9
↓ 18 callersMethodmkdir
(self, subdir=None)
smoke/phi/fluidformat.py:192
↓ 18 callersFunctionprint_progress_bar
(iteration, total, prefix='', suffix='', decimals=1, length=100, fill='█')
smoke/phi/solver/cuda/benchmarks/benchmark_utils.py:10
↓ 17 callersMethodexp
(self, x)
smoke/phi/math/base.py:86
↓ 17 callersMethodrun
(self, tasks, feed_dict=None, options=None, run_metadata=None, summary_key=None, time=None, merged_summary=Non
smoke/phi/tf/flow.py:29
↓ 16 callersFunctionextract
(a, t, x_shape)
smoke/ddpm/diffusion_2d.py:508
↓ 16 callersMethodstd
(self, x, axis=None)
smoke/phi/math/base.py:113
↓ 15 callersFunctionextract
(a, t, x_shape)
burgers/ddpm_burgers/model_utils.py:27
↓ 15 callersMethodview
(self, tasks, options=None, run_metadata=None, all_batches=False)
smoke/phi/tf/model.py:178
↓ 14 callersMethod__init__
(self, beta)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:110
↓ 14 callersMethod__init__
(self, beta)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:117
↓ 14 callersFunctiondefault
(val, d)
burgers/ddpm_burgers/model_utils.py:14
↓ 14 callersFunctionexists
(x)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:25
↓ 14 callersFunctionexists
(x)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:32
↓ 14 callersMethodmaximum
(self, a, b)
smoke/phi/math/base.py:74
↓ 14 callersMethodpad
(self, lower, upper=None, mode="symmetric")
smoke/phi/math/nd.py:508
↓ 14 callersMethodshape
(self, datasource)
smoke/phi/data/data.py:94
↓ 12 callersMethod__init__
( self, dim, init_dim = None, out_dim = None, dim_mults=(1, 2, 4, 8),
smoke/ddpm/diffusion_2d.py:377
↓ 12 callersFunctiondefault
(val, d)
smoke/ddpm/diffusion_2d.py:37
↓ 12 callersFunctionresidual_block
(y, nb_channels, kernel_size=(3, 3), _strides=(1, 1), activation=tf.nn.leaky_relu, _project
smoke/phi/tf/util.py:17
↓ 11 callersMethod__init__
(self, dim, dim_out, groups = 8, conv_2d = False)
burgers/ddpm_burgers/unet.py:130
↓ 11 callersMethodexpand_dims
(self, a, axis)
smoke/phi/math/base.py:92
↓ 11 callersFunctionextract
(a, t, x_shape)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:571
↓ 11 callersFunctionextract
(a, t, x_shape)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:579
↓ 10 callersMethod__init__
(self, dim, fn)
smoke/ddpm/modules.py:102
↓ 10 callersMethodflatten
(self, x)
smoke/phi/math/base.py:107
↓ 10 callersMethodmax
(self, x, axis=None)
smoke/phi/math/base.py:71
↓ 10 callersMethodupdate
(self, new_state, type)
smoke/phi/control/smoke_control.py:21
↓ 10 callersFunctionupsample2x
(tensor, interpolation="LINEAR")
smoke/phi/math/nd.py:284
↓ 9 callersMethodadd
(self, values)
smoke/phi/math/base.py:28
↓ 9 callersMethodceil
(self, x)
smoke/phi/math/base.py:65
↓ 9 callersMethoddivergence_free
(self, velocity, enforce_boundary_conditions=True, **kwargs)
smoke/phi/flow.py:318
↓ 9 callersFunctionexists
(x)
smoke/ddpm/diffusion_2d.py:31
↓ 9 callersMethodgather
(self, values, indices)
smoke/phi/math/base.py:104
↓ 9 callersMethodminimum
(self, a, b)
smoke/phi/math/base.py:77
↓ 9 callersMethodones_like
(self, tensor)
smoke/phi/math/base.py:49
↓ 9 callersFunctionto_dipole_format
(tensor)
smoke/phi/experimental.py:67
↓ 9 callersMethodwith_boundary_conditions
(self, velocity)
smoke/phi/flow.py:294
↓ 8 callersMethod__init__
(self, name, target_size, flags)
smoke/phi/data/data.py:274
↓ 8 callersMethod_create_mask
(self)
smoke/phi/flow.py:195
↓ 8 callersMethodconv
(self, tensor, kernel, padding="SAME")
smoke/phi/math/base.py:89
↓ 8 callersFunctionexists
(x)
burgers/ddpm_burgers/model_utils.py:11
↓ 8 callersFunctionget_envolve
(sim,pre_velocity,frame,control_write,vx=None,vy=None)
smoke/dataset/a_gen_test_128.py:288
↓ 8 callersFunctionget_envolve
Function: get next step velocity with indirect control Input: sim: FluidSimulation Object pre_velocity: StaggeredGrid previou
smoke/dataset/a_gen_train.py:363
↓ 8 callersFunctionget_envolve
(sim,pre_velocity,frame,control_write,vx=None,vy=None)
smoke/dataset/a_gen_test_64.py:304
↓ 8 callersFunctionget_real_vel
(vel)
smoke/dataset/a_gen_test_128.py:162
↓ 8 callersFunctionget_real_vel
Function: Get Real Velocity from Nomral Distribution Input: vel: float Output: real_vel: float
smoke/dataset/a_gen_train.py:189
↓ 8 callersFunctionget_real_vel
(vel)
smoke/dataset/a_gen_test_64.py:176
↓ 8 callersMethodsubpath
(self, name, create=False)
smoke/phi/fluidformat.py:106
↓ 8 callersMethodtile
(self, x, multiples)
smoke/phi/math/base.py:122
↓ 7 callersMethodisfinite
(self, x)
smoke/phi/math/base.py:119
↓ 7 callersMethodrestore
(self, dir, scope=None)
smoke/phi/tf/flow.py:175
↓ 6 callersMethodadd_scalar
(self, name, node)
smoke/phi/tf/model.py:131
↓ 6 callersFunctionbenchmark_pressure_solve
(solver, tests, dimension, dtype, warmup=10, testruns=50, accuracy=1e-5, batch_size=1,
smoke/phi/solver/cuda/benchmarks/benchmark_utils.py:68
↓ 6 callersFunctioncoef_to_tensor
(coef, pad = False)
smoke/wave_trans_2d.py:55
↓ 6 callersFunctiondefault
(val, d)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:34
↓ 6 callersFunctiondefault
(val, d)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:41
↓ 6 callersMethodfeed_dict
(self, iterator, training, subrange=None)
smoke/phi/tf/model.py:159
↓ 6 callersMethodget
(self, datasource, indices)
smoke/phi/data/data.py:100
↓ 6 callersFunctionget_target
is_wavelet==True & f==True: return [N*(1+3*3), 45(not padded!!), 64(padded!!)] is_wavelet==False & f==True: return:[N, 81(not padded!!), 128(
burgers/ddpm_burgers/test_util.py:156
↓ 6 callersMethodload
(self, milestone)
burgers/ddpm_burgers/train_diffusion.py:164
↓ 6 callersMethodpad_fluid
(self, fluid_mask)
smoke/phi/flow.py:418
↓ 6 callersMethodprepare
(self)
smoke/phi/tf/model.py:60
↓ 6 callersMethodsave
(self, milestone)
burgers/ddpm_burgers/train_diffusion.py:148
↓ 6 callersMethodstep
(self)
smoke/phi/tf/model.py:75
↓ 6 callersFunctiontensor_to_coef
input: [N, more than 5*8, padded, padded, padded] upsample_type: None (base) or space (super) or time (super) output: list [coarse, detai
smoke/wave_trans_2d.py:17
↓ 5 callersMethodat_centers
(self)
smoke/phi/math/nd.py:332
↓ 5 callersFunctionconv_function
(scope, constants_file=None)
smoke/phi/experimental.py:6
↓ 5 callersMethodexpand_dims
(self, a, axis)
smoke/phi/math/tensorflow_backend.py:150
↓ 5 callersFunctiongridIDXWithOffsetShifted
Converts coordinates of the simulation grid to indices of the extended mask grid with a shift to get the indices of the neighbors
smoke/phi/solver/cuda/src/laplace_op.cu.cc:16
↓ 5 callersFunctionindices_tensor
Returns an index tensor of the same spatial shape as the given tensor. Each index denotes the location within the tensor starting from zero. Indices
smoke/phi/math/nd.py:19
↓ 5 callersFunctioninit_sim
()
smoke/dataset/evaluate_solver.py:60
↓ 5 callersMethodpartition_execute
(self, n, start_frame_index, **kwargs)
smoke/phi/control/sequences.py:51
↓ 5 callersMethodresample
(self, inputs, sample_coords, interpolation="LINEAR", boundary="ZERO")
smoke/phi/math/base.py:43
↓ 5 callersMethodshow
(self, port=None, use_reloader=False)
smoke/phi/viz/dash_gui.py:535
↓ 5 callersMethodsolve_with_boundaries
(self, divergence, active_mask, fluid_mask, accuracy=1e-5, pressure_guess=None, # pressure_guess is not used
smoke/phi/solver/cuda/cuda.py:15
↓ 4 callersMethod__init__
(self, name, method, id)
smoke/phi/model.py:483
↓ 4 callersMethod_create_or_reset_mask
(self, old_mask)
smoke/phi/tf/flow.py:141
↓ 4 callersMethod_get_paddings
(self, rank, open)
smoke/phi/flow.py:432
next →1–100 of 1,208, ranked by callers