Update pytorch_example.py

This commit is contained in:
guanjihuan 2023-11-20 20:42:50 +08:00
parent 801768ab64
commit f28a409307

View File

@ -59,8 +59,8 @@ print(torch_tensor_2) # Torch张量不会被修改
# torch_tensor.numpy() 和 np.array() 的关系【前者共享内存,后者不共享内存】 # .numpy() 和 np.array() 的关系【前者共享内存,后者不共享内存】
print('\n\n\n\ntorch_tensor.numpy() 和 np.array() 的关系:') print('\n\n\n\n.numpy() 和 np.array() 的关系:')
torch_tensor = torch.tensor([1, 2, 3]) torch_tensor = torch.tensor([1, 2, 3])
numpy_array = torch_tensor.numpy() # 共享内存 numpy_array = torch_tensor.numpy() # 共享内存