import torch
x = torch.randn(1,3,224,224)
print(x[0].shape)
print(x[0::,...].shape)
torch.Size([3, 224, 224])
torch.Size([1, 3, 224, 224])
- 索引::加上，会保持维度不变。
torch——索引切片
阅读 46
2022-11-01
相关推荐
精彩评论（0）
0 0 举报
torch——索引切片
阅读 46
2022-11-01
import torch
x = torch.randn(1,3,224,224)
print(x[0].shape)
print(x[0::,...].shape)
torch.Size([3, 224, 224])
torch.Size([1, 3, 224, 224])
相关推荐
精彩评论（0）