在使用pytorch的时候我遇到的一些不熟悉的方法
在使用pytorch的时候我遇到的一些不熟悉的方法
tensor.size()[:-2] + (20,)
取除了倒数两个维度之前的维度,并且加上最后一个维度,比如说是(2, 3, 4, 5),变成了(2, 3, 20)
tensor.permute(0, 2, 1, 3)
对于四维的tensor,交换第二个和第三个维度
tensor.transpose(-1, -2)
对于一个tensor,交换倒数两个维度
跟permute类似,但是transpose只能交换两个维度,但是permute可以交换多
原创
2020-07-08 21:54:27 ·
1151 阅读 ·
0 评论