| D | test_sampled_softmax_loss_op.py | 23 def generate_test_data(num_classes, batch_size, sampled): argument
34 sampled_exp = np.full([len(sampled)], fill_value=0.5, dtype=np.float32)
35 sampled_values = (Tensor(sampled), Tensor(true_exp), Tensor(sampled_exp))
50 sampled=[4, 0, 2, 3])
76 sampled=[4, 5, 6, 3])
110 sampled=[4, 0, 2, 3])
|