pytorch 如何将Tensor标量转换为Tensor列表而不分离它

fafcakar  于 6个月前  发布在  其他
关注(0)|答案(1)|浏览(88)

我有一个名为x的tenor.long标量。如何在不将x转换为numpy的情况下创建Tensortensor[(x.detached().numpy())]的列表

hec6srdp

hec6srdp1#

如果我理解正确的话,你想添加另一个维度(通过转换为list,你将拥有一个像维度Tensor一样的可迭代容器/ Package 器;除非你想要任何特定于list的操作,否则大多数时候你想坚持使用tensor对象进行进一步处理)。
要添加一个dim(使Tensor为一维),可以使用unsqueeze

x.unsqueeze(dim=0)

字符串
view/reshape/index-ing也可以使用:

x.view(-1)
x.reshape(-1)
x[None]

相关问题