Strategy management

torch.multiprocessing.get_all_sharing_strategies()

返回一组由当前系统所支持的共享策略

torch.multiprocessing.get_sharing_strategy()

返回当前策略共享CPU中的张量。

torch.multiprocessing.set_sharing_strategy(new_strategy)

设置共享CPU张量的策略

参数: new_strategy(str)-被选中策略的名字。应当是get_all_sharing_strategies()中值当中的一个。

results matching ""

    No results matching ""