One-hot encoding tensors