小编sol*_*lst的帖子

tensor.size(-1) 中的 -1 是什么意思?

我在 Pytorch 文档中看到过类似的东西,

import torch

a = torch.tensor([1, 2])
a.size() # torch.Size([2])
a.size(-1) # 2
Run Code Online (Sandbox Code Playgroud)

这是如何运作的?我没有找到说明。谢谢,

python pytorch

2
推荐指数
1
解决办法
6075
查看次数

标签 统计

python ×1

pytorch ×1