HashEmbeddingBagCollection

tables

List[HashEmbeddingBagConfig]

必选

稀疏表配置文件列表。

参数范围参考HashEmbeddingBagConfig。

is_weighted

bool

可选

仅支持默认值为False。

device

str或者torch.device

必选

稀疏表的设备。

• 如果为str取值范围：
  • "npu"：npu设备。
  • "meta"：meta设备。
  • "cpu"：cpu设备。

    cpu设备不支持分布式表，只支持单机表。

• 如果为torch.device取值范围：
  • torch.device("npu")：npu设备。
  • torch.device("meta")：meta设备。
  • torch.device("cpu")：cpu设备。

    cpu设备不支持分布式表，只支持单机表。