Generating a model's confidence from a (1,2) pytorch tensor

前端 未结 0 650
梦如初夏
梦如初夏 2021-01-24 06:24

So in my classification neural network, my final tensor is

a = tensor([[ 546.3831, -796.4016]], grad_fn=AddmmBackward>). (There is a < in front of Addmm)

I

相关标签:
回答
  • 消灭零回复
提交回复
热议问题