Pytorch中一些函数的意义

此生再无相见时 提交于 2020-08-18 08:12:09

1. torch.clamp(tensor, min, max): 将tensor压缩到[min, max]

tmp_input = torch.randint(1, 125, (10,1)) tmp_output = torch.clamp(tmp_input, 50, 100) print(tmp_input) print(tmp_output)

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!