Мне нужно создать верхний треугольный тензор маскирования в pytorch. в тензорном потоке легко использовать matrix_band_part . это какой-либо эквивалент Pytorch для этой функции? это как numy triu_indices, но мне нужен тензор, а не просто матрица.