目录

快捷键

torch.mtia.get_rng_state¶

torch.mtia.get_rng_state(device='mtia')[source][source]¶

返回随机数生成器状态作为 ByteTensor。

参数:: device (torch.device 或 int，可选) – 返回 RNG 状态的设备。默认： 'mtia' （即， torch.device('mtia') ，当前 mtia 设备）。
返回类型:: 张量

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源