简单来说就是一个除法运算,被除数是一个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.]])
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)