Ctrl+K

pyarrow.Tensor

pyarrow.Tensor#

pyarrow.Tensor 是 \(n\) 维 pyarrow.Array，也就是张量。

import pyarrow as pa
import numpy as np

x = np.array([[2, 2, 4], [4, 5, 100]], np.int32)
y = pa.Tensor.from_numpy(x, dim_names=["dim1","dim2"])
y

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[1], line 1
----> 1 import pyarrow as pa
      2 import numpy as np
      4 x = np.array([[2, 2, 4], [4, 5, 100]], np.int32)

ModuleNotFoundError: No module named 'pyarrow'

pyarrow.Tensor.dim_name()(self, i)#

返回第 i 个张量维度的名称。

import pyarrow as pa
import numpy as np
x = np.array([[2, 2, 4], [4, 5, 100]], np.int32)
tensor = pa.Tensor.from_numpy(x, dim_names=["dim1","dim2"])
tensor.dim_name(0), tensor.dim_name(1)

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[2], line 1
----> 1 import pyarrow as pa
      2 import numpy as np
      3 x = np.array([[2, 2, 4], [4, 5, 100]], np.int32)

ModuleNotFoundError: No module named 'pyarrow'

pyarrow.Tensor.equals()(self, Tensor other)#

如果张量包含完全相同的数据，则返回 True。

import pyarrow as pa
import numpy as np
x = np.array([[2, 2, 4], [4, 5, 100]], np.int32)
tensor = pa.Tensor.from_numpy(x, dim_names=["dim1","dim2"])
y = np.array([[2, 2, 4], [4, 5, 10]], np.int32)
tensor2 = pa.Tensor.from_numpy(y, dim_names=["a","b"])
tensor.equals(tensor), tensor.equals(tensor2)

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[3], line 1
----> 1 import pyarrow as pa
      2 import numpy as np
      3 x = np.array([[2, 2, 4], [4, 5, 100]], np.int32)

ModuleNotFoundError: No module named 'pyarrow'

x1 = np.array([[2, 2, 4], [4, 5, 100]], np.int32)
x2 = np.array([[2, 2, 4], [4, 5, 100]], np.int32)
tensor1 = pa.Tensor.from_numpy(x1, dim_names=["dim11","dim12"])
tensor2 = pa.Tensor.from_numpy(x2, dim_names=["dim21","dim22"])
tensor1.equals(tensor2), tensor1 == tensor2

---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
Cell In[4], line 1
----> 1 x1 = np.array([[2, 2, 4], [4, 5, 100]], np.int32)
      2 x2 = np.array([[2, 2, 4], [4, 5, 100]], np.int32)
      3 tensor1 = pa.Tensor.from_numpy(x1, dim_names=["dim11","dim12"])

NameError: name 'np' is not defined

pyarrow.Tensor.to_numpy()#

x = np.array([[2, 2, 4], [4, 5, 100]], np.int32)
tensor = pa.Tensor.from_numpy(x, dim_names=["dim1","dim2"])
tensor.to_numpy()

---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
Cell In[5], line 1
----> 1 x = np.array([[2, 2, 4], [4, 5, 100]], np.int32)
      2 tensor = pa.Tensor.from_numpy(x, dim_names=["dim1","dim2"])
      3 tensor.to_numpy()

NameError: name 'np' is not defined
目录