def count_f32
import bendtorch-core@0.1.0.0/src/autograd/loss.bend as Lossimport Base
41# A count as an F32, rounded as PyTorch converts an integer count.
42def count_f32(+count: Nat) -> F32:
43 U32.to_f32(U32.from_nat(count))
Uses (4)
Used by (5)
- bendtorch-core@0.1.0.0/src/autograd/loss:mse_scale
- bendtorch-core@0.1.0.0/src/autograd/loss:total_over
- bendtorch-core@0.1.0.0/src/autograd/loss:classes
- bendtorch-core@0.1.0.0/src/autograd/tensor_backward:vjp_cross_entropy
- bendtorch-core@0.1.0.0/src/autograd/tensor_vjp_LAWS:tensor_cross_entropy_vjp_values