This document is relevant for: Trn1, Trn2, Trn3

nki.language.ones

nki.language.ones(shape, dtype, buffer=MemoryRegion.sbuf, name='')

Create a new tensor of given shape and dtype on the specified buffer, filled with ones.

((Similar to numpy.ones))

Warning

This API is experimental and may change in future releases.

Parameters:

• shape – the shape of the tensor.

• dtype – the data type of the tensor.

• buffer – the specific buffer (ie, sbuf, psum, hbm), defaults to sbuf.

• name – the name of the tensor.

Returns:

a new tensor allocated on the buffer.

This document is relevant for: Trn1, Trn2, Trn3