triu

Signature

nabla.triu(x, k=0)

Description

Return the upper triangular part of an array.

Parameters

x: Input array (batch, seq_len, seq_len) k: Diagonal offset (0 = main diagonal, > 0 = above, < 0 = below)

Returns

Upper triangular part of the input array