Table Of Contents
Table Of Contents

ones

mxnet.ndarray.ones(shape, ctx=None, dtype=None, **kwargs)[source]

Returns a new array filled with all ones, with the given shape and type.

Parameters:
  • shape (int or tuple of int or list of int) – The shape of the empty array.
  • ctx (Context, optional) – An optional device context. Defaults to the current default context (mxnet.context.current_context()).
  • dtype (str or numpy.dtype, optional) – An optional value type (default is float32).
  • out (NDArray, optional) – The output NDArray (default is None).
Returns:

A new array of the specified shape filled with all ones.

Return type:

NDArray

Examples

>>> mx.nd.ones(1).asnumpy()
array([ 1.], dtype=float32)
>>> mx.nd.ones((1,2), mx.gpu(0))
<NDArray 1x2 @gpu(0)>
>>> mx.nd.ones((1,2), dtype='float16').asnumpy()
array([[ 1.,  1.]], dtype=float16)