is_same_shape

paddle.sparse. is_same_shape ( x, y ) [源代码]

返回两个 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