产品 |
是否支持 |
---|---|
√ |
|
x |
|
x |
|
x |
|
x |
获取cache tensor。
1 | get_cache_tensors(cache: KvCache, tensor_index: int = 0) -> List[Tensor] |
参数名称 |
数据类型 |
取值说明 |
---|---|---|
cache |
要获取的tensor所在的KvCache。 |
|
tensor_index |
int |
要获取的tensor在Cache中的index。 |
1 | tensors = kv_cache_manager.get_cache_tensors(kv_cache, 0) |
本接口不支持并发调用。