pytorch 两个tensor的交集

  def tensor_intersect(self, t1, t2):
        # t1=t1.cuda()
        # t2=t2.cuda()
        indices = torch.zeros_like(t1, dtype = torch.bool, device = 'cuda')
        for elem in t2:
            indices = indices | (t1 == elem)  
            intersection = t1[indices]  
        return intersection

相关推荐

  1. pytorch tensor交集

    2024-01-03 17:18:07       52 阅读
  2. 数组交集

    2024-01-03 17:18:07       38 阅读
  3. 数组交集

    2024-01-03 17:18:07       39 阅读
  4. 349. 数组交集

    2024-01-03 17:18:07       29 阅读
  5. 【LeeCode】350.数组交集 II

    2024-01-03 17:18:07       55 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-01-03 17:18:07       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-01-03 17:18:07       100 阅读
  3. 在Django里面运行非项目文件

    2024-01-03 17:18:07       82 阅读
  4. Python语言-面向对象

    2024-01-03 17:18:07       91 阅读

热门阅读

  1. 通过 nvm 管理 Node 版本

    2024-01-03 17:18:07       65 阅读
  2. Unity 打包前,通过代码对 AndroidManifest 增删改查

    2024-01-03 17:18:07       63 阅读
  3. [嵌入式专栏](Qt - GUI框架)

    2024-01-03 17:18:07       60 阅读
  4. 机器学习的方法

    2024-01-03 17:18:07       55 阅读
  5. 机器学习模型评价指标

    2024-01-03 17:18:07       57 阅读
  6. 【Vue】响应式中的渲染 watcher

    2024-01-03 17:18:07       44 阅读
  7. (每天进步一点点-001)bash命令学习-case语法

    2024-01-03 17:18:07       59 阅读
  8. 关于Spring @Transactional事务传播机制详解

    2024-01-03 17:18:07       46 阅读