~/jendhub

def expanded

stelliferous@0.0.2.0 · tensor_select.bend:18

import stelliferous@0.0.2.0/tensor_select.bend as TensorSelect
import Base
import stelliferous@0.0.2.0/tensor.bend as Tensors
import stelliferous@0.0.2.0/tensor_layout.bend as Layout

18def expanded(same: Bool,shape: List<&2,U32>,tensor: Tensors.Tensor<F32>) -> Tensors.Tensor<F32>:
19  match same:
20    case True{}: tensor
21    case False{}: Layout.broadcast_to(~F32,tensor,shape)

Uses (6)

Used by (1)