pulsatrix/api/sinusoidal__timestep__embedding_8hpp_source.html

#pragma once


#include <cstdint>


#include "pulsatrix/device_backend.hpp"

#include "pulsatrix/tensor.hpp"


namespace pulsatrix {


[[nodiscard]] Tensor SinusoidalTimestepEmbedding(int64_t t, int64_t embedding_dim, DeviceBackend* backend,

                                                 float base = 10000.0f);


}  // namespace pulsatrix

pulsatrix::DeviceBackend
Vendor-agnostic compute/memory backend. CPUBackend, CUDABackend (Phase 1.5), and HIPBackend (Phase 1....
Definition device_backend.hpp:219

pulsatrix::Tensor
N-dimensional tensor. Owns its data buffer exclusively; a DeviceBackend* is injected (not owned) – th...
Definition tensor.hpp:29

device_backend.hpp
Abstract interface isolating vendor-specific memory/compute operations from Tensor/ComputationGraph.

pulsatrix
Definition acquisition_functions.hpp:16

pulsatrix::SinusoidalTimestepEmbedding
Tensor SinusoidalTimestepEmbedding(int64_t t, int64_t embedding_dim, DeviceBackend *backend, float base=10000.0f)
Standard Transformer-style sinusoidal encoding of a diffusion timestep t: emb[2i] = sin(t / base^(2i/...

tensor.hpp
N-dimensional tensor – owns a buffer via DeviceBackend*, RAII (Rule of Five).