PyTorch中的scatter()与scatter_()函数

仔细看了PyTorch的文档才搞懂这两个函数
PyTorch: torch.Tensor.scatter
另一个文档: pytorch_scatter

scatter()

这个是scatter_()的out-of-place版本,即函数修改的不是原tensor
在vscode里面看这个函数有两种:

	scatter(self: Tensor, dim: _int, index: Tensor, src: Tensor) -> Tensor
	# param dim:_int 是让输入第一个参数(?
	scatter(self: Tensor, dim: _int, index: Tensor, value: Number) -> Tensor
	# param dim:_int

两个的区别在于最后一个参数,可以用Tensor作为src进行填充,也可以指定某个数值作为填充

scatter_()

一句话总结:在一个tensor的基础上,在dim维上,根据index选择src的一些数填到原始的那个tensor里。
对于scatter,向原始tensor填数得到另外一个tensor

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值