diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py index d440eeaa54edc4c3d0934d445c6140568b6ccffc..b52d44955ed5a5a75c42656ba233febd224ad561 100644 --- a/mace/python/tools/converter_tool/transformer.py +++ b/mace/python/tools/converter_tool/transformer.py @@ -445,7 +445,7 @@ class Transformer(base_converter.ConverterInterface): gather_weights = self._consts[op.input[0]] mul_weight = ConverterUtil.get_arg(consumer_op, MaceKeyword.mace_scalar_input_str).f # noqa - gather_weights.float_data[:] = gather_weights.float_data * mul_weight # noqa + gather_weights.float_data[:] = [float_data * mul_weight for float_data in gather_weights.float_data] # noqa self.safe_remove_node(consumer_op, None, remove_input_tensor=True)