PyTorch torch.gt 関数


Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.gtPyTorch で要素ごとに大なり比較を行うための関数です。

関数定義

torch.gt(input, other, out)

使用例

実例

import torch

a = torch.tensor([1, 2, 3, 4])
b = torch.tensor([2, 2, 2, 2])

result = torch.gt(a, b)

print(result)

出力結果は:

tensor([False, False,  True,  True])

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張