pub struct CrossEntropy<D: Device> {
pub a: TensorRef<D>,
pub b: TensorRef<D>,
pub output: TensorRef<D>,
}Fields§
§a: TensorRef<D>§b: TensorRef<D>§output: TensorRef<D>Trait Implementations§
Source§impl<D: CoreDeviceOps> DeviceOperation<D> for CrossEntropy<D>
impl<D: CoreDeviceOps> DeviceOperation<D> for CrossEntropy<D>
Auto Trait Implementations§
impl<D> !RefUnwindSafe for CrossEntropy<D>
impl<D> !Send for CrossEntropy<D>
impl<D> !Sync for CrossEntropy<D>
impl<D> !UnwindSafe for CrossEntropy<D>
impl<D> Freeze for CrossEntropy<D>
impl<D> Unpin for CrossEntropy<D>
impl<D> UnsafeUnpin for CrossEntropy<D>where
TensorRef<D>: UnsafeUnpin,
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more