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

Method forward

guided_diffusion/unet.py:435–454  ·  view source on GitHub ↗

Apply QKV attention. :param qkv: an [N x (3 * H * C) x T] tensor of Qs, Ks, and Vs. :return: an [N x (H * C) x T] tensor after attention.

(self, qkv)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected