pytorch
文章平均质量分 70
_yuki_
一路打怪升级,直至见到大魔王
展开
-
Pytorch/Paddle topk 与 Numpy argpartition 函数应用
torch.topk 、paddle.topk、numpy.argpartition 函数应用原创 2022-12-21 15:28:17 · 746 阅读 · 1 评论 -
Pytorch & Paddle tensordot 函数应用
torch.tensordot & paddle.tensordot原创 2022-12-20 17:45:42 · 595 阅读 · 0 评论 -
pytorch scatter_ 函数应用
scatter()和scatter_()的作用是一样的,只不过 scatter() 不会直接修改原来的 Tensor,而 scatter_() 会PyTorch 中,一般函数加下划线代表直接在原来的 Tensor 上修改 (小知识)scatter(dim, index, src) 的参数有 3 个dim:沿着哪个维度进行索引 index:用来 scatter 的元素索引 src:用来 scatter 的源元素,可以是一个标量或一个张量这个 scatter可以理解成放置元素或...原创 2020-08-20 16:03:06 · 639 阅读 · 0 评论 -
pytorch torch.nn.functional.normalize 应用
torch.nn.functional.normalizetorch.nn.functional.normalize(input, p=2, dim=1, eps=1e-12, out=None)功能:将某一个维度除以那个维度对应的范数(默认是2范数)。原创 2020-08-07 16:32:29 · 1585 阅读 · 1 评论