Tensor processing for with masked artificial intelligence function behavior

A computer system comprising:

• at least one computing device;
• a set of one or more computer-readable storage media;
• and
• program instructions, collectively stored in the set of one or more computer-readable storage media, for causing the at least one computing device to perform computer operations including:
• executing an instruction, the executing the instruction including:
• obtaining an input tensor, the input tensor including a dimension of index size n;
• determining an element count, c, based on an indicator, the indicator specified by the instruction, and the element count specifying a number of vector elements on which to perform an artificial intelligence function;
• obtaining an input vector, of the input tensor, of size n;
• and
• performing the artificial intelligence function, the performing the artificial intelligence function including performing the artificial intelligence function on a first c number of elements of the input vector to provide a corresponding c number of elements of an output vector of index size n of an output tensor.

A computer-implemented method comprising:

• executing an instruction, the executing the instruction including:
• obtaining an input tensor, the input tensor including a dimension of index size n;
• determining an element count, c, based on an indicator, the indicator specified by the instruction, and the element count specifying a number of vector elements on which to perform an artificial intelligence function;
• obtaining an input vector, of the input tensor, of size n;
• and
• performing the artificial intelligence function, the performing the artificial intelligence function including performing the artificial intelligence function on a first c number of elements of the input vector to provide a corresponding c number of elements of an output vector of index size n of an output tensor.

A computer system comprising:

• at least one hardware accelerator to be used in executing an instruction, the executing the instruction including:
• obtaining an input tensor, the input tensor including a dimension of index size n;
• determining an element count, c, based on an indicator, the indicator specified by the instruction, and the element count specifying a number of vector elements on which to perform an artificial intelligence function, wherein the element count c is less than n;
• obtaining an input vector, of the input tensor, of size n;
• and
• performing the artificial intelligence function, the artificial intelligence function including a SOFTMAX function, and the performing the artificial intelligence function including performing the artificial intelligence function on a first c number of elements of the input vector to provide a corresponding c number of elements of an output vector of index size n of an output tensor, and setting each element, of the output vector, after the corresponding c number of elements of the output vector to a selected value.

A computer-implemented method comprising:

• executing an instruction, the executing the instruction including:
• obtaining an input tensor, the input tensor including a dimension of index size n;
• determining an element count, c, based on an indicator, the indicator specified by the instruction, and the element count specifying a number of vector elements on which to perform an artificial intelligence function, wherein the element count c is less than n;
• obtaining an input vector, of the input tensor, of size n;
• and
• performing the artificial intelligence function, the artificial intelligence function including a SOFTMAX function, and the performing the artificial intelligence function including performing the artificial intelligence function on a first c number of elements of the input vector to provide a corresponding c number of elements of an output vector of index size n of an output tensor, and setting each element, of the output vector, after the corresponding c number of elements of the output vector to a selected value.