本文共 3535 字，大约阅读时间需要 11 分钟。

Tensor ��������������������������� Tensor ��������������������������������������������������������������� Tensor ������������������������������������������������������������������������������

������������

��������������������������������� Tensor ��������������������������������������������� Tensor���������������������������������������������������������������������������������

������������ 2x2 ��� Tensor

��������������������� 2x2 ��� Tensor������������������������������ 0������������������ Tensor ������������������

[ A = \begin{bmatrix} 0 & 0 \ 0 & 0 \ \end{bmatrix} \quad B = \begin{bmatrix} 0 & 0 \ 0 & 0 \ \end{bmatrix} ]

������������������������

������������������������ 0 ������������������������������������������ Tensor ��������������������������������������� Tensor������������ Tensor ������������������ 4x2���

[ \text{������(A, B, axis=0)} = \begin{bmatrix} 0 & 0 & 0 & 0 \ 0 & 0 & 0 & 0 \ \end{bmatrix} ]

������������������������

������������������������ 1 ������������������������������ Tensor ��������������������������������������������������� 2x4���

[ \text{������(A, B, axis=1)} = \begin{bmatrix} 0 & 0 & 0 & 0 \ 0 & 0 & 0 & 0 \ \end{bmatrix} ]

��� 0 ��������� Stack

��������� 0 ��������� Stack ������������������������������������������������������ 2x2 ��� Tensor ��������������� 2x2x2 ��� 3D Tensor���

[ \text{stack(A, B, axis=0)} = \begin{bmatrix} 0 & 0 \ 0 & 0 \ 0 & 0 \ 0 & 0 \ \end{bmatrix} ]

��� 1 ��������� Stack

��������� 1 ��������� Stack ��������������������������������������������������������� 2x2 ��� Tensor ��������������� 2x2x2 ��� 3D Tensor���

[ \text{stack(A, B, axis=1)} = \begin{bmatrix} 0 & 0 \ 0 & 0 \ 0 & 0 \ 0 & 0 \ \end{bmatrix} ]

��� 2 ��������� Stack

��������� 2 ��������� Stack ������������������������������������������������������������ 2x2 ��� Tensor ��������������� 2x2x2 ��� 3D Tensor���

[ \text{stack(A, B, axis=2)} = \begin{bmatrix} 0 & 0 \ 0 & 0 \ 0 & 0 \ 0 & 0 \ \end{bmatrix} ]

������������

������������������������������������������������������������������������ Tensor ��������������������������� Tensor���������������������������������������������������������

������ chunk

��������� 0 ���������������������������

a = Tensor([1, 2, 3, 4], shape=(2,3))
 tensors = a.split([], dim=0)

��������������������������� 1x3 ��� Tensor���

[ \begin{bmatrix} 1 & 2 & 3 \ 4 & 0 & 0 \ \end{bmatrix} ]

��������� 1 ���������������

��������� 1 ���������������������������

a = Tensor([1, 2, 3, 4, 5, 6, 7, 8], shape=(2,4))
 tensors = a.split([], dim=1)

��������������������������� Tensor��� [ \begin{bmatrix} 1 & 2 & 3 & 4 \ 5 & 6 & 7 & 8 \ \end{bmatrix} ] ��������������������������������������� Tensor ������������������������������

[ a = \begin{bmatrix} 1 & 2 \ 3 & 4 \ 5 & 6 \ 7 & 8 \ 9 & 10 \ \end{bmatrix} ] ������������������������ [ \begin{bmatrix} 1 & 2 \ 3 & 4 \ 5 & 6 \ 7 & 8 \ \end{bmatrix}, \begin{bmatrix} 9 & 10 \ \end{bmatrix} ]

������ split

split ��������������������������������������������������������� 0 ������������������������������������������

a = Tensor([1, 2, 3, 4], shape=(2,2))
 tensors = a.split([1,1], dim=0)

��������������������������� 1x2 ��� Tensor���

������������

split ������������������������ list ������������������������������

a = Tensor([1, 2, 3, 4, 5, 6, 7, 8], shape=(4,))
 tensors = a.split([2, 2])

��������������������������� 2 ��������� Tensor���

��������������������������������������������������������� Tensor���������������������������������������Tensor ���������������������������������������������������������������������������������������������