您当前的位置: 首页 > 

IT之一小佬

暂无认证

  • 0浏览

    0关注

    1192博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

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

IT之一小佬 发布时间:2021-03-18 22:16:03 ,浏览量:0

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. 示例代码:
from torchvision import transforms
import numpy as np

data = np.random.randint(0, 255, size=12)
img = data.reshape(2, 2, 3)
print(img.shape)
img_tensor = transforms.ToTensor()(img)  # 转换成tensor
print(img_tensor)
print(img_tensor.shape)
print("*" * 20)
norm_img = transforms.Normalize((10, 10, 10), (1, 1, 1))(img_tensor)  # 进行规范化处理
print(norm_img)

运行效果:

原因:

pytorch1.5.0是可以的,但升级到1.6.0,发现tensor和int之间的除法不能直接用'/'。

解决方案:

对数据进行规范化处理

示例代码:

from torchvision import transforms
import numpy as np

data = np.random.randint(0, 255, size=12)
img = data.reshape(2, 2, 3)
print(img.shape)
img_tensor = transforms.ToTensor()(img)  # 转换成tensor
print(img_tensor)
print(img_tensor.shape)
print("*" * 20)
img_tensor = img_tensor.float()  # 加上这一行
norm_img = transforms.Normalize((10, 10, 10), (1, 1, 1))(img_tensor)  # 进行规范化处理
print(norm_img)

运行结果:

关注
打赏
1665675218
查看更多评论
立即登录/注册

微信扫码登录

0.0564s