“大于”运算符“>”对 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相同的（一个布尔张量）shape为it其中out[i]为真，如果it[i] > 0，否则为假。