masked_select

    该OP返回一个1-D 的Tensor, Tensor的值是根据 mask 对输入 x 进行选择的, 的数据类型是bool 。

    参数:

    • name (str,可选)- 具体用法请参见 ,一般无需设置,默认值为None。

    抛出异常:

    代码示例