This document is relevant for: Inf2, Trn1, Trn2

nki.language.fmod#

nki.language.fmod(x, y, dtype=None, mask=None, **kwargs)[source]#

Floor-mod of x / y, element-wise.

The remainder has the same sign as the dividend x. It is equivalent to the Matlab(TM) rem function and should not be confused with the Python modulus operator x % y.

((Similar to numpy.fmod))

Parameters:

  • x – a tile. If x is a scalar value it will be broadcast to the shape of y. x.shape and y.shape must be broadcastable to a common shape, that will become the shape of the output.

  • y – a tile or a scalar value. x.shape and y.shape must be broadcastable to a common shape, that will become the shape of the output.

  • dtype – (optional) data type to cast the output type to (see Supported Data Types for more information); if not specified, it will default to be the same as the data type of the input tiles, or whichever input type has the highest precision (see NKI Type Promotion for more information);

  • mask – (optional) a compile-time constant predicate that controls whether/how this instruction is executed (see NKI API Masking for details)

Returns:

a tile that has values x fmod y.

This document is relevant for: Inf2, Trn1, Trn2