pytorch
Evan_Tech
这个作者很懒,什么都没留下…
展开
-
PyTorch Tensor和numpy数组之间的转化
从Tensor到numpytensor = torch.Tensor([0,1]) # 如果参数不是一个list,而是一个int,将会返回该int长的内容未定义的Tensortensor_numpy = tensor.numpy()从numpy到Tensorarray = np.random.rand(3)torch.from_numpy(array)...原创 2020-02-08 16:04:46 · 1651 阅读 · 0 评论 -
pytorch检测cuda可用性
torch.cuda.is_available()原创 2020-02-07 21:05:12 · 5919 阅读 · 0 评论