Lambda匿名函数,这个操作要重要啦!
Transforms.Lambda
torch.stack
对张量进行拼接,不同的是需要先添加一个维度,然后再进行拼接,比如将2个3维的张量拼接成一个4维的张量;后面的语句是进行参数的解析。
上述代码段解读:从crops中一次取出一个元素,先把它转换成一个张量,最后转换好的饿张量是一个list列表的形式,然后这些张量被拼接成一个大的张量。
Lambda匿名函数,这个操作要重要啦!
Transforms.Lambda
torch.stack
对张量进行拼接,不同的是需要先添加一个维度,然后再进行拼接,比如将2个3维的张量拼接成一个4维的张量;后面的语句是进行参数的解析。
上述代码段解读:从crops中一次取出一个元素,先把它转换成一个张量,最后转换好的饿张量是一个list列表的形式,然后这些张量被拼接成一个大的张量。