The weight should be python built-in type or tensor? Or both work?
tensor
Both approximations should work.