monchin的博客

学习pytorch中，看到文档里关于torch.nonzero的介绍和举例，一维的那个例子还好理解，二维的就不是很理解了，不明白为什么会出现两个[0,1,2,3]，于是稍微研究了一下，搞明白了是怎么回事。先看文档的介绍：“返回一个包含输入 input 中非零元素索引的张量.输出张量中的每行包含 input 中非零元素的索引.如果输入 input 有 n 维,则输出的索引张量 out 的 size ...

torch.multinomial(input, num_samples,replacement=False, out=None) → LongTensor作用是对input的每一行做n_samples次取值，输出的张量是每一次取值时input张量对应行的下标。输入是一个input张量，一个取样数量，和一个布尔值replacement。input张量可以看成一个权重张量，每一个元素代表其在该行中的...