jax.Array.round

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

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

请参阅jax.numpy.round()以获取完整文档。

参数:

• self (Array)

• decimals (int)

• out (None)

返回类型:

Array