파이썬 PyTorch에서 행렬의 곱을 수행하는 방법
numpy에서는 다음과 같은 간단한 행렬 곱셈을 수행 할 수 있습니다. a = numpy.arange(2*3).reshape(3,2) b = numpy.arange(2).reshape(2,1) print(a) print(b) print(a.dot(b)) 그러나 PyTorch Tensor로 이것을 시도하면 작동하지 않습니다. a = torch.Tensor([[1, 2, 3], [1, 2, 3]]).view(-1, 2) b = torch.Tensor([[2, 1]]).view(2, -1) print(a) print(a.size()) print(b) print(b.size()) print(torch.dot(a, b)) 이 코드는 다음 오류를 발생시킵니다. RuntimeError : inconsistent te..
2020. 10. 18.
파이썬 NumPy 다차원 배열의 i 번째 열에 액세스하는 방법은 무엇입니까?
내가 가지고 있다고 가정합니다. test = numpy.array([[1, 2], [3, 4], [5, 6]]) test [i] 는 배열의 이 줄을 가져옵니다 (예 : [1, 2] ). ith 열에 액세스하려면 어떻게해야합니까? (예 : [1, 3, 5] ). 또한 이것은 비용이 많이 드는 작업입니까? 해결 방법 >>> test[:,0] array([1, 3, 5]) 비슷하게, >>> test[1,:] array([3, 4]) 참조 페이지 https://stackoverflow.com/questions/4455076
2020. 10. 17.