.nu​​mpy() 函数有什么作用？

Question

我尝试在线搜索文档，但找不到任何可以给我答案的内容。.numpy()函数有什么作用？给出的示例代码是：

y_true = []
for X_batch, y_batch in mnist_test:
    y_true.append(y_batch.numpy()[0].tolist())

Answer 1

在 Pytorch 和 Tensorflow 中，该.numpy()方法都非常简单。它将tensor对象转换为numpy.ndarray对象。这隐含地意味着转换后的张量现在将在 CPU 上进行处理。