MCPcopy
hub / github.com/NVIDIA/TensorRT-LLM / unbind

Method unbind

tensorrt_llm/functional.py:516–520  ·  view source on GitHub ↗

See functional.unbind.

(self, dim=0)

Source from the content-addressed store, hash-verified

514 return select(self, dim, index)
515
516 def unbind(self, dim=0):
517 '''
518 See functional.unbind.
519 '''
520 return unbind(self, dim)
521
522 def repeat(self, sizes):
523 '''

Callers 12

test_unbind_1Method · 0.95
assert_closeFunction · 0.80
test_dora_scalingMethod · 0.80
forwardMethod · 0.80
_bmm_moe_down_split_hookFunction · 0.80
forwardMethod · 0.80
vision_encoder_forwardFunction · 0.80
rotate_halfMethod · 0.80

Calls 1

unbindFunction · 0.85

Tested by 5

test_unbind_1Method · 0.76
assert_closeFunction · 0.64
test_dora_scalingMethod · 0.64