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...
继续阅读 >