- 博客(1)
- 收藏
- 关注
RSS订阅原创 pytorch框架中，sum()函数中axis = 的理解
首先创建一个三维张量。A = torch.arange(20*2,dtype=torch.float32).reshape(2, 5, 4)A.shape, A.sum(), A输出的结果为。(torch.Size([2, 5, 4]), tensor(780.), tensor([[[ 0., 1., 2., 3.], [ 4., 5., 6., 7.], [ 8., 9., 10., 11.], [12.,
2022-04-05 21:29:08
2254
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人