分类:编程语言
1.torch.eq(input,other,out=None)说明:比较元素是否相等,第二个参数可以是一个数,或者是第一个参数同类型形状的张量参数:input(Tensor)----待比较张量other(Tenosrorfloat)----比较张量或者数out(Tensor,可选的)----输出张量返回值:一个torch.ByteTensor张量,包含了每个位置的比较结果(相等为1,不等为0)>>>a=torch.Tensor([[1,2],[3,4]])>>>b=torch.Tensor([[1,1],[4,4]])>>>torch.eq(a,b)tens...
继续阅读 >
1.字面理解:torch.cat是将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起。2.例子理解>>>importtorch>>>A=torch.ones(2,3)#2x3的张量(矩阵)>>>Atensor([[1.,1.,1.],[1.,1.,1.]])>>>B=2*torch.ones(4,3)#4x3的张量(矩阵)>>>Btensor([[2.,2.,2.],[2.,2.,2.],[2.,2.,2.],[2.,2.,2.]])>>>C=torch.cat((A,B),0)#按...
继续阅读 >
我就废话不多说了,大家还是直接看代码吧~importtorchinput_tensor=torch.tensor([1,2,3,4,5])print(input_tensor>3)mask=(input_tensor>3).nonzero()print(mask)print(input_tensor.index_select(0,mask))tensor([0,0,0,1,1],dtype=torch.uint8)tensor([3,4])tensor([4,5])补充知识:pytorchtensor筛选满足条件的行或列(使用与或)我就废话不多说了,大家还是直接看代码吧~importtorchx=torch.linspace(1,8,st...
继续阅读 >
max找出tensor的行或者列最大的值:找出每行的最大值:importtorchoutputs=torch.FloatTensor([[1],[2],[3]])print(torch.max(outputs.data,1))输出:(tensor([1.,2.,3.]),tensor([0,0,0]))找出每列的最大值:importtorchoutputs=torch.FloatTensor([[1],[2],[3]])print(torch.max(outputs.data,0))输出结果:(tensor([3.]),tensor([2]))Tensor比较eq相等:importtorchoutputs=torch.FloatTensor([[1],[2],[3]])targe...
继续阅读 >
在利用torch.max函数和F.Ssoftmax函数时,对应该设置什么维度,总是有点懵,遂总结一下:首先看看二维tensor的函数的例子:importtorchimporttorch.nn.functionalasFinput=torch.randn(3,4)print(input)tensor([[-0.5526,-0.0194,2.1469,-0.2567],[-0.3337,-0.9229,0.0376,-0.0801],[1.4721,0.1181,-2.6214,1.7721]])b=F.softmax(input,dim=0)#按列SoftMax,列和为1print(b)tensor([[0.1018,0.3918,...
继续阅读 >