陈大佬的python 使用笔记

.transpose()

功能:改变序列的顺序。

# 改变序列的顺序,原始的为(0,1),这里使用(1,0)就是进行了转置
preds = preds.transpose(1, 0)

.contiguous().view(-1)

功能:有些tensor并不是占用一整块内存,而是由不同的数据块组成,而tensor的view()操作依赖于内存是整块的,这时只需要执行contiguous()这个函数,把tensor变成在内存中连续分布的形式。

#在使用view()前,需要先使用contiguous()把tensor变成在内存中连续分布的形式
preds = preds.transpose(1, 0).contiguous().view(-1)

view()

功能:修改尺寸。其中参数-1代表该项的值有其他项决定。

image = image.view(1, *image.size())

猜你喜欢

转载自blog.csdn.net/chen1234520nnn/article/details/107003858