我知道 CPU、GPU 和 TPU。但是,这是我第一次从PyTorch 关于 MODULE 的文档中了解 XPU 。
xpu(device=None)
Moves all model parameters and buffers to the XPU.
This also makes associated parameters and buffers different objects. So it should be called before constructing optimizer if the module will live on XPU while being optimized.
Note: This method modifies the module in-place.
Parameters
device (int, optional) – if specified, all parameters will be copied to that device
Returns
self
Return type
Module
CPU代表中央处理器。GPU 代表图形处理单元。TPU 代表张量处理单元。
我们大多数人都知道,这些处理单元在人工智能的一些计算密集型领域(包括深度学习)的研究中非常有用。所以,我想知道它是否XPU在人工智能研究中也有用,因为它在 PyTorch 中使用。
从上下文来看,我可以说它PU代表processing unit. 但我不知道是什么X。
XPU 的完整形式是什么?我在哪里可以详细了解 XPU?