MCPcopy Create free account
hub / github.com/ImprintLab/MedSegDiff / forward

Method forward

guided_diffusion/unet.py:1067–1105  ·  view source on GitHub ↗

Apply the model to an input batch. :param x: an [N x C x ...] Tensor of inputs. :param timesteps: a 1-D batch of timesteps. :param y: an [N] Tensor of labels, if class-conditional. :return: an [N x C x ...] Tensor of outputs.

(self, x, timesteps, y=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 2

highway_forwardMethod · 0.95
timestep_embeddingFunction · 0.85

Tested by

no test coverage detected