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

Method unsqueeze

tensorrt_llm/functional.py:462–466  ·  view source on GitHub ↗

See functional.squeeze.

(self, dim)

Source from the content-addressed store, hash-verified

460 return squeeze(self, dim, zero_is_placeholder)
461
462 def unsqueeze(self, dim):
463 '''
464 See functional.squeeze.
465 '''
466 return unsqueeze(self, dim)
467
468 def log(self):
469 '''

Callers 15

build_alibi_tensorFunction · 0.95
compaction_torchFunction · 0.80
upcast_from_mxfp_torchFunction · 0.80
test_gpt_attentionMethod · 0.80
tile_beam_widthMethod · 0.80
torch_execMethod · 0.80
test_mambaMethod · 0.80
test_loadersMethod · 0.80
check_accuracyFunction · 0.80

Calls 1

unsqueezeFunction · 0.85

Tested by 15

test_gpt_attentionMethod · 0.64
tile_beam_widthMethod · 0.64
torch_execMethod · 0.64
test_mambaMethod · 0.64
test_loadersMethod · 0.64
check_accuracyFunction · 0.64
test_unpack_gen_dataMethod · 0.64
test_beams2treeMethod · 0.64
quant_dequant_intFunction · 0.64