def get(self,keys: Union[str, List[str]],tensors: Union[torch.Tensor, List[torch.Tensor], List[List[torch.Tensor]]],**kwargs) -> List[bool]
函数功能
读取一个或一组键的张量数据;通常支持**原位填充**到给定的 `tensors`(作为输出缓冲区)。
函数原型
def get(self,keys: Union[str, List[str]],tensors: Union[torch.Tensor, List[torch.Tensor], List[List[torch.Tensor]]],**kwargs ) -> List[bool]:
参数说明
参数名 |
参数类型 |
说明 |
|---|---|---|
keys |
Union[str, List[str]] |
|
tensors |
Union[Tensor, List[Tensor], List[List[Tensor]]] |
|
**kwargs |
- |
|
返回值
List[bool]:表示从池化后端中获取当前的keys对应tensors并原位填充成功/失败的bool值,每个bool对应每个key的结果,若keys为str,返回值仍为List[bool],其长度为1。
父主题: class MemPool