欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

RuntimeError: Integer division of tensors using div or / is no longer supported, and in a future rel

程序员文章站 2022-03-04 14:10:21
...

RuntimeError: Integer division of tensors using div or / is no longer supported, and in a future release div will perform true division as in Python 3. Use true_divide or floor_divide (// in Python) instead.

对于tensor A和整数n之间的除法:

result = A / n # not supported in torch 1.6.0

solution

result = torch.true_divide(A, n)