PyTorch(1)：安装和基本函数介绍

最新推荐文章于 2025-06-09 20:27:05 发布

闲人_Yty

最新推荐文章于 2025-06-09 20:27:05 发布

阅读量221

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签： pytorch 神经网络

本文链接：https://blog.youkuaiyun.com/qq_40326280/article/details/112297711

PyTorch(1)：安装和基本函数介绍

PyTorch学习笔记。
内容包含神经网络的介绍，PyTorch的安装、基本函数等内容。

PyTorch安装

PyTorch官网：https://pytorch.org/ （2020/6/22）

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

Torch vs Numpy

torch ≈ 神经网络中的numpy

np_data = np.arange(6).reshape((2, 3))
torch_data = torch.from_numpy(np_data) # np 转化为 torch
tensor2array = torch_data.numpy()      # torch 转化为 np

# abs
data = [-1, -2, 1, 2]
tensor = torch.FloatTensor(data)  # 32-bit floating point
print(
    '\nabs',
    '\nnumpy: ', np.abs(data),          # [1 2 1 2]
    '\ntorch: ', torch.abs(tensor)      # [1 2 1 2]
)

# sin
print(
    '\nsin',
    '\nnumpy: ', np.sin(data),      # [-0.84147098 -0.90929743  0.84147098  0.90929743]
    '\ntorch: ', torch.sin(tensor)  # [-0.8415 -0.9093  0.8415  0.9093]
)

# mean
print(
    '\nmean',
    '\nnumpy: ', np.mean(data)