不能直接用list.numpy()
但 torch.Tensor 直接转 numpy.ndarray:
'''
inputs[batch] -> <class 'torch.Tensor'>
'''
mat = inputs[batch].numpy()
'''
mat -> <class 'numpy.ndarray'>
'''
本文介绍了如何在PyTorch中通过简便方式将Tensor转换为numpy.ndarray类型,这对于数据预处理和与现有库交互非常实用。
不能直接用list.numpy()
但 torch.Tensor 直接转 numpy.ndarray:
'''
inputs[batch] -> <class 'torch.Tensor'>
'''
mat = inputs[batch].numpy()
'''
mat -> <class 'numpy.ndarray'>
'''
您可能感兴趣的与本文相关的镜像
PyTorch 2.7
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
5749
3288

被折叠的 条评论
为什么被折叠?