pytorch學習（一）numpy與torch對比

torch類似於numpy，可以理解為pytorch框架下的numpy，因此很多numpy的運算適用於torch

1、numpy可以轉化為torch，torch也可以轉化為numpy

import torch

import numpy as np
np_data = np.arange(6).reshape((2, 3))
torch_data = torch.from_numpy(np_data)
torch2array = torch.numpy(torch_data)
print(
'\nnp_data', np_data,
'\ntorch_data', torch_data,
'\ntorch2array', torch2array,
)

np_data [[0, 1, 2],

[3, 4, 5]]
torch_data 
0 1 2
3 4 5
torch2array [[0, 1, 2],
[3, 4, 5]]

運算可以參考官網的英文文件

1）絕對值 # abs

torch.abs()

torch.mean()

矩陣相乘

# numpy的矩陣相乘為

np.matmul(data, data)
data.dot(data)
# torch的矩陣相乘為
torch.mm(tensor, tensor)
tensor.dot(tensor) # 則對矩陣元素對應相乘後相加

Pytorch學習筆記之通過numpy實現線性擬合

通過使用numpy庫編寫簡單的gradient descent資料位於附件之中 import torch from torch import autograd import numpy as np import matplotlib.pyplot as plt torch關於求導的簡單運用 x to...

numpy學習（一） numpy基礎

此文為學習 理解numpy，numpy簡單入門教程整理的學習筆記 numpy是乙個功能強大的python庫，主要用於對多維陣列執行計算。numpy這個詞 於兩個單詞 numerical和python。a np.array 0,1,2,3,4 a np.array 0,1,2,3,4 1,2,3,4,...

Pytorch之Tensor和Numpy之間的轉換

最重要的區別t.tensor和t.tensor 不論輸入的型別是什麼，t.tensor 都會進行資料拷貝，不會共享記憶體 t.tensor 與numpy共享記憶體，但當numpy的資料型別和tensor的型別不一樣的時候，資料會被複製，不會共享記憶體。可使用t.from numpy 或者t.deta...