diff options
Diffstat (limited to 'candle-examples/examples/custom-ops')
| -rw-r--r-- | candle-examples/examples/custom-ops/main.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/custom-ops/main.rs b/candle-examples/examples/custom-ops/main.rs index 63bcd83a..7f7a3f26 100644 --- a/candle-examples/examples/custom-ops/main.rs +++ b/candle-examples/examples/custom-ops/main.rs @@ -89,7 +89,7 @@ fn main() -> anyhow::Result<()> { let device = candle_examples::device(args.cpu)?; let t = Tensor::arange(0f32, 14f32, &device)?.reshape((2, 7))?; println!("{t}"); - let t = t.custom_op1(LayerNorm { eps: 1e-5 })?; + let t = t.apply_op1(LayerNorm { eps: 1e-5 })?; println!("{t}"); Ok(()) } |