Pytorch tensor操作
https://www.cnblogs.com/jeshy/p/11366269.html
我們需要明確一下,torch.Tensor()是python類,更明確地說,是默認張量類型。torch.FloatTensor()的別名,torch.Tensor([1,2])會調用Tensor類的構造函數init,生成單精度浮點類型的張量。如上右圖。而torch.tensor()僅僅是python函數:https://pytorch.org/docs/stable/torch.html#torch.tensor ,函數原型是:
其中data可以是:list, tuple, NumPy ndarray, scalar和其他類型。torch.tensor會從data中的數據部分做拷貝(而不是直接引用),根據原始數據類型生成相應的torch.LongTensor、torch.FloatTensor和torch.DoubleTensor。
這里再說一下torch.empty(),根據 https://pytorch.org/docs/stable/torch.html?highlight=empty#torch.empty ,我們可以生成指定類型、指定設備以及其他參數的張量,由於torch.Tensor()只能指定數據類型為torch.float,所以torch.Tensor()可以看做torch.empty()的一個特殊情況。
最后放一個小彩蛋:可看到: