昇腾社区首页
中文
注册
开发者
下载

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]]

  • str:表示写入单个条目。
  • List[str]:表示批量写入,长度需与“tensors”一致。

tensors

Union[Tensor, List[Tensor], List[List[Tensor]]]

  • Tensor:单key对应单个Tensor。
  • List[Tensor]:多key对应等长的List[Tensor]。
  • List[List[Tensor]]:二维列表用于多分片/多段写入。

**kwargs

-

  • overwrite (bool):是否允许覆盖已存在的key(默认由实现决定)。
  • ttl (float|int): 过期时间/秒(若后端支持)。
  • sync (bool): 是否同步落盘/刷写。
  • shard (Any): 分片策略或标识。
  • 其他实现相关参数。

返回值

  • List[bool]:表示向池化后端中写入当前的keys,对应tensors成功/失败的bool值,每个bool对应每个key的结果,若keys为str,返回值仍为List[bool],其长度为1。