Function burn_wgpu::kernel::cast

source · 
pub fn cast<InputElem: WgpuElement, OutputElem: WgpuElement, const D: usize>(
    tensor: WgpuTensor<InputElem, D>
) -> WgpuTensor<OutputElem, D>
Expand description

Cast a tensor to the given element type.