【04】Lambda匿名函数的用法

Lambda匿名函数,这个操作要重要啦!

Transforms.Lambda

在这里插入图片描述
torch.stack 对张量进行拼接,不同的是需要先添加一个维度,然后再进行拼接,比如将2个3维的张量拼接成一个4维的张量;后面的语句是进行参数的解析。

上述代码段解读:从crops中一次取出一个元素,先把它转换成一个张量,最后转换好的饿张量是一个list列表的形式,然后这些张量被拼接成一个大的张量。

你可能感兴趣的:(深度学习碎碎念,python,pytorch)