pytorch - "greater than"运算符 ">"对 PyTorch 张量意味着什么？

Question

我有一个张量 it 定义为:

import torch
it = torch.tensor([0,  0,  0,  0,  0,  0,  0,  0,  0,  0], device='cuda:0')

根据这个定义，it > 0 是什么意思？

Answer 1

使用 >运算符与使用 torch.gt() 相同功能。

换句话说，

it > 0

和

一样

torch.gt(it, 0)

它返回一个 ByteTensor ( bool 张量)相同的shape作为 it在哪里 out[i]如果 it[i] > 0 为真否则为 False。