pytorch 多个 tensor 按位取与 / 按位取或

Kkkkaii

于 2022-12-24 21:00:39 发布

阅读量959

点赞数

分类专栏： PyTorch 文章标签： pytorch 深度学习 python 经验分享

本文链接：https://blog.csdn.net/hugvgngj/article/details/128431405

版权

PyTorch 专栏收录该内容

12 篇文章

订阅专栏

pytorch 多个 tensor 按位取与 / 按位取或

问题

pytorch 中 torch.logical_or 只接受两个 input，但假如有一系列 inputs 都想按位取与该怎么做

解法

也还是参考 MEAN 的 code
CONDITIONAL ANTIBODY DESIGN AS 3D EQUIVARIANT GRAPH TRANSLATION

def sequential_or(*tensors):
    res = tensors[0]
    for mat in tensors[1:]:
        res = torch.logical_or(res, mat)
    return res
    ```

其实实现上很粗暴，还是一个个按位 `torch.logical_or`。就是说代码看起来更整洁了，这样写也确实挺漂亮的，但是没有做到高效实现。
这里就是贴出来到时候可以直接抄

调用的时候就是 `sequential_or(tensor1, tensor2, tensor3, ...)` tensor 数量可变