我尝试在线搜索文档,但找不到任何可以给我答案的内容。什么.numpy()
功能呢?给出的示例代码是:
y_true = []
for X_batch, y_batch in mnist_test:
y_true.append(y_batch.numpy()[0].tolist())
最佳答案
在 Pytorch 和 Tensorflow 中,.numpy()
方法非常简单。它转换 tensor
对象变成 numpy.ndarray
目的。这隐含地意味着转换后的张量现在将在 CPU 上进行处理。
关于numpy - .numpy() 函数有什么作用?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63968868/