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

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

  • str:表示读取单个条目。
  • List[str]:表示批量读取,顺序与返回/填充顺序一致。

tensors

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

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

**kwargs

-

  • allow_missing (bool):允许缺失项并返回占位/跳过(默认 False)。
  • verify_shape (bool):是否强校验形状/dtype(默认 True)。
  • shard (Any):分片策略或标识。
  • 其他实现相关参数(如 pin_memory、非阻塞等)。

返回值

List[bool]:表示从池化后端中获取当前的keys对应tensors并原位填充成功/失败的bool值,每个bool对应每个key的结果,若keys为str,返回值仍为List[bool],其长度为1。