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

Method forward

guided_diffusion/unet.py:289–299  ·  view source on GitHub ↗

Apply the block to a Tensor, conditioned on a timestep embedding. :param x: an [N x C x ...] Tensor of features. :param emb: an [N x emb_channels] Tensor of timestep embeddings. :return: an [N x C x ...] Tensor of outputs.

(self, x, emb)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 1

checkpointFunction · 0.85

Tested by

no test coverage detected