diffusion模型张量