MCPcopy Create free account
hub / github.com/AlphaGPU/leetgpu-challenges / OutTensor

Class OutTensor

challenges/core/challenge_base.py:44–50  ·  view source on GitHub ↗

Output buffer (by shape): materialized empty where outputs are written in place (torch), omitted where they're returned functionally (jax).

Source from the content-addressed store, hash-verified

42
43
44class OutTensor:
45 """Output buffer (by shape): materialized empty where outputs are written in
46 place (torch), omitted where they're returned functionally (jax)."""
47
48 def __init__(self, shape, dtype="float32"):
49 self.shape = tuple(shape)
50 self.dtype = dtype
51
52
53class ChallengeBase(ABC):

Calls

no outgoing calls

Tested by

no test coverage detected