- 博客(4)
- 收藏
- 关注
原创 pytorch max() 用法
b:要输出的内容类型;b=0时输出最大值数据,b=1时输出最大值索引。tensor:要进行操作的数据(张量)a:要进行操作的维度。
2022-11-21 20:30:20
1246
1
原创 torch.sort() 用法
torch.sort(input, dim=-1, descending=False, stable=False, *, out=None)-> (Tensor, LongTensor)延指定的维度对tensor的元素进行排序。参数:返回值:(, ) 的元组,Tensor是排序后的tensor,LongTensor是原始tensor中元素索引组成的tensor例:......
2022-06-21 19:55:28
823
原创 python中的shape, shape[0], shape[1]
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...
2022-04-22 21:25:25
1794
原创 np.linspace() 函数用法
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,必须为非负
2022-04-22 17:29:13
14323
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人