Pytorch
Fujh-nk
这个作者很懒,什么都没留下…
展开
-
Pytorch-Tensor(张量)
Tensor的基本数值类型 Tensor(张量)的创建 1. 直接创建 torch.ones(int[, int[, int[, ...]]]) -> 创建一个1填充的矩阵 torch.zeros(int[, int[, int[, ...]]], dtype=type) -> 创建一个0填充的矩阵,并指定数据类型 torch.empty(int[, int[, i...原创 2019-08-18 23:30:08 · 779 阅读 · 0 评论 -
Pytorch-squeeze()和unsqueeze()
1.squeeze(dim) 其中 dim 从 0 开始计数。 函数效果为将参数对应的维去掉,需满足该维中只有一个维度小于该维的维, 举个例子就是 view((1, 2, 3)) 中,只有 1 那个维才可以被 squeeze() 降维。 初始化一个三维的 Tensor : # coding=utf-8 import torch x = torch.arange(6).view((1...原创 2019-08-27 09:09:09 · 341 阅读 · 0 评论