2つのTensor配列の要素ごとの最大値・最小値を取得するtorch.maximum、torch.minimum、torch.fmax、torch.fmin【PyTorch】

2つのTensor配列の要素ごとの最大値を取得するmaximum、fmax torch.maximumとtorch.fmaxの第一引数と第二引数に、Tensor配列を指定すると要素ごとの最大値を取得できる。 import torch a = torch.tensor((1, 2, -1)) b = torch.tensor((3, 0, 4)) print(torch.maximum(a, b)) # tensor([3, 2, 4]) print(torch.fmax(a, b)) # tensor([3, 2, 4])