Python NumPy 张量整数转换:四舍五入、向上取整、向下取整
将一个非整数张量转换成整数张量可以通过四舍五入、向上取整或向下取整等方式进行。
以下是使用 Python 中的 numpy 库进行转换的示例代码:
- 四舍五入:
import numpy as np
# 创建一个非整数张量
tensor = np.array([1.2, 2.7, 3.5, 4.9])
# 使用round函数进行四舍五入转换
integer_tensor = np.round(tensor)
print(integer_tensor)
输出结果:
[1. 3. 4. 5.]
- 向上取整:
import numpy as np
# 创建一个非整数张量
tensor = np.array([1.2, 2.7, 3.5, 4.9])
# 使用ceil函数进行向上取整转换
integer_tensor = np.ceil(tensor)
print(integer_tensor)
输出结果:
[2. 3. 4. 5.]
- 向下取整:
import numpy as np
# 创建一个非整数张量
tensor = np.array([1.2, 2.7, 3.5, 4.9])
# 使用floor函数进行向下取整转换
integer_tensor = np.floor(tensor)
print(integer_tensor)
输出结果:
[1. 2. 3. 4.]
以上示例代码使用了 numpy 库中的 round、ceil 和 floor 函数,分别实现了四舍五入、向上取整和向下取整的功能。根据具体需求,选择合适的转换方式即可。
原文地址: https://www.cveoy.top/t/topic/bRxO 著作权归作者所有。请勿转载和采集!