fix type promotion TypeError in tensor_utils.py

pull/3883/head
Wang Xin 10 months ago
parent a1f9339181
commit 2194bcdeab

@ -177,8 +177,9 @@ def th_accuracy(pad_outputs: paddle.Tensor,
Returns: Returns:
float: Accuracy value (0.0 - 1.0). float: Accuracy value (0.0 - 1.0).
""" """
pad_pred = pad_outputs.view(pad_targets.shape[0], pad_targets.shape[1], pad_pred = pad_outputs.reshape(
pad_outputs.shape[1]).argmax(2) [pad_targets.shape[0], pad_targets.shape[1],
pad_outputs.shape[1]]).argmax(2)
mask = pad_targets != ignore_label mask = pad_targets != ignore_label
#TODO(Hui Zhang): sum not support bool type #TODO(Hui Zhang): sum not support bool type
# numerator = paddle.sum( # numerator = paddle.sum(
@ -248,7 +249,7 @@ def st_reverse_pad_list(ys_pad: paddle.Tensor,
# >>> tensor([[ 2, 1, 0], # >>> tensor([[ 2, 1, 0],
# >>> [ 2, 1, 0], # >>> [ 2, 1, 0],
# >>> [ 0, -1, -2]]) # >>> [ 0, -1, -2]])
index = index * seq_mask index = index * seq_mask.astype(index.dtype)
# >>> index # >>> index
# >>> tensor([[2, 1, 0], # >>> tensor([[2, 1, 0],

Loading…
Cancel
Save