import torch
import numpy as np
#abs 绝对值转换
data = [-1,-2,-3,4]
tensor = torch.FloatTensor(data) #32bit浮点数
print(
'\nabs',
'\nnumpy',np.abs(data),
'\ntorch',torch.abs(tensor),
)
print(
'\nsin',
'\nnumpy',np.sin(data),
'\ntorch',torch.sin(tensor),
)
print(
'\nmean',
'\nnumpy',np.mean(data),
'\ntorch',torch.mean(tensor),
)
data = [[1,2], [3,4]]
tensor = torch.FloatTensor(data) # 转换成32位浮点 tensor
print(
'\nnumpy',np.matmul(data,data),
'\ntorch',torch.mm(tensor,tensor),
)
data = np.array([1,2,3,4])
tensor = torch.FloatTensor(data)
print(
'\nnumpy',data.dot(data),
'\ntorch',tensor.dot(tensor), # torch 会转换成 [1,2,3,4].dot([1,2,3,4) = 30.0
)
#dot()的使用 https://docs.scipy.org/doc/numpy/reference/generated/numpy.dot.html
#numpy a.dot(b) = np.dot(a,b) np.dot(a,b) != np.dot(b,a)
#一维数组返回内积
data = [1,2,3,4]
data = np.array(data)
print(
'\nnumpy',data.dot(data),
)
#二维数组(矩阵)返回矩阵积
data = [[1,2], [3,4]]
data = np.array(data)
print(
'\nnumpy',data.dot(data),
)
#pytorch
#仅支持一维数据
data = [1,2,3,4]
data = np.array(data)
tensor = torch.FloatTensor(data)
print(
'\ntorch',tensor.dot(tensor),
)
#二维会报错
# data = [[1,2], [3,4]]
# data = np.array(data)
# tensor = torch.FloatTensor(data)
# print(
# '\ntorch',tensor.dot(tensor),
# )
莫烦pytorch学习01:pytorch与numpy之间的转换
最新推荐文章于 2024-06-23 09:16:45 发布