Thanks – that makes sense. I must have missed the log probability part, although it seems weird not giving an option to specify probabilities directly on input.
And thanks for pointing out the FloatTensor issue. I guess torch.tensor() is the recommended way now.