luv_camping的博客

b：要输出的内容类型；b=0时输出最大值数据，b=1时输出最大值索引。tensor：要进行操作的数据(张量)a：要进行操作的维度。

torch.sort(input, dim=-1, descending=False, stable=False, *, out=None)-> (Tensor, LongTensor)延指定的维度对tensor的元素进行排序。参数:返回值:(, ) 的元组，Tensor是排序后的tensor，LongTensor是原始tensor中元素索引组成的tensor例:......

shape: 获取数组的形状假设是一个二维数组shape[0]: 获取数组的行数shape[1]: 获取数组的列数例：import numpy as npx = np.array([[1, 2, 3, 4], [4, 3, 2, 1]])x.shape>>> (2, 4)x.shape[0]>>> 2x.shape[1]>>> 4...

numpy.linspace | np.linspace 函数用法np.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0)在间隔start和stop之间返回num个均匀间隔的数据。Parameters: start: array_like 序列的起始值 stop: array_like 序列的结束值 num: int, optional 要生成的样本数，默认值为50，必须为非负