operand – 要重塑形状的 BCOO 数组。
new_sizes (Sequence[int]) – 指定结果形状的整数序列。最终数组的大小必须与输入的大小匹配。必须指定此项,以使批次维度、稀疏维度和稠密维度不会混合。
dimensions (Sequence[int] | None | None) – 可选的整数序列,指定输入形状的置换顺序。如果指定,则长度必须与 operand.shape
匹配。此外,维度必须仅在 mat 的相似维度之间进行置换:批次维度、稀疏维度和稠密维度不能置换。
mat (BCOO)