def pk_b2
import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorForkimport Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA
132def pk_b2(r: TA.PK) -> B2<F32, TA.T8>:
133 match r:
134 case TA.PK{a, t}:
135 B2{a, t}
import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorForkimport Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA
132def pk_b2(r: TA.PK) -> B2<F32, TA.T8>:
133 match r:
134 case TA.PK{a, t}:
135 B2{a, t}