Pytorch
c-minus
对机器学习、深度学习等领域方面的研究颇有兴趣
展开
-
Pytorch系列(1):torch.gather()
torch.gather作用:收集输入的特定维度指定位置的数值参数:input(tensor): 待操作数。不妨设其维度为(x1, x2, …, xn)dim(int): 待操作的维度。index(LongTensor): 如何对input进行操作。其维度有限定,例如当dim=i时,index的维度为(x1, x2, …y, …,xn),...原创 2019-05-16 11:43:46 · 49503 阅读 · 9 评论 -
PyTorch系列(2):tensor操作大全
torch.is_tensor(obj):判断obj是否是tensor类型import torchx = torch.randn(1,2)y = x.numpy()print(torch.is_tensor(x))print(torch.is_tensor(y))TrueFalsetorch.is_storage(obj): 判断obj是否是pytorch storage 对象...原创 2019-05-16 14:45:15 · 1514 阅读 · 0 评论 -
PyTorch系列(3):tensor生成操作大全
下面参数中的省略号代表dtype、requires_grad等常规参数。一、torch.tensor(data, dtype=None, device=None, requires_grad=False, pin_memory=False) → Tensor1、用data创建一个tensor2、torch.tensor会从data出复制数据,意味着生成的tensor不会和data共享内存(...原创 2019-05-16 16:44:03 · 12226 阅读 · 0 评论