TORCH.TRIU

torch.triu(input, diagonal=0, *, out=None) → Tensor

返回一个上三角矩阵

参数：

input：输入的张量
diagonal：对角线

>>> a = torch.randn(3, 3)
>>> a
tensor([[ 0.2309,  0.5207,  2.0049],[ 0.2072, -1.0680,  0.6602],[ 0.3480, -0.5211, -0.4573]])
>>> torch.triu(a)
tensor([[ 0.2309,  0.5207,  2.0049],[ 0.0000, -1.0680,  0.6602],[ 0.0000,  0.0000, -0.4573]])
>>> torch.triu(a, diagonal=1)
tensor([[ 0.0000,  0.5207,  2.0049],[ 0.0000,  0.0000,  0.6602],[ 0.0000,  0.0000,  0.0000]])
>>> torch.triu(a, diagonal=-1)
tensor([[ 0.2309,  0.5207,  2.0049],[ 0.2072, -1.0680,  0.6602],[ 0.0000, -0.5211, -0.4573]])>>> b = torch.randn(4, 6)
>>> b
tensor([[ 0.5876, -0.0794, -1.8373,  0.6654,  0.2604,  1.5235],[-0.2447,  0.9556, -1.2919,  1.3378, -0.1768, -1.0857],[ 0.4333,  0.3146,  0.6576, -1.0432,  0.9348, -0.4410],[-0.9888,  1.0679, -1.3337, -1.6556,  0.4798,  0.2830]])
>>> torch.triu(b, diagonal=1)
tensor([[ 0.0000, -0.0794, -1.8373,  0.6654,  0.2604,  1.5235],[ 0.0000,  0.0000, -1.2919,  1.3378, -0.1768, -1.0857],[ 0.0000,  0.0000,  0.0000, -1.0432,  0.9348, -0.4410],[ 0.0000,  0.0000,  0.0000,  0.0000,  0.4798,  0.2830]])
>>> torch.triu(b, diagonal=-1)
tensor([[ 0.5876, -0.0794, -1.8373,  0.6654,  0.2604,  1.5235],[-0.2447,  0.9556, -1.2919,  1.3378, -0.1768, -1.0857],[ 0.0000,  0.3146,  0.6576, -1.0432,  0.9348, -0.4410],[ 0.0000,  0.0000, -1.3337, -1.6556,  0.4798,  0.2830]])

TORCH.TRIU相关推荐

pytorch torch.triu
API torch.triu(input, diagonal=0, out=None) → Tensor 返回上三角矩阵参数描述 input (Tensor) the input tensor. ...
PyTorch 笔记（09）— Tensor 线性代数计算（torch.trace、torch.diag、torch.mm、torch.dot、torch.inverse逆矩阵、转置）
1. 常用函数常用线性表函数如下表所示: 2. 使用示例 2.1 torch.trace In [22]: import torch as tIn [23]: a = t.arange(1, 10) ...
（超级详细）numpy与torch用法对比手册
文章目录前言生成类操作采样类数学运算操作判断类(布尔类)操作矩阵形状操作其他操作前言由于numpy比较基础和通用,但是GPU上跑实验必须使用tensor,故还是直接用torch里面的 ...
【PyTorch】torch.nn.Transformer解读与应用
nn.TransformerEncoderLayer 这个类是transformer encoder的组成部分,代表encoder的一个层,而encoder就是将transformerEncoderL ...
torch.nn.Transformer解读与应用
nn.TransformerEncoderLayer 这个类是transformer encoder的组成部分,代表encoder的一个层,而encoder就是将transformerEncoderL ...
python torch exp_PyTorch官方中文文档：torch
torch 包 torch 包含了多维张量的数据结构以及基于其上的多种数学操作.另外,它也提供了多种工具,其中一些可以更有效地对张量和任意类型进行序列化. 它有CUDA 的对应实现,可以在NVIDIA ...
pytorch 定义torch类型数据_PyTorch官方中文文档：torch.Tensor
torch.Tensor torch.Tensor是一种包含单一数据类型元素的多维矩阵. Torch定义了七种CPU tensor类型和八种GPU tensor类型: Data tyoe CPU te ...
torch(七)、Math operations(2)
参考 torch(七).Math operations(2) - 云+社区 - 腾讯云目录 Spectral Ops torch.fft(input, signal_ndim, normalized ...
Transformer的PyTorch实现
Google 2017年的论文 Attention is all you need 阐释了什么叫做大道至简!该论文提出了Transformer模型,完全基于Attention mechanism,抛弃 ...

TORCH.TRIU

TORCH.TRIU相关推荐

最新文章

热门文章