摘要: import torch from torch import nn import torch.nn.functional as F a = torch.tensor([[1,2,3],[4,5,6]]) torch.sum(a) #tensor(21) per_col_sum = torch.sum 阅读全文
posted @ 2021-04-08 15:49 crazybird123 阅读(143) 评论(0) 推荐(0) 编辑