首页pytorch张量转数组

pytorch张量转数组

时间: 2023-10-12 17:09:19 浏览: 87

pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换

1, 创建pytorch 的Tensor张量： torch.rand((3,224,224)) #创建随机值的三维张量，大小为（3,224,224） torch.Tensor([3,2]) #创建张量，[3,2] 2, cpu上的tensor和GPU即pytorch创建的tensor的相互转化 b = a.cpu() # GPU → CPU a = b.cuda() #CPU → GPU 3, tensor和numpy的转化 b = a.numpy() # tensor转化为 numpy数组 a = b.from_numpy() # numpy数组转化为tensor 4, torch的

可以使用PyTorch张量的`.numpy()`方法将其转换为NumPy数组。例如： ```python import torch # 创建一个3x3的张量 tensor = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) # 将张量转换为数组 array = tensor.numpy() print(array) # 输出: # array([[1, 2, 3], # [4, 5, 6], # [7, 8, 9]]) ``` 注意，将PyTorch张量转换为NumPy数组会共享底层数据。这意味着，如果修改NumPy数组，则原始张量也会被修改，反之亦然。

阅读全文