标签： indexing pytorch
我需要在pytorch中创建上三角遮盖张量。在tensorflow中，使用matrix_band_part很容易。此功能是否有任何pytorch等效项？ 就像numpy triu_indices一样，但我需要张量，而不仅仅是矩阵。