Pytorch框架

Purple Coder2026-02-18 12:28

张量 = 多维数组 / 多维数据块

CUDA就是英伟达为自家 GPU 量身定做的「沟通桥梁」------ 它提供了一套编程接口（API）和工具，让程序员能直接指挥 GPU 去做这些并行计算

Pytorch（深度学习框架）

张量

【黑马程序员AI大模型《神经网络与深度学习》全套视频课程，涵盖Pytorch深度学习框架、BP神经网络、CNN图像分类算法及RNN文本生成算法】 https://www.bilibili.com/video/BV1c5yrBcEEX/?p=8\&share_source=copy_web\&vd_source=0caeacd6c3217ba41c56ea47a129e168

（用于表示图片）

标量张量（只有一个值）

书写

【黑马程序员AI大模型《神经网络与深度学习》全套视频课程，涵盖Pytorch深度学习框架、BP神经网络、CNN图像分类算法及RNN文本生成算法】 https://www.bilibili.com/video/BV1c5yrBcEEX/?p=9\&share_source=copy_web\&vd_source=0caeacd6c3217ba41c56ea47a129e168

创建

python 复制代码

torch.tensor
torch.Tensor
torch.IntTensor、torch.FloatTensor、torch.DoubleTensor

#导包
import torch

def dm01():
    t1 = torch.tensor(10)
    print(f't1: {t1}')



if _name_ = '_main_':
    dm01()