def put(self,keys: Union[str, List[str]],tensors: Union[torch.Tensor, List[torch.Tensor], List[List[torch.Tensor]]],**kwargs) -> List[bool]
函数功能
写入或覆盖一个或一组键对应的张量数据。
函数原型
def put(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