pytorch常用API

· torchvision.utils.make_grid():

make_grid的作用是将若干幅图像拼成一幅图像。其中padding的作用就是子图像与子图像之间的pad有多宽。

pytorch常用API

· torch.clamp():

将输入input张量每个元素的夹紧到区间 [min,max][min,max],并返回结果到一个新张量。

上一篇:【智能指针】


下一篇:2022-1-25