第一章 pytorch(入门)
PyTorch是一个基于 Python 的科学计算包,它主要有两个用途:
1.类似Numpy但是能利用GPU加速
2.一个非常灵活和快速的用于深度学习的研究平台
numpy与pytorch对比与转换
pytorch 中常用的函数
1.tensor.mean()
tensor.mean()
dim=0表示计算第一维度,等于1等于计算第二维度,如果不指定,则计算张量中所有的数的均值
2.torch.argmax()
我们想要求每一行最大的列标号,我们就要指定dim=1,表示我们不要列了,保留行的size就可以了。
假如我们想求每一列的最大行标,就可以指定dim=0,表示我们不要行了。
例题:
张量学习
Tensors与Numpy中的 ndarrays类似,但是在PyTorch中 Tensors 可以使用GPU进行计算.
1.torch.empty(5,3) 未初始化