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

Method select

tensorrt_llm/functional.py:510–514  ·  view source on GitHub ↗

See functional.select.

(self, dim, index)

Source from the content-addressed store, hash-verified

508 return split(self, split_size_or_sections, dim)
509
510 def select(self, dim, index):
511 '''
512 See functional.select.
513 '''
514 return select(self, dim, index)
515
516 def unbind(self, dim=0):
517 '''

Callers 9

test_select_from_intMethod · 0.80
copyTextFunction · 0.80
prepare_text_inputsFunction · 0.80
_logits_to_probsMethod · 0.80
get_calib_dataloaderFunction · 0.80
forwardMethod · 0.80
create_padded_xMethod · 0.80

Calls 1

selectFunction · 0.85

Tested by 2

test_select_from_intMethod · 0.64