2020年6月27日
torch.max()使用讲解
摘要： output = torch.max(input, dim) input输入的是一个tensor dim是max函数索引的维度0/1，0是每列的最大值，1是每行的最大值 实例： import torch import numpy as np import matplotlib.pyplot as p 阅读全文
posted @ 2020-06-27 19:49 九叶草 阅读(5007) 评论(0) 推荐(0) 编辑