2022年3月15日
torch中的几个函数（min()、max()、prod()、clamp_()、where（））
摘要： torch.min()、torch.max()、torch.prod() 这两个函数很好理解，就是求张量中的最小值和最大值以及相乘 1.在这两个函数中如果没有指定维度的话，那么默认是将张量中的所有值进行比较，输出最大值或者最小值或是所有值相乘。 2.而当指定维度之后，会将对应维度的数据进行比较，同时 阅读全文
posted @ 2022-03-15 10:37 maple_hx 阅读(1986) 评论(0) 推荐(0)