I am wondering while calling this function named forward you were passing the wrong (type) argument we should pass the argument with type Tensor or if you share more about this it would be better