Pythonではnumpyとtorch.tensor形式におけるmax関数の用法の実験

869 ワード

a=np.array([[1,2,3,4],[5,6,7,8],[-1,-4,-6,-9],[-1,3,-7,4]])

c,d=a.max(1)
Traceback (most recent call last):

  File "", line 1, in 
    c,d=a.max(1)

ValueError: too many values to unpack (expected 2)




a.max(1)
Out[12]: array([ 4,  8, -1,  4])

a.max(0)
Out[13]: array([5, 6, 7, 8])

_,c=a.max(1)
Traceback (most recent call last):

  File "", line 1, in 
    _,c=a.max(1)

ValueError: too many values to unpack (expected 2)




a=torch.from_numpy(a)

a
Out[16]: 

 1  2  3  4
 5  6  7  8
-1 -4 -6 -9
-1  3 -7  4
[torch.IntTensor of size 4x4]

_,c=a.max(1)

_
Out[18]: 

 4
 8
-1
 4
[torch.IntTensor of size 4]

c
Out[19]: 

 3
 3
 0
 3
[torch.LongTensor of size 4]