This document is relevant for: Inf2, Trn1, Trn1n

nki.language.full#

nki.language.full(shape, fill_value, dtype, buffer=None, name='', **kwargs)[source]#

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

((Similar to numpy.full))

Parameters:

  • shape – the shape of the tensor.

  • fill_value – the initial value of the tensor.

  • dtype – the data type of the tensor (see Supported Data Types for more information).

  • 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: Inf2, Trn1, Trn1n