详解PyTorch中Tensor的高阶操作-创新互联
条件选取:torch.where(condition, x, y) → Tensor
返回从 x 或 y 中选择元素的张量,取决于 condition
操作定义:
举个例子:
>>> import torch >>> c = randn(2, 3) >>> c tensor([[ 0.0309, -1.5993, 0.1986], [-0.0699, -2.7813, -1.1828]]) >>> a = torch.ones(2, 3) >>> a tensor([[1., 1., 1.], [1., 1., 1.]]) >>> b = torch.zeros(2, 3) >>> b tensor([[0., 0., 0.], [0., 0., 0.]]) >>> torch.where(c > 0, a, b) tensor([[1., 0., 1.], [0., 0., 0.]])
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。
文章标题:详解PyTorch中Tensor的高阶操作-创新互联
路径分享:http://pwwzsj.com/article/djdcio.html