Pytorch框架之ceil/floor 取整方法

2023-07-24 01:46| 来源: 网络整理| 查看: 265

tensor的ceil|floor 方法 ceil - 向上取整floor - 向下取整ceil_ 与 floor_

QQ:3020889729 小蔡 ceil - 向上取整

返回一个向上取整后的张量–tensor.ceil() >> return new_tensor

''' 不修改自身 ''' x = torch.linspace(-3, 3, 10) print('before_op:\n', x) y = x.ceil() print('after_op:\n', x) print('the op_data:\n', y)

在这里插入图片描述

floor - 向下取整

返回一个向下取整后的张量–tensor.floor() >> return new_tensor

''' 不修改自身 ''' x = torch.linspace(-3, 3, 10) print('before_op:\n', x) y = x.floor() print('after_op:\n', x) print('the op_data:\n', y)

在这里插入图片描述

ceil_ 与 floor_

向上取整后替换原张量，并返回当前张量 --tensor.ceil_() >> return tensor

''' 修改自身 ''' x = torch.linspace(-3, 3, 10) print('before_op:\n', x) y = x.ceil_() print('after_op:\n', x) print('the op_data:\n', y)

在这里插入图片描述

向下取整后替换原张量，并返回当前张量 --tensor.floor_() >> return tensor

''' 修改自身 ''' x = torch.linspace(-3, 3, 10) print('before_op:\n', x) y = x.floor_() print('after_op:\n', x) print('the op_data:\n', y)

在这里插入图片描述

【本文地址】

今日新闻