is_same_shape¶
返回两个 Tensor 形状比较的结果,判断输入 x
与输入 y
的形状是否相同,支持 DenseTensor、SparseCsrTensor 与 SparseCooTensor 之间任意两种的形状比较。
参数¶
x (Tensor) - 输入的 Tensor,类型为 DenseTensor、SparseCsrTensor 与 SparseCooTensor 之一。
y (Tensor) - 输入的 Tensor,类型为 DenseTensor、SparseCsrTensor 与 SparseCooTensor 之一。
返回¶
bool,两个 Tensor 形状比较的结果,相同为 True,不同为 False。
代码示例¶
import paddle
x = paddle.rand([2, 3, 8])
y = paddle.rand([2, 3, 8])
y = y.to_sparse_csr()
z = paddle.rand([2, 5])
paddle.sparse.is_same_shape(x, y)
# True
paddle.sparse.is_same_shape(x, z)
# False