是前面1234的重新复习,还是一些基础语法知识
主要参考:
PyTorch 源码解读之 nn.Module:核心网络模块接口详解 - 知乎
一、张量Tensor
各种数据类型都被成为张量,阶指的是高维数组中数的个数,即形状
传入列表 数组 都可以转为tensor
1、tensor的创建:
import torch
import numpy as np
t1=torch.Tensor([1,2,3])
#将array类型转换为torch类型
array1=np.arange(12).reshape(3,4)
array1
torch.Tensor(array1)
#用torch的api创建tensor
torch.empty(3,4)#空数组(里面有随机垃圾数据填充)
torch.zeros(3,4)
torch.ones(3,4)
torch.rand(3,4)#三行四列01区间的浮点型tensor
torch.randint(3,10,(2,2))#创建3-10区间的,形状为两行两列的tensor
#或者写详细一些torch.randint(low=3,high=10,size=(2,2))
torch.randn(3,4)#创建均值为0方差为1的自然分布
**关于array与arange
#一个参数 默认起点0,步长为1 输出:[0 1 2]
a = np.arange(3)
#两个参数 默认步长为1 输出[3 4 5 6 7