一、torch.cat()函数 熟悉C字符串的同学们应该都用过strcat（）函数，这个函数在C/C++程序中用于连接2个C字符串。在pytorch中，同样有这样的函数，那就是torch.cat()函数. 先上源码定义：torch.cat(tensors,dim=0,out=None ...

chunk方法可以对张量分块，返回一个张量列表： torch.chunk(tensor, chunks, dim=0) → List of Tensors Splits a tensor into a specific number of chunks. Last chunk ...

NLLLoss 和 CrossEntropyLoss 在图片单标签分类时，输入m张图片，输出一个m*N的Tensor，其中N是分类个数。比如输入3张图片，分3类，最后的输出是一个3*3的Tens ...

torch.clamp(input, min, max, out=None) → Tensor 将输入input张量每个元素的夹紧到区间 [min,max][min,max]，并返回结果到一个新张量。 操作定义如下： | min, if x_i < miny_i ...

一、_, predicted = torch.max(outputs.data, 1) 那么，这里的 下划线_ 表示什么意思？ 首先，torch.max()这个函数返回的是两个值，第一个值是具体的value（我们用下划线_表示），第二个值是value所在的index（也就是predicted ...

这个函数是用来求tensor中某个dim的前k大或者前k小的值以及对应的index。 用法 比如，三行两列，3个样本，2个类别。 ...

转载请注明出处： http://www.cnblogs.com/darkknightzh/p/6221633.html torch中的apply函数通过可以不断遍历model的各个模块。实际上其使用的是深度优先算法。 其具体代码如下所示（代码见torch/install/share ...

1、torch.cat()：是将两个张量（tensor）拼接在一起。 2、tensor.expand_as()：把一个tensor变成和函数括号内一样形状的tensor 3、tensor.narrow(dim,index,number)：dim-取行/列;index-从索引 ...