也许使用
einsum？
>>> x = np.array([[1, 2], [3, 4], [5,6]])>>> np.einsum('ij...,i...->ij...',x,x)array([[[ 1, 2], [ 2, 4]], [[ 9, 12], [12, 16]], [[25, 30], [30, 36]]])
也许使用
einsum？
>>> x = np.array([[1, 2], [3, 4], [5,6]])>>> np.einsum('ij...,i...->ij...',x,x)array([[[ 1, 2], [ 2, 4]], [[ 9, 12], [12, 16]], [[25, 30], [30, 36]]])名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
