evox.algorithms.so.pso_variants.utils

模块内容

函数

min_by	找到具有最小键的值。
random_select_from_mask	随机从 1D 掩码中使用均匀噪声随机选择 s 个元素。

API

evox.algorithms.so.pso_variants.utils.min_by(values: List[torch.Tensor], keys: List[torch.Tensor]) → Tuple[torch.Tensor, torch.Tensor]

找到具有最小键的值。

参数:

• values -- 一个张量或一个列表的张量

• keys -- 一个张量或与值对应的张量列表。

返回:

具有最小键的值及其对应的键。

evox.algorithms.so.pso_variants.utils.random_select_from_mask(mask: torch.Tensor, count: int, dim: int = -1) → torch.Tensor

随机从 1D 掩码中使用均匀噪声随机选择 s 个元素。

参数:

• key -- 用于随机种子的张量。

• mask -- 一个形状为 (N,) 的张量，包含 {0, 1}。

• s -- 要选择的元素数量。

返回:

一个新的掩码张量，其中恰好有 s 个元素设置为 1。