Pytorch函数例子详解
文章平均质量分 77
深度学习代码上常见的torch函数用法讲解。
江川轻寒
学学数学和code
展开
-
【Pytorch】torch.cat() Pytorch中对于张量拼接函数的用法和示例以及dim=-1的应用场景
对于Pytorch.cat()函数中的dim详细举例解释。举例中的张量从形状(2,3)到(2,2,3)扩展维度均有例子理解。dim=-1的情况及其优点,以及其应用场景原创 2024-03-21 16:03:51 · 759 阅读 · 1 评论 -
【Numpy】np.random.uniform()均匀分布中的采样函数用法
从打印结果来看生成了一个三维的numpy数组,所以size的输入限制与numpy中大部分的输入限制一致,size控制返回值的形状,一个不太严谨的说法,可以参考shape函数,size接受的传参与shape函数的返回值相同。使用了list储存的维度信息,返回的同样是根据list的维度信息的numpy数组,与元组的类似,其可接受的参数与shape返回的形状中的数值也是相同的。low和high构成采样的数值区域,size为返回的样本数目规则,返回的为numpy数组类型。原创 2024-01-11 11:48:59 · 732 阅读 · 1 评论