USTC_SC的博客

文章目录1.指定运行设备2.稀疏矩阵3.四则运算4.矩阵运算5.幂、开方、对数运算6.in-place操作7.广播机制8.取整、取余、比较、top-k9.判定异常值10.三角函数11.统计函数1.指定运行设备pytorch可以指定数据运行在cpu或者gpu上，以下为例。import torch#dev = torch.device("cpu")dev = torch.device("cuda")a = torch.tensor([2, 2], dtype=to

文章目录创建Tensor1.基本创建2.创建特殊值3.创建随机值和特定序列Tensor4.Numpy转Tensor5.Tensor转Numpy创建Tensorpytorch学习的最基础的学习就从创建张量开始。1.基本创建import torch#根据值创建a = torch.Tensor([[1, 2],[3, 4]])print(a)#根据形状创建，随机初始b = torch.Tensor(2, 2)print(b)d = torch.tensor(((1, 2), (3, 4