pytorch学习记录:torch.div

pytorch学习记录:torch.div,第1张

pytorch doc介绍


简单来说就是一个除法运算,被除数是一个tensor,除数可以是一个数字,也可以是一个tensor。

import torch
x = torch.tensor([ 0.3810,  1.2774, -0.2972, -0.3719,  0.4637])
torch.div(x, 0.5)
# tensor([ 0.7620,  2.5548, -0.5944, -0.7438,  0.9274])

a = torch.tensor([[-0.3711, -1.9353, -0.4605, -0.2917],
                  [ 0.1815, -1.0111,  0.9805, -1.5923],
                  [ 0.1062,  1.4581,  0.7759, -1.2344],
                  [-0.1830, -0.0313,  1.1908, -1.4757]])
b = torch.tensor([ 0.8032,  0.2930, -0.8113, -0.2308])
torch.div(a, b)
# tensor([[-0.4620, -6.6051,  0.5676,  1.2639],
#         [ 0.2260, -3.4509, -1.2086,  6.8990],
#         [ 0.1322,  4.9764, -0.9564,  5.3484],
#         [-0.2278, -0.1068, -1.4678,  6.3938]])

torch.div(a, b, rounding_mode='trunc')
# tensor([[-0., -6.,  0.,  1.],
#         [ 0., -3., -1.,  6.],
#         [ 0.,  4., -0.,  5.],
#         [-0., -0., -1.,  6.]])

torch.div(a, b, rounding_mode='floor')
# tensor([[-1., -7.,  0.,  1.],
#         [ 0., -4., -2.,  6.],
#         [ 0.,  4., -1.,  5.],
#         [-1., -1., -2.,  6.]])

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/langs/736736.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-28
下一篇 2022-04-28

发表评论

登录后才能评论

评论列表(0条)

保存