jax.Array.round

内容

jax.Array.round#

abstract Array.round(decimals=0, out=None)[source]#

将数组元素四舍五入到给定的十进制位数。

有关完整文档,请参阅 jax.numpy.round()

参数:
  • self (Array)

  • decimals (int)

  • out (None)

返回类型:

Array