jax.Array.round

abstract Array.round(decimals=0, out=None)

将数组元素四舍五入到给定的小数位数。

完整文档请参考jax.numpy.round()。

参数:

• self (Array)

• decimals (int)

• out (None)

返回类型:

Array