小编jwe*_*324的帖子

Pytorch 中是否有一种方法可以以可以反向传播的方式计算唯一值的数量?

给定以下张量(这是网络的结果 [注意 grad_fn]):

tensor([121., 241., 125.,   1., 108., 238., 125., 121.,  13., 117., 121., 229.,
        161.,  13.,   0., 202., 161., 121., 121.,   0., 121., 121., 242., 125.],
       grad_fn=<MvBackward>)
Run Code Online (Sandbox Code Playgroud)

我们将定义为:

tensor([121., 241., 125.,   1., 108., 238., 125., 121.,  13., 117., 121., 229.,
        161.,  13.,   0., 202., 161., 121., 121.,   0., 121., 121., 242., 125.],
       grad_fn=<MvBackward>)
Run Code Online (Sandbox Code Playgroud)

我想定义一个计算每个值出现次数的操作,该操作将输出以下张量:

tensor([2, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, …
Run Code Online (Sandbox Code Playgroud)

counting backpropagation pytorch

5
推荐指数
2
解决办法
7182
查看次数

标签 统计

backpropagation ×1

counting ×1

pytorch ×1