pytorch基操03-取整取余操作

import torch
torch.manual_seed(1234)

a=torch.empty(size=(2,3)).normal_(10,10)
print('original :',a)

# 向下取整
print('floor :',torch.floor(a))
# 向上取整
print('ceil :',torch.ceil(a))
# 四舍五入
print('round :',torch.round(a))
# 只取整数
print('trunc :',torch.trunc(a))
# 只取小数
print('frac :',torch.frac(a))
# 取余数
print('% :',a%2)

original : tensor([[10.4613, 14.0240, -0.1153],
        [12.1674,  3.8773, 15.0361]])
floor : tensor([[10., 14., -1.],
        [12.,  3., 15.]])
ceil : tensor([[11., 15., -0.],
        [13.,  4., 16.]])
round : tensor([[10., 14., -0.],
        [12.,  4., 15.]])
trunc : tensor([[10., 14., -0.],
        [12.,  3., 15.]])
frac : tensor([[ 0.4613,  0.0240, -0.1153],
        [ 0.1674,  0.8773,  0.0361]])
% : tensor([[0.4613, 0.0240, 1.8847],
        [0.1674, 1.8773, 1.0361]])