[ torch 参数更多]torch.triu_indices

torch.triu_indices

torch.triu_indices(row,col,offset=0,*,dtype=torch.long,device='cpu',layout=torch.strided)

paddle.triu_indices

paddle.triu_indices(row,col=None,offset=0,dtype='int64')

PyTorch 相比 Paddle 支持更多其他参数，具体如下：

参数映射

PyTorch PaddlePaddle 备注
row row 表示输入 Tensor 的行数。
col col 表示输入 Tensor 的列数。
offset offset 表示从指定二维平面中获取对角线的位置。如果 offset = 0 ，取主对角线；如果 offset > 0 ，取主对角线右上的对角线；如果 offset < 0 ，取主对角线左下的对角线。
dtype dtype 表示数据类型。
device - 表示 Tensor 存放设备位置，Paddle 无此参数，需要转写。
layout - 表示布局方式， Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。

转写示例

device: Tensor 的设备

# PyTorch 写法
y = torch.triu_indices(row,col,offset,dtype,device=torch.device('cpu'))

# Paddle 写法
y = paddle.triu_indices(row,col,offset,dtype)
y.cpu()