jax.numpy.zerosΒΆ

jax.numpy.zeros(shape, dtype=None)[source]ΒΆ

Return a new array of given shape and type, filled with zeros.

LAX-backend implementation of zeros().

Original docstring below.

Parameters
  • shape (int or tuple of ints) – Shape of the new array, e.g., (2, 3) or 2.

  • dtype (data-type, optional) – The desired data-type for the array, e.g., numpy.int8. Default is numpy.float64.

Returns

out – Array of zeros with the given shape, dtype, and order.

Return type

ndarray