python return加上for用法

问题遇到的现象和发生背景

今天在学李沐的动手学深度学习中遇到一个return加上for循环的用法不明白

img

return 返回的是一个列表[]
其实就是把text_labels里面的值给返回了,这是python的for循环

就是类似于下面两种方式返回结果是一样的


# 方式一
y = [i for i in range(10)]
return y

# 方式二
return [i for i in range(10)]

这是python中一种特殊的语法,叫做列表推导式,表示对列表逐个元素处理,并把处理过后的元素组成列表返回。
语法如下所示,可以参考 https://showmeai.tech/tutorials/56?articleId=83 中的讲解