随笔分类 - Pytorch
摘要：Cat对数据沿着某一维度进行拼接。cat后数据的总维数不变.比如下面代码对两个2维tensor（分别为2*3,1*3）进行拼接，拼接完后变为3*3还是2维的tensor。import torch torch.manual_seed(1) x = torch.randn(2,3) y = torch.randn(1,3) print(x,y)结果：0.6614 0.2669 0.0617 0.621...
阅读全文
摘要：permute(dims)，常用的维度转换方法将tensor的维度换位 参数：dim(int)---换位顺序>>>x = torch.randn(2,3,5)>>>x.size()torch.size([2,3,5])>>>x.permute(2,0,1).size()torch.size([5,2,3])contiguous()contiguous：view只能用在contiguous...
阅读全文
该文被密码保护。
该文被密码保护。
摘要：一、Pytorch介绍Pytorch 是Torch在Python上的衍生物和Tensorflow相比：Pytorch建立的神经网络是动态的，而Tensorflow建立的神经网络是静态的Tensorflow的高度工业化，它的底层代码很难看懂官网：http://pytorch.org/Pytorch主要有两个模块：一个是torch,一个是torchvision,torch是主模块，用来搭建神经网络。t...
阅读全文
