Single bounded output layer

1

Hi,

I am looking for an output layer which has a single output that is bounded between (0,1).

I tried nn.linear, but its not bounded. nn.softmax doesnt work as its single output, its value is always 1.

2

You could try torch.sigmoid.