API接口 |
说明 |
|---|---|
torch_npu.npu.get_rng_state |
Torch_npu提供随机数相关的部分接口，具体可参考torch.cuda。 |
torch_npu.npu.set_rng_state |
|
torch_npu.npu.get_rng_state_all |
|
torch_npu.npu.set_rng_state_all |
|
torch_npu.npu.manual_seed |
|
torch_npu.npu.manual_seed_all |
|
torch_npu.npu.seed |
|
torch_npu.npu.seed_all |
|
torch_npu.npu.initial_seed |