MCPcopy Index your code
hub / github.com/NVIDIA/TensorRT-LLM / shape

Method shape

tensorrt_llm/functional.py:270–274  ·  view source on GitHub ↗

The shape of the tensor.

(self)

Source from the content-addressed store, hash-verified

268
269 @property
270 def shape(self):
271 '''
272 The shape of the tensor.
273 '''
274 return self.size()
275
276 @shape.setter
277 def shape(self, shape):

Callers 5

test_slice_implicitMethod · 0.45
test_alibi_biasesMethod · 0.45
test_expand_implicitMethod · 0.45
test_gatherMethod · 0.45
run_inferenceFunction · 0.45

Calls 1

sizeMethod · 0.95

Tested by 4

test_slice_implicitMethod · 0.36
test_alibi_biasesMethod · 0.36
test_expand_implicitMethod · 0.36
test_gatherMethod · 0.36