diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py index 6cae50dc2d9aa3b4e72b826371d38538b5061844..6ce3beba976adffbfa1ed8895a050dba99fd5e97 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)