R/CompareMethods_torch.R

In GPUmatrix: Basic Linear Algebra with GPU

setMethod("Compare",
          c(e1="gpu.matrix.torch", e2="ANY"),
          function(e1, e2)
          {
            op = .Generic[[1]]
            castMatrix <- castTypeOperations_torch(e1,e2)
            e1 <- castMatrix[[1]]
            e2 <- castMatrix[[2]]
            dtype(e2) <- dtype(e1)
            switch(op,
                   '==' = {
                     return(as.matrix((e1@gm == e2@gm)$cpu()))
                   },
                   '>' = {
                     return(as.matrix((e1@gm > e2@gm)$cpu()))
                   },
                   '<' = {
                     return(as.matrix((e1@gm < e2@gm)$cpu()))
                   },
                   '!=' = {
                     return(as.matrix((e1@gm != e2@gm)$cpu()))
                   },
                   '<='={
                     return(as.matrix((e1@gm <= e2@gm)$cpu()))
                   },
                   '>=' = {
                     return(as.matrix((e1@gm >= e2@gm)$cpu()))
                   }
            )
          }
)

setMethod("Compare",
          c(e1="ANY", e2="gpu.matrix.torch"),
          function(e1, e2)
          {
            op = .Generic[[1]]
            castMatrix <- castTypeOperations_torch(e1,e2)
            e1 <- castMatrix[[1]]
            e2 <- castMatrix[[2]]
            dtype(e1) <- dtype(e2)

            switch(op,
                   '==' = {
                     return(as.matrix((e1@gm == e2@gm)$cpu()))
                   },
                   '>' = {
                     return(as.matrix((e1@gm > e2@gm)$cpu()))
                   },
                   '<' = {
                     return(as.matrix((e1@gm < e2@gm)$cpu()))
                   },
                   '!=' = {
                     return(as.matrix((e1@gm != e2@gm)$cpu()))
                   },
                   '<='={
                     return(as.matrix((e1@gm <= e2@gm)$cpu()))
                   },
                   '>=' = {
                     return(as.matrix((e1@gm >= e2@gm)$cpu()))
                   }
            )
          }
)