1 numpy与CUDA之间的转换1.tensor张量与numpy相互转换tensor >numpy import ... import torch tensor = torch.randn(2, 2) print(tensor.type()) ...
確定! 回上一頁