如果使用torch，代码可以怎么写

使用 PyTorch 来实现一个类 ChatGPT 的 AI 可以参考以下代码：

import torch
from transformers import GPT2Tokenizer, GPT2LMHeadModel

# 加载预训练模型和分词器
tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')

# 设置模型为评估模式
model.eval()

# 定义函数生成回复
def generate_response(input_text, max_length=50):
    # 对输入文本进行分词和编码
    input_ids = tokenizer.encode(input_text, return_tensors='pt')
    
    # 生成回复
    sample_output = model.generate(
        input_ids, 
        do_sample=True, 
        max_length=max_length, 
        top_k=50, 
        top_p=0.95,
        pad_token_id=tokenizer.eos_token_id
    )
    
    # 解码生成的回复
    generated_text = tokenizer.decode(sample_output[0], skip_special_tokens=True)
    
    return generated_text

接下来可以调用 `generate_response` 函数生成 AI 的回复，例如：

input_text = "你好"
response = generate_response(input_text)
print(response)

这样就可以得到 AI 的回复了。