torch.mtia
MTIA后端在外独立实现，此处仅定义接口。
此包提供了在Python中访问MTIA后端的接口。
|
用于选择指定流的上下文管理器。 |
|
|
返回当前选定设备的索引。 |
|
|
返回给定设备当前选择的 |
|
|
返回指定设备的默认 |
|
|
返回可用的MTIA设备的数量。 |
|
|
若MTIA设备可用，返回true |
|
|
返回 PyTorch 的 MTIA 状态是否已经初始化。 |
|
|
为指定设备返回MTIA内存分配器统计信息的字典。 |
|
|
设置当前设备。 |
|
|
设置当前流。这是一个用于设置流的包装 API。 |
|
|
使用上下文管理器 StreamContext 来选择指定的流。 |
|
|
等待MTIA设备上所有流中所有作业的完成。 |
|
|
用于更改选定设备的上下文管理器。 |
|
|
设定随机数生成器的状态。 |
|
|
返回随机数生成器的状态，形式为ByteTensor。 |
|