- 博客(1)
- 收藏
- 关注
原创 torch.sum()
import torch a = torch.randint(0,3,(2,3,4)) print(a.size()) print(a) b = torch.sum(a, 0) # 按将第一个纬度的值相加 c = torch.sum(a, 1) # 按将第二个纬度的值相加 print('b', b) print('c', c)
2020-09-05 23:12:04 217
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人