2022年1月18日
torch.triu()
摘要： torch.triu(input, diagonal=0, out=None) → Tensor 返回矩阵上三角部分，其余部分定义为0。 Parameters: input (Tensor) – the input tensor diagonal (int, optional) – the diag 阅读全文
posted @ 2022-01-18 14:47 ArdenWang 阅读(333) 评论(0) 推荐(0) 编辑