张量乘法

张量乘法是指两个张量之间的乘法运算，它是一种常见的张量运算，在神经网络和深度学习等领域中有着广泛的应用。在进行张量乘法时，需要满足两个张量的维度匹配，即其中一个张量的最后一个维度的大小等于另一个张量的倒数第二个维度的大小，例如，一个形状为(m, n, k)的张量A和一个形状为(n, k, p)的张量B之间的乘法可以表示为：

C = A @ B

其中，C表示结果张量，它的形状为(m, n, p)，且在每一个位置上的元素都等于对应位置上A和B的元素之和。在numpy中，可以使用@符号或dot函数进行张量乘法运算，例如：

import numpy as np

# 创建两个形状分别为(2,3,4)和(3,4,5)的张量
A = np.random.randn(2,3,4)
B = np.random.randn(3,4,5)

# 进行张量乘法运算
C = A @ B

# 输出结果张量的形状
print(C.shape)  # 输出(2,3,5)

在神经网络中，张量乘法通常用于计算输入张量和权重张量之间的乘积，以及前向传播和反向传播等操作。