2021-12-16から1日間の記事一覧
torch.maxとtorch.argmaxがゴタゴタしていたのでメモ torch.maxは1つ目に最大値、2つ目に最大値の入っているindexを返す。 torch.argmaxはtorch.maxの2つ目のみを返す。 axisを指定しないと1行にreshapeされた後にargmaxが行われる。 import torch a = torch…
torch.maxとtorch.argmaxがゴタゴタしていたのでメモ torch.maxは1つ目に最大値、2つ目に最大値の入っているindexを返す。 torch.argmaxはtorch.maxの2つ目のみを返す。 axisを指定しないと1行にreshapeされた後にargmaxが行われる。 import torch a = torch…