示例#1
0
 private static extern IntPtr THSTensor_max(IntPtr src, AllocatePinnedArray allocator, long dimension, bool keep_dim);
示例#2
0
 private static extern void THSNN_Module_get_named_parameters(HType module, AllocatePinnedArray allocator1, AllocatePinnedArray allocator2);
示例#3
0
 static extern void THSTensor_broadcast_tensors(IntPtr tensor, long length, AllocatePinnedArray allocator);
示例#4
0
 private static extern void THSNN_Optimizer_getParameters(HType module, AllocatePinnedArray allocator);
示例#5
0
 private static extern void THSAutograd_grad(
     IntPtr outputs, long oLength,
     IntPtr inputs, long iLength,
     IntPtr grad_outs, long gLength,
     bool retain_graph, bool create_graph, bool allow_unused,
     AllocatePinnedArray allocator);