课程6 relax.Var报错

[ShuRaymond]

A = relax.Var("A", (128, 128), relax.DynTensorType(2, "float32"))
B = relax.Var("B", (128, 128), relax.DynTensorType(2, "float32"))

TypeError Traceback (most recent call last)
in <cell line: 1>()
----> 1 A = relax.Var("A", (128, 128), relax.DynTensorType(2, "float32"))
2 B = relax.Var("B", (128, 128), relax.DynTensorType(2, "float32"))

/usr/local/lib/python3.10/dist-packages/tvm/relax/expr.py in init(self, name_hint, struct_info, span)
414 struct_info = tvm.runtime.convert_to_object(struct_info)
415 if not isinstance(struct_info, StructInfo):
--> 416 raise TypeError(
417 "struct_info needs to be an instance of StructInfo. "
418 "If you attempt to pass in shape, "

TypeError: struct_info needs to be an instance of StructInfo. If you attempt to pass in shape, use relax.TensorStructInfo(shape, dtype).

[hanquanjushi]

excuse me , have you already solve it ?

[Jvanlo]

relax.Var("A", relax.TensorStructInfo((128, 128), "float32"))

[hanquanjushi]

thank you so much!