不能直接用list.numpy()
但 torch.Tensor 直接转 numpy.ndarray:
'''
inputs[batch] -> <class 'torch.Tensor'>
'''
mat = inputs[batch].numpy()
'''
mat -> <class 'numpy.ndarray'>
'''
不能直接用list.numpy()
但 torch.Tensor 直接转 numpy.ndarray:
'''
inputs[batch] -> <class 'torch.Tensor'>
'''
mat = inputs[batch].numpy()
'''
mat -> <class 'numpy.ndarray'>
'''