ggml_timestep_embedding method

Pointer<ggml_tensor> ggml_timestep_embedding(
  1. Pointer<ggml_context> ctx,
  2. Pointer<ggml_tensor> timesteps,
  3. int dim,
  4. int max_period,
)

Ref: https://github.com/CompVis/stable-diffusion/blob/main/ldm/modules/diffusionmodules/util.py#L151 timesteps: N, return: N, dim

Implementation

ffi.Pointer<ggml_tensor> ggml_timestep_embedding(
  ffi.Pointer<ggml_context> ctx,
  ffi.Pointer<ggml_tensor> timesteps,
  int dim,
  int max_period,
) {
  return _ggml_timestep_embedding(
    ctx,
    timesteps,
    dim,
    max_period,
  );
}