常用的Tensor操作

Posted monkeyt

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了常用的Tensor操作相关的知识,希望对你有一定的参考价值。

常用的Tensor操作

1、通过tensor.view方法可以调整tensor的形状,但必须保证调整去前后元素总数一致。view不会修改自身的数据,返回新的tensor与原tensor共享内存,即更改其中的一个,另一个也会跟这改变。
2、实际中经常需要添加或减少某一维度,可用squeeze和unsqueeze这两函数。

import torch as t
a = t.arange(0,6)
b = a.view(2,3)     #调整tensor的形状,不会修改自身数据
c = a.view(-1,3)    #-1自动计算大小
‘‘‘
a = tensor([0, 1, 2, 3, 4, 5])
b = tensor([[0, 1, 2],
        [3, 4, 5]])
c = tensor([[0, 1, 2],
        [3, 4, 5]])‘‘‘
b = b.unsqueeze(1)      #在第一维(下标从0开始)上增加“1”
‘‘‘tensor([[[0, 1, 2]],

        [[3, 4, 5]]])‘‘‘
c = c.unsqueeze(-2)
‘‘‘tensor([[[0, 1, 2]],

        [[3, 4, 5]]])‘‘‘
d = b.view(1,1,1,2,3)
‘‘‘tensor([[[[[0, 1, 2],
           [3, 4, 5]]]]])‘‘‘
d = d.squeeze(0)    #压缩第0维的“1”
‘‘‘tensor([[[0, 1, 2],
         [3, 4, 5]]])‘‘‘
d = d.squeeze()     #所有维度为“1”的压缩
‘‘‘tensor([[0, 1, 2],
        [3, 4, 5]])‘‘‘

3、resize是另一种可用调整size的方法,但与view不同,它可以修改tensor的尺寸。如果新尺寸超过原尺寸,会自动分配新的内存空间,而如果新尺寸小于原尺寸,则之前的数据依旧会被保存。
b.resize_(1,3)      #resize_调整size,新尺寸小于原尺寸,则之前的保留
‘‘‘tensor([[0, 1, 2]])‘‘‘
b.resize_(3,3)      #resize_调整size,新尺寸大于原尺寸,会自动分配新的内存空间
‘‘‘tensor([[                  0,                   1,                   2],
        [                  3,                   4,                   5],
        [7526747973031060338, 7306812055932138085, 8389969046800051066]])‘‘‘

2018-10-23 17:59:12

以上是关于常用的Tensor操作的主要内容,如果未能解决你的问题,请参考以下文章

常用的Tensor操作

常用Tensor操作

Pytorch中的tensor常用操作

PyTorch常用知识总结

PyTorch常用知识总结

Pytorch基础重新巩固