如何分支TorchScript版本？

问题描述

我有一个TorchScript，并且想根据炬的版本进行分支：

import torch

def test_div():
    a = torch.tensor([10,20])
    b = torch.tensor([5,5])
    return torch.div(a,b) \
      if torch.__version__ < '1.6.0' else torch.floor_divide(a,b)

print(torch.__version__)
torch.jit.script(test_div)

但是，该示例不适用于没有floor_divide的旧版本：

1.2.0
RuntimeError:
object has no attribute floor_divide

如何隐藏旧版本（floor_divide）中的__version__ < '1.6.0'呼叫？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）