RuntimeError: Boolean value of Tensor with more than one value is ambiguous

Question

1.前提、実現したいこと
256×256ピクセルの特徴マップから1点を選び、ラベルを見てその1点と同じクラスだったら1、それ以外の場合は-1を入れた256×256のmapを作っています。

2.発生している問題、エラーメッセージ
ラベルを見てその1点と同じクラスだったら1、それ以外の場合は0を入れた256×256のmapなら、

# targets[:, w, h].view(-1, 1, 1) <- 選んだ点
# targets <- 256×256の特徴map
map = 1.0 * (targets[:, w, h].view(-1, 1, 1) == targets)

のコードで実行でせることができました。0ではなくで-1にするときのコードの書き方に困っています。

3.該当のソースコード
このコードだと、以下のようなエラーメッセージが出ます。

if targets[:, w, h].view(-1, 1, 1) == targets:
            map = 1

        else:
            map = -1

# エラーメッセージ
File "main2.py", line 341, in <module>
    train_loss1, train_loss2, train_accuracy = train(epoch) # train
  File "main2.py", line 108, in train
    loss1 = criterion1(output, targets)
  File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/home/hotta1080/moeka/Unet/Loss_a.py", line 88, in forward
    if targets[:, w, h].view(-1, 1, 1) == targets:
RuntimeError: Boolean value of Tensor with more than one value is ambiguous

4.自分で調べたことや試したこと
テンソルをif文に渡すときに起こるエラーだと書いてあったので、if文に変えたことで起こってしまったエラーだと思うですが対処に困っています。

5.補足情報
targetsはラベルで[batch size, W, H]

Answer 1

• if文に変えたことで起こってしまったエラーだと思うですが対処に困っています。

はい、そのとおりです。

map = 1.0 * (targets[:, w, h].view(-1, 1, 1) == targets)

で1か0が入るというなら、

map = 2.0 * (targets[:, w, h].view(-1, 1, 1) == targets) - 1.0

で1と-1が入るでしょう。