pytorch使用scatter构建onehot向量,scatter不仅可以用来构建one-hot向量,也可以用来对矩阵进行修改。
scatter表示按照你的要求修改tensor某些位置的元素值。scatter_表示直接在原来的tensor上修改,而scatter是会将原来的tensor复制一份然后再修改。
dim
:表示按照哪一个维度修改,dim=1
表示按照行修改
index
:表示被修改元素位于行的第几列。
value
:表示将这个位置的值替换为多少
由dim和index
两个参数就可确定要修改元素的位置。
pytorch 构建one-hot向量
最新推荐文章于 2022-12-28 10:59:01 发布