当前位置：网站首页>Torch slice maintenance

Torch slice maintenance

2022-07-26 04:44:00 【Fun of construction】 

>>> import torch
>>> a=torch.ones((3,3)) 
>>> a[0].shape
torch.Size([3])
>>> a[0,0].shape
torch.Size([])
>>> a[0,0:1].shape
torch.Size([1])
>>> a[0,2:].shape
torch.Size([1])
>>> a[0,2:3].shape 
torch.Size([1])
>>>
原网站

版权声明
本文为[Fun of construction]所创，转载请带上原文链接，感谢
https://yzsam.com/2022/201/202207182057080088.html

边栏推荐

猜你喜欢

随机推荐