torch.Tensor.index_add_函数,pytorch中的tf.unsorted_segment_sum
ref:https://pytorch.org/docs/1.4.0/tensors.html?highlight=index_add_#torch.Tensor.index_add_https://blog.csdn.net/weixin_44289071/article/details/103882658torch.Tensor.index_add_能实现指定行或列的内容相加的功能,类似于tensorflow中tf.unsorted_segment_sum函数,可以用在比如实例分割中进行特征聚
原创
2021-05-10 11:48:59 ·
5695 阅读 ·
2 评论