As suggested in this thread 9, it would be interesting to know, what kind of values you are using for the target tensor.
Are you dealing with float features or do you have a class target for each sample?